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, ])