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()