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')
