[python] Add markup option to disassembler

[oota-llvm.git] / bindings / python / llvm / tests / test_disassembler.py

diff --git a/bindings/python/llvm/tests/test_disassembler.py b/bindings/python/llvm/tests/test_disassembler.py
index 6eb11a23dc183f0c2313f0c448d0dc294011f011..46d12f7056267d629024160435f21f58d33f9000 100644 (file)
--- a/bindings/python/llvm/tests/test_disassembler.py
+++ b/bindings/python/llvm/tests/test_disassembler.py
@@ -1,62 +1,39 @@
-from unittest import expectedFailure
-from unittest import skip
-
 from .base import TestBase
-from ..disassembler import DisassemblerByteArraySource
-from ..disassembler import DisassemblerFileSource
-from ..disassembler import Disassembler
-from ..object import ObjectFile
+
+from ..disassembler import Disassembler, Option_UseMarkup
 
 class TestDisassembler(TestBase):
-    def test_simple(self):
+    def test_instantiate(self):
+         Disassembler('i686-apple-darwin9')
+
+    def test_basic(self):
         sequence = '\x67\xe3\x81' # jcxz -127
         triple = 'i686-apple-darwin9'
 
-        source = DisassemblerByteArraySource(sequence)
-
-        disassembler = Disassembler(triple, source)
-        instructions = list(disassembler.get_instructions())
-
-        self.assertEqual(len(instructions), 1)
-
-        i = instructions[0]
-        self.assertEqual(str(i), '\tjcxz\t-127\n')
-        self.assertEqual(i.byte_size, 3)
-        self.assertEqual(i.id, 1032)
-        self.assertTrue(i.is_branch)
-        self.assertFalse(i.is_move)
-        self.assertEqual(i.branch_target_id, 0)
-
-        tokens = list(i.get_tokens())
-        self.assertEqual(len(tokens), 4)
-        token = tokens[0]
-        self.assertEqual(str(token), 'jcxz')
-        self.assertFalse(token.is_whitespace)
-        self.assertFalse(token.is_punctuation)
-        self.assertTrue(token.is_opcode)
-        self.assertFalse(token.is_literal)
-        self.assertFalse(token.is_register)
+        disassembler = Disassembler(triple)
 
-        self.assertTrue(tokens[1].is_whitespace)
+        count, s = disassembler.get_instruction(sequence)
+        self.assertEqual(count, 3)
+        self.assertEqual(s, '\tjcxz\t-127')
 
-        operands = list(i.get_operands())
-        self.assertEqual(len(operands), 1)
+    def test_get_instructions(self):
+        sequence = '\x67\xe3\x81\x01\xc7' # jcxz -127; addl %eax, %edi
 
-        # TODO implement operand tests
+        disassembler = Disassembler('i686-apple-darwin9')
 
-    @skip('This test is horribly broken and probably not even correct.')
-    def test_read_instructions(self):
-        filename = self.get_test_binary()
-        o = ObjectFile(filename=filename)
+        instructions = list(disassembler.get_instructions(sequence))
+        self.assertEqual(len(instructions), 2)
 
-        for symbol in o.get_symbols():
-            address = symbol.address
-            offset = symbol.file_offset
-            size = symbol.size
+        self.assertEqual(instructions[0], (0, 3, '\tjcxz\t-127'))
+        self.assertEqual(instructions[1], (3, 2, '\taddl\t%eax, %edi'))
 
-            source = DisassemblerFileSource(filename, offset, length=size,
-                                            start_address=address)
+    def test_set_options(self):
+        sequence = '\x10\x40\x2d\xe9'
+        triple = 'arm-linux-android'
 
-            disassembler = Disassembler('x86-generic-gnu-linux', source)
-            for instruction in disassembler.get_instructions():
-                print instruction
+        disassembler = Disassembler(triple)
+        disassembler.set_options(Option_UseMarkup)
+        count, s = disassembler.get_instruction(sequence)
+        print s
+        self.assertEqual(count, 4)
+        self.assertEqual(s, '\tpush\t{<reg:r4>, <reg:lr>}')