unit-testing - 如何对 Go Gin 处理程序函数进行单元测试?
问题描述
我有这样的控制器功能....
func GetMaterialByFilter(c *gin.Context) {
queryParam := weldprogs.QueryParam{}
c.BindQuery(&queryParam)
materialByFilter, getErr := services.WeldprogService.GetMaterialByFilter(&queryParam)
if getErr != nil {
//TODO : Handle user creation error
c.JSON(getErr.Status, getErr)
return
}
c.JSON(http.StatusOK, materialByFilter)
}
QueryParam Struct 是这样的..
type QueryParam struct {
Basematgroup_id []string `form:"basematgroup_id"`
License_id []string `form:"license_id"`
Diameter_id []string `form:"diameter_id"`
Gasgroup_id []string `form:"gasgroup_id"`
Wiregroup_id []string `form:"wiregroup_id"`
Wiremat_id []string `form:"wiremat_id"`
}
我的测试功能是这样的..
func TestGetMaterialByFilter(t *testing.T) {
w := httptest.NewRecorder()
c, _ := gin.CreateTestContext(w)
GetMaterialByFilter(c)
assert.Equal(t, 200, w.Code)
var got gin.H
err := json.Unmarshal(w.Body.Bytes(), &got)
if err != nil {
t.Fatal(err)
}
fmt.Println(got)
assert.Equal(t, got, got)
}
在运行此测试时,它给了我以下错误
panic: runtime error: invalid memory address or nil pointer dereference [recovered]
panic: runtime error: invalid memory address or nil pointer dereference
[signal SIGSEGV: segmentation violation code=0x1 addr=0x10 pc=0x97f626]
但是当我在我的控制器函数中注释掉 c.BindQuery() 行时,它成功地运行了我的测试函数。我在这里做错了什么?我可以以某种方式模拟 c.BindQuery 函数吗?
解决方案
要测试涉及 HTTP 请求的操作,您必须实际初始化 an*http.Request
并将其设置为 Gin 上下文。要专门测试c.BindQuery
它足以正确初始化请求的URL
and URL.RawQuery
:
func mockGin() (*gin.Context, *httptest.ResponseRecorder) {
w := httptest.NewRecorder()
c, _ := gin.CreateTestContext(w)
// test request, must instantiate a request first
req := &http.Request{
URL: &url.URL{},
Header: make(http.Header), // if you need to test headers
}
// example: req.Header.Add("Accept", "application/json")
// request query
testQuery := weldprogs.QueryParam{/* init fields */}
q := req.URL.Query()
for _, s := range testQuery.Basematgroup_id {
q.Add("basematgroup_id", s)
}
// ... repeat for other fields as needed
// must set this, since under the hood c.BindQuery calls
// `req.URL.Query()`, which calls `ParseQuery(u.RawQuery)`
req.URL.RawQuery = q.Encode()
// finally set the request to the gin context
c.Request = req
return c, w
}
如果您需要模拟 JSON 绑定,请参阅此答案。
services.WeldprogService.GetMaterialByFilter(&queryParam)
无法按原样测试服务调用。为了可测试,它必须(理想情况下)是一个接口,并以某种方式作为处理程序的依赖项注入。
假设它已经是一个接口,要使其可注入,您要么需要它作为处理程序参数——但这会迫使你更改处理程序的签名——或者将其设置为 Gin 上下文值:
func GetMaterialByFilter(c *gin.Context) {
//...
weldprogService := mustGetService(c)
materialByFilter, getErr := weldprogService.GetMaterialByFilter(&queryParam)
// ...
}
func mustGetService(c *gin.Context) services.WeldprogService {
svc, exists := c.Get("svc_context_key")
if !exists {
panic("service was not set")
}
return svc.(services.WeldprogService)
}
然后你可以在你的单元测试中模拟它:
type mockSvc struct {
}
// have 'mockSvc' implement the interface
func TestGetMaterialByFilter(t *testing.T) {
w := httptest.NewRecorder()
c, _ := gin.CreateTestContext(w)
// now you can set mockSvc into the test context
c.Set("svc_context_key", &mockSvc{})
GetMaterialByFilter(c)
// ...
}
推荐阅读
- c++ - Boost ASIO - 如何在 TCP 消息前面附加标头
- r - 基于查找表使用 dplyr 重新编码矩阵值
- android - 新创建的服务的 AndroidP-SELinux 权限被拒绝
- r - 将系统时间转换为纽约时区
- sql - 在所有具有给定名称的列的表中,根据不是某个字母的条件替换某个位置的字符
- python - 将张量移动到 cuda 设备会导致 Pytorch 中的非法内存访问
- reactjs - 无法涵盖具有 props 的 React 组件的单元测试用例
- node.js - 如何使用 Node.js 从 Twilio whatsapp API 下载图像
- javascript - 为什么我的 JS 函数参数未定义?
- powerbi - 如何从另一个没有关系的维度表中获取列