import pychrysalide
from chrysacase import ChrysalideTestCase
from pychrysalide.arch import ArchInstruction


class TestProcessor(ChrysalideTestCase):
    """TestCase for arch.ArchProcessor."""


    def testAbstractClass(self):
        """Forbid instruction class instance."""

        with self.assertRaisesRegex(RuntimeError, 'pychrysalide.arch.ArchInstruction is an abstract class'):
            ins = ArchInstruction()


    def testInstructionBasicImplementation(self):
        """Implement basic custom instructions."""


        class TodoInstruction(ArchInstruction):

            def __init__(self):
                super().__init__(0x123)


        ins = TodoInstruction()

        with self.assertRaisesRegex(NotImplementedError, 'unexpected NULL value as encoding'):
            print(ins.encoding)

        with self.assertRaisesRegex(NotImplementedError, 'unexpected NULL value as keyword'):
            print(ins.keyword)


        class CustomInstruction(ArchInstruction):

            def __init__(self):
                super().__init__(0x123)

            def _get_encoding(self):
                return 'custom'

            def _get_keyword(self):
                return 'kw'


        ins = CustomInstruction()

        self.assertEqual('custom', ins.encoding)
        self.assertEqual('kw', ins.keyword)