def test_load_plugins(self): with mock.patch('compose.plugin_manager.os.path.isdir') as mock_isdir, \ mock.patch('compose.plugin_manager.os.path.isfile') as mock_isfile, \ mock.patch('compose.plugin_manager.os.listdir') as mock_listdir, \ mock.patch('imp.load_source') as mock_load_source, \ mock.patch.object(Plugin, '__init__') as mock_plugin, \ mock.patch('os.walk') as mock_walk: mock_walk.return_value = [] mock_plugin.return_value = None plugin_manager = PluginManager('') plugin_manager.plugin_classes = None mock_load_source.return_value = self._get_helper_class({ 'plugin': Plugin }) mock_isdir.return_value = True mock_isfile.return_value = True mock_listdir.return_value = ['plugin_1', 'plugin_2'] plugin_manager.load_config('', {}) loaded_plugins = plugin_manager._load_plugins() self.assertEquals('plugin_1' in loaded_plugins.keys(), True) self.assertEquals('plugin_2' in loaded_plugins.keys(), True) self.assertEquals(isinstance(loaded_plugins['plugin_1'], Plugin), True) self.assertEquals(isinstance(loaded_plugins['plugin_2'], Plugin), True)
def get_loaded_plugin_manager(self): with mock.patch('compose.plugin_manager.os.path.isdir') as mock_isdir, \ mock.patch('compose.plugin_manager.os.path.isfile') as mock_isfile, \ mock.patch('compose.plugin_manager.os.listdir') as mock_listdir, \ mock.patch('imp.load_source') as mock_load_source, \ mock.patch.object(Plugin, '__init__') as mock_plugin, \ mock.patch('os.walk') as mock_walk: mock_walk.return_value = [] mock_plugin.return_value = None mock_load_source.return_value = self._get_helper_class({ 'plugin': Plugin }) mock_isdir.return_value = True mock_isfile.return_value = True mock_listdir.return_value = ['plugin_1', 'plugin_2'] plugin_manager = PluginManager('plugin_folder') plugin_manager.load_config('', {}) return plugin_manager
def test_load_config(self): plugin_manager = PluginManager('') plugin_manager.load_config('', {}) self.assertEquals(plugin_manager.config, False)