使用依赖注入测试包装函数

Question

我有这个功能，我需要在测试中模拟， 我能够使用 http 模拟包按预期模拟它，但现在我有正在调用的函数 到HttpReq 方法，这里我不能使用http mock 包 我阅读了依赖注入并尝试了一些方法，但我无法完全做到，

这是函数

type params struct {
    cs     string
    ci     string
    method string
    url    string
}

// I added this struct but not sure if it's needed ... probably for test purpose but not sure how to use it.

type Impl struct {
 client *http.Client
}

func (i *Impl) HttpReq(p *params) ([]byte, error) {
    httpClient := i.client
    req, err := http.NewRequest(p.method, p.url, nil)
    if err != nil {
        fmt.Sprintf(err)
    }
    req.SetBasicAuth(p.cs, p.ci)
    res, err := httpClient.Do(req)
    if err != nil {
        fmt.Sprintf(err)
    }
    t, err := ioutil.ReadAll(res.Body)
    if err != nil {
        fmt.Sprintf(err)
    }
    defer res.Body.Close()
    return t, nil
}

这是我尝试过的

我已经创建了界面

type Req interface {
    HttpReq(params) ([]byte, error)
}

现在我创建了一个包含接口的结构体

type Service struct {
    req Req
}

这是我需要测试的功能

func (c *Service) execute(cli Connection , args []string) (error, []byte) {
    sk, err := c.doSomthing(cli, args)

    sc, err := c.doSometing2(serviceK, []string{"url", "cl", "ct"})

    cc := strings.Fields(serviceCredentials)

    // ----------Here is what I need to mock ----------
    t, err := c.req.HttpReq(params{cs: cc[1],
        ci:     cc[2],
        method: http.MethodPost,
        url:    cc[0],})
    return err, t
}

知道如何为这个函数运行测试吗？？？我为此苦苦挣扎。

Answer 1

独立于原始问题，您不应为每个请求创建新的 HTTP 客户端。客户端维护一个连接池，应该尽可能的复用。

您可以解决这个问题，并通过注入 HTTP 客户端继续使用现有的模拟服务器。

另请注意，问题中的接口定义与实现不匹配。这两个方法签名不一样：

HttpReq(params) ([]byte, error)  // Req
HttpReq(*params) ([]byte, error) // Impl

选择一个。我可能会在这里使用非指针类型。 And upper case initials are idiomatic in Go（HTTPReq，不是HttpReq）。

将客户端添加到Impl类型并在HTTPReq中使用：

type Impl struct {
    client *http.Client
}

func (i *Impl) HTTPReq(p params) ([]byte, error) {
    req, err := http.NewRequest(p.method, p.url, nil)
    if err != nil {
        return nil, err
    }
    req.SetBasicAuth(p.cs, p.ci)

    res, err := i.client.Do(req)
    if err != nil {
        return nil, err
    }
    defer res.Body.Close()

    return ioutil.ReadAll(res.Body)
}

服务类型不必改变。

在测试中，只需将测试客户端注入Impl 值：

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

func TestService_execute(t *testing.T) {
    var testHandler http.Handler // TODO

    srv := httptest.NewServer(testHandler)
    defer srv.Close()

    client := srv.Client()
    tp := client.Transport.(*http.Transport)

    // Direct all requests to the test server, no matter what the request URL is.
    tp.DialContext = func(ctx context.Context, network, addr string) (net.Conn, error) {
        // Note that we ignore the network and addr parameters, since these are 
        // derived from the request URL and are unrelated to the test server.

        srvAddr := srv.Listener.Addr()
        return (&net.Dialer{}).DialContext(ctx, srvAddr.Network(), srvAddr.String())
    }

    svc := &Service{
        req: &Impl{
            client: client,
        },
    }

    svc.execute(/* ... */)

    // assert request, response, etc.
}

Answer 2

由于 Service struct 已经有一个 req 接口，在测试期间使用满足 req 接口的 mock 初始化服务对象。 像这样
https://stackoverflow.com/a/53805535/3968921