如何在pytest中模拟elasticsearch对象？

import pytest
from elasticsearch import Elasticsearch

def search_data(es_client, index_name, query):
    # 使用elasticsearch对象进行搜索操作
    result = es_client.search(index=index_name, body=query)
    return result

def test_search_data(mocker):
    # 创建一个模拟的elasticsearch对象
    es_mock = mocker.Mock(spec=Elasticsearch)
    
    # 设置模拟对象的返回值
    es_mock.search.return_value = {
        "hits": {
            "total": 2,
            "hits": [
                {"_source": {"name": "John"}},
                {"_source": {"name": "Alice"}}
            ]
        }
    }
    
    # 调用被测试函数，并传入模拟的elasticsearch对象
    result = search_data(es_mock, "my_index", {"query": {"match_all": {}}})
    
    # 断言结果是否符合预期
    assert result["hits"]["total"] == 2
    assert len(result["hits"]["hits"]) == 2
    assert result["hits"]["hits"][0]["_source"]["name"] == "John"
    assert result["hits"]["hits"][1]["_source"]["name"] == "Alice"