17370845950

go 项目中测试 http 服务时，需避免在测试文件中定义 `main()` 函数，并确保测试函数以大写 `test` 开头、属于同一 `package main`，才能被 `go test` 正确识别和执行。

在 Go 中为基于 net/http 的 Web 应用编写测试，核心原则是：测试文件（如 beacon_test.go）必须与主程序文件（如 beacon.go）位于同一包（package main），且不能包含 func main() —— 因为 main 函数仅允许在一个可执行包中出现一次。

你遇到的错误：

./beacon_test.go:11: main redeclared in this block
    previous declaration at ./beacon.go:216

正是因为 beacon_test.go 复制了官方 httptest 示例中的完整可运行程序（含 func main()），而该示例本意是独立演示，不可直接作为测试文件使用。

✅ 正确做法如下：

1. 保持包声明一致：beacon.go 和 beacon_test.go 都应以 package main 开头；
2. 移除 main() 函数：测试文件中不写 func main()，改用标准测试函数签名；
3. 命名规范：测试函数必须以 Test 开头（首字母大写），接收 *testing.T 参数；
4. 使用 httptest.NewServer 或 httptest.NewRecorder：模拟 HTTP 服务或捕获 handler 输出。

例如，假设 beacon.go 包含一个简单 HTTP handler：

// beacon.go
package main

import (
    "fmt"
    "net/http"
)

func handler(w http.ResponseWriter, r *http.Request) {
    fmt.Fprint(w, "OK")
}

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

对应的测试文件应写为：

// beacon_test.go
package main

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

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

    handler(w, req)

    if w.Code != http.StatusOK {
        t.Errorf("expected status OK; got %v", w.Code)
    }
    if w.Body.String() != "OK" {
        t.Errorf("expected body 'OK'; got %q", w.Body.String())
    }
}

⚠️ 注意事项：

• 不要将 httptest.NewServer 用于单元测试 handler 逻辑（它启动真实监听端口，适合集成测试）；优先用 httptest.NewRecorder 进行快速、隔离的 handler 单元测试；
• 若需测试整个 HTTP 服务生命周期（如路由、中间件、端口绑定），再考虑 NewServer + http.Client 组合，并记得调用 server.Close() 清理资源；
• 运行测试：在项目根目录执行 go test -v（-v 显示详细输出），确保当前目录下只有 main 包的 .go 文件。

总结：Go 的测试机制依赖严格的命名与包规则——测试函数不是“可执行程序”，而是由 go test 框架驱动的回调。摒弃复制粘贴完整 main 示例的习惯，转而聚焦于 testing.T 驱动的断言逻辑，才能写出健壮、可维护的 HTTP 测试代码。