def test_handle_script_query_with_params(self): scriptMock = MagicMock() getAttrMock = MagicMock(return_value=scriptMock) subject = QueryHandler(QueryParser('script arg1:a arg2:b')) with patch('query_handler.getattr', getAttrMock): response = subject.handle_query(self.aliases) scriptMock.assert_called_with(arg1='a', arg2='b')
def test_handle_backup_search_query(self): searchEngineMock = MagicMock(return_value='foo_') with patch('settings.Settings.getSetting', searchEngineMock): subject = QueryHandler(QueryParser('needs_search')) response = subject.handle_query(self.aliases) self.assertEqual(response.status_code, 302) self.assertEqual(response.location, 'foo_needs_search')
def get_route_for_alias(query): handler = QueryHandler(QueryParser(query)) return handler.handle_query(aliases)
def test_handle_alias_query(self): subject = QueryHandler(QueryParser('foo')) response = subject.handle_query(self.aliases) self.assertEqual(response.status_code, 302) self.assertEqual(response.location, 'bar')