def test__esp_add_query_dict(self):
esp = ElasticsearchProcessor()
page = 1
page_size = 20
query = {
"query": {
"match": {
"_all": "foobar"
}
}
}
# ElasticsearchProcessor internally sets the from/size parameters
# on the query; we need to compare with those values included
query_with_size = query.copy()
query_with_size.update({
'from': (page - 1) * page_size,
'size': page_size
})
esp.add_search(query.copy())
ddtools.assert_equal(esp.bulk_search_data[0], {})
ddtools.assert_equal(esp.bulk_search_data[1], query_with_size)
esp.reset()
esp.add_search(query.copy(), index='blog')
ddtools.assert_equal(esp.bulk_search_data[0], {'index': 'blog'})
ddtools.assert_equal(esp.bulk_search_data[1], query_with_size)
esp.reset()
esp.add_search(query.copy(), index='blog', doc_type='posts')
ddtools.assert_equal(esp.bulk_search_data[0], {'index': 'blog', 'type': 'posts'})
ddtools.assert_equal(esp.bulk_search_data[1], query_with_size)
def test__esp_reset(self):
esp = ElasticsearchProcessor()
self.assertTrue(len(esp.bulk_search_data) == 0)
self.assertTrue(len(esp.page_ranges) == 0)
esp.add_search(self.form)
self.assertFalse(len(esp.bulk_search_data) == 0)
self.assertFalse(len(esp.page_ranges) == 0)
esp.reset()
self.assertTrue(len(esp.bulk_search_data) == 0)
self.assertTrue(len(esp.page_ranges) == 0)