我在 Go 中构建了一个快速简单的 API 来查询 ElasticSearch。现在我知道它可以完成，我想通过添加测试来正确地完成它。我已经抽象了一些我的代码，以便它可以进行单元测试，但是我在模拟弹性库时遇到了一些问题，因此我认为最好尝试一个简单的案例来模拟它。

import (
    "encoding/json"
    "github.com/olivere/elastic"
    "net/http"
)
...
func CheckBucketExists(name string, client *elastic.Client) bool {
    exists, err := client.IndexExists(name).Do()
    if err != nil {
        panic(err)
    }
    return exists
}

现在测试...

  import (
      "fmt"
      "github.com/stretchr/testify/assert"
      "github.com/stretchr/testify/mock"
      "testing"
  )

type MockClient struct {
      mock.Mock
  }

  func (m *MockClient) IndexExists(name string) (bool, error) {
      args := m.Mock.Called()
      fmt.Println("This is a thing")
      return args.Bool(0), args.Error(1)
  }

  func TestMockBucketExists(t *testing.T) {
      m := MockClient{}
      m.On("IndexExists", "thisuri").Return(true)

>>    r := CheckBucketExists("thisuri", m)
      assert := assert.New(t)
      assert.True(r, true)
  }

我遇到了以下错误:不能在 CheckBucketExists 的参数中使用 m(类型 MockClient)作为类型 *elastic.Client。

我假设这是我使用 elastic.client 类型的基础，但我仍然是一个菜鸟。

最佳答案

这是一个老问题，但也找不到解决方案。 不幸的是，这个库是使用结构实现的，这使得模拟它一点也不简单，所以我找到的选项是:

(1) 将所有 elastic.SearchResult 方法包装在您自己的接口(interface)上并“代理”调用，因此您最终得到如下结果:

type ObjectsearchESClient interface {
    // ... all methods... 
    Do(context.Context) (*elastic.SearchResult, error)
}


// NewObjectsearchESClient returns a new implementation of ObjectsearchESClient
func NewObjectsearchESClient(cluster *config.ESCluster) (ObjectsearchESClient, error) {
    esClient, err := newESClient(cluster)
    if err != nil {
        return nil, err
    }

    newClient := objectsearchESClient{
        Client: esClient,
    }
    return &newClient, nil
}
// ... all methods... 
func (oc *objectsearchESClient) Do(ctx context.Context) (*elastic.SearchResult, error) {
    return oc.searchService.Do(ctx)
}

然后像模拟应用程序的其他模块一样模拟此接口(interface)和响应。

(2) 另一个选项就像在 this blog post 中指出的那样这是使用 httptest.Server

为此，我模拟了处理程序，包括模拟来自“HTTP 调用”的响应

func mockHandler () http.HandlerFunc{
    return func(w http.ResponseWriter, r *http.Request) {
        resp := `{
            "took": 73,
            "timed_out": false,
            ... json ... 
            "hits": [... ]
            ...json ... ,
            "aggregations": { ... }
        }`

        w.Write([]byte(resp))
    }
}

然后你创建一个虚拟的 elastic.Client 结构体

func mockClient(url string) (*elastic.Client, error) {
    client, err := elastic.NewSimpleClient(elastic.SetURL(url))
    if err != nil {
        return nil, err
    }
    return client, nil
}

在这种情况下，我有一个构建我的 elastic.SearchService 并返回它的库，所以我使用 HTTP 如下:

    ... 
    ts := httptest.NewServer(mockHandler())
    defer ts.Close()
    esClient, err := mockClient(ts.URL)
    ss := elastic.NewSearchService(esClient)
    mockLibESClient := es_mock.NewMockSearcherClient(mockCtrl)
    mockLibESClient.EXPECT().GetEmployeeSearchServices(ctx).Return(ss, nil)

其中 mockLibESClient 是我提到的库，我们 stub mockLibESClient.GetEmployeeSearchServices 方法，使其返回 SearchService，它将返回预期的有效负载。

注意:为了创建模拟 mockLibESClient，我使用了 https://github.com/golang/mock

我发现这很复杂，但在我看来，“包裹”elastic.Client 需要更多工作。

问题:我尝试通过使用 https://github.com/vburenin/ifacemaker 创建接口(interface)来模拟它，然后使用 https://github 模拟该接口(interface)。 com/golang/mock 并有点使用它，但是在尝试返回接口(interface)而不是结构时我一直遇到兼容性错误，我根本不是 Go expect 所以可能我需要了解类型转换 a能够像那样解决它会更好一些。因此，如果你们中有人知道如何使用它，请告诉我。

关于unit-testing - Golang 模拟 Elastic，我们在Stack Overflow上找到一个类似的问题： https://stackoverflow.com/questions/28454337/