echo c.Get("пользователь") не работает при тестировании env

Я пытаюсь протестировать API golang, созданный на основе echo-фреймворка / маршрутизатора. У меня есть следующий тест.....

func TestLogout(t *testing.T) {
loadConfig()
db := stubDBs(t)
Convey("When you post to /logout", t, func() {
    Convey("with a valid token, you should get aa success msg and be logged out", func() {
        e := echo.New()
        e.Use(middleware.JWTWithConfig(middleware.JWTConfig{
            SigningKey:  []byte("secret"),
            TokenLookup: "query:user",
        }))
        req, err := http.NewRequest(echo.POST, "auth/logout", strings.NewReader(""))
        if err == nil {
            req.Header.Set(echo.HeaderAuthorization, fmt.Sprintf("Bearer %v", Token))
            rec := httptest.NewRecorder()
            c := e.NewContext(req, rec)
            Logout(db, Models.Redis)(c)
            body := GetJsonBody(rec.Body.String())
            error := body.Path("error").Data()
            msg := body.Path("error").Data().(string)
            pw := body.Path("data.user.password").Data().(string)
            token := body.Path("data.user.token").Data()
            So(error, ShouldEqual, nil)
            So(msg, ShouldEqual, "Logged Out")
            So(rec.Code, ShouldEqual, 200)
            So(pw, ShouldEqual, "")
            So(token, ShouldEqual, nil)
        }

    })
})
}

и в контроллере....

//Logout ...
func Logout(db *sqlx.DB, r *redis.Client) echo.HandlerFunc {
    return func(c echo.Context) error {
        var user Models.User
        log.Println(c.Get("user")) //<-----This Logs NIL only on testing
        if c.Get("user") == nil {
            res := createAuthErrorResponse(user, "Invalid Token")
            return c.JSON(http.StatusOK, res)
        }
        token := c.Get("user").(*jwt.Token)
        err := Models.Redis.Del(token.Raw).Err()
        if err != nil {
            handleErr(err)
            res := createAuthErrorResponse(user, "Token not in storage")
            return c.JSON(http.StatusOK, res)
        }
        user.Password = ""
        user.Token = null.StringFrom("")
        res := createAuthSuccessResponse(user, "Logged Out.")
        return c.JSON(http.StatusOK, res)
    }
}

Любой совет о том, как я могу проверить эту функциональность в рамках чо? Такое поведение я бы не ожидал (c.Get("пользователь") не действует так же при тестировании, как в live env).

1 ответ

Через много боли я смог понять это. Вот рабочий код. Объяснение, чтобы следовать.

        Convey("with a valid token, you should get a success msg and be logged out", func() {
        e := echo.New()
        req, err := http.NewRequest(echo.POST, "auth/logout", strings.NewReader(""))
        if err == nil {
            req.Header.Set(echo.HeaderAuthorization, fmt.Sprintf("Bearer %v", Token))
            rec := httptest.NewRecorder()
            c := e.NewContext(req, rec)
            middleware.JWTWithConfig(middleware.JWTConfig{
                SigningKey: []byte("secret"),
            })(Logout(db, Models.Redis))(c)
            body := GetJsonBody(rec.Body.String())
            error := body.Path("error").Data()
            msg := body.Path("msg").Data().(string)
            pw := body.Path("data.user.password").Data().(string)
            token := body.Path("data.user.token").Data()
            So(error, ShouldEqual, nil)
            So(msg, ShouldEqual, "Logged Out")
            So(rec.Code, ShouldEqual, 200)
            So(pw, ShouldEqual, "")
            So(token, ShouldEqual, nil)
        }

    })

Я правильно установил свой заголовок, но не выполнял промежуточное ПО JWT echo, что означало, что мой токен не был проверен и стал доступным с помощью команды c.Get().

Ключевым моментом здесь является то, что тестирование промежуточного программного обеспечения echo без запуска сервера требует от вас выполнения функции промежуточного программного обеспечения, которая затем принимает функцию-обработчик (это ваша функция, которую вы хотите протестировать), а затем получает контекст. Таким образом, функция должна выглядеть следующим образом, чтобы проверить ваши промежуточные маршруты. Если это все заглавные буквы, вам нужно заменить его на ваш конкретный случай.

A: middleware.MIDDLEWARE_FUNC(NECESSARY_CONFIGS) // <--- this returns echo.MiddlewareFunc

B: (YOUR_HANDLER(req, res)) // <---echo.MiddlewareFunc takes an echo.HanlderFunc as its only argument and you pass your handler you test req, res

(c) // <--Finally, you call the resulting compose function with your test context

A(B())(c)
Другие вопросы по тегам