diff --git a/tests/test_manticore.py b/tests/test_manticore.py index c81fe43..1639bdc 100644 --- a/tests/test_manticore.py +++ b/tests/test_manticore.py @@ -20,6 +20,17 @@ class ManticoreTest(unittest.TestCase): pass self.assertTrue(tmp in self.m._hooks[entry]) + def test_hook(self): + self.m.context['x'] = 0 + + @self.m.hook(None) + def tmp(state): + self.m.context['x'] = 1 + self.m.terminate() + self.m.run() + + self.assertEqual(self.m.context['x'], 1) + def test_hook_dec_err(self): with self.assertRaises(TypeError): @self.m.hook('0x00400e40')