def setUp(self):
super(SearchTests, self).setUp()
self.fake_org_id = 'orgId001'
self.test_search_result = {'whateverField': 'something'}
self.test_total_hits = 666
self.test_id = 13
self.test_es_search_results = {
'hits': {
'hits': [
{
'_source': self.test_search_result,
'_id': self.test_id
}
],
'total': self.test_total_hits
},
'aggregations': {
'categories': {
'buckets': [
{
'key': 'health'
},
{
'key': 'science'
}
]
},
'formats': {
'buckets': [
{
'key': 'csv'
}
]
}
}
}
self._search_obj = DataSetSearch()
self._search_obj._translator.translate = self._mock_translate = MagicMock()
self._search_obj._elastic_search.search = self._mock_es_search = MagicMock()
self.request_context = self.app.test_request_context('/rest/datasets')
self.request_context.push()
def __init__(self):
super(DataSetCountResource, self).__init__()
self._elastic_search = Elasticsearch('{}:{}'.format(
self._config.elastic.elastic_hostname,
self._config.elastic.elastic_port))
self._search = DataSetSearch()
