如何在单元测试中按顺序执行Go函数？

“纵有疾风来，人生不言弃”，这句话送给正在学习Golang的朋友们，也希望在阅读本文《如何在单元测试中按顺序执行Go函数？》后，能够真的帮助到大家。我也会在后续的文章中，陆续更新Golang相关的技术文章，有好的建议欢迎大家在评论留言，非常感谢！

我是 go 新手，想为这个小型 api 编写单元测试：

• 登录请求。
• 注销请求。 我希望它们能够按顺序执行，并且两个请求都会成功。

但是，当我执行testapi时，最后一个断言总是错误的，并告诉我当前用户（abc@abc）未登录。我知道它们是并行运行的（因此在处理注销请求，后端cookie尚未存储此用户名），但我不知道如何重写，以便登录请求始终在注销之前发生请求。

我不想浪费你的时间，但我用谷歌搜索了很长一段时间，但没有找到适合我的情况的解决方案。 非常感谢您的帮助！

func PostJson(uri string, param map[string]string, router *gin.Engine) *httptest.ResponseRecorder {
    jsonByte, _ := json.Marshal(param)
    req := httptest.NewRequest("POST", uri, bytes.NewReader(jsonByte))
    w := httptest.NewRecorder()
    router.ServeHTTP(w, req)
    return w
}

func TestAPI(t *testing.T) {
    server := setUpServer()
    var w *httptest.ResponseRecorder

    param := make(map[string]string)
    param["email"] = "abc@abc"
    param["password"] = "123"

    param2 := make(map[string]string)
    param2["email"] = "abc@abc"

    urlLogin := "/login"
    w = PostJson(urlLogin, param, server)
    assert.Equal(t, 200, w.Code)
    assert.Equal(t, w.Body.String(), "{\"msg\":\"login success\",\"status\":\"success\"}")

    urlLogout := "/logout"
    w = PostJson(urlLogout, param2, server)
    assert.Equal(t, 200, w.Code)
    assert.Equal(t, w.Body.String(), "{\"msg\":\"logout success\",\"status\":\"success\"}")

正确答案

单个测试中的语句按顺序运行

单个测试中的语句按顺序运行。这些语句都在同一个顶级 testapi 测试函数中，因此它们按顺序运行。

关于 cookie

cookie 由前端持有，其内容不可信任

不存在“后端”cookie 这样的东西。 cookie 是来自后端的响应标头，客户端将其包含在后续请求的请求标头中。

这意味着客户可以完全控制其 cookie 的内容。一般来说，cookie 包含一个不可猜测的随机值来识别用户，该值与后端会话存储中的用户数据相关联。

如果您确实将用户名放入 cookie 中，然后信任该用户名，那么您的应用程序的安全性很容易被绕过（我只是在发出请求之前在 cookie 中设置我想要的任何用户的名称） .

gin 应该提供一个安全的会话管理系统，也许由 redis 之类的东西支持。确保您正在使用它，而不是真正将用户名放入 cookie 中。

客户端必须存储 cookie 并将其包含在后续请求中

您的 http 客户端有责任在后续请求中包含先前响应的 cookie。您没有执行任何操作来将登录时的响应 cookie 包含在注销时的请求 cookie 中。因此注销请求是“未登录”。

您需要的是一个“cookie jar”，这是一个常见术语，表示为后续请求存储和包含 cookie 的机制。其中一个由 https://pkg.go.dev/net/http/cookiejar 提供，可以添加到 http.client，但您实际上并未使用 http.client - 您直接根据请求调用服务器处理程序。

简单选项：让 http.client 为您处理请求 cookie：

重构您的测试以使用 testhttp 运行“真正的”http 服务器，然后进行 针对服务器的“真实”请求。

使用 https://pkg.go.dev/net/http/httptest#Server 在本地主机上启动测试 http 服务器，并使用您从路由器提供的处理程序。

使用 jar 创建 http.client (https://pkg.go.dev/net/http#Client.Jar)

然后只需重用 http 客户端来发出请求，包括 cookie。

https://pkg.go.dev/net/http/cookiejar#example-New 有一个很好的例子来说明如何做到这一点。

更多工作：使用 responserecorder 将 cookie 存储在您的案例中，并直接调用 http 响应处理程序

如果您想做 http.client 可以为您做的事情，请在测试级别创建一个 cookie jar。

jar, err := cookiejar.new(&cookiejar.options{publicsuffixlist: publicsuffix.list})
if err != nil {
    t.fail(err)
}

在每个请求中包含 jar 中的 cookie，并返回将响应中的 cookie 添加到 jar 中。执行此操作的明显位置是 postjson 代码，jar 可以作为附加参数传递给该代码。

在发出请求之前，获取 url 为 https://pkg.go.dev/net/http#CookieJar.Cookies 的 jar cookie。然后通过迭代 cookie 列表并为每个 cookie 调用 req 的 https://pkg.go.dev/net/http#Request.AddCookie 将这些 cookie 添加到请求中。

然后像现在一样“提出”请求。

响应返回后，您必须将cookie添加到cookie jar中。

您可以在响应记录器上通过 https://pkg.go.dev/net/http/httptest#ResponseRecorder.Result 访问响应 cookie，这将返回 * http.response。然后，您可以根据该回复致电 https://pkg.go.dev/net/http#Response.Cookies。

使用 https://pkg.go.dev/net/http#CookieJar.SetCookies 将这些 cookie 添加到罐子中。

代码可能类似于：

func PostJson(uri string, param map[string]string, router *gin.Engine, jar http.CookieJar) *httptest.ResponseRecorder {
    jsonByte, _ := json.Marshal(param)
    req := httptest.NewRequest("POST", uri, bytes.NewReader(jsonByte))
    for _, cookie := range jar.Cookies(uri) {
       req.AddCookie(cookie)
    }
    w := httptest.NewRecorder()
    router.ServeHTTP(w, req)
    jar.SetCookies(uri, w.Result().Cookies())
    return w
}

关于问题中可重现代码的价值的最后说明。

如果您有 made your code something I could run without significant modification，我会花时间实施和测试我建议的更改，然后我可以给您一个可行的示例。因此，下次当您有一个很好的堆栈溢出问题时，请花时间投资我们可以运行的完整代码，也许在 a Go Playground share 中。除了更容易提供高质量的答案之外，这对您来说也是极好的实践！

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