怎样测试Golang的HTTP服务使用httptest包模拟请求-Golang-PHP中文网

怎样测试Golang的HTTP服务使用httptest包模拟请求

P粉602998670

发布： 2025-08-15 09:32:02

原创

704人浏览过

使用 httptest 可以在不启动真实服务器的情况下测试 go 的 http 服务，通过构造请求、捕获响应、断言结果三步完成测试；2. httptest.newrequest 创建模拟请求，httptest.newrecorder 记录响应，直接调用处理器函数进行测试；3. 可测试不同场景如参数缺失或不同 http 方法；4. 若使用 http.servemux，可通过 mux.servehttp 测试路由匹配；5. 运行 go test -v 执行测试并验证结果，确保处理器行为正确。

怎样测试Golang的HTTP服务使用httptest包模拟请求

在 Go 中测试 HTTP 服务时，

net/http/httptest

登录后复制

包是非常有用的工具。它允许你模拟 HTTP 请求和响应，而无需真正启动网络服务，从而实现快速、可靠的单元测试。

下面是如何使用

httptest

登录后复制

测试一个简单的 Golang HTTP 服务的完整示例。

1. 编写一个简单的 HTTP 服务

假设我们有一个处理

/hello

登录后复制

路由的简单服务：

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

// main.go
package main

import (
    "fmt"
    "net/http"
)

func helloHandler(w http.ResponseWriter, r *http.Request) {
    fmt.Fprintf(w, "Hello, %s!", r.URL.Query().Get("name"))
}

登录后复制

我们注册这个处理器：

func main() {
    http.HandleFunc("/hello", helloHandler)
    http.ListenAndServe(":8080", nil)
}

登录后复制

2. 使用

httptest

登录后复制

编写测试

我们为

helloHandler

登录后复制

编写测试，验证它是否正确返回响应。

析稿Ai写作

科研人的高效工具：AI论文自动生成，十分钟万字，无限大纲规划写作思路。

142

查看详情

// main_test.go
package main

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

func TestHelloHandler(t *testing.T) {
    // 创建一个请求
    req := httptest.NewRequest("GET", "/hello?name=Gopher", nil)

    // 创建一个响应记录器
    w := httptest.NewRecorder()

    // 调用处理器
    helloHandler(w, req)

    // 获取响应
    resp := w.Result()
    defer resp.Body.Close()

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

    // 检查响应体
    expected := "Hello, Gopher!"
    if body := w.Body.String(); body != expected {
        t.Errorf("期望响应体 %q，实际得到 %q", expected, body)
    }
}

登录后复制

3. 关键组件说明

```
httptest.NewRequest(method, url, body)
```
登录后复制
创建一个模拟的 HTTP 请求。第三个参数是请求体（如 POST 数据），如果是 GET 可以传
```
nil
```
登录后复制
。
```
httptest.NewRecorder()
```
登录后复制
返回一个
```
*httptest.ResponseRecorder
```
登录后复制
，它实现了
```
http.ResponseWriter
```
登录后复制
接口，可以记录响应的状态码、头信息和响应体，便于后续断言。
直接调用 handler
因为 Go 的处理器是函数，可以直接传入
```
w
```
登录后复制
和
```
req
```
登录后复制
进行调用，不需要启动服务器。

4. 测试其他场景

你可以轻松测试各种情况，比如参数缺失、不同 HTTP 方法等。

func TestHelloHandler_NoName(t *testing.T) {
    req := httptest.NewRequest("GET", "/hello", nil)
    w := httptest.NewRecorder()

    helloHandler(w, req)

    expected := "Hello, !"
    if body := w.Body.String(); body != expected {
        t.Errorf("期望 %q，实际 %q", expected, body)
    }
}

登录后复制

5. 测试路由（使用

http.ServeMux

登录后复制

）

如果你使用了

ServeMux

登录后复制

，也可以测试路由是否正确匹配：

func TestRouter(t *testing.T) {
    mux := http.NewServeMux()
    mux.HandleFunc("/hello", helloHandler)

    req := httptest.NewRequest("GET", "/hello?name=World", nil)
    w := httptest.NewRecorder()

    mux.ServeHTTP(w, req)

    if w.Code != http.StatusOK {
        t.Errorf("期望状态码 %d，实际 %d", http.StatusOK, w.Code)
    }

    expected := "Hello, World!"
    if w.Body.String() != expected {
        t.Errorf("期望 %q，实际 %q", expected, w.Body.String())
    }
}

登录后复制

6. 运行测试

在项目目录下运行：

go test -v

登录后复制

你应该看到类似输出：

=== RUN   TestHelloHandler
--- PASS: TestHelloHandler (0.00s)
=== RUN   TestHelloHandler_NoName
--- PASS: TestHelloHandler_NoName (0.00s)
PASS
ok      your-module-name    0.001s

登录后复制

基本上就这些。使用