@@ -72,31 +72,26 @@ class TestResolveBangDefinition(SearxTestCase):
72class TestGetBangDefinitionAndAutocomplete(SearxTestCase):
72class TestGetBangDefinitionAndAutocomplete(SearxTestCase):
74 def test_found(self):
74 def test_found(self):
76 bang_definition, new_autocomplete = get_bang_definition_and_autocomplete('exam', external_bangs_db=TEST_DB)
75 bang_definition, new_autocomplete = get_bang_definition_and_autocomplete('exam', external_bangs_db=TEST_DB)
77 self.assertEqual(bang_definition, TEST_DB['trie']['exam']['*'])
76 self.assertEqual(bang_definition, TEST_DB['trie']['exam']['*'])
78 self.assertEqual(new_autocomplete, ['example'])
77 self.assertEqual(new_autocomplete, ['example'])
80 def test_found_optimized(self):
79 def test_found_optimized(self):
82 bang_definition, new_autocomplete = get_bang_definition_and_autocomplete('example', external_bangs_db=TEST_DB)
80 bang_definition, new_autocomplete = get_bang_definition_and_autocomplete('example', external_bangs_db=TEST_DB)
83 self.assertEqual(bang_definition, TEST_DB['trie']['exam']['ple'])
81 self.assertEqual(bang_definition, TEST_DB['trie']['exam']['ple'])
84 self.assertEqual(new_autocomplete, [])
82 self.assertEqual(new_autocomplete, [])
86 def test_partial(self):
84 def test_partial(self):
88 bang_definition, new_autocomplete = get_bang_definition_and_autocomplete('examp', external_bangs_db=TEST_DB)
85 bang_definition, new_autocomplete = get_bang_definition_and_autocomplete('examp', external_bangs_db=TEST_DB)
89 self.assertEqual(bang_definition, None)
86 self.assertEqual(bang_definition, None)
90 self.assertEqual(new_autocomplete, ['example'])
87 self.assertEqual(new_autocomplete, ['example'])
92 def test_partial2(self):
89 def test_partial2(self):
94 bang_definition, new_autocomplete = get_bang_definition_and_autocomplete('sea', external_bangs_db=TEST_DB)
90 bang_definition, new_autocomplete = get_bang_definition_and_autocomplete('sea', external_bangs_db=TEST_DB)
95 self.assertEqual(bang_definition, TEST_DB['trie']['sea']['*'])
91 self.assertEqual(bang_definition, TEST_DB['trie']['sea']['*'])
96 self.assertEqual(new_autocomplete, ['search', 'searching', 'seascapes', 'season'])
92 self.assertEqual(new_autocomplete, ['search', 'searching', 'seascapes', 'season'])
98 def test_error(self):
94 def test_error(self):
100 bang_definition, new_autocomplete = get_bang_definition_and_autocomplete('error', external_bangs_db=TEST_DB)
95 bang_definition, new_autocomplete = get_bang_definition_and_autocomplete('error', external_bangs_db=TEST_DB)
101 self.assertEqual(bang_definition, None)
96 self.assertEqual(bang_definition, None)
102 self.assertEqual(new_autocomplete, [])
97 self.assertEqual(new_autocomplete, [])
@@ -114,7 +109,6 @@ class TestExternalBangJson(SearxTestCase):
114 self.assertEqual(result, None)
109 self.assertEqual(result, None)
116 def test_get_bang_url(self):
111 def test_get_bang_url(self):
118 url = get_bang_url(SearchQuery('test', engineref_list=[], external_bang='example'), external_bangs_db=TEST_DB)
112 url = get_bang_url(SearchQuery('test', engineref_list=[], external_bang='example'), external_bangs_db=TEST_DB)
119 self.assertEqual(url, 'https://example.com/test')
113 self.assertEqual(url, 'https://example.com/test')