如何在单元测试中利用 main 函数测试 Gin？

本篇文章主要是结合我之前面试的各种经历和实战开发中遇到的问题解决经验整理的，希望这篇《如何在单元测试中利用 main 函数测试 Gin？》对你有很大帮助！欢迎收藏，分享给更多的需要的朋友学习~

我想对杜松子酒控制器功能进行单元测试。

package controllers

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

    "github.com/gin-gonic/gin"
    "github.com/go-playground/assert/v2"
)

func testgetdivisions(t *testing.t) {
    tt := []struct {
        name         string
        enterpriseid string
        code         int
    }{
        {name: "enterprise id exists", enterpriseid: "1", code: 200},
        {name: "enterprise id cannot be string", enterpriseid: "string", code: 404},
    }
    w := httptest.newrecorder()
    c, _ := gin.createtestcontext(w)

    for _, tc := range tt {
        c.params = []gin.param{
            {
                key:   "enterprise_id",
                value: tc.enterpriseid,
            },
        }
        getdivisions(c) // error from here
        assert.equal(t, tc.code, w.code, tc.name)
    }
}

问题是，每当我在测试文件中使用 getdivision 函数时都会出现如下错误：

--- fail: testgetdivisions (0.00s)
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=0x28 pc=0x43c7438]

找不到问题出在哪里。我可以将测试上下文作为参数提供给函数吗？

这是我的 getdivisions 处理程序：

// Get divisions of an enterprise by given id. Default id = "1"
func GetDivisions(c *gin.Context) {
    var divisions []models.Division

    if err := config.DB.Find(&divisions).Error; err != nil {
        c.JSON(http.StatusInternalServerError, err.Error())
        return
    }

    c.JSON(http.StatusOK, gin.H{
        "divisions": divisions,
    })
}

正确答案

似乎 config.DB.Find(&divisions) 中发生了错误，因为我认为从数据库读取并写入变量时会发生无效内存错误...您能显示 config.DB.Find( 的代码吗？ ）。 （如果在 GitHub 上，可以给我发送链接）

今天带大家了解了的相关知识，希望对你有所帮助；关于Golang的技术知识我们会一点点深入介绍，欢迎大家关注golang学习网公众号，一起学习编程~