diff options
author | Lars Wirzenius <liw@liw.fi> | 2012-02-22 17:23:04 +0000 |
---|---|---|
committer | Lars Wirzenius <liw@liw.fi> | 2012-02-22 17:23:04 +0000 |
commit | 4560577fce7e64fdd393d9e54dac237f35e8a25f (patch) | |
tree | 11e364bc87e98cc236d24f843d72f8a49c22f5da /cliapp/hookmgr_tests.py | |
parent | d9f7efcde37f8208b8679b5aae0f1d38875ccc73 (diff) | |
download | cliapp-4560577fce7e64fdd393d9e54dac237f35e8a25f.tar.gz |
Add a plugin system
This code was originally written separately, which is why there's no
history of it in the cliapp version control repository. The code
is derived from code I wrote for obnam.
Diffstat (limited to 'cliapp/hookmgr_tests.py')
-rw-r--r-- | cliapp/hookmgr_tests.py | 59 |
1 files changed, 59 insertions, 0 deletions
diff --git a/cliapp/hookmgr_tests.py b/cliapp/hookmgr_tests.py new file mode 100644 index 0000000..9957d00 --- /dev/null +++ b/cliapp/hookmgr_tests.py @@ -0,0 +1,59 @@ +# Copyright (C) 2009-2012 Lars Wirzenius +# +# This program is free software; you can redistribute it and/or modify +# it under the terms of the GNU General Public License as published by +# the Free Software Foundation; either version 2 of the License, or +# (at your option) any later version. +# +# This program is distributed in the hope that it will be useful, +# but WITHOUT ANY WARRANTY; without even the implied warranty of +# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the +# GNU General Public License for more details. +# +# You should have received a copy of the GNU General Public License along +# with this program; if not, write to the Free Software Foundation, Inc., +# 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA. + + +import unittest + +from cliapp import HookManager, FilterHook + + +class HookManagerTests(unittest.TestCase): + + def setUp(self): + self.hooks = HookManager() + self.hooks.new('foo', FilterHook()) + + def callback(self, *args, **kwargs): + self.args = args + self.kwargs = kwargs + + def test_has_no_tests_initially(self): + hooks = HookManager() + self.assertEqual(hooks.hooks, {}) + + def test_adds_new_hook(self): + self.assert_(self.hooks.hooks.has_key('foo')) + + def test_adds_callback(self): + self.hooks.add_callback('foo', self.callback) + self.assertEqual(self.hooks.hooks['foo'].callbacks, [self.callback]) + + def test_removes_callback(self): + cb_id = self.hooks.add_callback('foo', self.callback) + self.hooks.remove_callback('foo', cb_id) + self.assertEqual(self.hooks.hooks['foo'].callbacks, []) + + def test_calls_callbacks(self): + self.hooks.add_callback('foo', self.callback) + self.hooks.call('foo', 'bar', kwarg='foobar') + self.assertEqual(self.args, ('bar',)) + self.assertEqual(self.kwargs, { 'kwarg': 'foobar' }) + + def test_call_returns_value_of_callbacks(self): + self.hooks.new('bar', FilterHook()) + self.hooks.add_callback('bar', lambda data: data + 1) + self.assertEqual(self.hooks.call('bar', 1), 2) + |