如何在Golang测试中模拟一个进行外部调用的中间件？

Question

有一个go-chi中间件正在进行外部调用以授权请求。我需要嘲笑它的作用。

细节：

我有这个测试：

func Test_HTTP_UserSet_Create_InvalidAction(t *testing.T) {

    requestBody :=

        `{

            "name": "test",

            "action": "TEST"

        }`

    req, _ := http.NewRequest("POST", "/1234/users", bytes.NewBuffer([]byte(requestBody)))

    recorder := setupInvalidActionRecorder(t, req)

    if status := recorder.Code; status != http.StatusBadRequest {

        t.Errorf("Status code expected to be %d but got %d", http.StatusBadRequest, status)

        return

    }

}

以上setupUnknownErrorRecorder是：

func setupUnknownErrorRecorder(t *testing.T, request *http.Request) *httptest.ResponseRecorder {

    recorder := httptest.NewRecorder()

    c := resty.New()

    resource := users.NewResource(

        httpClient.NewHttpClient(log, &config.Server{}, &logger.LogRecord{}, c),

    )

    resource.Routes().ServeHTTP(recorder, request)

    return recorder

}

我得到的结果是：

=== RUN   Test_HTTP_UserSet_Create_InvalidAction

    invalidAction_test.go:71: Status code expected to be 400 but got 401

--- FAIL: Test_HTTP_UserSet_Create_InvalidAction (0.00s)

c这个结果是预期的，因为服务器正在使用我传递给的客户端进行外部调用NewResource。

有一个go-chi中间件正在使用此客户端进行外部调用。

我的问题是：如何让这个 http 调用始终返回 200 而不是真正进行调用。又名如何嘲笑这个电话？

编辑： 上面httpClient的接口是：

type HttpClient struct {

    logger          *logger.Logger

    config          *config.Server

    instrumentation *instrumentation

    record          *logger.LogRecord

    restyClient     *resty.Client

}

func NewHttpClient(

    logger *logger.Logger,

    config *config.Server,

    record *logger.LogRecord,

    restyClient *resty.Client,

) HttpClient {

    return HttpClient{

        instrumentation: newInstrumentation(logger, record),

        config:          config,

        logger:          logger,

        record:          record,

        restyClient:     restyClient,

    }

}

Answer 1

可测试代码的一个非常基本的规则是，您不能只在其他服务中创建新服务。您需要注入它，并准备一个接口（如果返回的内容还没有接口的话）resty.New。这样你就可以在测试中注入你的模拟。然后你可以使用例如https://pkg.go.dev/github.com/stretchr/testify/mock来生成模拟并说明你的模拟服务应该返回什么值。评论后更新：type HttpClient struct {&nbsp; &nbsp; logger&nbsp; &nbsp; &nbsp; &nbsp; &nbsp; *logger.Logger&nbsp; &nbsp; config&nbsp; &nbsp; &nbsp; &nbsp; &nbsp; *config.Server&nbsp; &nbsp; instrumentation *instrumentation&nbsp; &nbsp; record&nbsp; &nbsp; &nbsp; &nbsp; &nbsp; *logger.LogRecord&nbsp; &nbsp; restyClient&nbsp; &nbsp; &nbsp;*resty.Client // <- this is the thing you want to change from a pointer to resty.Client to an interface}检查restyClient您在代码中使用了哪些方法，并创建包含它们的新接口，例如：type MyRestyClient interface {&nbsp; &nbsp;FirstUsedMethod(..)&nbsp; &nbsp;&nbsp; &nbsp;...}并将 restyClient 声明交换为type HttpClient struct {&nbsp; &nbsp; ...&nbsp; &nbsp; restyClient&nbsp; &nbsp; &nbsp;MyRestyClient之后，您可以使用我之前粘贴的链接中的模拟和命令为您生成模拟。稍后，您只需在测试中设置模拟：restyMock := new(RestyMock)restyMock.On("MethodYouExpect").Return(returnValueOfYourChoice)然后您将准备好的模拟注入到您的测试服务中。