加入 Gitee
与超过 1200万 开发者一起发现、参与优秀开源项目,私有仓库也完全免费 :)
免费加入
文件
克隆/下载
request_id_test.go 771 Bytes
一键复制 编辑 原始数据 按行查看 历史
aesoper 提交于 2020-05-19 21:00 . init
/**
* @Author: aesoper
* @Description: 请求Id中间件测试
* @File: request_id_test
* @Version: 1.0.0
* @Date: 2020/5/19 19:13
*/
package gin_middleware
import (
"gitee.com/gin-ecosystem/gin-middleware/consts"
"github.com/gin-gonic/gin"
"github.com/stretchr/testify/assert"
"net/http"
"net/http/httptest"
"testing"
)
func TestNewRequestID(t *testing.T) {
e := gin.New()
e.GET("/", NewRequestID(RequestIDConfig{
Skipper: DefaultSkipper,
Generator: func() string {
return "test-header"
},
}), func(c *gin.Context) {
c.String(200, "test")
return
})
req := httptest.NewRequest(http.MethodGet, "/", nil)
rec := httptest.NewRecorder()
e.ServeHTTP(rec, req)
assert.Equal(t, rec.Header().Get(consts.HeaderXRequestID), "test-header")
}
马建仓 AI 助手
尝试更多
代码解读
代码找茬
代码优化