From 772c048d01c7585fd60afca1ce30a1914e6e5b4a Mon Sep 17 00:00:00 2001 From: Marc Abonce Seguin Date: Wed, 28 Feb 2018 22:30:48 -0600 Subject: refactor engine's search language handling Add match_language function in utils to match any user given language code with a list of engine's supported languages. Also add language_aliases dict on each engine to translate standard language codes into the custom codes used by the engine. --- tests/unit/engines/test_google.py | 7 +++++++ 1 file changed, 7 insertions(+) (limited to 'tests/unit/engines/test_google.py') diff --git a/tests/unit/engines/test_google.py b/tests/unit/engines/test_google.py index ecd1ed4d9..33556cc7a 100644 --- a/tests/unit/engines/test_google.py +++ b/tests/unit/engines/test_google.py @@ -15,6 +15,8 @@ class TestGoogleEngine(SearxTestCase): return response def test_request(self): + google.supported_languages = ['en', 'fr', 'zh-CN'] + query = 'test_query' dicto = defaultdict(dict) dicto['pageno'] = 1 @@ -31,6 +33,11 @@ class TestGoogleEngine(SearxTestCase): self.assertIn('google.co', params['url']) self.assertIn('en', params['headers']['Accept-Language']) + dicto['language'] = 'zh' + params = google.request(query, dicto) + self.assertIn('google.com', params['url']) + self.assertIn('zh-CN', params['headers']['Accept-Language']) + def test_response(self): self.assertRaises(AttributeError, google.response, None) self.assertRaises(AttributeError, google.response, []) -- cgit v1.2.3