在go单元测试中模拟elasticsearch

// main.go
package main

import (
    "fmt"
    "github.com/elastic/go-elasticsearch/v8"
)

func SearchDocuments(client *elasticsearch.Client, query string) ([]string, error) {
    // 实际的 Elasticsearch 搜索逻辑
    // ...
    // 返回搜索结果
    return []string{"document1", "document2"}, nil
}

// main_test.go
package main

import (
    "testing"

    "github.com/elastic/go-elasticsearch/v8"
    "github.com/stretchr/testify/assert"
)

// 模拟 Elasticsearch 客户端
type MockElasticsearchClient struct{}

// 模拟 Elasticsearch 搜索方法
func (m *MockElasticsearchClient) Search(query string) ([]string, error) {
    // 模拟搜索结果
    return []string{"mocked_document1", "mocked_document2"}, nil
}

func TestSearchDocuments(t *testing.T) {
    // 创建模拟的 Elasticsearch 客户端
    mockClient := &MockElasticsearchClient{}

    // 调用被测试的函数，传入模拟的客户端
    result, err := SearchDocuments(mockClient, "test query")

    // 断言结果是否符合预期
    assert.NoError(t, err)
    assert.Equal(t, []string{"mocked_document1", "mocked_document2"}, result)
}