def test_range_filter(self):
with self.patch_send_request() as send_request:
send_request.return_value = '{}'
q = self.searcher.search()
qf = QueryFilter()
qf.add_filter(RangeFilter('price', 'price_unit', gather_stats=True,
_local_params=LocalParams(cache=False),
type=Float))
params = {
'price__gte': '100',
'price__lte': ['nan', '200'],
'price': '66',
}
q = qf.apply(q, params)
raw_query = force_unicode(q)
self.assertIn('fq={!cache=false tag=price}'
'price_unit:[100.0 TO *]', raw_query)
self.assertIn('fq={!cache=false tag=price}'
'price_unit:[* TO 200.0]', raw_query)
self.assertNotIn('fq={!cache=false tag=price}'
'price_unit:"66.0"', raw_query)
results = q.results
with self.patch_send_request() as send_request:
send_request.return_value = '''
{
"response": {
"numFound": 800,
"start":0,
"docs":[]
},
"stats": {
"stats_fields": {
"price_unit": {
"min": 3.5,
"max": 892.0,
"count": 1882931,
"missing": 556686,
"sum": 5.677964302447648E13,
"sumOfSquares": 2.452218850256837E26,
"mean": 3.0154924967763808E7,
"stddev": 1.1411980204045008E10
}
}
}
}'''
qf.process_results(results)
price_filter = qf.get_filter('price')
self.assertEqual(price_filter.from_value, 100)
self.assertEqual(price_filter.to_value, 200)
self.assertEqual(price_filter.min, 3.5)
self.assertEqual(price_filter.max, 892.0)
def test_range_filter(self):
with self.patch_send_request() as send_request:
send_request.return_value = "{}"
q = self.searcher.search()
qf = QueryFilter()
qf.add_filter(
RangeFilter(
"price", "price_unit", gather_stats=True, _local_params=LocalParams(cache=False), type=Float
)
)
params = {"price__gte": "100", "price__lte": ["nan", "200"], "price": "66"}
q = qf.apply(q, params)
raw_query = force_unicode(q)
self.assertIn("fq={!cache=false tag=price}" "price_unit:[100.0 TO *]", raw_query)
self.assertIn("fq={!cache=false tag=price}" "price_unit:[* TO 200.0]", raw_query)
self.assertNotIn("fq={!cache=false tag=price}" 'price_unit:"66.0"', raw_query)
results = q.results
with self.patch_send_request() as send_request:
send_request.return_value = """
{
"response": {
"numFound": 800,
"start":0,
"docs":[]
},
"stats": {
"stats_fields": {
"price_unit": {
"min": 3.5,
"max": 892.0,
"count": 1882931,
"missing": 556686,
"sum": 5.677964302447648E13,
"sumOfSquares": 2.452218850256837E26,
"mean": 3.0154924967763808E7,
"stddev": 1.1411980204045008E10
}
}
}
}"""
qf.process_results(results)
price_filter = qf.get_filter("price")
self.assertEqual(price_filter.from_value, 100)
self.assertEqual(price_filter.to_value, 200)
self.assertEqual(price_filter.min, 3.5)
self.assertEqual(price_filter.max, 892.0)
def test_filter(self):
q = self.searcher.search()
qf = QueryFilter()
qf.add_filter(Filter("country", _local_params=LocalParams(tag="cc")))
params = {"country": ["us", "ru"]}
q = qf.apply(q, params)
raw_query = force_unicode(q)
self.assertIn('fq={!tag=cc,country}(country:"us" OR country:"ru")', raw_query)
def test_filter(self):
q = self.searcher.search()
qf = QueryFilter()
qf.add_filter(Filter('country',
_local_params=LocalParams(tag='cc')))
params = {
'country': ['us', 'ru'],
}
q = qf.apply(q, params)
raw_query = force_unicode(q)
self.assertIn('fq={!tag=cc,country}(country:"us" OR country:"ru")', raw_query)
def test_ordering_filter(self):
with self.patch_send_request() as send_request:
send_request.return_value = '''
{
"facet_counts": {
"facet_queries": {
"date_created__today": 28,
"date_created__week_ago": 105,
"dist__d5": 0,
"dist__d10": 12,
"dist__d20": 40
},
"facet_fields": {},
"facet_dates":{},
"facet_ranges":{}
}
}'''
q = self.searcher.search()
qf = QueryFilter()
qf.add_ordering(
OrderingFilter(
'sort',
OrderingValue('-score', '-score'),
OrderingValue('price', 'price'),
OrderingValue('-price', '-price')))
params = {
'sort': '-price',
}
q = qf.apply(q, params)
raw_query = force_unicode(q)
self.assertIn('sort=price desc', raw_query)
ordering_filter = qf.ordering_filter
self.assertEqual(ordering_filter.get_value('-price').selected, True)
self.assertEqual(ordering_filter.get_value('-price').direction, OrderingValue.DESC)
def test_ordering_filter(self):
with self.patch_send_request() as send_request:
send_request.return_value = """
{
"facet_counts": {
"facet_queries": {
"date_created__today": 28,
"date_created__week_ago": 105,
"dist__d5": 0,
"dist__d10": 12,
"dist__d20": 40
},
"facet_fields": {},
"facet_dates":{},
"facet_ranges":{}
}
}"""
q = self.searcher.search()
qf = QueryFilter()
qf.add_ordering(
OrderingFilter(
"sort",
OrderingValue("-score", "-score"),
OrderingValue("price", "price"),
OrderingValue("-price", "-price"),
)
)
params = {"sort": "-price"}
q = qf.apply(q, params)
raw_query = force_unicode(q)
self.assertIn("sort=price desc", raw_query)
ordering_filter = qf.ordering_filter
self.assertEqual(ordering_filter.get_value("-price").selected, True)
self.assertEqual(ordering_filter.get_value("-price").direction, OrderingValue.DESC)
def test_X(self):
self.assertEqual(force_unicode(X(status=0)), "(AND: ('status', 0))")
self.assertEqual(
force_unicode(X(status=0) & X(company_status__in=[0, 6])),
"(AND: ('status', 0), ('company_status__in', [0, 6]))")
self.assertEqual(force_unicode(X(status=0) | X(company_status=0)),
"(OR: ('status', 0), ('company_status', 0))")
self.assertEqual(force_unicode(X(with_photo=True)),
"(AND: ('with_photo', True))")
self.assertEqual(
force_unicode(
X(date_created__gt=datetime(2012, 5, 17, 14, 35, 41, 794880))),
"(AND: ('date_created__gt', datetime.datetime(2012, 5, 17, 14, 35, 41, 794880)))"
)
self.assertEqual(force_unicode(X(price__lt=1000)),
"(AND: ('price__lt', 1000))")
self.assertEqual(force_unicode(X(price__gte=100) & X(price__lte=1000)),
"(AND: ('price__gte', 100), ('price__lte', 1000))")
self.assertEqual(force_unicode(X(price__between=[500, 1000])),
"(AND: ('price__between', [500, 1000]))")
self.assertEqual(force_unicode(X(price__range=[2, 10])),
"(AND: ('price__range', [2, 10]))")
self.assertEqual(
force_unicode(
X(category__in=[1, 2, 3, 4, 5])
& (X(status=0) | X(status=5)
| X(status=1) & X(company_status=6))),
"(AND: ('category__in', [1, 2, 3, 4, 5]), (OR: ('status', 0), ('status', 5), (AND: ('status', 1), ('company_status', 6))))"
)
self.assertEqual(force_unicode(~X(status=1)),
"(AND: (NOT (AND: ('status', 1))))")
self.assertEqual(force_unicode(~X(status__in=[1, 2, 3])),
"(AND: (NOT (AND: ('status__in', [1, 2, 3]))))")
def test_local_params(self):
self.assertEqual(force_unicode(LocalParams({'cache': False})),
'{!cache=false}')
self.assertEqual(
force_unicode(LocalParams(LocalParams({'cache': False}))),
'{!cache=false}')
self.assertEqual(force_unicode(LocalParams({'ex': 'tag'}, key='tag')),
'{!ex=tag key=tag}')
self.assertEqual(
force_unicode(
LocalParams({
'ex': 'tag',
'key': 'tag'
}, key='category')), '{!ex=tag key=category}')
self.assertEqual(force_unicode(LocalParams(ex=['f1', 'f2'])),
'{!ex=f1,f2}')
self.assertEqual(force_unicode(LocalParams('frange', l=0, u=5)),
'{!frange l=0 u=5}')
self.assertEqual(
force_unicode(LocalParams(['geofilt', ('d', 10), ('key', 'd10')])),
'{!geofilt d=10 key=d10}')
self.assertEqual(
force_unicode(
LocalParams({
'type': 'join',
'from': 'id',
'to': 'manu_id'
})), '{!join from=id to=manu_id}')
self.assertEqual(force_unicode(LocalParams()), '')
self.assertEqual(force_unicode(LocalParams(None)), '')
self.assertEqual(force_unicode(LocalParams('dismax', v='OR test')),
"""{!dismax v='or test'}""")
self.assertEqual(force_unicode(LocalParams('dismax', v='"test"')),
"""{!dismax v='\\"test\\"'}""")
self.assertEqual(force_unicode(LocalParams('dismax', v='test\'')),
"""{!dismax v='test\\\\\''}""")
self.assertRaises(ValueError, LocalParams, '{dismax}', v='test')
self.assertRaises(ValueError, LocalParams, ['dismax', ('!v', 'test')])
self.assertEqual(
force_unicode(
LocalParams('dismax',
qf='name',
v=X(SafeUnicode('"nokia lumia"'))
| X(SafeUnicode('"nokia n900"')))),
"""{!dismax qf=name v='(\\"nokia lumia\\" OR \\"nokia n900\\")'}"""
)
lp = LocalParams('dismax', bf=func.linear('rank', 100, 0), v='$q1')
lp.update(LocalParams(qf='name^10 description'))
lp.add('pf', 'name')
lp.add('ps', 2)
self.assertIn('type', lp)
self.assertIn('v', lp)
self.assertFalse('q' in lp)
self.assertEqual(lp['type'], 'dismax')
self.assertEqual(
force_unicode(lp),
"{!dismax bf='linear(rank,100,0)' v=$q1 qf='name^10 description' pf=name ps=2}"
)
def test_pivot_filter(self):
with self.patch_send_request() as send_request:
send_request.return_value = '''
{
"facet_counts": {
"facet_pivot": {
"manu": [
{
"field": "manufacturer",
"value": "samsung",
"count": 100,
"pivot": [
{
"field": "model",
"value": "note",
"count": 66,
"pivot": [
{
"field": "discount",
"value": true,
"count": 11
},
{
"field": "discount",
"value": false,
"count": 55
}
]
},
{
"field": "model",
"value": "S4",
"count": 44,
"pivot": [
{
"field": "discount",
"value": true,
"count": 11
},
{
"field": "discount",
"value": false,
"count": 33
}
]
}
]
},
{
"field": "manufacturer",
"value": "nokia",
"count": 1,
"pivot": [
{
"field": "model",
"value": "n900",
"count": 1,
"pivot": [
{
"field": "discount",
"value": false,
"count": 1
}
]
}
]
},
{
"field": "manufacturer",
"value": "lenovo",
"count": 5,
"pivot": [
{
"field": "model",
"value": "p770",
"count": 4
}
]
}
]
},
"facet_fields": {},
"facet_queries": {},
"facet_dates": {},
"facet_ranges": {}
}
}'''
obj_mapper = Mock(wraps=_obj_mapper)
q = self.searcher.search()
qf = QueryFilter()
qf.add_filter(
PivotFilter(
'manu',
FacetPivotFilter('manufacturer', mincount=1, ensure_selected_values=True),
FacetPivotFilter('model', limit=5, instance_mapper=obj_mapper,
get_title=cap_filter_value),
FacetPivotFilter('discount', missing=True, type=Boolean)
)
)
params = {
'manu': ['samsung:note', 'nokia:n900:false', 'nokia:n900:null', 'noname:', 10],
'manu__gte': '100',
}
q = qf.apply(q, params)
raw_query = force_unicode(q)
self.assertIn('facet=true', raw_query)
self.assertIn('facet.pivot={!key=manu ex=manu}manufacturer,model,discount', raw_query)
self.assertIn('f.manufacturer.facet.mincount=1', raw_query)
self.assertIn('f.model.facet.limit=5', raw_query)
self.assertIn('fq={!tag=manu}'
'((manufacturer:"samsung" AND model:"note") '
'OR (manufacturer:"nokia" AND model:"n900" AND discount:"false") '
'OR (manufacturer:"nokia" AND model:"n900" AND (*:* NOT discount:[* TO *])) '
'OR (manufacturer:"noname" AND model:"") '
'OR manufacturer:"10")',
raw_query)
qf.process_results(q.results)
manu_filter = qf.get_filter('manu')
self.assertEqual(len(manu_filter.all_values), 5)
self.assertEqual(len(manu_filter.selected_values), 4)
self.assertEqual(len(manu_filter.values), 1)
self.assertEqual(manu_filter.all_values[0].filter_name, 'manu')
self.assertEqual(manu_filter.all_values[0].value, 'samsung')
self.assertEqual(manu_filter.all_values[0].title, 'samsung')
self.assertEqual(manu_filter.all_values[0].filter_value, 'samsung')
self.assertEqual(manu_filter.all_values[0].count, 100)
self.assertEqual(manu_filter.all_values[0].selected, True)
self.assertEqual(manu_filter.all_values[0].pivot.all_values[0].filter_name, 'manu')
self.assertEqual(manu_filter.all_values[0].pivot.all_values[0].value, 'note')
self.assertEqual(manu_filter.all_values[0].pivot.all_values[0].title, 'Note')
self.assertEqual(manu_filter.all_values[0].pivot.all_values[0].filter_value, 'samsung:note')
self.assertEqual(manu_filter.all_values[0].pivot.all_values[0].instance, ('note', 'note note'))
self.assertEqual(manu_filter.all_values[0].pivot.all_values[0].count, 66)
self.assertEqual(manu_filter.all_values[0].pivot.all_values[0].count_plus, '66')
self.assertEqual(manu_filter.all_values[0].pivot.all_values[0].selected, True)
self.assertEqual(len(manu_filter.all_values[0].pivot.all_values[0].pivot.all_values), 2)
self.assertEqual(manu_filter.all_values[0].pivot.all_values[0].pivot.all_values[0].value, True)
self.assertEqual(manu_filter.all_values[0].pivot.all_values[0].pivot.all_values[0].title, 'True')
self.assertEqual(manu_filter.all_values[0].pivot.all_values[0].pivot.all_values[0].filter_value, 'samsung:note:true')
self.assertEqual(manu_filter.all_values[0].pivot.all_values[0].pivot.all_values[0].count, 11)
self.assertEqual(manu_filter.all_values[0].pivot.all_values[0].pivot.all_values[0].selected, False)
self.assertEqual(manu_filter.all_values[0].pivot.all_values[0].pivot.all_values[0].pivot, None)
self.assertEqual(manu_filter.all_values[0].pivot.all_values[0].pivot.all_values[1].value, False)
self.assertEqual(manu_filter.all_values[0].pivot.all_values[0].pivot.all_values[1].title, 'False')
self.assertEqual(manu_filter.all_values[0].pivot.all_values[0].pivot.all_values[1].filter_value, 'samsung:note:false')
self.assertEqual(manu_filter.all_values[0].pivot.all_values[0].pivot.all_values[1].count, 55)
self.assertEqual(manu_filter.all_values[0].pivot.all_values[0].pivot.all_values[1].selected, False)
self.assertEqual(manu_filter.all_values[0].pivot.all_values[0].pivot.all_values[1].pivot, None)
self.assertEqual(manu_filter.all_values[0].pivot.all_values[1].filter_name, 'manu')
self.assertEqual(manu_filter.all_values[0].pivot.all_values[1].value, 'S4')
self.assertEqual(manu_filter.all_values[0].pivot.all_values[1].title, 'S4')
self.assertEqual(manu_filter.all_values[0].pivot.all_values[1].filter_value, 'samsung:S4')
self.assertEqual(manu_filter.all_values[0].pivot.all_values[1].instance, ('S4', 'S4 S4'))
self.assertEqual(manu_filter.all_values[0].pivot.all_values[1].count, 44)
self.assertEqual(manu_filter.all_values[0].pivot.all_values[1].count_plus, '+44')
self.assertEqual(manu_filter.all_values[0].pivot.all_values[1].selected, False)
self.assertEqual(len(manu_filter.all_values[0].pivot.all_values[1].pivot.all_values), 2)
self.assertEqual(manu_filter.all_values[0].pivot.all_values[1].pivot.all_values[0].value, True)
self.assertEqual(manu_filter.all_values[0].pivot.all_values[1].pivot.all_values[0].title, 'True')
self.assertEqual(manu_filter.all_values[0].pivot.all_values[1].pivot.all_values[0].filter_value, 'samsung:S4:true')
self.assertEqual(manu_filter.all_values[0].pivot.all_values[1].pivot.all_values[0].count, 11)
self.assertEqual(manu_filter.all_values[0].pivot.all_values[1].pivot.all_values[0].count_plus, '11')
self.assertEqual(manu_filter.all_values[0].pivot.all_values[1].pivot.all_values[0].selected, False)
self.assertEqual(manu_filter.all_values[0].pivot.all_values[1].pivot.all_values[0].pivot, None)
self.assertEqual(manu_filter.all_values[0].pivot.all_values[1].pivot.all_values[1].value, False)
self.assertEqual(manu_filter.all_values[0].pivot.all_values[1].pivot.all_values[1].title, 'False')
self.assertEqual(manu_filter.all_values[0].pivot.all_values[1].pivot.all_values[1].filter_value, 'samsung:S4:false')
self.assertEqual(manu_filter.all_values[0].pivot.all_values[1].pivot.all_values[1].count_plus, '33')
self.assertEqual(manu_filter.all_values[0].pivot.all_values[1].pivot.all_values[1].selected, False)
self.assertEqual(manu_filter.all_values[0].pivot.all_values[1].pivot.all_values[1].pivot, None)
self.assertEqual(manu_filter.all_values[1].value, 'nokia')
self.assertEqual(manu_filter.all_values[1].title, 'nokia')
self.assertEqual(manu_filter.all_values[1].filter_value, 'nokia')
self.assertEqual(manu_filter.all_values[1].count, 1)
self.assertEqual(manu_filter.all_values[1].selected, True)
self.assertEqual(len(manu_filter.all_values[1].pivot.all_values), 1)
self.assertEqual(manu_filter.all_values[1].pivot.all_values[0].value, 'n900')
self.assertEqual(manu_filter.all_values[1].pivot.all_values[0].title, 'N900')
self.assertEqual(manu_filter.all_values[1].pivot.all_values[0].filter_value, 'nokia:n900')
self.assertEqual(manu_filter.all_values[1].pivot.all_values[0].instance, ('n900', 'n900 n900'))
self.assertEqual(manu_filter.all_values[1].pivot.all_values[0].count, 1)
self.assertEqual(manu_filter.all_values[1].pivot.all_values[0].selected, True)
self.assertEqual(len(manu_filter.all_values[1].pivot.all_values[0].pivot.all_values), 1)
self.assertEqual(manu_filter.all_values[1].pivot.all_values[0].pivot.all_values[0].value, False)
self.assertEqual(manu_filter.all_values[1].pivot.all_values[0].pivot.all_values[0].title, 'False')
self.assertEqual(manu_filter.all_values[1].pivot.all_values[0].pivot.all_values[0].filter_value, 'nokia:n900:false')
self.assertEqual(manu_filter.all_values[1].pivot.all_values[0].pivot.all_values[0].count, 1)
self.assertEqual(manu_filter.all_values[1].pivot.all_values[0].pivot.all_values[0].count_plus, '1')
self.assertEqual(manu_filter.all_values[1].pivot.all_values[0].pivot.all_values[0].selected, True)
self.assertEqual(manu_filter.all_values[2].value, 'lenovo')
self.assertEqual(manu_filter.all_values[2].title, 'lenovo')
self.assertEqual(manu_filter.all_values[2].filter_value, 'lenovo')
self.assertEqual(manu_filter.all_values[2].count, 5)
self.assertEqual(manu_filter.all_values[2].selected, False)
self.assertEqual(manu_filter.all_values[3].value, 'noname')
self.assertEqual(manu_filter.all_values[3].title, 'noname')
self.assertEqual(manu_filter.all_values[3].filter_value, 'noname')
self.assertEqual(manu_filter.all_values[3].count, None)
self.assertEqual(manu_filter.all_values[3].count_plus, '')
self.assertEqual(manu_filter.all_values[3].selected, True)
self.assertEqual(manu_filter.all_values[4].value, '10')
self.assertEqual(manu_filter.all_values[4].title, '10')
self.assertEqual(manu_filter.all_values[4].filter_value, '10')
self.assertEqual(manu_filter.all_values[4].count, None)
self.assertEqual(manu_filter.all_values[4].count_plus, '')
self.assertEqual(manu_filter.all_values[4].selected, True)
self.assertEqual(obj_mapper.call_count, 1)
def cap_filter_value(fv):
return force_unicode(fv.value).capitalize()
def test_facet_query_filter(self):
with self.patch_send_request() as send_request:
send_request.return_value = """
{
"facet_counts": {
"facet_queries": {
"date_created__today": 28,
"date_created__week_ago": 105,
"dist__d5": 0,
"dist__d10": 12,
"dist__d20": 40
},
"facet_fields": {},
"facet_dates":{},
"facet_ranges":{}
}
}"""
q = self.searcher.search()
qf = QueryFilter()
qf.add_filter(
FacetQueryFilter(
"date_created",
FacetQueryFilterValue(
"today",
X(date_created__gte="NOW/DAY-1DAY"),
_local_params=LocalParams(ex="test"),
title="Only new",
help_text="Documents one day later",
),
FacetQueryFilterValue("week_ago", X(date_created__gte="NOW/DAY-7DAY"), title="Week ago"),
)
)
qf.add_filter(
FacetQueryFilter(
"dist",
FacetQueryFilterValue("d5", None, _local_params=LocalParams("geofilt", d=5, tag="d5")),
FacetQueryFilterValue("d10", None, _local_params=LocalParams("geofilt", d=10, tag="d10")),
FacetQueryFilterValue("d20", None, _local_params=LocalParams("geofilt", d=20, tag="d20")),
select_multiple=False,
)
)
qf.add_ordering(
OrderingFilter(
"sort",
OrderingValue("-score", "-score"),
OrderingValue("price", "price"),
OrderingValue("-price", "-price"),
)
)
params = {"cat": ["5", "13"], "country": ["us", "ru"], "date_created": "today", "dist": "d10"}
q = qf.apply(q, params)
raw_query = force_unicode(q)
self.assertIn("facet=true", raw_query)
self.assertIn(
"facet.query={!ex=test,date_created key=date_created__today}date_created:[NOW/DAY-1DAY TO *]", raw_query
)
self.assertIn(
"facet.query={!key=date_created__week_ago ex=date_created}date_created:[NOW/DAY-7DAY TO *]", raw_query
)
self.assertIn("facet.query={!geofilt d=5 tag=d5 key=dist__d5 ex=dist}", raw_query)
self.assertIn("facet.query={!geofilt d=10 tag=d10 key=dist__d10 ex=dist}", raw_query)
self.assertIn("facet.query={!geofilt d=20 tag=d20 key=dist__d20 ex=dist}", raw_query)
self.assertIn("fq={!tag=date_created}date_created:[NOW/DAY-1DAY TO *]", raw_query)
self.assertIn("fq={!geofilt d=10 tag=d10,dist}", raw_query)
qf.process_results(q.results)
date_created_filter = qf.get_filter("date_created")
self.assertEqual(date_created_filter.get_value("today").count, 28)
self.assertEqual(date_created_filter.get_value("today").count_plus, "28")
self.assertEqual(date_created_filter.get_value("today").selected, True)
self.assertEqual(date_created_filter.get_value("today").title, "Only new")
self.assertEqual(date_created_filter.get_value("today").opts["help_text"], "Documents one day later")
self.assertEqual(date_created_filter.get_value("week_ago").count, 105)
self.assertEqual(date_created_filter.get_value("week_ago").count_plus, "105")
dist_filter = qf.get_filter("dist")
self.assertEqual(dist_filter.get_value("d5").count, 0)
self.assertEqual(dist_filter.get_value("d5").selected, False)
self.assertEqual(dist_filter.get_value("d10").count, 12)
self.assertEqual(dist_filter.get_value("d10").selected, True)
self.assertEqual(dist_filter.get_value("d20").count, 40)
self.assertEqual(dist_filter.get_value("d20").selected, False)
def test_facet_filter(self):
with self.patch_send_request() as send_request:
send_request.return_value = '''
{
"facet_counts": {
"facet_queries": {},
"facet_fields": {
"cat": [
"100", 500,
"5", 10,
"2", 5,
"1", 2,
null, 4
],
"region": [
"kiev", 42,
"bucha", 18,
null, 2
]
},
"facet_dates": {},
"facet_ranges": {}
}
}'''
q = self.searcher.search()
qf = QueryFilter()
qf.add_filter(
CategoryFilter(
'cat', 'category', mincount=1,
type=Integer,
ensure_selected_values=True,
_local_params={'cache': False, 'ex': ('test',)}))
qf.add_filter(FacetFilter('region', missing=True, get_title=cap_filter_value))
params = {
'cat': ['5', '13', 'null'],
'region': ['kiev'],
}
q = qf.apply(q, params)
raw_query = force_unicode(q)
self.assertIn('facet=true', raw_query)
self.assertIn('facet.field={!cache=false ex=test,cat key=cat}category', raw_query)
self.assertIn('f.category.facet.mincount=1', raw_query)
self.assertIn('facet.field={!key=region ex=region}region', raw_query)
self.assertIn('f.region.facet.missing=true', raw_query)
self.assertIn('fq={!cache=false ex=test tag=cat}(category:"5" OR category:"13" OR (*:* NOT category:[* TO *]))', raw_query)
qf.process_results(q.results)
category_filter = qf.get_filter('cat')
self.assertIsInstance(category_filter, CategoryFilter)
self.assertEqual(category_filter.name, 'cat')
self.assertEqual(category_filter.field, 'category')
self.assertEqual(len(category_filter.all_values), 6)
self.assertEqual(len(category_filter.values), 3)
self.assertEqual(len(category_filter.selected_values), 3)
self.assertIsInstance(category_filter.all_values[0], CategoryFilterValue)
self.assertEqual(category_filter.all_values[0].filter_name, 'cat')
self.assertEqual(category_filter.all_values[0].value, 100)
self.assertEqual(category_filter.all_values[0].filter_value, '100')
self.assertEqual(category_filter.all_values[0].count, 500)
self.assertEqual(category_filter.all_values[0].count_plus, '+500')
self.assertEqual(category_filter.all_values[0].selected, False)
self.assertEqual(category_filter.all_values[0].title, '100')
self.assertEqual(category_filter.all_values[1].value, 5)
self.assertEqual(category_filter.all_values[1].filter_value, '5')
self.assertEqual(category_filter.all_values[1].count, 10)
self.assertEqual(category_filter.all_values[1].count_plus, '10')
self.assertEqual(category_filter.all_values[1].selected, True)
self.assertEqual(category_filter.all_values[2].value, 2)
self.assertEqual(category_filter.all_values[2].filter_value, '2')
self.assertEqual(category_filter.all_values[2].count, 5)
self.assertEqual(category_filter.all_values[2].count_plus, '+5')
self.assertEqual(category_filter.all_values[2].selected, False)
self.assertEqual(category_filter.all_values[3].value, 1)
self.assertEqual(category_filter.all_values[3].filter_value, '1')
self.assertEqual(category_filter.all_values[3].count, 2)
self.assertEqual(category_filter.all_values[3].count_plus, '+2')
self.assertEqual(category_filter.all_values[3].selected, False)
self.assertEqual(category_filter.all_values[4].value, None)
self.assertEqual(category_filter.all_values[4].filter_value, 'null')
self.assertEqual(category_filter.all_values[4].count, 4)
self.assertEqual(category_filter.all_values[4].count_plus, '4')
self.assertEqual(category_filter.all_values[4].selected, True)
self.assertEqual(category_filter.all_values[5].value, 13)
self.assertEqual(category_filter.all_values[5].filter_value, '13')
self.assertEqual(category_filter.all_values[5].count, None)
self.assertEqual(category_filter.all_values[5].count_plus, '')
self.assertEqual(category_filter.all_values[5].selected, True)
region_filter = qf.get_filter('region')
self.assertEqual(len(region_filter.all_values), 3)
self.assertEqual(len(region_filter.values), 2)
self.assertEqual(len(region_filter.selected_values), 1)
self.assertEqual(region_filter.all_values[0].filter_name, 'region')
self.assertEqual(region_filter.all_values[0].value, 'kiev')
self.assertEqual(region_filter.all_values[0].title, 'Kiev')
self.assertEqual(region_filter.all_values[0].filter_value, 'kiev')
self.assertEqual(region_filter.all_values[0].count, 42)
self.assertEqual(region_filter.all_values[0].count_plus, '42')
self.assertEqual(region_filter.all_values[1].value, 'bucha')
self.assertEqual(region_filter.all_values[1].title, 'Bucha')
self.assertEqual(region_filter.all_values[1].filter_value, 'bucha')
self.assertEqual(region_filter.all_values[1].count, 18)
self.assertEqual(region_filter.all_values[1].count_plus, '+18')
self.assertEqual(region_filter.all_values[2].value, None)
self.assertEqual(region_filter.all_values[2].title, 'None')
self.assertEqual(region_filter.all_values[2].filter_value, 'null')
self.assertEqual(region_filter.all_values[2].count, 2)
self.assertEqual(region_filter.all_values[2].count_plus, '+2')
def cap_filter_value(fv):
return force_unicode(fv.value).capitalize()
def test_pivot_filter(self):
with self.patch_send_request() as send_request:
send_request.return_value = """
{
"facet_counts": {
"facet_pivot": {
"manu": [
{
"field": "manufacturer",
"value": "samsung",
"count": 100,
"pivot": [
{
"field": "model",
"value": "note",
"count": 66,
"pivot": [
{
"field": "discount",
"value": true,
"count": 11
},
{
"field": "discount",
"value": false,
"count": 55
}
]
},
{
"field": "model",
"value": "S4",
"count": 44,
"pivot": [
{
"field": "discount",
"value": true,
"count": 11
},
{
"field": "discount",
"value": false,
"count": 33
}
]
}
]
},
{
"field": "manufacturer",
"value": "nokia",
"count": 1,
"pivot": [
{
"field": "model",
"value": "n900",
"count": 1,
"pivot": [
{
"field": "discount",
"value": false,
"count": 1
}
]
}
]
},
{
"field": "manufacturer",
"value": "lenovo",
"count": 5,
"pivot": [
{
"field": "model",
"value": "p770",
"count": 4
}
]
}
]
},
"facet_fields": {},
"facet_queries": {},
"facet_dates": {},
"facet_ranges": {}
}
}"""
obj_mapper = Mock(wraps=_obj_mapper)
q = self.searcher.search()
qf = QueryFilter()
qf.add_filter(
PivotFilter(
"manu",
FacetPivotFilter("manufacturer", mincount=1, ensure_selected_values=True),
FacetPivotFilter("model", limit=5, instance_mapper=obj_mapper, get_title=cap_filter_value),
FacetPivotFilter("discount", missing=True, type=Boolean),
)
)
params = {
"manu": ["samsung:note", "nokia:n900:false", "nokia:n900:null", "noname:", 10],
"manu__gte": "100",
}
q = qf.apply(q, params)
raw_query = force_unicode(q)
self.assertIn("facet=true", raw_query)
self.assertIn("facet.pivot={!key=manu ex=manu}manufacturer,model,discount", raw_query)
self.assertIn("f.manufacturer.facet.mincount=1", raw_query)
self.assertIn("f.model.facet.limit=5", raw_query)
self.assertIn(
"fq={!tag=manu}"
'((manufacturer:"samsung" AND model:"note") '
'OR (manufacturer:"nokia" AND model:"n900" AND discount:"false") '
'OR (manufacturer:"nokia" AND model:"n900" AND (*:* NOT discount:[* TO *])) '
'OR (manufacturer:"noname" AND model:"") '
'OR manufacturer:"10")',
raw_query,
)
qf.process_results(q.results)
manu_filter = qf.get_filter("manu")
self.assertEqual(len(manu_filter.all_values), 5)
self.assertEqual(len(manu_filter.selected_values), 4)
self.assertEqual(len(manu_filter.values), 1)
self.assertEqual(manu_filter.all_values[0].filter_name, "manu")
self.assertEqual(manu_filter.all_values[0].value, "samsung")
self.assertEqual(manu_filter.all_values[0].title, "samsung")
self.assertEqual(manu_filter.all_values[0].filter_value, "samsung")
self.assertEqual(manu_filter.all_values[0].count, 100)
self.assertEqual(manu_filter.all_values[0].selected, True)
self.assertEqual(manu_filter.all_values[0].pivot.all_values[0].filter_name, "manu")
self.assertEqual(manu_filter.all_values[0].pivot.all_values[0].value, "note")
self.assertEqual(manu_filter.all_values[0].pivot.all_values[0].title, "Note")
self.assertEqual(manu_filter.all_values[0].pivot.all_values[0].filter_value, "samsung:note")
self.assertEqual(manu_filter.all_values[0].pivot.all_values[0].instance, ("note", "note note"))
self.assertEqual(manu_filter.all_values[0].pivot.all_values[0].count, 66)
self.assertEqual(manu_filter.all_values[0].pivot.all_values[0].count_plus, "66")
self.assertEqual(manu_filter.all_values[0].pivot.all_values[0].selected, True)
self.assertEqual(len(manu_filter.all_values[0].pivot.all_values[0].pivot.all_values), 2)
self.assertEqual(manu_filter.all_values[0].pivot.all_values[0].pivot.all_values[0].value, True)
self.assertEqual(manu_filter.all_values[0].pivot.all_values[0].pivot.all_values[0].title, "True")
self.assertEqual(
manu_filter.all_values[0].pivot.all_values[0].pivot.all_values[0].filter_value, "samsung:note:true"
)
self.assertEqual(manu_filter.all_values[0].pivot.all_values[0].pivot.all_values[0].count, 11)
self.assertEqual(manu_filter.all_values[0].pivot.all_values[0].pivot.all_values[0].selected, False)
self.assertEqual(manu_filter.all_values[0].pivot.all_values[0].pivot.all_values[0].pivot, None)
self.assertEqual(manu_filter.all_values[0].pivot.all_values[0].pivot.all_values[1].value, False)
self.assertEqual(manu_filter.all_values[0].pivot.all_values[0].pivot.all_values[1].title, "False")
self.assertEqual(
manu_filter.all_values[0].pivot.all_values[0].pivot.all_values[1].filter_value, "samsung:note:false"
)
self.assertEqual(manu_filter.all_values[0].pivot.all_values[0].pivot.all_values[1].count, 55)
self.assertEqual(manu_filter.all_values[0].pivot.all_values[0].pivot.all_values[1].selected, False)
self.assertEqual(manu_filter.all_values[0].pivot.all_values[0].pivot.all_values[1].pivot, None)
self.assertEqual(manu_filter.all_values[0].pivot.all_values[1].filter_name, "manu")
self.assertEqual(manu_filter.all_values[0].pivot.all_values[1].value, "S4")
self.assertEqual(manu_filter.all_values[0].pivot.all_values[1].title, "S4")
self.assertEqual(manu_filter.all_values[0].pivot.all_values[1].filter_value, "samsung:S4")
self.assertEqual(manu_filter.all_values[0].pivot.all_values[1].instance, ("S4", "S4 S4"))
self.assertEqual(manu_filter.all_values[0].pivot.all_values[1].count, 44)
self.assertEqual(manu_filter.all_values[0].pivot.all_values[1].count_plus, "+44")
self.assertEqual(manu_filter.all_values[0].pivot.all_values[1].selected, False)
self.assertEqual(len(manu_filter.all_values[0].pivot.all_values[1].pivot.all_values), 2)
self.assertEqual(manu_filter.all_values[0].pivot.all_values[1].pivot.all_values[0].value, True)
self.assertEqual(manu_filter.all_values[0].pivot.all_values[1].pivot.all_values[0].title, "True")
self.assertEqual(
manu_filter.all_values[0].pivot.all_values[1].pivot.all_values[0].filter_value, "samsung:S4:true"
)
self.assertEqual(manu_filter.all_values[0].pivot.all_values[1].pivot.all_values[0].count, 11)
self.assertEqual(manu_filter.all_values[0].pivot.all_values[1].pivot.all_values[0].count_plus, "11")
self.assertEqual(manu_filter.all_values[0].pivot.all_values[1].pivot.all_values[0].selected, False)
self.assertEqual(manu_filter.all_values[0].pivot.all_values[1].pivot.all_values[0].pivot, None)
self.assertEqual(manu_filter.all_values[0].pivot.all_values[1].pivot.all_values[1].value, False)
self.assertEqual(manu_filter.all_values[0].pivot.all_values[1].pivot.all_values[1].title, "False")
self.assertEqual(
manu_filter.all_values[0].pivot.all_values[1].pivot.all_values[1].filter_value, "samsung:S4:false"
)
self.assertEqual(manu_filter.all_values[0].pivot.all_values[1].pivot.all_values[1].count_plus, "33")
self.assertEqual(manu_filter.all_values[0].pivot.all_values[1].pivot.all_values[1].selected, False)
self.assertEqual(manu_filter.all_values[0].pivot.all_values[1].pivot.all_values[1].pivot, None)
self.assertEqual(manu_filter.all_values[1].value, "nokia")
self.assertEqual(manu_filter.all_values[1].title, "nokia")
self.assertEqual(manu_filter.all_values[1].filter_value, "nokia")
self.assertEqual(manu_filter.all_values[1].count, 1)
self.assertEqual(manu_filter.all_values[1].selected, True)
self.assertEqual(len(manu_filter.all_values[1].pivot.all_values), 1)
self.assertEqual(manu_filter.all_values[1].pivot.all_values[0].value, "n900")
self.assertEqual(manu_filter.all_values[1].pivot.all_values[0].title, "N900")
self.assertEqual(manu_filter.all_values[1].pivot.all_values[0].filter_value, "nokia:n900")
self.assertEqual(manu_filter.all_values[1].pivot.all_values[0].instance, ("n900", "n900 n900"))
self.assertEqual(manu_filter.all_values[1].pivot.all_values[0].count, 1)
self.assertEqual(manu_filter.all_values[1].pivot.all_values[0].selected, True)
self.assertEqual(len(manu_filter.all_values[1].pivot.all_values[0].pivot.all_values), 1)
self.assertEqual(manu_filter.all_values[1].pivot.all_values[0].pivot.all_values[0].value, False)
self.assertEqual(manu_filter.all_values[1].pivot.all_values[0].pivot.all_values[0].title, "False")
self.assertEqual(
manu_filter.all_values[1].pivot.all_values[0].pivot.all_values[0].filter_value, "nokia:n900:false"
)
self.assertEqual(manu_filter.all_values[1].pivot.all_values[0].pivot.all_values[0].count, 1)
self.assertEqual(manu_filter.all_values[1].pivot.all_values[0].pivot.all_values[0].count_plus, "1")
self.assertEqual(manu_filter.all_values[1].pivot.all_values[0].pivot.all_values[0].selected, True)
self.assertEqual(manu_filter.all_values[2].value, "lenovo")
self.assertEqual(manu_filter.all_values[2].title, "lenovo")
self.assertEqual(manu_filter.all_values[2].filter_value, "lenovo")
self.assertEqual(manu_filter.all_values[2].count, 5)
self.assertEqual(manu_filter.all_values[2].selected, False)
self.assertEqual(manu_filter.all_values[3].value, "noname")
self.assertEqual(manu_filter.all_values[3].title, "noname")
self.assertEqual(manu_filter.all_values[3].filter_value, "noname")
self.assertEqual(manu_filter.all_values[3].count, None)
self.assertEqual(manu_filter.all_values[3].count_plus, "")
self.assertEqual(manu_filter.all_values[3].selected, True)
self.assertEqual(manu_filter.all_values[4].value, "10")
self.assertEqual(manu_filter.all_values[4].title, "10")
self.assertEqual(manu_filter.all_values[4].filter_value, "10")
self.assertEqual(manu_filter.all_values[4].count, None)
self.assertEqual(manu_filter.all_values[4].count_plus, "")
self.assertEqual(manu_filter.all_values[4].selected, True)
self.assertEqual(obj_mapper.call_count, 1)
def test_X(self):
self.assertEqual(force_unicode(X(status=0)),
"(AND: ('status', 0))")
self.assertEqual(force_unicode(X(status=0) & X(company_status__in=[0,6])),
"(AND: ('status', 0), ('company_status__in', [0, 6]))")
self.assertEqual(force_unicode(X(status=0) | X(company_status=0)),
"(OR: ('status', 0), ('company_status', 0))")
self.assertEqual(force_unicode(X(with_photo=True)),
"(AND: ('with_photo', True))")
self.assertEqual(force_unicode(X(date_created__gt=datetime(2012, 5, 17, 14, 35, 41, 794880))),
"(AND: ('date_created__gt', datetime.datetime(2012, 5, 17, 14, 35, 41, 794880)))")
self.assertEqual(force_unicode(X(price__lt=1000)),
"(AND: ('price__lt', 1000))")
self.assertEqual(force_unicode(X(price__gte=100) & X(price__lte=1000)),
"(AND: ('price__gte', 100), ('price__lte', 1000))")
self.assertEqual(force_unicode(X(price__between=[500, 1000])),
"(AND: ('price__between', [500, 1000]))")
self.assertEqual(force_unicode(X(price__range=[2, 10])),
"(AND: ('price__range', [2, 10]))")
self.assertEqual(force_unicode(X(category__in=[1, 2, 3, 4, 5]) & (X(status=0) | X(status=5) | X(status=1) & X(company_status=6))),
"(AND: ('category__in', [1, 2, 3, 4, 5]), (OR: ('status', 0), ('status', 5), (AND: ('status', 1), ('company_status', 6))))")
self.assertEqual(force_unicode(~X(status=1)),
"(AND: (NOT (AND: ('status', 1))))")
self.assertEqual(force_unicode(~X(status__in=[1, 2, 3])),
"(AND: (NOT (AND: ('status__in', [1, 2, 3]))))")
def test_facet_query_filter(self):
with self.patch_send_request() as send_request:
send_request.return_value = '''
{
"facet_counts": {
"facet_queries": {
"date_created__today": 28,
"date_created__week_ago": 105,
"dist__d5": 0,
"dist__d10": 12,
"dist__d20": 40
},
"facet_fields": {},
"facet_dates":{},
"facet_ranges":{}
}
}'''
q = self.searcher.search()
qf = QueryFilter()
qf.add_filter(
FacetQueryFilter(
'date_created',
FacetQueryFilterValue(
'today',
X(date_created__gte='NOW/DAY-1DAY'),
_local_params=LocalParams(ex='test'),
title='Only new',
help_text='Documents one day later'),
FacetQueryFilterValue(
'week_ago',
X(date_created__gte='NOW/DAY-7DAY'),
title='Week ago')))
qf.add_filter(
FacetQueryFilter(
'dist',
FacetQueryFilterValue(
'd5',
None, _local_params=LocalParams('geofilt', d=5, tag='d5')),
FacetQueryFilterValue(
'd10',
None, _local_params=LocalParams('geofilt', d=10, tag='d10')),
FacetQueryFilterValue(
'd20',
None, _local_params=LocalParams('geofilt', d=20, tag='d20')),
select_multiple=False))
qf.add_ordering(
OrderingFilter(
'sort',
OrderingValue('-score', '-score'),
OrderingValue('price', 'price'),
OrderingValue('-price', '-price')))
params = {
'cat': ['5', '13'],
'country': ['us', 'ru'],
'date_created': 'today',
'dist': 'd10',
}
q = qf.apply(q, params)
raw_query = force_unicode(q)
self.assertIn('facet=true', raw_query)
self.assertIn('facet.query={!ex=test,date_created key=date_created__today}date_created:[NOW/DAY-1DAY TO *]', raw_query)
self.assertIn('facet.query={!key=date_created__week_ago ex=date_created}date_created:[NOW/DAY-7DAY TO *]', raw_query)
self.assertIn('facet.query={!geofilt d=5 tag=d5 key=dist__d5 ex=dist}', raw_query)
self.assertIn('facet.query={!geofilt d=10 tag=d10 key=dist__d10 ex=dist}', raw_query)
self.assertIn('facet.query={!geofilt d=20 tag=d20 key=dist__d20 ex=dist}', raw_query)
self.assertIn('fq={!tag=date_created}date_created:[NOW/DAY-1DAY TO *]', raw_query)
self.assertIn('fq={!geofilt d=10 tag=d10,dist}', raw_query)
qf.process_results(q.results)
date_created_filter = qf.get_filter('date_created')
self.assertEqual(date_created_filter.get_value('today').count, 28)
self.assertEqual(date_created_filter.get_value('today').count_plus, '28')
self.assertEqual(date_created_filter.get_value('today').selected, True)
self.assertEqual(date_created_filter.get_value('today').title, 'Only new')
self.assertEqual(date_created_filter.get_value('today').opts['help_text'], 'Documents one day later')
self.assertEqual(date_created_filter.get_value('week_ago').count, 105)
self.assertEqual(date_created_filter.get_value('week_ago').count_plus, '105')
dist_filter = qf.get_filter('dist')
self.assertEqual(dist_filter.get_value('d5').count, 0)
self.assertEqual(dist_filter.get_value('d5').selected, False)
self.assertEqual(dist_filter.get_value('d10').count, 12)
self.assertEqual(dist_filter.get_value('d10').selected, True)
self.assertEqual(dist_filter.get_value('d20').count, 40)
self.assertEqual(dist_filter.get_value('d20').selected, False)
def test_local_params(self):
self.assertEqual(force_unicode(LocalParams({'cache': False})),
'{!cache=false}')
self.assertEqual(force_unicode(LocalParams(LocalParams({'cache': False}))),
'{!cache=false}')
self.assertEqual(force_unicode(LocalParams({'ex': 'tag'}, key='tag')),
'{!ex=tag key=tag}')
self.assertEqual(force_unicode(LocalParams({'ex': 'tag', 'key': 'tag'}, key='category')),
'{!ex=tag key=category}')
self.assertEqual(force_unicode(LocalParams(ex=['f1', 'f2'])),
'{!ex=f1,f2}')
self.assertEqual(force_unicode(LocalParams('frange', l=0, u=5)),
'{!frange l=0 u=5}')
self.assertEqual(force_unicode(LocalParams(['geofilt', ('d', 10), ('key', 'd10')])),
'{!geofilt d=10 key=d10}')
self.assertEqual(force_unicode(LocalParams({'type': 'join', 'from': 'id', 'to': 'manu_id'})),
'{!join from=id to=manu_id}')
self.assertEqual(force_unicode(LocalParams()), '')
self.assertEqual(force_unicode(LocalParams(None)), '')
self.assertEqual(force_unicode(LocalParams('dismax', v='OR test')),
"""{!dismax v='or test'}""")
self.assertEqual(force_unicode(LocalParams('dismax', v='"test"')),
"""{!dismax v='\\"test\\"'}""")
self.assertEqual(force_unicode(LocalParams('dismax', v='test\'')),
"""{!dismax v='test\\\\\''}""")
self.assertRaises(ValueError, LocalParams, '{dismax}', v='test')
self.assertRaises(ValueError, LocalParams, ['dismax', ('!v', 'test')])
self.assertEqual(
force_unicode(LocalParams('dismax', qf='name',
v=X(SafeUnicode('"nokia lumia"')) | X(SafeUnicode('"nokia n900"')))),
"""{!dismax qf=name v='(\\"nokia lumia\\" OR \\"nokia n900\\")'}""")
lp = LocalParams('dismax', bf=func.linear('rank', 100, 0), v='$q1')
lp.update(LocalParams(qf='name^10 description'))
lp.add('pf', 'name')
lp.add('ps', 2)
self.assertIn('type', lp)
self.assertIn('v', lp)
self.assertFalse('q' in lp)
self.assertEqual(lp['type'], 'dismax')
self.assertEqual(
force_unicode(lp),
"{!dismax bf='linear(rank,100,0)' v=$q1 qf='name^10 description' pf=name ps=2}")
def test_facet_filter(self):
with self.patch_send_request() as send_request:
send_request.return_value = """
{
"facet_counts": {
"facet_queries": {},
"facet_fields": {
"cat": [
"100", 500,
"5", 10,
"2", 5,
"1", 2,
null, 4
],
"region": [
"kiev", 42,
"bucha", 18,
null, 2
]
},
"facet_dates": {},
"facet_ranges": {}
}
}"""
q = self.searcher.search()
qf = QueryFilter()
qf.add_filter(
CategoryFilter(
"cat",
"category",
mincount=1,
type=Integer,
ensure_selected_values=True,
_local_params={"cache": False, "ex": ("test",)},
)
)
qf.add_filter(FacetFilter("region", missing=True, get_title=cap_filter_value))
params = {"cat": ["5", "13", "null"], "region": ["kiev"]}
q = qf.apply(q, params)
raw_query = force_unicode(q)
self.assertIn("facet=true", raw_query)
self.assertIn("facet.field={!cache=false ex=test,cat key=cat}category", raw_query)
self.assertIn("f.category.facet.mincount=1", raw_query)
self.assertIn("facet.field={!key=region ex=region}region", raw_query)
self.assertIn("f.region.facet.missing=true", raw_query)
self.assertIn(
'fq={!cache=false ex=test tag=cat}(category:"5" OR category:"13" OR (*:* NOT category:[* TO *]))',
raw_query,
)
qf.process_results(q.results)
category_filter = qf.get_filter("cat")
self.assertIsInstance(category_filter, CategoryFilter)
self.assertEqual(category_filter.name, "cat")
self.assertEqual(category_filter.field, "category")
self.assertEqual(len(category_filter.all_values), 6)
self.assertEqual(len(category_filter.values), 3)
self.assertEqual(len(category_filter.selected_values), 3)
self.assertIsInstance(category_filter.all_values[0], CategoryFilterValue)
self.assertEqual(category_filter.all_values[0].filter_name, "cat")
self.assertEqual(category_filter.all_values[0].value, 100)
self.assertEqual(category_filter.all_values[0].filter_value, "100")
self.assertEqual(category_filter.all_values[0].count, 500)
self.assertEqual(category_filter.all_values[0].count_plus, "+500")
self.assertEqual(category_filter.all_values[0].selected, False)
self.assertEqual(category_filter.all_values[0].title, "100")
self.assertEqual(category_filter.all_values[1].value, 5)
self.assertEqual(category_filter.all_values[1].filter_value, "5")
self.assertEqual(category_filter.all_values[1].count, 10)
self.assertEqual(category_filter.all_values[1].count_plus, "10")
self.assertEqual(category_filter.all_values[1].selected, True)
self.assertEqual(category_filter.all_values[2].value, 2)
self.assertEqual(category_filter.all_values[2].filter_value, "2")
self.assertEqual(category_filter.all_values[2].count, 5)
self.assertEqual(category_filter.all_values[2].count_plus, "+5")
self.assertEqual(category_filter.all_values[2].selected, False)
self.assertEqual(category_filter.all_values[3].value, 1)
self.assertEqual(category_filter.all_values[3].filter_value, "1")
self.assertEqual(category_filter.all_values[3].count, 2)
self.assertEqual(category_filter.all_values[3].count_plus, "+2")
self.assertEqual(category_filter.all_values[3].selected, False)
self.assertEqual(category_filter.all_values[4].value, None)
self.assertEqual(category_filter.all_values[4].filter_value, "null")
self.assertEqual(category_filter.all_values[4].count, 4)
self.assertEqual(category_filter.all_values[4].count_plus, "4")
self.assertEqual(category_filter.all_values[4].selected, True)
self.assertEqual(category_filter.all_values[5].value, 13)
self.assertEqual(category_filter.all_values[5].filter_value, "13")
self.assertEqual(category_filter.all_values[5].count, None)
self.assertEqual(category_filter.all_values[5].count_plus, "")
self.assertEqual(category_filter.all_values[5].selected, True)
region_filter = qf.get_filter("region")
self.assertEqual(len(region_filter.all_values), 3)
self.assertEqual(len(region_filter.values), 2)
self.assertEqual(len(region_filter.selected_values), 1)
self.assertEqual(region_filter.all_values[0].filter_name, "region")
self.assertEqual(region_filter.all_values[0].value, "kiev")
self.assertEqual(region_filter.all_values[0].title, "Kiev")
self.assertEqual(region_filter.all_values[0].filter_value, "kiev")
self.assertEqual(region_filter.all_values[0].count, 42)
self.assertEqual(region_filter.all_values[0].count_plus, "42")
self.assertEqual(region_filter.all_values[1].value, "bucha")
self.assertEqual(region_filter.all_values[1].title, "Bucha")
self.assertEqual(region_filter.all_values[1].filter_value, "bucha")
self.assertEqual(region_filter.all_values[1].count, 18)
self.assertEqual(region_filter.all_values[1].count_plus, "+18")
self.assertEqual(region_filter.all_values[2].value, None)
self.assertEqual(region_filter.all_values[2].title, "None")
self.assertEqual(region_filter.all_values[2].filter_value, "null")
self.assertEqual(region_filter.all_values[2].count, 2)
self.assertEqual(region_filter.all_values[2].count_plus, "+2")
