时间戳

首页 > 解决方案 > 如何在 golang 中测试 http.Redirect 目标 URL

go - 如何在 golang 中测试 http.Redirect 目标 URL

问题描述

我已经在 go 处理程序中设置了一个 http 重定向，并且我正在尝试验证目标 URL 是否按预期组成。我目前正在使用 httptest 进行相关的单元测试。

在处理程序中：

url := fmt.Sprintf("%s%s?token=%s", domain, someURL, token)
http.Redirect(w, r, url, 302)

当前的单元测试只能验证响应代码，不能验证目标 URL：

resp := httptest.NewRecorder()
router.ServeHTTP(resp, req)
assert.Equal(t, 302, resp.Code)

我期待类似的东西

assert.Equal(t, expectedURL, resp.redirectURL)

标签： gotesting

解决方案

断言Location标头符合预期：

assert.Equal(t, expectedURL, resp.HeaderMap.Get("Location"))

推荐阅读