"""
Test some lldb command abbreviations.
"""

from __future__ import print_function


import lldb
from lldbsuite.test.decorators import *
from lldbsuite.test.lldbtest import *
from lldbsuite.test import lldbutil


class DisassemblyTestCase(TestBase):

    mydir = TestBase.compute_mydir(__file__)
    NO_DEBUG_INFO_TESTCASE = True

    def test(self):
        self.build()
        target, _, _, bkpt = lldbutil.run_to_source_breakpoint(self,
                "Set a breakpoint here", lldb.SBFileSpec("main.cpp"))
        self.runCmd("dis -f")
        disassembly_with_break = self.res.GetOutput().splitlines()

        self.assertTrue(target.BreakpointDelete(bkpt.GetID()))

        self.runCmd("dis -f")
        disassembly_without_break = self.res.GetOutput().splitlines()

        # Make sure all assembly instructions are the same as instructions
        # with the breakpoint removed.
        self.assertEqual(len(disassembly_with_break),
                         len(disassembly_without_break))
        for dis_inst_with, dis_inst_without in \
                zip(disassembly_with_break, disassembly_without_break):
            inst_with = dis_inst_with.split(':')[-1]
            inst_without = dis_inst_without.split(':')[-1]
            self.assertEqual(inst_with, inst_without)
