需用httptest.NewRecorder包装ResponseWriter并传给中间件,构造含正确Path/Method的请求;通过包装器字段断言status/size,用defer+time.Now()验证耗时,重定向log.SetOutput至bytes.Buffer后断言日志内容。

怎么用 httptest 测试日志中间件的响应状态和耗时
日志中间件本身不修改响应体,但必须准确捕获 status、size 和 latency。测试重点不是“有没有打日志”,而是“日志字段是否真实反映 HTTP 交互结果”。
- 必须用
httptest.NewRecorder()包装原始http.ResponseWriter,再传给中间件包装后的 handler - 构造请求时别漏掉
r.URL.Path和r.Method,否则日志里 path 和 method 会是空或默认值 - 如果中间件内部调用了
log.Printf或zap.Info,测试时无法断言日志内容——得改用testify/mock或重定向log.SetOutput到bytes.Buffer - 验证耗时不能靠
time.Sleep模拟,而应检查中间件是否在next.ServeHTTP前后正确记录了time.Now()差值
为什么直接测 log.Print 输出会失败
Go 的 log 包默认输出到 os.Stderr,单元测试中无法直接捕获。更关键的是:日志中间件的典型结构是先记录“start”,再调 next.ServeHTTP,最后记录“end”——但如果你没让 next 实际执行(比如传了个空 handler),那“end”日志根本不会触发。
- 错误写法:
log.Print("start"); next(w, r); log.Print("end")—— 中间件 panic 或next提前 return 时,“end”永远不会执行 - 正确做法:用
defer+ 时间戳差值,确保“end”逻辑一定运行(哪怕 panic) - 测试时用
log.SetOutput(&buf)替换输出目标,再用strings.Contains(buf.String(), "200")断言关键字段
如何验证自定义 responseWriter 是否正确捕获 status 和 size
所有靠谱的日志中间件都依赖一个包装了 http.ResponseWriter 的类型来获取真实状态码和写入字节数。这个包装器必须被中间件创建并传给 next.ServeHTTP,否则 status 始终是 0,size 始终是 0。
使用 uber-go/dig 在 Golang 中实现依赖注入——基于反射容器,提供 Provide/Invoke、dig.In/dig.Out 参数与结果对象,支持命名依赖等
- 必须在
WriteHeader方法里显式赋值rw.status = status,不能只靠Write里的 fallback(因为 404/500 等非 200 响应不会进Write) -
Write方法里要判断rw.status == 0才 fallback 到 200,这是 HTTP 规范要求:未显式调用WriteHeader时,默认状态是 200 - 测试时可构造一个返回 404 的
http.HandlerFunc,喂给中间件,然后断言包装器的status字段是否等于 404 - 别复用同一个包装器实例——每个请求必须 new 一个,否则并发下 status/size 会错乱
Gin 框架下测日志中间件要注意什么
Gin 的日志中间件通常接收 *gin.Context,而不是标准 http.Handler,所以不能直接用 net/http/httptest 套路。你得先初始化一个 gin.Engine,注册中间件,再用 httptest.NewRequest 发请求过去。
立即学习“go语言免费学习笔记(深入)”;
- 必须调
gin.SetMode(gin.TestMode),否则默认的调试日志会刷屏,干扰你自己的日志断言 - 别忘了注册路由:
router.GET("/test", handler),否则 404 不是中间件问题,是路由没配 - Gin 的
c.Writer.Status()是安全的,它内部已经做了包装,但你写的中间件如果自己实现responseWriter,仍需确保它和 Gin 的 writer 兼容(即嵌套c.Writer而不是裸http.ResponseWriter) - 如果中间件里用了
c.Next(),测试时必须保证后续 handler 至少执行一次c.Abort()或c.Status(404),否则 status 可能还是 0
time.Since(start) 可能因调度延迟出现毫秒级抖动,导致断言失败。真要严格验证耗时,得用 gomonkey 打桩 time.Now,或者只断言是否大于 0,而非具体数值。

















