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/pluginmgr_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/pluginmgr_tests.py')
-rw-r--r-- | cliapp/pluginmgr_tests.py | 122 |
1 files changed, 122 insertions, 0 deletions
diff --git a/cliapp/pluginmgr_tests.py b/cliapp/pluginmgr_tests.py new file mode 100644 index 0000000..8c49633 --- /dev/null +++ b/cliapp/pluginmgr_tests.py @@ -0,0 +1,122 @@ +# 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 PluginManager + + +class PluginManagerInitialStateTests(unittest.TestCase): + + def setUp(self): + self.pm = PluginManager() + + def test_locations_is_empty_list(self): + self.assertEqual(self.pm.locations, []) + + def test_plugins_is_empty_list(self): + self.assertEqual(self.pm.plugins, []) + + def test_application_version_is_zeroes(self): + self.assertEqual(self.pm.application_version, '0.0.0') + + def test_plugin_files_is_empty(self): + self.assertEqual(self.pm.plugin_files, []) + + def test_plugin_arguments_is_empty(self): + self.assertEqual(self.pm.plugin_arguments, []) + + def test_plugin_keyword_arguments_is_empty(self): + self.assertEqual(self.pm.plugin_keyword_arguments, {}) + + +class PluginManagerTests(unittest.TestCase): + + def setUp(self): + self.pm = PluginManager() + self.pm.locations = ['test-plugins', 'not-exist'] + self.pm.plugin_arguments = ('fooarg',) + self.pm.plugin_keyword_arguments = { 'bar': 'bararg' } + + self.files = sorted(['test-plugins/hello_plugin.py', + 'test-plugins/aaa_hello_plugin.py', + 'test-plugins/oldhello_plugin.py', + 'test-plugins/wrongversion_plugin.py']) + + def test_finds_the_right_plugin_files(self): + self.assertEqual(self.pm.find_plugin_files(), self.files) + + def test_plugin_files_attribute_implicitly_searches(self): + self.assertEqual(self.pm.plugin_files, self.files) + + def test_loads_hello_plugin(self): + plugins = self.pm.load_plugins() + self.assertEqual(len(plugins), 1) + self.assertEqual(plugins[0].name, 'Hello') + + def test_plugins_attribute_implicitly_searches(self): + self.assertEqual(len(self.pm.plugins), 1) + self.assertEqual(self.pm.plugins[0].name, 'Hello') + + def test_initializes_hello_with_correct_args(self): + plugin = self.pm['Hello'] + self.assertEqual(plugin.foo, 'fooarg') + self.assertEqual(plugin.bar, 'bararg') + + def test_calls_setup_method_at_load_time(self): + plugin = self.pm['Hello'] + self.assert_(plugin.setup_called) + + def test_raises_keyerror_for_unknown_plugin(self): + self.assertRaises(KeyError, self.pm.__getitem__, 'Hithere') + + def test_enable_plugins_enables_all_plugins(self): + enabled = set() + for plugin in self.pm.plugins: + plugin.enable = lambda: enabled.add(plugin) + self.pm.enable_plugins() + self.assertEqual(enabled, set(self.pm.plugins)) + + def test_disable_plugins_disables_all_plugins(self): + disabled = set() + for plugin in self.pm.plugins: + plugin.disable = lambda: disabled.add(plugin) + self.pm.disable_plugins() + self.assertEqual(disabled, set(self.pm.plugins)) + + +class PluginManagerCompatibleApplicationVersionTests(unittest.TestCase): + + def setUp(self): + self.pm = PluginManager() + self.pm.application_version = '1.2.3' + + def test_rejects_zero(self): + self.assertFalse(self.pm.compatible_version('0')) + + def test_rejects_two(self): + self.assertFalse(self.pm.compatible_version('2')) + + def test_rejects_one_two_four(self): + self.assertFalse(self.pm.compatible_version('1.2.4')) + + def test_accepts_one(self): + self.assert_(self.pm.compatible_version('1')) + + def test_accepts_one_two_three(self): + self.assert_(self.pm.compatible_version('1.2.3')) + |