本文介绍如何解决 Martini 框架中因 httptest.ResponseRecorder 缺失 CloseNotify() 方法导致的测试 panic 问题，并提供可直接运行的兼容性修复方案。
本文介绍如何解决 martini 框架中因 `httptest.responserecorder` 缺失 `closenotify()` 方法导致的测试 panic 问题，并提供可直接运行的兼容性修复方案。
Martini 已于多年前停止维护（见其 GitHub README），但若项目仍依赖该框架实现反向代理逻辑，测试时会遇到一个典型兼容性问题：Martini 内部强制将 http.ResponseWriter 断言为 http.CloseNotifier 接口，而标准库中的 httptest.ResponseRecorder 并未实现该方法（自 Go 1.8 起 CloseNotifier 已被弃用并移除），从而触发 panic：
[martini] PANIC: interface conversion: *httptest.ResponseRecorder is not http.CloseNotifier: missing method CloseNotify
根本原因在于 Martini 的设计缺陷——它未定义组合接口，而是直接假设响应写入器同时支持写入与连接关闭通知。虽然现代 Go 已废弃 CloseNotifier（推荐使用 Request.Context().Done()），但为兼容旧版 Martini，需手动构造一个满足其期望的包装类型。
以下是一个轻量、线程安全的修复方案：定义 closeNotifyingRecorder 结构体，内嵌 *httptest.ResponseRecorder 并实现 CloseNotify() 方法：
type closeNotifyingRecorder struct {
*httptest.ResponseRecorder
closed chan bool
}
func newCloseNotifyingRecorder() *closeNotifyingRecorder {
return &closeNotifyingRecorder{
httptest.NewRecorder(),
make(chan bool, 1),
}
}
func (c *closeNotifyingRecorder) close() {
select {
case c.closed <- true:
default:
}
}
func (c *closeNotifyingRecorder) CloseNotify() <-chan bool {
return c.closed
}注意：closed 通道使用带缓冲的 chan bool（容量为 1），并在 close() 中使用 select+default 避免重复写入导致 panic；CloseNotify() 返回只读通道，符合原接口语义。
完整可运行测试示例如下：
package main
import (
"net/http"
"net/http/httptest"
"net/http/httputil"
"net/url"
"testing"
"github.com/go-martini/martini"
"github.com/stretchr/testify/assert"
)
// ... 上述 closeNotifyingRecorder 定义 ...
func TestReverseProxy(t *testing.T) {
// 启动模拟后端服务
backendResponse := "I am the backend"
backend := httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
w.Header().Set("Content-Type", "text/plain")
w.Write([]byte(backendResponse))
}))
defer backend.Close()
backendURL, _ := url.Parse(backend.URL)
// 构建 Martini 反向代理路由
m := martini.Classic()
m.Get("/", func(w http.ResponseWriter, r *http.Request) {
proxy := httputil.NewSingleHostReverseProxy(backendURL)
proxy.ServeHTTP(w, r)
})
// 执行集成测试
req, _ := http.NewRequest("GET", "/", nil)
res := newCloseNotifyingRecorder()
m.ServeHTTP(res, req)
// 验证响应状态与内容
assert.Equal(t, 200, res.Code, "expected HTTP 200 OK")
assert.Equal(t, backendResponse, res.Body.String(), "response body mismatch")
}✅ 关键注意事项：
- 此方案仅用于遗留 Martini 项目测试，新项目强烈建议迁移至 net/http 原生 Handler、Gin、Echo 或 Fiber 等活跃维护的框架；
- CloseNotify() 在本例中仅为满足 Martini 类型断言，实际测试中无需主动调用 close() —— Martini 会在内部触发该通道；
- 若需验证响应头（如 Content-Type），请确保后端 handler 显式设置（如示例所示）；
- Martini 的 Classic() 默认启用日志中间件，若测试输出干扰 CI，可通过 martini.New() 手动组装精简中间件链。
综上，通过封装 ResponseRecorder 实现 CloseNotifier，即可绕过 Martini 的类型限制，安全、可靠地完成反向代理逻辑的端到端测试。
