如何模拟Go的ElasticSearchSDK？

jljoyd4f 于 2023-10-14 发布在 Go

关注(0)|答案(1)|浏览(113)

我在go中开发了一个API，它使用SDK v8与ElasticSearch进行通信。但是我在模拟单元测试的弹性方法时遇到了困难
我有以下功能：

func DocumentExists(id string) (bool, error) {
  exists, err := elastic.Client.Exists("my_index", "doc_id")
  if err != nil {
    return false, err
  }
  if exists.StatusCode != 200 {
    return false, nil
  }
  return true, nil
}

此函数仅检查弹性文件中是否存在文档。elastic.Client是全局提供的，可在应用程序中的任何位置使用，并且具有以下类型：

import  elastic "github.com/elastic/go-elasticsearch/v8"
var (
  Client             *elastic.Client
)

你可以试着用不同的方式来嘲笑，用嘲笑来证明。但没有成功。
我怎么能这么做？我应该改变我获得客户的方式吗？利用合同？
在我看来，在这种情况下，使用合同是没有必要的。

go

来源：https://stackoverflow.com/questions/77208629/how-to-mock-the-elastic-search-sdk-for-go

1条答案

按热度按时间

o3imoua41#

Elasticsearch Go客户端的GitHub存储库中的官方example演示了如何为Elasticsearch客户端创建模拟。此过程主要需要使用替代HTTP传输的配置调用NewClient。

type MockTransport struct {
    Response    *http.Response
    RoundTripFn func(req *http.Request) (*http.Response, error)
}
func (t *MockTransport) RoundTrip(req *http.Request) (*http.Response, error) {
    return t.RoundTripFn(req)
}
func TestStore(t *testing.T) {
    mockTransport := MockTransport{
        Response: &http.Response{
            StatusCode: http.StatusOK,
            Body:       io.NopCloser(strings.NewReader(`{}`)),
            Header:     http.Header{"X-Elastic-Product": []string{"Elasticsearch"}},
        },
    }
    mockTransport.RoundTripFn = func(req *http.Request) (*http.Response, error) { return mockTransport.Response, nil }
    client, err := elasticsearch.NewClient(elasticsearch.Config{
        Transport: &mockTransport,
    })
    if err != nil {
        t.Fatalf("Error creating Elasticsearch client: %s", err)
    }
    // Now you can use the client for testing.
}

展开查看全部

赞(0）回复(0）举报 2023-10-14

我来回答

如何模拟Go的ElasticSearchSDK？

1条答案

相关问题

热门标签

最新问答