需要帮助查明中间件测试(Golang)中零引用错误的原因

Question

我在为 http 处理程序编写单元测试时遇到了"无效内存地址或零指针取消引用"错误 - 特别是身份验证中间件。

以下是要测试的函数：

func Authenticate(next http.Handler) http.Handler {
return http.HandlerFunc(func(w http.ResponseWriter, req *http.Request) {
var err error
ck, err := req.Cookie("session")
if err != nil {
w.Header().Add("Authorization", "expired/invalid")
next.ServeHTTP(w, req)
return
}
r, err := env.Db.VerifySession(ck.Value)
if err != nil {
http.Error(w, http.StatusText(500), http.StatusInternalServerError)
return
}
if r == "expired/invalid" {
ck.MaxAge = -1
http.SetCookie(w, ck)
w.Header().Add("Authorization", r)
next.ServeHTTP(w, req)
return
}
w.Header().Add("Authorization", r)
next.ServeHTTP(w, req)
})
}

引发错误的特定测试用例：

rec = httptest.NewRecorder()
req, _ = http.NewRequest("GET", "/", nil)
c := &http.Cookie{
Name:  "session",
Value: "success",
}
req.AddCookie(c)
routes.Authenticate(http.NotFoundHandler()).ServeHTTP(rec, req)
assert.Equal("admin", rec.Header().Get("Authorization"))

env的模拟版本。.DB。用于测试的验证会话：

func (mdb *mockdb) VerifySession(val string) (string, error) {
switch val {
case "success":
return "admin", nil
case "failure":
return "expired/invalid", nil
default:
return "", errors.New("500")
}
}

我已经为 Authenticate(( 的第一部分编写了一个通过测试：如果未找到"会话"cookie，则测试确认响应标头已适当更新。但是，当我传入"会话"cookie时，与上面的测试用例一样，Authenticate(( 将运行到r, err := env.Db.VerifySession(ck.Value).尝试在调试模式下单步跳过或进入该函数调用会导致引发 nil 指针错误。

我不认为问题出在环境上。db 变量，因为我已经使用相同的变量在我的非中间件处理程序上成功运行了一套测试，没有问题。

我还可以通过调试确认"会话"cookie确实存在。

有关这里发生的事情的任何指导将不胜感激。

Answer 1

没关系，我意识到问题是什么。对于我的其他路由，我将 env 变量作为参数传入。我需要找到一种方法来为中间件提供对它的访问。

我想我会将中间件定义为 env 上的一种方法。