41
52
"""docs() should return something reasonable about the Hooks."""
42
53
class MyHooks(Hooks):
44
hooks = MyHooks("bzrlib.tests.test_hooks", "some_hooks")
55
hooks = MyHooks("breezy.tests.test_hooks", "some_hooks")
45
56
hooks['legacy'] = []
46
57
hooks.add_hook('post_tip_change',
47
"Invoked after the tip of a branch changes. Called with "
48
"a ChangeBranchTipParams object.", (1, 4))
58
"Invoked after the tip of a branch changes. Called with "
59
"a ChangeBranchTipParams object.", (1, 4))
49
60
hooks.add_hook('pre_tip_change',
50
"Invoked before the tip of a branch changes. Called with "
51
"a ChangeBranchTipParams object. Hooks should raise "
52
"TipChangeRejected to signal that a tip change is not permitted.",
61
"Invoked before the tip of a branch changes. Called with "
62
"a ChangeBranchTipParams object. Hooks should raise "
63
"TipChangeRejected to signal that a tip change is not permitted.",
54
65
self.assertEqualDiff(
78
89
"signal that a tip change is not permitted.\n", hooks.docs())
80
91
def test_install_named_hook_raises_unknown_hook(self):
81
hooks = Hooks("bzrlib.tests.test_hooks", "some_hooks")
82
self.assertRaises(errors.UnknownHook, hooks.install_named_hook, 'silly',
92
hooks = Hooks("breezy.tests.test_hooks", "some_hooks")
93
self.assertRaises(UnknownHook, hooks.install_named_hook, 'silly',
85
96
def test_install_named_hook_appends_known_hook(self):
86
hooks = Hooks("bzrlib.tests.test_hooks", "some_hooks")
97
hooks = Hooks("breezy.tests.test_hooks", "some_hooks")
87
98
hooks['set_rh'] = []
88
99
hooks.install_named_hook('set_rh', None, "demo")
89
100
self.assertEqual(hooks['set_rh'], [None])
91
102
def test_install_named_hook_and_retrieve_name(self):
92
hooks = Hooks("bzrlib.tests.test_hooks", "somehooks")
103
hooks = Hooks("breezy.tests.test_hooks", "somehooks")
93
104
hooks['set_rh'] = []
94
105
hooks.install_named_hook('set_rh', None, "demo")
95
106
self.assertEqual("demo", hooks.get_hook_name(None))
97
108
def test_uninstall_named_hook(self):
98
hooks = Hooks("bzrlib.tests.test_hooks", "some_hooks")
109
hooks = Hooks("breezy.tests.test_hooks", "some_hooks")
99
110
hooks.add_hook('set_rh', "Set revision history", (2, 0))
100
111
hooks.install_named_hook('set_rh', None, "demo")
101
112
self.assertEqual(1, len(hooks["set_rh"]))
114
125
self.assertEqual(1, len(hooks["set_rh"]))
116
127
def test_uninstall_named_hook_unknown_callable(self):
117
hooks = Hooks("bzrlib.tests.test_hooks", "some_hooks")
128
hooks = Hooks("breezy.tests.test_hooks", "some_hooks")
118
129
hooks.add_hook('set_rh', "Set revision hsitory", (2, 0))
119
130
self.assertRaises(KeyError, hooks.uninstall_named_hook, "set_rh",
122
133
def test_uninstall_named_hook_raises_unknown_hook(self):
123
hooks = Hooks("bzrlib.tests.test_hooks", "some_hooks")
124
self.assertRaises(errors.UnknownHook, hooks.uninstall_named_hook,
134
hooks = Hooks("breezy.tests.test_hooks", "some_hooks")
135
self.assertRaises(UnknownHook, hooks.uninstall_named_hook, 'silly', "")
127
137
def test_uninstall_named_hook_old_style(self):
128
hooks = Hooks("bzrlib.tests.test_hooks", "some_hooks")
138
hooks = Hooks("breezy.tests.test_hooks", "some_hooks")
129
139
hooks["set_rh"] = []
130
140
hooks.install_named_hook('set_rh', None, "demo")
131
141
self.assertRaises(errors.UnsupportedOperation,
132
hooks.uninstall_named_hook, "set_rh", "demo")
142
hooks.uninstall_named_hook, "set_rh", "demo")
134
hooks = Hooks("bzrlib.tests.test_hooks", "TestHooks.hooks")
144
hooks = Hooks("breezy.tests.test_hooks", "TestHooks.hooks")
136
146
def test_install_lazy_named_hook(self):
137
147
# When the hook points are not yet registered the hook is
138
# added to the _lazy_hooks dictionary in bzrlib.hooks.
148
# added to the _lazy_hooks dictionary in breezy.hooks.
139
149
self.hooks.add_hook('set_rh', "doc", (0, 15))
140
set_rh = lambda: None
141
install_lazy_named_hook('bzrlib.tests.test_hooks',
142
'TestHooks.hooks', 'set_rh', set_rh, "demo")
151
def set_rh(): return None
152
install_lazy_named_hook('breezy.tests.test_hooks',
153
'TestHooks.hooks', 'set_rh', set_rh, "demo")
143
154
set_rh_lazy_hooks = _mod_hooks._lazy_hooks[
144
('bzrlib.tests.test_hooks', 'TestHooks.hooks', 'set_rh')]
145
self.assertEquals(1, len(set_rh_lazy_hooks))
146
self.assertEquals(set_rh, set_rh_lazy_hooks[0][0].get_obj())
147
self.assertEquals("demo", set_rh_lazy_hooks[0][1])
155
('breezy.tests.test_hooks', 'TestHooks.hooks', 'set_rh')]
156
self.assertEqual(1, len(set_rh_lazy_hooks))
157
self.assertEqual(set_rh, set_rh_lazy_hooks[0][0].get_obj())
158
self.assertEqual("demo", set_rh_lazy_hooks[0][1])
148
159
self.assertEqual(list(TestHooks.hooks['set_rh']), [set_rh])
150
set_rh = lambda: None
161
def set_rh(): return None
152
163
def test_install_named_hook_lazy(self):
153
hooks = Hooks("bzrlib.tests.hooks", "some_hooks")
164
hooks = Hooks("breezy.tests.hooks", "some_hooks")
154
165
hooks['set_rh'] = HookPoint("set_rh", "doc", (0, 15), None)
155
hooks.install_named_hook_lazy('set_rh', 'bzrlib.tests.test_hooks',
156
'TestHooks.set_rh', "demo")
166
hooks.install_named_hook_lazy('set_rh', 'breezy.tests.test_hooks',
167
'TestHooks.set_rh', "demo")
157
168
self.assertEqual(list(hooks['set_rh']), [TestHooks.set_rh])
159
170
def test_install_named_hook_lazy_old(self):
160
171
# An exception is raised if a lazy hook is raised for
161
172
# an old style hook point.
162
hooks = Hooks("bzrlib.tests.hooks", "some_hooks")
173
hooks = Hooks("breezy.tests.hooks", "some_hooks")
163
174
hooks['set_rh'] = []
164
175
self.assertRaises(errors.UnsupportedOperation,
165
hooks.install_named_hook_lazy,
166
'set_rh', 'bzrlib.tests.test_hooks', 'TestHooks.set_rh',
176
hooks.install_named_hook_lazy,
177
'set_rh', 'breezy.tests.test_hooks', 'TestHooks.set_rh',
169
180
def test_valid_lazy_hooks(self):
170
181
# Make sure that all the registered lazy hooks are referring to existing
171
182
# hook points which allow lazy registration.
172
for key, callbacks in _mod_hooks._lazy_hooks.iteritems():
183
for key, callbacks in _mod_hooks._lazy_hooks.items():
173
184
(module_name, member_name, hook_name) = key
174
185
obj = pyutils.get_named_object(module_name, member_name)
175
self.assertEquals(obj._module, module_name)
176
self.assertEquals(obj._member_name, member_name)
186
self.assertEqual(obj._module, module_name)
187
self.assertEqual(obj._member_name, member_name)
177
188
self.assertTrue(hook_name in obj)
178
189
self.assertIs(callbacks, obj[hook_name]._callbacks)
265
278
def test_known_hooks_key_to_object(self):
266
279
self.assertIs(branch.Branch.hooks,
267
known_hooks_key_to_object(('bzrlib.branch', 'Branch.hooks')))
280
known_hooks_key_to_object(('breezy.branch', 'Branch.hooks')))
269
282
def test_known_hooks_key_to_parent_and_attribute(self):
270
283
self.assertEqual((branch.Branch, 'hooks'),
271
known_hooks.key_to_parent_and_attribute(
272
('bzrlib.branch', 'Branch.hooks')))
284
known_hooks.key_to_parent_and_attribute(
285
('breezy.branch', 'Branch.hooks')))
273
286
self.assertEqual((branch, 'Branch'),
274
known_hooks.key_to_parent_and_attribute(
275
('bzrlib.branch', 'Branch')))
287
known_hooks.key_to_parent_and_attribute(
288
('breezy.branch', 'Branch')))