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 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 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)
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")
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 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(self.app)
self.event_manager = self.plugin_manager._event_manager
# Database db = SQLAlchemy() # Login login_manager = LoginManager() # Mail mail = Mail() # Caching cache = Cache() # Redis redis_store = Redis() # Debugtoolbar debugtoolbar = DebugToolbarExtension() # Migrations migrate = Migrate() # Themes themes = Themes() # PluginManager plugin_manager = PluginManager() # Babel babel = Babel()
def setUp(self):
self.app = Flask(__name__)
self.app.config['TESTING'] = True
self.plugin_manager = PluginManager()
def test_init_app(self):
plugin_manager = PluginManager()
plugin_manager.init_app(self.app)
self.assertIsInstance(plugin_manager, PluginManager)
from oauthlib.oauth2 import WebApplicationClient
from lib.Authentication import AuthenticationHandler
from lib.Config import Configuration
from lib.DatabaseSchemaChecker import SchemaChecker, DatabaseSchemaError
from lib.LogHandler import AppLogger
from lib.Toolkit import isURL
from lib.User import User
auth_handler = AuthenticationHandler()
logging.setLoggerClass(AppLogger)
login_manager = LoginManager()
plugins = PluginManager()
app = None
token_blacklist = None
socketio = None
oidcClient = None
config = None
ACCESS_EXPIRES = timedelta(minutes=15)
REFRESH_EXPIRES = timedelta(days=30)
def create_app(version, run_path):
global app, token_blacklist, socketio, oidcClient, config