class PluginManagerTestCase(unittest.TestCase): def setUp(self): global CURRENT_DIR plugin_dir = os.path.join(CURRENT_DIR, 'plugins') self.pm = PluginManager(plugin_dir, None) def test_discover(self): self.assertEqual(len(self.pm), 0) self.pm.discover() self.assertEqual(len(self.pm), 1) def test_magic_method_contains(self): global PLUGIN_VALID_NAME, PLUGIN_NOT_VALID_NAME self.pm.discover() self.assertEqual(PLUGIN_VALID_NAME in self.pm, True) self.assertEqual(PLUGIN_NOT_VALID_NAME in self.pm, False) def test_magic_method_getitem(self): global PLUGIN_VALID_NAME, PLUGIN_NOT_VALID_NAME self.pm.discover() self.assertTrue(self.pm['test_plugin.plugin']) self.assertRaises(KeyError, self.pm.__getitem__, PLUGIN_NOT_VALID_NAME)
def setUp(self): global CURRENT_DIR plugin_dir = os.path.join(CURRENT_DIR, 'plugins') self.pm = PluginManager(plugin_dir, None)