def test_build_aggregation_query_with_category(self):
query = "eu gene"
multi_match_fields = [1,2]
category = "locus"
es_query = build_es_search_query(query, multi_match_fields)
aggs = {}
for c in self.category_filters[category]:
aggs[c[1]] = {'terms': {'field': c[1] + '.raw', 'size': 999}}
self.assertEqual({
'query': es_query,
'size': 0,
'aggs': aggs
}, build_aggregation_query(es_query, category, self.category_filters))
def test_build_aggregation_query_with_emtpy_category(self):
query = "eu gene"
multi_match_fields = [1,2]
category = ""
es_query = build_es_search_query(query, multi_match_fields)
self.assertEqual({
'query': es_query,
'size': 0,
'aggs': {
'categories': {
'terms': {'field': 'category', 'size': 50}
},
'feature_type': {
'terms': {'field': 'feature_type', 'size': 50}
}
}
}, build_aggregation_query(es_query, category, self.category_filters))
