python - 如何模拟 Elasticsearch Python？

Question

我需要模拟 elasticsearch 调用，但我不确定如何在 python 单元测试中模拟它们。我看到了这个名为ElasticMock的框架。我尝试按照文档中指示的方式使用它，但它给了我很多错误。是这里 : https://github.com/vrcmarcos/elasticmock

我的问题是，还有其他方法来模拟 Elasticsearch 调用吗？

这似乎也没有答案:Mock elastic search data 。这只是表明实际进行集成测试而不是单元测试，这不是我想要的: Unit testing elastic search inside Django app .

有人能指出我正确的方向吗？我从来没有用 ElasticSearch 来 mock 过东西。

Answer 1

你必须模拟你需要的属性或方法，例如:

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}

为了了解您需要模拟的路径，只需使用 IDE 探索该库即可。当您已经了解其中一个时，您就可以轻松找到其他的。