Auth: support JWT Authentication (#29995)

pkg/middleware/testing.go

@@ -24,12 +24,14 @@ type scenarioContext struct {
 	resp                 *httptest.ResponseRecorder
 	apiKey               string
 	authHeader           string
+	jwtAuthHeader        string
 	tokenSessionCookie   string
 	respJson             map[string]interface{}
 	handlerFunc          handlerFunc
 	defaultHandler       macaron.Handler
 	url                  string
 	userAuthTokenService *auth.FakeUserAuthTokenService
+	jwtAuthService       *models.FakeJWTService
 	remoteCacheService   *remotecache.RemoteCache
 	cfg                  *setting.Cfg
 	sqlStore             *sqlstore.SQLStore
@@ -53,6 +55,11 @@ func (sc *scenarioContext) withAuthorizationHeader(authHeader string) *scenarioC
 	return sc
 }
 
+func (sc *scenarioContext) withJWTAuthHeader(jwtAuthHeader string) *scenarioContext {
+	sc.jwtAuthHeader = jwtAuthHeader
+	return sc
+}
+
 func (sc *scenarioContext) fakeReq(method, url string) *scenarioContext {
 	sc.t.Helper()
 
@@ -95,6 +102,11 @@ func (sc *scenarioContext) exec() {
 		sc.req.Header.Set("Authorization", sc.authHeader)
 	}
 
+	if sc.jwtAuthHeader != "" {
+		sc.t.Logf(`Adding header "%s: %s"`, sc.cfg.JWTAuthHeaderName, sc.jwtAuthHeader)
+		sc.req.Header.Set(sc.cfg.JWTAuthHeaderName, sc.jwtAuthHeader)
+	}
+
 	if sc.tokenSessionCookie != "" {
 		sc.t.Log(`Adding cookie`, "name", sc.cfg.LoginCookieName, "value", sc.tokenSessionCookie)
 		sc.req.AddCookie(&http.Cookie{