How to mock elastic search python?

Question

I need to mock elasticsearch calls, but I am not sure how to mock them in my python unit tests. I saw this framework called ElasticMock. I tried using it the way indicated in the documentation and it gave me plenty of errors. It is here : https://github.com/vrcmarcos/elasticmock

My question is, is there any other way to mock elastic search calls?

This doesn't seem to have an answer either: Mock elastic search data. And this just indicates to actually do integration tests rather than unit tests, which is not what I want: Unit testing elastic search inside Django app.

Can anyone point me in the right direction? I have never mocked things with ElasticSearch.

Answer 1

After looking at the decorator source code, the trick for me was to reference Elasticsearch with the module:

import elasticsearch
...
elasticsearch.Elasticsearch(...

instead of

from elasticsearch import Elasticsearch
...
Elasticsearch(...

Answer 2

You have to mock the attr or method you need, for example:

import mock

with mock.patch("elasticsearch.Elasticsearch.search") as mocked_search, \
                mock.patch("elasticsearch.client.IndicesClient.create") as mocked_index_create:

            mocked_search.return_value = "pipopapu"
            mocked_index_create.return_value = {"acknowledged": True}

In order to know the path you need to mock, just explore the lib with your IDE. When you already know one you can easily find the others.