1
# Copyright (C) 2008 Canonical Ltd
1
# Copyright (C) 2008, 2009, 2011, 2012, 2016 Canonical Ltd
3
3
# This program is free software; you can redistribute it and/or modify
4
4
# it under the terms of the GNU General Public License as published by
19
19
from bzrlib.branch import Branch
20
20
from bzrlib.tests import TestCaseWithTransport
23
25
class TestHooks(TestCaseWithTransport):
25
27
def _check_hooks_output(self, command_output, hooks):
26
28
for hook_type in Branch.hooks:
27
29
s = "\n ".join(hooks.get(hook_type, ["<no hooks installed>"]))
28
self.assert_("%s:\n %s" % (hook_type, s) in command_output)
30
self.assertTrue("%s:\n %s" % (hook_type, s) in command_output)
30
32
def test_hooks_with_no_hooks(self):
31
33
self.make_branch('.')
37
39
def test_hooks_with_unnamed_hook(self):
38
40
self.make_branch('.')
40
Branch.hooks.install_named_hook('set_rh', foo, None)
42
Branch.hooks.install_named_hook('post_push', foo, None)
41
43
out, err = self.run_bzr('hooks')
42
self._check_hooks_output(out, {'set_rh': ["No hook name"]})
44
self._check_hooks_output(out, {'post_push': ["No hook name"]})
44
46
def test_hooks_with_named_hook(self):
45
47
self.make_branch('.')
47
49
name = "Foo Bar Hook"
48
Branch.hooks.install_named_hook('set_rh', foo, name)
50
Branch.hooks.install_named_hook('post_push', foo, name)
49
51
out, err = self.run_bzr('hooks')
50
self._check_hooks_output(out, {'set_rh': [name]})
52
self._check_hooks_output(out, {'post_push': [name]})
52
54
def test_hooks_no_branch(self):
53
55
self.run_bzr('hooks')
57
def test_hooks_lazy_with_unnamed_hook(self):
60
Branch.hooks.install_named_hook_lazy('post_push',
61
'bzrlib.tests.blackbox.test_hooks',
64
out, err = self.run_bzr('hooks')
65
self._check_hooks_output(out, {'post_push': ["No hook name"]})
67
def test_hooks_lazy_with_named_hook(self):
70
Branch.hooks.install_named_hook_lazy('post_push',
71
'bzrlib.tests.blackbox.test_hooks',
74
out, err = self.run_bzr('hooks')
75
self._check_hooks_output(out, {'post_push': ["hook has a name"]})