golang httptest用法是什么

PHPz

发布时间：2023-04-14 13:45:24

1205人浏览过

来源于php中文网

原创

近年来，随着 golang 语言的普及和高效性，它越来越受到开发者们的关注和喜爱。golang 提供了一套简单易用的测试框架，其中就包括了 httptest 库。在本文章中，我将会介绍 httptest 的基本概念、用法和注意事项，希望能对 golang 开发者们在接下来的测试工作中有所帮助。

httptest 简介

在 Golang 的测试框架中，httptest 的主要作用就是帮助测试者模拟 HTTP 请求和响应，从而方便地对服务端的 API 接口进行测试。它不依赖于网络和实际的 HTTP 服务器，而是使用 Golang 提供的 net/http 包中的功能，模拟出请求和响应的整个流程。

httptest 常用方法

（1）创建测试服务器

在 httptest 中，我们需要通过 NewServer 方法创建一个测试服务器。该方法的参数需要传入一个实现了 http.Handler 接口的 HandlerFunc。例如，我们可以创建一个处理 GET 请求的 HandlerFunc，并通过它创建一个测试服务器：

func TestGetFoo(t *testing.T) {
    srv := httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
        w.WriteHeader(http.StatusOK)
        fmt.Fprintln(w, "Hello, World!")
    }))
    defer srv.Close()

    resp, err := http.Get(srv.URL)
    if err != nil {
        t.Fatal(err)
    }
    defer resp.Body.Close()

    if resp.StatusCode != http.StatusOK {
        t.Errorf("expected %d but got %d", http.StatusOK, resp.StatusCode)
    }
}

（2）发送请求

立即学习“go语言免费学习笔记（深入）”；

有了测试服务器之后，我们就可以通过 Client 来向服务器发送请求。Client 是 Golang 提供的一个 HTTP 请求客户端。例如，我们可以创建一个处理 POST 请求的 HandlerFunc，并通过它创建一个测试服务器。然后发送一个 POST 请求，包含 Json 数据：

func TestPostJson(t *testing.T) {
    srv := httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
        body, _ := ioutil.ReadAll(r.Body)
        w.WriteHeader(http.StatusOK)
        fmt.Fprintln(w, string(body))
    }))
    defer srv.Close()

    payload := struct {
        Name string `json:"name"`
        Age  int    `json:"age"`
    }{
        Name: "Robert",
        Age:  30,
    }
    buf, _ := json.Marshal(payload)
    resp, err := http.Post(srv.URL, "application/json", bytes.NewReader(buf))
    if err != nil {
        t.Fatal(err)
    }
    defer resp.Body.Close()

    if resp.StatusCode != http.StatusOK {
        t.Errorf("expected %d but got %d", http.StatusOK, resp.StatusCode)
    }
    body, _ := ioutil.ReadAll(resp.Body)
    if string(body) != `{"name":"Robert","age":30}` {
        t.Errorf("expected %s but got %s", `{"name":"Robert","age":30}`, string(body))
    }
}

（3）设置请求头和响应头

在测试中，请求头和响应头也是很重要的一部分。我们可以通过 Header 方法来设置请求头和响应头。例如，我们可以创建一个处理 GET 请求的 HandlerFunc，并通过它创建一个测试服务器。然后设置请求头和响应头：

PPT.CN,PPTCN,PPT.CN是什么,PPT.CN官网,PPT.CN如何使用

一键操作，智能生成专业级PPT

下载

func TestHeader(t *testing.T) {
    srv := httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
        w.Header().Set("X-test", "hello")
        w.WriteHeader(http.StatusOK)
        fmt.Fprintln(w, "Hello, World!")
    }))
    defer srv.Close()

    req, err := http.NewRequest("GET", srv.URL, nil)
    if err != nil {
        t.Fatal(err)
    }
    req.Header.Add("X-test", "world")
    resp, err := http.DefaultClient.Do(req)
    if err != nil {
        t.Fatal(err)
    }
    defer resp.Body.Close()

    if resp.StatusCode != http.StatusOK {
        t.Errorf("expected %d but got %d", http.StatusOK, resp.StatusCode)
    }
    if resp.Header.Get("X-test") != "hello" {
        t.Errorf("expected %s but got %s", "hello", resp.Header.Get("X-test"))
    }
}

（4）处理 HTTPS 请求

在实际的项目中，很多 HTTP 接口都需要通过 HTTPS 进行安全传输。在 httptest 中，我们也可以通过 NewTLSServer 方法来创建一个支持 HTTPS 的测试服务器。新建的服务器使用自签名证书，因此不应该在生产环境中使用。例如，我们可以创建一个处理 POST 请求的 HandlerFunc，并通过它创建一个支持 HTTPS 的测试服务器。然后发送一个 POST 请求，包含 Json 数据：

func TestTlsPostJson(t *testing.T) {
    srv := httptest.NewTLSServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
        body, _ := ioutil.ReadAll(r.Body)
        w.WriteHeader(http.StatusOK)
        fmt.Fprintln(w, string(body))
    }))
    defer srv.Close()

    payload := struct {
        Name string `json:"name"`
        Age  int    `json:"age"`
    }{
        Name: "Robert",
        Age:  30,
    }
    buf, _ := json.Marshal(payload)
    client := srv.Client()

    resp, err := client.Post(srv.URL, "application/json", bytes.NewReader(buf))
    if err != nil {
        t.Fatal(err)
    }
    defer resp.Body.Close()

    if resp.StatusCode != http.StatusOK {
        t.Errorf("expected %d but got %d", http.StatusOK, resp.StatusCode)
    }
    body, _ := ioutil.ReadAll(resp.Body)
    if string(body) != `{"name":"Robert","age":30}` {
        t.Errorf("expected %s but got %s", `{"name":"Robert","age":30}`, string(body))
    }
}