From 8354ded7e0ae0b2d439fa8b87afacd6a3f04f024 Mon Sep 17 00:00:00 2001 From: Mark Mossberg Date: Thu, 25 May 2017 16:22:08 -0400 Subject: [PATCH] Test hook api (#274) --- tests/test_manticore.py | 11 +++++++++++ 1 file changed, 11 insertions(+) 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')