def test_unregister(self):
class GoodHook(HookBase):
""""""
hook = Hook()
hook.register(GoodHook)
self.assertListEqual(hook._registry, [GoodHook, ])
hook.unregister(GoodHook)
self.assertListEqual(hook._registry, [])
# calling unregister again should do nothing
hook.unregister(GoodHook)
self.assertListEqual(hook._registry, [])
def test_register(self):
class BadHook:
""""""
class GoodHook(HookBase):
""""""
hook = Hook()
self.assertRaises(AssertionError, hook.register, BadHook)
self.assertRaises(AssertionError, hook.register, GoodHook("req"))
hook.register(GoodHook)
self.assertListEqual(hook._registry, [GoodHook, ])
