1
# Copyright (C) 2007 Canonical Ltd
3
# This program is free software; you can redistribute it and/or modify
4
# it under the terms of the GNU General Public License as published by
5
# the Free Software Foundation; either version 2 of the License, or
6
# (at your option) any later version.
8
# This program is distributed in the hope that it will be useful,
9
# but WITHOUT ANY WARRANTY; without even the implied warranty of
10
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
11
# GNU General Public License for more details.
13
# You should have received a copy of the GNU General Public License
14
# along with this program; if not, write to the Free Software
15
# Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA
17
"""Tests for the core Hooks logic."""
19
from bzrlib.hooks import (
22
from bzrlib.errors import (
26
from bzrlib.tests import TestCase
29
class TestHooks(TestCase):
31
def test_install_hook_raises_unknown_hook(self):
32
"""install_hook should raise UnknownHook if a hook is unknown."""
34
self.assertRaises(UnknownHook, hooks.install_hook, 'silly', None)
36
def test_install_hook_appends_known_hook(self):
37
"""install_hook should append the callable for known hooks."""
40
hooks.install_hook('set_rh', None)
41
self.assertEqual(hooks['set_rh'], [None])
43
def test_install_named_hook_raises_unknown_hook(self):
46
self.assertRaises(UnknownHook, hooks.install_named_hook, 'silly',
49
def test_install_named_hook_appends_known_hook(self):
52
hooks.install_named_hook('set_rh', None, "demo")
53
self.assertEqual(hooks['set_rh'], [None])
55
def test_install_named_hook_and_retrieve_name(self):
58
hooks.install_named_hook('set_rh', None, "demo")
59
self.assertEqual("demo", hooks.get_hook_name(None))
61
def test_name_hook_and_retrieve_name(self):
62
"""name_hook puts the name in the names mapping."""
65
hooks.install_hook('set_rh', None)
66
hooks.name_hook(None, 'demo')
67
self.assertEqual("demo", hooks.get_hook_name(None))
69
def test_get_unnamed_hook_name_is_unnamed(self):
72
hooks.install_hook('set_rh', None)
73
self.assertEqual("No hook name", hooks.get_hook_name(None))