import os import logging try: import unittest.mock as mock except ImportError: import mock from senpy.extensions import Senpy from flask import Flask from flask.ext.testing import TestCase class ExtensionsTest(TestCase): def create_app(self): self.app = Flask("test_extensions") self.dir = os.path.join(os.path.dirname(__file__), "..") self.senpy = Senpy(plugin_folder=self.dir) self.senpy.init_app(self.app) self.senpy.activate_plugin("Dummy", sync=True) return self.app def test_init(self): """ Initialising the app with the extension. """ assert hasattr(self.app, "senpy") tapp = Flask("temp app") self.senpy.init_app(tapp) assert hasattr(tapp, "senpy") def test_discovery(self): """ Discovery of plugins in given folders. """ # noinspection PyProtectedMember assert self.dir in self.senpy._search_folders print self.senpy.plugins assert "Dummy" in self.senpy.plugins def test_enabling(self): """ Enabling a plugin """ self.senpy.activate_all(sync=True) assert len(self.senpy.plugins) == 2 assert self.senpy.plugins["Sleep"].is_activated def test_disabling(self): """ Disabling a plugin """ self.senpy.deactivate_all(sync=True) assert self.senpy.plugins["Dummy"].is_activated == False assert self.senpy.plugins["Sleep"].is_activated == False def test_default(self): """ Default plugin should be set """ assert self.senpy.default_plugin assert self.senpy.default_plugin == "Dummy" def test_analyse(self): """ Using a plugin """ with mock.patch.object(self.senpy.plugins["Dummy"], "analyse") as mocked: self.senpy.analyse(algorithm="Dummy", input="tupni", output="tuptuo") self.senpy.analyse(input="tupni", output="tuptuo") mocked.assert_any_call(input="tupni", output="tuptuo", algorithm="Dummy") mocked.assert_any_call(input="tupni", output="tuptuo") for plug in self.senpy.plugins: self.senpy.deactivate_plugin(plug, sync=True) resp = self.senpy.analyse(input="tupni") logging.debug("Response: {}".format(resp)) assert resp["status"] == 400 def test_filtering(self): """ Filtering plugins """ assert len(self.senpy.filter_plugins(name="Dummy")) > 0 assert not len(self.senpy.filter_plugins(name="notdummy")) assert self.senpy.filter_plugins(name="Dummy", is_activated=True) self.senpy.deactivate_plugin("Dummy", sync=True) assert not len(self.senpy.filter_plugins(name="Dummy", is_activated=True))