go - 在 Go 包函数中模拟函数

Question

我正在尝试模拟在我的 Go 代码中的 API 函数调用中使用的 HTTP 客户端。

import (
    "internal.repo/[...]/http"
    "encoding/json"
    "strings"
    "github.com/stretchr/testify/require"
)

func CreateResource(t *testing.T, url string, bodyReq interface{}, username string, password string, resource string) []byte {
    bodyReqJSON, err := json.Marshal(bodyReq)

    if err != nil {
        panic(err)
    }

    headers := make(map[string]string)
    headers["Content-Type"] = "application/json"

    logger.Logf(t, "*************************** CREATE a temporary test %s ***************************", resource)

    // this func below should be mocked
    statusCode, body := http.POST(t, url, bodyReqJSON, headers, username, password)

    require.Equal(t, statusCode, 201, "******ERROR!! A problem occurred while creating %s. Body: %s******", resource, strings.TrimSpace(string(body)))

    return body
}

我想模拟我的 http.POST 函数，它是内部 HTTP 包的一部分，这样我就不需要实际进行在线调用，也不需要离线隔离测试。

是否有替代方法来依赖注入(inject)模拟结构来实现假设的 HTTP 接口(interface)？

你会怎么做这样的事情？

Answer 1

这是解决方案，感谢@Peter。

import (
    "net/http"
    "net/http/httptest"
    "testing"

    "github.com/stretchr/testify/assert"
)

func TestCreateResource(t *testing.T) {
    t.Run("successful", func(t *testing.T) {
        server := httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
            w.WriteHeader(201)
        }))
        defer server.Close()

        o := CreateResource(t, server.URL, nil, "admin", "password", "resource")
        assert.Equal(t, []byte{}, o)
    })
}