from collections import defaultdict from datetime import datetime import mock from searx.engines import currency_convert from searx.testing import SearxTestCase class TestCurrencyConvertEngine(SearxTestCase): def test_request(self): query = b'test_query' dicto = defaultdict(dict) dicto['pageno'] = 1 self.assertFalse(currency_convert.is_accepted(query, dicto)) query = b'convert 10 Pound Sterlings to United States Dollars' self.assertTrue(currency_convert.is_accepted(query, dicto)) params = currency_convert.request(query, dicto) self.assertIn('url', params) self.assertIn('finance.google.com', params['url']) self.assertIn('GBP', params['url']) self.assertIn('USD', params['url']) def test_response(self): dicto = defaultdict(dict) dicto['amount'] = float(10) dicto['from'] = "GBP" dicto['to'] = "USD" dicto['from_name'] = "pound sterling" dicto['to_name'] = "United States dollar" response = mock.Mock(text='a,b,c,d', search_params=dicto) self.assertEqual(currency_convert.response(response), []) body = "0.5 {}".format(dicto['to']) response = mock.Mock(text=body, search_params=dicto) results = currency_convert.response(response) self.assertEqual(type(results), list) self.assertEqual(len(results), 1) self.assertEqual(results[0]['answer'], '10.0 GBP = 5.0 USD, 1 GBP (pound sterling)' + ' = 0.5 USD (United States dollar)') target_url = 'https://finance.google.com/finance?q={}{}'.format( dicto['from'], dicto['to']) self.assertEqual(results[0]['url'], target_url)