Example #1
0
class TestCommandBytesUtil(unittest.TestCase):
    def setUp(self):
        self.cmd = CMDS(b'\x34', b'\x23')
        self.cmd2 = CMDS(b'\x34')
        self.cmd_map = CommandBytesMap(True)

    def test_cmd(self):
        self.assertEqual(hash(self.cmd), hash(b'\x34' + b'\x23'))

    def test_map(self):
        self.cmd_map.register(self.cmd, True)
        self.assertTrue(self.cmd_map.get(self.cmd))

    def test_raise(self):
        self.assertRaises(KeyError, self.cmd_map.get, self.cmd2)
Example #2
0
 def setUp(self):
     self.cmd = CMDS(b'\x34', b'\x23')
     self.cmd2 = CMDS(b'\x34')
     self.cmd_map = CommandBytesMap(True)