replace log.Fatal with 500 error mesagge.

Signed-off-by: Bo-Yi Wu <[email protected]>

Author: appleboy

auth_jwt.go

@@ -4,7 +4,6 @@ import (
 	"errors"
 	"github.com/gin-gonic/gin"
 	"gopkg.in/dgrijalva/jwt-go.v3"
-	"log"
 	"net/http"
 	"strings"
 	"time"
@@ -65,17 +64,6 @@ type Login struct {
 
 // MiddlewareInit initialize jwt configs.
 func (mw *GinJWTMiddleware) MiddlewareInit() error {
-	if mw.Realm == "" {
-		return errors.New("realm is required")
-	}
-
-	if mw.Authenticator == nil {
-		return errors.New("authenticator is required")
-	}
-
-	if mw.Key == nil {
-		return errors.New("secret key is required")
-	}
 
 	if mw.SigningAlgorithm == "" {
 		mw.SigningAlgorithm = "HS256"
@@ -100,13 +88,28 @@ func (mw *GinJWTMiddleware) MiddlewareInit() error {
 		}
 	}
 
+	if mw.Realm == "" {
+		return errors.New("realm is required")
+	}
+
+	if mw.Authenticator == nil {
+		return errors.New("authenticator is required")
+	}
+
+	if mw.Key == nil {
+		return errors.New("secret key is required")
+	}
+
 	return nil
 }
 
 // MiddlewareFunc makes GinJWTMiddleware implement the Middleware interface.
 func (mw *GinJWTMiddleware) MiddlewareFunc() gin.HandlerFunc {
 	if err := mw.MiddlewareInit(); err != nil {
-		log.Fatal(err.Error())
+		return func(c *gin.Context) {
+			mw.unauthorized(c, http.StatusInternalServerError, err.Error())
+			return
+		}
 	}
 
 	return func(c *gin.Context) {
@@ -286,6 +289,11 @@ func (mw *GinJWTMiddleware) parseToken(c *gin.Context) (*jwt.Token, error) {
 }
 
 func (mw *GinJWTMiddleware) unauthorized(c *gin.Context, code int, message string) {
+
+	if mw.Realm == "" {
+		mw.Realm = "gin jwt"
+	}
+
 	c.Header("WWW-Authenticate", "JWT realm="+mw.Realm)
 	c.Abort()
 

auth_jwt_test.go

@@ -130,6 +130,25 @@ func ginHandler(auth *GinJWTMiddleware) *gin.Engine {
 	return r
 }
 
+func TestInternalServerError(t *testing.T) {
+	// the middleware to test
+	authMiddleware := &GinJWTMiddleware{}
+
+	handler := ginHandler(authMiddleware)
+
+	r := gofight.New()
+
+	r.GET("/auth/hello").
+		Run(handler, func(r gofight.HTTPResponse, rq gofight.HTTPRequest) {
+			data := []byte(r.Body.String())
+
+			message, _ := jsonparser.GetString(data, "message")
+
+			assert.Equal(t, "realm is required", message)
+			assert.Equal(t, http.StatusInternalServerError, r.Code)
+		})
+}
+
 func TestLoginHandler(t *testing.T) {
 
 	// the middleware to test