def test__query(self): # create an object to use for an ES query response es_response = { '_shards': { 'failed': 0, 'successful': 1, 'total': 1 }, 'hits': { 'hits': [{ '_id': 'foo', '_index': 'elasticdict', '_score': 0.30685282, '_source': { 'foo': 'bar' }, '_type': 'data' }], 'max_score': 0.30685282, 'total': 1, }, 'timed_out': False, 'took': 1 } # Create a Mock response object config = { 'status_code': 200, 'json.return_value': es_response, } mock_response = Mock(**config) self.mock_requests.get.return_value = mock_response d = ElasticDict() results = d._query('bar') eq_(results, [{u'foo': u'bar'}]) expected_query_json = ('{"query": {"filtered": {"query": ' '{"query_string": {"query": "bar"}}}}}') self.mock_requests.get.assert_called_once_with( 'http://localhost:9200/elasticdict/data/_search', data=expected_query_json)
def test__query(self): # create an object to use for an ES query response es_response = { '_shards': {'failed': 0, 'successful': 1, 'total': 1}, 'hits': { 'hits': [{ '_id': 'foo', '_index': 'elasticdict', '_score': 0.30685282, '_source': {'foo': 'bar'}, '_type': 'data' }], 'max_score': 0.30685282, 'total': 1, }, 'timed_out': False, 'took': 1 } # Create a Mock response object config = { 'status_code': 200, 'json.return_value': es_response, } mock_response = Mock(**config) self.mock_requests.get.return_value = mock_response d = ElasticDict() results = d._query('bar') eq_(results, [{u'foo': u'bar'}]) expected_query_json = ('{"query": {"filtered": {"query": ' '{"query_string": {"query": "bar"}}}}}') self.mock_requests.get.assert_called_once_with( 'http://localhost:9200/elasticdict/data/_search', data=expected_query_json )