进行单元测试时如何使用 http 作为依赖项

学习Golang要努力，但是不要急！今天的这篇文章《进行单元测试时如何使用 http 作为依赖项》将会介绍到等等知识点，如果你想深入学习Golang，可以关注我！我会持续更新相关文章的，希望对大家都能有所帮助！

我有以下功能，可以按预期工作。 现在我想对其运行单元测试。

func httpclient(cc []string,method http) ([]byte, error) {
    httpclient := http.client{}
    req, err := http.newrequest(http.methodpost, c[0]+"/oauth/token", nil)
    if err != nil {
        fmt.error(err)
    }
    //here we are passing user and password
    req.setbasicauth(c[1], c[2])
    res, err := httpclient.do(req)
    if err != nil {
        fmt.error(err)
    }
    val, err := ioutil.readall(res.body)
    if err != nil {
        fmt.error(err)
    }
    defer res.body.close()
    return val, nil
}

问题是我需要一些东西来伪造 http 调用， 我尝试了以下操作，更改方法签名并将方法和 url 作为参数。 问题是我不知道如何伪造 post 方法

这是更改的代码，使测试更容易

func httpclient(cc []string, method string, url string) ([]byte, error) {
    httpclient := http.client{}
    req, err := http.newrequest(method, url, nil)
    if err != nil {
        return nil, errors.wrap(err, "failed to execute http request")
    }
    //here we are passing user and password
    req.setbasicauth(cc[1], cc[2])
    res, err := httpclient.do(req)
    if err != nil {
        fmt.error(err)
    }
    val, err := ioutil.readall(res.body)
    if err != nil {
        fmt.error(err)
    }
    defer res.body.close()
    return val, nil
}

这是我尝试的测试，但仍然不起作用，因为我需要以某种方式伪造post方法

func Test_httpClient(t *testing.T) {
    type args struct {
        params    []string
        method string
        url    string
    }
    tests := []struct {
        name    string
        args    args
        want    []byte
        wantErr bool
    }{
        {
            name:    "httpCallTest",
            args:{{"fakeuser", "fakepasswword"},{"post"},{"/any/url"}},
            want:    nil,
            wantErr: false,
        },

    }
    for _, tt := range tests {
        t.Run(tt.name, func(t *testing.T) {
            got, err := httpClient(tt.args.params, tt.args.method, tt.args.url)
            if (err != nil) != tt.wantErr {
                t.Errorf("httpClient() error = %v, wantErr %v", err, tt.wantErr)
                return
            }
            if !reflect.DeepEqual(got, tt.want) {
                t.Errorf("httpClient() = %v, want %v", got, tt.want)
            }
        })
    }
}

解决方案

我建议查看 httptest 包。它有一个非常适合您的假 http 服务器。

func test_httpclient(t *testing.t){
  var called bool
  defer func(){
    if !called{
      t.fatal("expected endpoint to be called")
    }
  }()
  expectedvalue = "some-value"

  server := httptest.newserver(http.handlerfunc(func(w http.responsewriter, r *http.request){
    called = true
    u,p,ok := r.basicauth()
    if !ok || u != "fakeuser" || p != "fakepassword" {
      t.fatal("wrong auth")
    }
    w.write([]byte(expectedvalue))
  })

  val, err := httpclient(
   []string{"fakeuser", "fakepasswword"}, 
   http.methodpost, 
   server.url,
  )

  if err != nil{
    t.fatal(err)
  }

  if val != expectedvalue {
    t.fatalf("expected %q to equal %q", val, expectedvalue)
  }
}

我会使用类似以下代码的内容。用法是newtestserver(address).run()

import (
    "github.com/gorilla/mux"
    "net/http"
    "sync"
    "time"
)

type testserver struct {
    addr   string
    router *mux.router
}

// addr can be e.g. `localhost:3030`
// -> your client will do request on `localhost:3030/some/endpoint/path`
func newtestserver(addr string) *testserver {
    router := mux.newrouter()
    router.handlefunc(
        "/some/endpoint/path",
        handleendpoint,
    ).methods(http.methodpost)

    return &testserver{
        addr:   addr,
        router: router,
    }
}

func handleendpoint(writer http.responsewriter, request *http.request) {
    // your implementation
}

func (server *testserver) run() {
    wg := sync.waitgroup{}
    wg.add(1)
    go func() {
        wg.done()
        if err := http.listenandserve(server.addr, server.router); err != nil {
            panic(err)
        }
    }()
    wg.wait()                          // wait for goroutine starts
    time.sleep(100 * time.millisecond) // wait for listening starts
}

但我认为测试仍然无法通过。

// this will panic
// requires 2nd and 3rd item in an array
req.SetBasicAuth(cc[1], cc[2])

本篇关于《进行单元测试时如何使用 http 作为依赖项》的介绍就到此结束啦，但是学无止境，想要了解学习更多关于Golang的相关知识，请关注golang学习网公众号！