Golang如何模拟HTTP请求进行测试_Golang HTTP请求模拟测试实践_技术教程

使用httptest可创建模拟HTTP服务器进行单元测试。1. 通过httptest.NewServer模拟接口返回JSON数据，验证状态码和响应内容；2. 可模拟404、500、超时等异常场景，确保客户端正确处理错误；3. 替换默认Client为自定义Client，实现对超时和请求行为的控制；4. 封装APIClient并注入模拟服务器URL，隔离外部依赖，提升测试可靠性与效率。

在Go语言开发中，对HTTP请求进行测试是保证服务稳定性的关键环节。直接调用外部API不利于测试的可重复性和效率，因此需要模拟HTTP请求。通过net/http/httptest包，我们可以轻松创建虚拟的HTTP服务器来模拟接口行为，从而安全、高效地完成单元测试。

使用 httptest 模拟HTTP服务器

Go标准库中的httptest包提供了NewServer方法，可以快速启动一个临时的HTTP服务用于测试。这个服务不会绑定真实端口，仅在测试过程中存在，适合模拟第三方API或内部服务。

以下是一个简单的例子，模拟一个返回JSON数据的GET接口：

func TestAPICall(t *testing.T) {
    // 创建模拟服务器
    server := httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
        w.Header().Set("Content-Type", "application/json")
        w.WriteHeader(http.StatusOK)
        fmt.Fprintln(w, `{"message": "success", "code": 200}`)
    }))
    defer server.Close()

    // 使用模拟服务器地址发起请求
    resp, err := http.Get(server.URL)
    if err != nil {
        t.Fatal(err)
    }
    defer resp.Body.Close()

    if resp.StatusCode != http.StatusOK {
        t.Errorf("期望状态码 200，实际得到 %d", resp.StatusCode)
    }

    var data map[string]interface{}
    if err := json.NewDecoder(resp.Body).Decode(&data); err != nil {
        t.Fatal(err)
    }

    if msg := data["message"]; msg != "success" {
        t.Errorf("期望 message 为 success，实际为 %v", msg)
    }
}

模拟不同状态码和错误场景

真实的网络请求可能遇到各种异常情况，比如404、500、超时等。我们可以通过控制模拟服务器的响应来覆盖这些边界条件。

常见测试场景包括：

返回404 Not Found，验证客户端是否正确处理资源不存在的情况
返回500 Internal Server Error，测试重试逻辑或错误提示
模拟网络超时，可通过time.Sleep配合带超时的http.Client实现
返回无效JSON，检查反序列化是否安全

server := httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
    http.Error(w, "Internal Server Error", http.StatusInternalServerError)
}))
defer server.Close()

替换默认Client以增强控制力

为了更灵活地控制请求行为（如设置超时、拦截请求），建议将http.DefaultClient替换为自定义的http.Client。这样可以在测试中注入模拟服务器，在生产环境中使用正常配置。

示例：封装可配置的HTTP客户端

type APIClient struct {
    Client *http.Client
    BaseURL string
}

func (c *APIClient) GetStatus() (*Status, error) {
    resp, err := c.Client.Get(c.BaseURL + "/status")
    if err != nil {
        return nil, err
    }
    defer resp.Body.Close()

    var status Status
    if err := json.NewDecoder(resp.Body).Decode(&status); err != nil {
        return nil, err
    }
    return &status, nil
}

测试时传入httptest服务器的URL和自定义Client，即可完全隔离外部依赖。