Fix parsing of quoted words

This commit is contained in:
Kovid Goyal
2022-04-12 19:31:56 +05:30
parent 8f92c594f2
commit 93a7b220c9
2 changed files with 10 additions and 5 deletions

View File

@@ -8,20 +8,23 @@ from . import BaseTest
class TestSQP(BaseTest):
def test_search_query_parser(self):
from kitty.search_query_parser import search
from kitty.search_query_parser import search, ParseException
locations = 'id'
universal_set = {1, 2, 3, 4, 5}
def get_matches(location, query, candidates):
return {x for x in candidates if query == str(x)}
def t(q, expected):
def t(q, expected=set()):
actual = search(q, locations, universal_set, get_matches)
self.ae(actual, expected)
t('id:1', {1})
t('id:"1"', {1})
t('id:1 and id:1', {1})
t('id:1 or id:2', {1, 2})
t('id:1 and id:2', set())
t('id:1 and id:2')
t('not id:1', universal_set - {1})
t('(id:1 or id:2) and id:1', {1})
self.assertRaises(ParseException, t, '1')
self.assertRaises(ParseException, t, '"id:1"')