summaryrefslogtreecommitdiff
path: root/tests/unit/engines/test_wikipedia.py
diff options
context:
space:
mode:
Diffstat (limited to 'tests/unit/engines/test_wikipedia.py')
-rw-r--r--tests/unit/engines/test_wikipedia.py12
1 files changed, 10 insertions, 2 deletions
diff --git a/tests/unit/engines/test_wikipedia.py b/tests/unit/engines/test_wikipedia.py
index 21d0225ed..316b12bc5 100644
--- a/tests/unit/engines/test_wikipedia.py
+++ b/tests/unit/engines/test_wikipedia.py
@@ -8,7 +8,8 @@ from searx.testing import SearxTestCase
class TestWikipediaEngine(SearxTestCase):
def test_request(self):
- wikipedia.supported_languages = ['fr', 'en']
+ wikipedia.supported_languages = ['fr', 'en', 'no']
+ wikipedia.language_aliases = {'nb': 'no'}
query = 'test_query'
dicto = defaultdict(dict)
@@ -25,10 +26,17 @@ class TestWikipediaEngine(SearxTestCase):
self.assertIn('Test_Query', params['url'])
self.assertNotIn('test_query', params['url'])
- dicto['language'] = 'xx'
+ dicto['language'] = 'nb'
+ params = wikipedia.request(query, dicto)
+ self.assertIn('no.wikipedia.org', params['url'])
+ dicto['language'] = 'all'
params = wikipedia.request(query, dicto)
self.assertIn('en', params['url'])
+ dicto['language'] = 'xx'
+ params = wikipedia.request(query, dicto)
+ self.assertIn('en.wikipedia.org', params['url'])
+
def test_response(self):
dicto = defaultdict(dict)
dicto['language'] = 'fr'