def test_sort(self):
args = [
('sort=a', [{'field': 'a', 'direction': 'desc'}]),
('sort=a:desc', [{'field': 'a', 'direction': 'desc'}]),
('sort=a:asc', [{'field': 'a', 'direction': 'asc'}]),
('sort=asc:a', [{'field': 'asc:a', 'direction': 'desc'}]),
('sort=a:invalid', [{'field': 'a:invalid', 'direction': 'desc'}]),
('sort=a, b,', [
{'field': 'a', 'direction': 'desc'},
{'field': 'b', 'direction': 'desc'},
]),
('sort=a,b:asc,c:desc,d,e:nothing,f:asc', [
{'field': 'a', 'direction': 'desc'},
{'field': 'b', 'direction': 'asc'},
{'field': 'c', 'direction': 'desc'},
{'field': 'd', 'direction': 'desc'},
{'field': 'e:nothing', 'direction': 'desc'},
{'field': 'f', 'direction': 'asc'},
]),
]
def sort_list(el):
return '{}{}'.format(el['field'], el['direction'])
for sort_arg, res in args:
s = Selector(request_obj=FakeRequest(args=sort_arg))
self.assertListEqual(
sorted(s.sort(), key=sort_list),
sorted(res, key=sort_list)
)
def test_sort_only(self):
args = [
('sort=a,b,c', ['a'], ['a']),
('sort=a,b,c', ['a', 'b'], ['a', 'b']),
('sort=a', ['a', 'b'], ['a']),
]
for arg, only, expected in args:
s = Selector(request_obj=FakeRequest(args=arg))
received = [f['field'] for f in s.sort(only=only)]
self.assertListEqual(sorted(received), sorted(expected))
def test_filter_only(self):
args = [
('key1=1&key2=2&key3', ['key1'], ['key1']),
('key1&key2=le:2&key3', ['key1', 'key2'], ['key1', 'key2']),
('key1', ['key1', 'key2'], ['key1']),
]
for arg, only, expected in args:
s = Selector(request_obj=FakeRequest(args=arg))
received = [f['field'] for f in s.filter(only=only)]
print(arg)
self.assertListEqual(sorted(received), sorted(expected))
def test_offset(self):
offsets = {
'offset=0': 0,
'offset=1': 1,
'offset=10': 10,
'offset=12345678': 12345678,
'offset=-1': None,
'offset=a': None,
'offset=': None,
'offset=TEST': None,
}
for offset, res in offsets.items():
s = Selector(request_obj=FakeRequest(args=offset))
self.assertEquals(s.offset(), res)
def test_limit(self):
limits = {
'limit=0': 0,
'limit=1': 1,
'limit=10': 10,
'limit=12345678': 12345678,
'limit=-1': None,
'limit=a': None,
'limit=': None,
'limit=TEST': None,
}
for lim, res in limits.items():
s = Selector(request_obj=FakeRequest(args=lim))
self.assertEquals(s.limit(), res)
def test_filter(self):
filters = [
(
'key_1=1&key_2=%202&key_3=3%20&key_4=%204%20&key_5= eq : 5 ',
[
{'field': 'key_1', 'op': 'eq', 'value': '1'},
{'field': 'key_2', 'op': 'eq', 'value': '2'},
{'field': 'key_3', 'op': 'eq', 'value': '3'},
{'field': 'key_4', 'op': 'eq', 'value': '4'},
{'field': 'key_5', 'op': 'eq', 'value': '5'},
]
),
(
'time=ge:10:30&time=le:14:30',
[
{'field': 'time', 'op': 'le', 'value': '14:30'},
{'field': 'time', 'op': 'ge', 'value': '10:30'},
]
),
(
'a=:a&b=b:b&c=invalid:c',
[
{'field': 'a', 'op': 'eq', 'value': ':a'},
{'field': 'b', 'op': 'eq', 'value': 'b:b'},
{'field': 'c', 'op': 'eq', 'value': 'invalid:c'},
]
),
(
'eq1=eq:eq1&eq2=eq2&le=le:le<=lt:lt>=gt:gt&ge=ge:ge&ne=ne:ne',
[
{'field': 'eq1', 'op': 'eq', 'value': 'eq1'},
{'field': 'eq2', 'op': 'eq', 'value': 'eq2'},
{'field': 'le', 'op': 'le', 'value': 'le'},
{'field': 'lt', 'op': 'lt', 'value': 'lt'},
{'field': 'gt', 'op': 'gt', 'value': 'gt'},
{'field': 'ge', 'op': 'ge', 'value': 'ge'},
{'field': 'ne', 'op': 'ne', 'value': 'ne'},
]
),
(
'a=in:1&b=in:1,&c=in:1,2&d=in:1,2,3&e=in: 1,%20 2 ,%203&f=in:',
[
{'field': 'a', 'op': 'in', 'value': ['1']},
{'field': 'b', 'op': 'in', 'value': ['1']},
{'field': 'c', 'op': 'in', 'value': ['1', '2']},
{'field': 'd', 'op': 'in', 'value': ['1', '2', '3']},
{'field': 'e', 'op': 'in', 'value': ['1', '2', '3']},
{'field': 'f', 'op': 'in', 'value': []},
]
),
]
def sort_list(el):
if isinstance(el['value'], list):
value = ','.join(sorted(el['value']))
else:
value = el['value']
return '{}{}{}'.format(el['field'], el['op'], value)
for filt, res in filters:
s = Selector(request_obj=FakeRequest(args=filt))
self.assertListEqual(
sorted(s.filter(), key=sort_list),
sorted(res, key=sort_list)
)
def test_empty(self):
s = Selector(request_obj=FakeRequest())
self.assertEquals(s.filter(), [])
self.assertEquals(s.sort(), {})
self.assertIsNone(s.limit())