class PluginManagerTests(unittest.TestCase):
def setUp(self):
self.app = Flask(__name__)
self.app.config['TESTING'] = True
self.plugin_manager = PluginManager()
self.plugin_manager.init_app(self.app)
def test_find_plugins(self):
found_plugins = self.plugin_manager.find_plugins()
self.assertEqual(len(found_plugins), 3)
expected_plugins = [
'TestOnePlugin', 'TestTwoPlugin', 'TestThreePlugin'
]
self.assertEquals(sorted(found_plugins), sorted(expected_plugins))
def test_plugins(self):
self.plugin_manager._plugins = None
self.assertEquals(self.plugin_manager._plugins, None)
self.assertEquals(sorted(self.plugin_manager.plugins.keys()),
["test1", "test2"])
def test_all_plugins(self):
self.plugin_manager._all_plugins = None
self.assertEquals(self.plugin_manager._all_plugins, None)
self.assertEquals(sorted(self.plugin_manager.all_plugins.keys()),
["test1", "test2", "test3"])
class PluginManagerGetPlugins(unittest.TestCase):
def setUp(self):
self.app = Flask(__name__)
self.app.config['TESTING'] = True
self.plugin_manager = PluginManager()
self.plugin_manager.init_app(self.app)
def test_get_enabled_plugins(self):
with self.app.test_request_context():
plugins = get_enabled_plugins()
self.assertEquals(set(plugins),
set(self.plugin_manager.plugins.values()))
def test_get_all_plugins(self):
with self.app.test_request_context():
plugins = get_all_plugins()
self.assertEquals(len(plugins), 3)
def test_get_plugin(self):
with self.app.test_request_context():
plugin = get_plugin("test1")
self.assertEquals(plugin, self.plugin_manager.plugins["test1"])
def test_get_plugin_from_all(self):
with self.app.test_request_context():
plugin = get_plugin_from_all("test3")
self.assertFalse(plugin.enabled)
class PluginManagerManagementMethodsTests(unittest.TestCase):
def setUp(self):
self.app = Flask(__name__)
self.app.config['TESTING'] = True
self.plugin_manager = PluginManager()
self.plugin_manager.init_app(self.app)
# Call load_plugins to reload them without calling setup_plugins()
self.plugin_manager.load_plugins()
def test_setup_plugins(self):
plugin = self.plugin_manager.plugins["test1"]
self.assertFalse(plugin.setup_called)
self.plugin_manager.setup_plugins()
self.assertTrue(plugin.setup_called)
def test_install_plugins(self):
plugin = self.plugin_manager.plugins["test1"]
self.assertFalse(plugin.install_called)
self.plugin_manager.install_plugins()
self.assertTrue(plugin.install_called)
def test_uninstall_plugins(self):
plugin = self.plugin_manager.plugins["test1"]
self.assertFalse(plugin.uninstall_called)
self.plugin_manager.uninstall_plugins()
self.assertTrue(plugin.uninstall_called)
class PluginManagerOnePluginTests(unittest.TestCase):
"""Tests the plugin.setup(), plugin.enabled()... methods"""
def setUp(self):
self.app = Flask(__name__)
self.app.config['TESTING'] = True
self.plugin_manager = PluginManager()
self.plugin_manager.init_app(self.app)
def tearDown(self):
# Make sure plugin "test1" is enabled
plugin_path = self.plugin_manager.all_plugins["test1"].path
if os.path.exists(os.path.join(plugin_path, "DISABLED")):
os.remove(os.path.join(plugin_path, "DISABLED"))
self.plugin_manager.load_plugins()
def test_plugin_disable_and_enable(self):
plugin = self.plugin_manager.plugins["test1"]
self.assertFalse(plugin.disable())
plugin = self.plugin_manager.all_plugins["test1"]
self.assertTrue(plugin.enable())
def test_plugin_setup(self):
plugin = self.plugin_manager.plugins["test1"]
plugin.setup()
self.assertTrue(plugin.setup_called)
def test_plugin_install(self):
plugin = self.plugin_manager.plugins["test1"]
plugin.install()
self.assertTrue(plugin.install_called)
def test_plugin_uninstall(self):
plugin = self.plugin_manager.plugins["test1"]
plugin.uninstall()
self.assertTrue(plugin.uninstall_called)
def setUp(self):
self.app = Flask(__name__)
self.app.config['TESTING'] = True
self.plugin_manager = PluginManager()
self.plugin_manager.init_app(self.app)
# Call load_plugins to reload them without calling setup_plugins()
self.plugin_manager.load_plugins()
class PluginManagerOtherDirectoryTests(unittest.TestCase):
def setUp(self):
self.app = Flask(__name__)
self.app.config['TESTING'] = True
self.plugin_manager = PluginManager()
def test_wrong_plugin(self):
# should raise an exception because the plugin in the "plugs" folder
# have set the __plugin__ variable not correctly.
with self.assertRaises(PluginError):
self.plugin_manager.init_app(self.app, plugin_folder="plugs")
class PluginManagerEnableDisableTests(unittest.TestCase):
def setUp(self):
self.app = Flask(__name__)
self.app.config['TESTING'] = True
self.plugin_manager = PluginManager()
self.plugin_manager.init_app(self.app)
def test_disable_plugins(self):
plugin = self.plugin_manager.plugins["test1"]
self.assertTrue(plugin.enabled)
self.assertEquals(self.plugin_manager.disable_plugins([plugin]), 1)
self.assertTrue(os.path.exists(os.path.join(plugin.path, "DISABLED")))
def test_enable_plugins(self):
plugin = self.plugin_manager.all_plugins["test1"]
self.assertFalse(plugin.enabled)
self.plugin_manager.enable_plugins([plugin])
self.assertTrue(plugin.enabled)
def setUp(self):
self.app = Flask(__name__)
self.app.config['TESTING'] = True
self.plugin_manager = PluginManager()
def test_class_init(self):
plugin_manager = PluginManager(self.app)
self.assertIsInstance(plugin_manager, PluginManager)
def test_init_app(self):
plugin_manager = PluginManager()
plugin_manager.init_app(self.app)
self.assertIsInstance(plugin_manager, PluginManager)
from spotipo_plugins import PluginManager
plugin_manager = PluginManager()
def configure(app):
plugin_manager.init_app(app, plugin_folder="modules")