def test_add_plugin_views_to_compare_view_missing_key(self): cr = CompareRoutes(AppMock(), None) plugin_views = [('plugin_1', b'<plugin view 1>'), ('plugin_2', b'<plugin view 2>')] compare_view = 'xxxxxyyyyy' result = cr._add_plugin_views_to_compare_view(compare_view, plugin_views) assert result == compare_view
def test_add_to_compare_basket(self): with self.frontend.app.test_request_context(): assert 'uids_for_comparison' not in session CompareRoutes.add_to_compare_basket(self.frontend, 'test') assert 'uids_for_comparison' in session assert isinstance(session['uids_for_comparison'], dict) assert 'test' in session['uids_for_comparison']
def test_insert_plugin_into_view_at_index(): view = '------><------' plugin = 'o' index = view.find('<') assert CompareRoutes._insert_plugin_into_view_at_index(plugin, view, 0) == 'o------><------' assert CompareRoutes._insert_plugin_into_view_at_index(plugin, view, index) == '------>o<------' assert CompareRoutes._insert_plugin_into_view_at_index(plugin, view, len(view) + 10) == '------><------o' assert CompareRoutes._insert_plugin_into_view_at_index(plugin, view, -10) == view
def test_get_compare_plugin_views(): cr = CompareRoutes(AppMock(), None) compare_result = {'plugins': {}} result = cr._get_compare_plugin_views(compare_result) assert result == ([], []) compare_result = {'plugins': {'plugin_1': None, 'plugin_2': None}} plugin_views, plugins_without_view = cr._get_compare_plugin_views(compare_result) assert plugin_views == [('plugin_1', b'<plugin 1 view>')] assert plugins_without_view == ['plugin_2']
def test_remove_all_from_compare_basket(self): with self.frontend.app.test_request_context(): session['uids_for_comparison'] = [TEST_FW.uid, TEST_FW_2.uid] session.modified = True assert 'uids_for_comparison' in session assert TEST_FW.uid in session['uids_for_comparison'] assert TEST_FW_2.uid in session['uids_for_comparison'] CompareRoutes.remove_all_from_compare_basket(self.frontend, 'some_uid') assert TEST_FW.uid not in session['uids_for_comparison'] assert TEST_FW_2.uid not in session['uids_for_comparison']
def test_remove_from_compare_basket(self): with self.frontend.app.test_request_context(): session['uids_for_comparison'] = [ TEST_FW.get_uid(), TEST_FW_2.get_uid() ] session.modified = True assert 'uids_for_comparison' in session assert TEST_FW.get_uid() in session['uids_for_comparison'] assert TEST_FW_2.get_uid() in session['uids_for_comparison'] CompareRoutes._remove_from_compare_basket('some_uid', TEST_FW.get_uid()) assert TEST_FW.get_uid() not in session['uids_for_comparison'] assert TEST_FW_2.get_uid() in session['uids_for_comparison']
def test_add_plugin_views_to_compare_view(): cr = CompareRoutes(AppMock(), None) plugin_views = [ ('plugin_1', b'<plugin view 1>'), ('plugin_2', b'<plugin view 2>') ] key = '{# individual plugin views #}' compare_view = 'xxxxx{}yyyyy'.format(key) key_index = compare_view.find(key) result = cr._add_plugin_views_to_compare_view(compare_view, plugin_views) for plugin, view in plugin_views: assert 'elif plugin == \'{}\''.format(plugin) in result assert view.decode() in result assert key_index + len(key) <= result.find(view.decode()) < result.find('yyyyy')
def test_remove_from_compare_basket(self): with self.frontend.app.test_request_context(): CompareRoutes.add_to_compare_basket(self.frontend, TEST_FW.uid) CompareRoutes.add_to_compare_basket(self.frontend, TEST_FW_2.uid) assert 'uids_for_comparison' in session assert TEST_FW.uid in session['uids_for_comparison'] assert TEST_FW_2.uid in session['uids_for_comparison'] CompareRoutes.remove_from_compare_basket(self.frontend, 'some_uid', TEST_FW.uid) assert TEST_FW.uid not in session['uids_for_comparison'] assert TEST_FW_2.uid in session['uids_for_comparison']
def _setup_app(self): self.app = Flask(__name__) self.app.config.from_object(__name__) Flask.secret_key = os.urandom(24) RestBase(app=self.app, config=self.config) AjaxRoutes(self.app, self.config) AnalysisRoutes(self.app, self.config) CompareRoutes(self.app, self.config) DatabaseRoutes(self.app, self.config) IORoutes(self.app, self.config) MiscellaneousRoutes(self.app, self.config) StatisticRoutes(self.app, self.config) FilterClass(self.app, self.program_version, self.config)
def _setup_app(self): self.app = Flask(__name__) self.app.config.from_object(__name__) self.app.config['SECRET_KEY'] = os.urandom(24) add_config_from_configparser_to_app(self.app, self.config) self.user_db, self.user_datastore = add_flask_security_to_app(self.app) AjaxRoutes(self.app, self.config) AnalysisRoutes(self.app, self.config) CompareRoutes(self.app, self.config) DatabaseRoutes(self.app, self.config) IORoutes(self.app, self.config) MiscellaneousRoutes(self.app, self.config) StatisticRoutes(self.app, self.config) UserManagementRoutes(self.app, self.config, user_db=self.user_db, user_db_interface=self.user_datastore) rest_base = RestBase(app=self.app, config=self.config) PluginRoutes(self.app, self.config, api=rest_base.api) FilterClass(self.app, self.program_version, self.config)
def test_get_compare_view(): cr = CompareRoutes(AppMock(), None) result = cr._get_compare_view([]) assert '>General information<' in result assert '--- plugin results ---' in result