diff options
| author | Alexandre Flament <alex@al-f.net> | 2022-01-12 22:00:18 +0100 |
|---|---|---|
| committer | GitHub <noreply@github.com> | 2022-01-12 22:00:18 +0100 |
| commit | 687bdef4108f9d76e5b4dc3d847479ba257626c8 (patch) | |
| tree | 963f04ea17af3ba3abad0e60ccb8dc5dbaff41eb /tests/unit | |
| parent | 94736d7465ddf2ac2628a61039aa58106559d618 (diff) | |
| parent | a7c3c352d836e5c75ae4b6d84b2566f3d21c7b13 (diff) | |
Merge pull request #740 from return42/fix-bang
[fix] get_bang_url: handle ambiguous !!bangs without error
Diffstat (limited to 'tests/unit')
| -rw-r--r-- | tests/unit/test_external_bangs.py | 20 |
1 files changed, 13 insertions, 7 deletions
diff --git a/tests/unit/test_external_bangs.py b/tests/unit/test_external_bangs.py index 698ce36c6..794edf159 100644 --- a/tests/unit/test_external_bangs.py +++ b/tests/unit/test_external_bangs.py @@ -1,4 +1,10 @@ -from searx.external_bang import get_node, resolve_bang_definition, get_bang_url, get_bang_definition_and_autocomplete +from searx.external_bang import ( + get_node, + resolve_bang_definition, + get_bang_url, + get_bang_definition_and_autocomplete, + LEAF_KEY, +) from searx.search import SearchQuery, EngineRef from tests import SearxTestCase @@ -7,12 +13,12 @@ TEST_DB = { 'trie': { 'exam': { 'ple': '//example.com/' + chr(2) + chr(1) + '0', - '*': '//wikipedia.org/wiki/' + chr(2) + chr(1) + '0', + LEAF_KEY: '//wikipedia.org/wiki/' + chr(2) + chr(1) + '0', }, 'sea': { - '*': 'sea' + chr(2) + chr(1) + '0', + LEAF_KEY: 'sea' + chr(2) + chr(1) + '0', 'rch': { - '*': 'search' + chr(2) + chr(1) + '0', + LEAF_KEY: 'search' + chr(2) + chr(1) + '0', 'ing': 'searching' + chr(2) + chr(1) + '0', }, 's': { @@ -31,7 +37,7 @@ class TestGetNode(SearxTestCase): 'trie': { 'exam': { 'ple': 'test', - '*': 'not used', + LEAF_KEY: 'not used', } } } @@ -71,7 +77,7 @@ class TestResolveBangDefinition(SearxTestCase): class TestGetBangDefinitionAndAutocomplete(SearxTestCase): def test_found(self): bang_definition, new_autocomplete = get_bang_definition_and_autocomplete('exam', external_bangs_db=TEST_DB) - self.assertEqual(bang_definition, TEST_DB['trie']['exam']['*']) + self.assertEqual(bang_definition, TEST_DB['trie']['exam'][LEAF_KEY]) self.assertEqual(new_autocomplete, ['example']) def test_found_optimized(self): @@ -86,7 +92,7 @@ class TestGetBangDefinitionAndAutocomplete(SearxTestCase): def test_partial2(self): bang_definition, new_autocomplete = get_bang_definition_and_autocomplete('sea', external_bangs_db=TEST_DB) - self.assertEqual(bang_definition, TEST_DB['trie']['sea']['*']) + self.assertEqual(bang_definition, TEST_DB['trie']['sea'][LEAF_KEY]) self.assertEqual(new_autocomplete, ['search', 'searching', 'seascapes', 'season']) def test_error(self): |