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

时间:2018-06-02 16:30:04

标签: unit-testing go null runtime-error

我遇到了无效的内存地址或无指针取消引用"为我的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.VerifySession的模拟版本:

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()的第一部分写了一个通过测试:如果没有" session"找到cookie后,测试确认响应标题已得到适当更新。但是,当我通过"会话" cookie,与上面的测试用例一样,Authenticate()将最多运行r, err := env.Db.VerifySession(ck.Value)。尝试在调试模式下跳过或进入该函数调用会导致抛出nil指针错误。

我不认为问题是env.Db变量,因为我已成功运行一套测试,我的非中间件处理程序使用相同的变量而没有问题。

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

非常感谢任何有关此处发生的事情的指导。

1 个答案:

答案 0 :(得分:0)

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

我想我会将中间件定义为env。

上的方法