使用 Elastic 进行 Golang 模拟

Question

我在 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)

  }

我因以下错误而屈服：cannot use m (type MockClient) as type *elastic.Client in argument to CheckBucketExists.

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