Golang如何测试API请求错误处理-Golang-PHP中文网

Golang如何测试API请求错误处理

P粉602998670

发布： 2025-10-11 19:49:01

原创

509人浏览过

答案是使用 httptest 和自定义 RoundTripper 模拟错误响应与网络故障，通过依赖注入传递 HTTP 客户端，测试时验证错误类型与消息。

golang如何测试api请求错误处理

测试 Go 中 API 请求的错误处理，关键在于模拟各种异常情况并验证程序是否能正确响应。不需要依赖真实网络请求，使用标准库和接口抽象就能写出可靠、可维护的测试。

使用 net/http/httptest 模拟 HTTP 服务

Go 的 httptest 包可以创建临时的 HTTP 服务，用来模拟后端返回错误状态码或网络问题。

比如你想测试当远程 API 返回 500 或超时，你的客户端能否正确处理：

定义一个简单的客户端：

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

func FetchData(client *http.Client, url string) error {
    resp, err := client.Get(url)
    if err != nil {
        return fmt.Errorf("request failed: %w", err)
    }
    defer resp.Body.Close()

    if resp.StatusCode != http.StatusOK {
        return fmt.Errorf("unexpected status: %d", resp.StatusCode)
    }
    return nil
}

登录后复制

在测试中用 httptest.NewServer 模拟返回 500：

func TestFetchData_ServerError(t *testing.T) {
    server := httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
        w.WriteHeader(http.StatusInternalServerError)
    }))
    defer server.Close()

    client := &http.Client{}
    err := FetchData(client, server.URL)

    if err == nil {
        t.Fatal("expected error, got nil")
    }
    if !strings.Contains(err.Error(), "unexpected status: 500") {
        t.Errorf("wrong error message: %v", err)
    }
}

登录后复制

模拟网络失败（如连接超时）

你可以通过自定义 RoundTripper 来模拟完全的网络故障，比如连接被拒绝或超时。

例如，让所有请求都返回“connection refused”：

千图设计室AI海报

千图网旗下的智能海报在线设计平台

172

查看详情

type failingRoundTripper struct{}

func (f *failingRoundTripper) RoundTrip(*http.Request) (*http.Response, error) {
    return nil, errors.New("connection refused")
}

func TestFetchData_NetworkFailure(t *testing.T) {
    client := &http.Client{
        Transport: &failingRoundTripper{},
    }

    err := FetchData(client, "https://example.com")

    if err == nil {
        t.Fatal("expected network error")
    }
    if !strings.Contains(err.Error(), "request failed") {
        t.Errorf("error should wrap network error")
    }
}

登录后复制

注入可控的 HTTP 客户端

为了便于测试，不要在函数内部直接使用 http.DefaultClient。应该把 *http.Client 作为参数传入，或者使用接口依赖注入。

这样你可以在测试中替换为 mock 客户端或自定义 RoundTripper，灵活控制行为。

常见做法：

将 Client 作为结构体字段
在测试时替换 Transport 实现
使用 httptest.Server 提供预设响应

测试错误类型与消息

除了判断是否有错，还应检查错误类型或内容是否符合预期，便于上层逻辑做判断。

例如你可以定义自定义错误类型：

type HTTPError struct {
    StatusCode int
}

func (e *HTTPError) Error() string {
    return fmt.Sprintf("HTTP error: %d", e.StatusCode)
}

登录后复制

然后在处理非 2xx 响应时返回它，并在测试中用 errors.As 验证：

var httpErr *HTTPError
if errors.As(err, &httpErr) {
    if httpErr.StatusCode != 500 {
        t.Errorf("expected 500, got %d", httpErr.StatusCode)
    }
} else {
    t.Error("expected HTTPError")
}

登录后复制

基本上就这些。核心是隔离外部依赖，用 httptest 和自定义 Transport 控制输入，再断言错误输出。不复杂但容易忽略细节。

以上就是Golang如何测试API请求错误处理的详细内容，更多请关注php中文网其它相关文章！