test_core.py revision 653212fdd1f5b5eea1c5b7d4d28b3f6c8fd05bba
1from .base import TestBase 2from ..core import OpCode 3from ..core import MemoryBuffer 4 5class TestCore(TestBase): 6 def test_opcode(self): 7 self.assertTrue(hasattr(OpCode, 'Ret')) 8 self.assertTrue(isinstance(OpCode.Ret, OpCode)) 9 self.assertEqual(OpCode.Ret.value, 1) 10 11 op = OpCode.from_value(1) 12 self.assertTrue(isinstance(op, OpCode)) 13 self.assertEqual(op, OpCode.Ret) 14 15 def test_memory_buffer_create_from_file(self): 16 source = self.get_test_binary() 17 18 MemoryBuffer(filename=source) 19 20 def test_memory_buffer_failing(self): 21 with self.assertRaises(Exception): 22 MemoryBuffer(filename="/hopefully/this/path/doesnt/exist") 23 24 def test_memory_buffer_len(self): 25 source = self.get_test_file() 26 m = MemoryBuffer(filename=source) 27 self.assertEqual(len(m), 50) 28