Added tests

TykTechnologies · May 30, 2018 · b178227 · b178227
2 parents 3aa2858 + 8725fa9
commit b178227
Showing 1 changed file with 22 additions and 18 deletions.
diff --git a/mw_jwt_test.go b/mw_jwt_test.go
@@ -514,6 +514,7 @@ func TestJWTSessionRSAWithRawSourceInvalidPolicyID(t *testing.T) {
 }
 
 func TestJWTSessionExpiresAtValidationConfigs(t *testing.T) {
+	const expiry = "exp"
 
 	ts := newTykTestServer()
 	defer ts.Close()
@@ -530,6 +531,7 @@ func TestJWTSessionExpiresAtValidationConfigs(t *testing.T) {
 
 	pID := createPolicy()
 
+	// This test is successful by definition
 	t.Run("Expiry_After_now--Valid_jwt", func(t *testing.T) {
 		spec.JWTDisableExpiresAtValidation = false //Default value
 		spec.JWTExpiresAtValidationSkew = 0        //Default value
@@ -538,7 +540,7 @@ func TestJWTSessionExpiresAtValidationConfigs(t *testing.T) {
 		jwtToken := createJWKToken(func(t *jwt.Token) {
 			t.Claims.(jwt.MapClaims)["policy_id"] = pID
 			t.Claims.(jwt.MapClaims)["user_id"] = "user123"
-			t.Claims.(jwt.MapClaims)["exp"] = time.Now().Add(+time.Second).Unix() //the jwt will be 1 second ahead
+			t.Claims.(jwt.MapClaims)[expiry] = time.Now().Add(+time.Second).Unix()
 		})
 		authHeaders := map[string]string{"authorization": jwtToken}
 		ts.Run(t, test.TestCase{
@@ -556,7 +558,7 @@ func TestJWTSessionExpiresAtValidationConfigs(t *testing.T) {
 		jwtToken := createJWKToken(func(t *jwt.Token) {
 			t.Claims.(jwt.MapClaims)["policy_id"] = pID
 			t.Claims.(jwt.MapClaims)["user_id"] = "user123"
-			t.Claims.(jwt.MapClaims)["exp"] = time.Now().Add(-time.Second).Unix() //the jwt will be 1 second ahead
+			t.Claims.(jwt.MapClaims)[expiry] = time.Now().Add(-time.Second).Unix() //the jwt will be 1 second ahead
 		})
 		authHeaders := map[string]string{"authorization": jwtToken}
 		ts.Run(t, test.TestCase{
@@ -574,7 +576,7 @@ func TestJWTSessionExpiresAtValidationConfigs(t *testing.T) {
 		jwtToken := createJWKToken(func(t *jwt.Token) {
 			t.Claims.(jwt.MapClaims)["policy_id"] = pID
 			t.Claims.(jwt.MapClaims)["user_id"] = "user123"
-			t.Claims.(jwt.MapClaims)["exp"] = time.Now().Add(-time.Second).Unix()
+			t.Claims.(jwt.MapClaims)[expiry] = time.Now().Add(-time.Second).Unix()
 		})
 		authHeaders := map[string]string{"authorization": jwtToken}
 		ts.Run(t, test.TestCase{
@@ -590,7 +592,7 @@ func TestJWTSessionExpiresAtValidationConfigs(t *testing.T) {
 		jwtToken := createJWKToken(func(t *jwt.Token) {
 			t.Claims.(jwt.MapClaims)["policy_id"] = pID
 			t.Claims.(jwt.MapClaims)["user_id"] = "user123"
-			t.Claims.(jwt.MapClaims)["exp"] = time.Now().Add(-time.Second).Unix()
+			t.Claims.(jwt.MapClaims)[expiry] = time.Now().Add(-time.Second).Unix()
 		})
 		authHeaders := map[string]string{"authorization": jwtToken}
 		ts.Run(t, test.TestCase{
@@ -617,6 +619,7 @@ func TestJWTSessionIssueAtValidationConfigs(t *testing.T) {
 
 	pID := createPolicy()
 
+	// This test is successful by definition
 	t.Run("IssuedAt_Before_now-no_skew--Valid_jwt", func(t *testing.T) {
 		spec.JWTDisableIssuedAtValidation = false
 		spec.JWTIssuedAtValidationSkew = 0
@@ -635,39 +638,39 @@ func TestJWTSessionIssueAtValidationConfigs(t *testing.T) {
 		})
 	})
 
-	t.Run("IssueAt-JWTDisableIssuedAtValidation--valid_jwt", func(t *testing.T) {
-		spec.JWTDisableIssuedAtValidation = true
-		spec.JWTIssuedAtValidationSkew = 1000 // This value doesn't matter since validation is disabled
+	t.Run("IssueAt-After_now-no_skew--Invalid_jwt", func(t *testing.T) {
+		spec.JWTDisableIssuedAtValidation = false
+		spec.JWTIssuedAtValidationSkew = 0
+
 		loadAPI(spec)
 
 		jwtToken := createJWKToken(func(t *jwt.Token) {
 			t.Claims.(jwt.MapClaims)["policy_id"] = pID
 			t.Claims.(jwt.MapClaims)["user_id"] = "user123"
-			t.Claims.(jwt.MapClaims)[issueAt] = time.Now().Add(time.Second).Unix()
+			t.Claims.(jwt.MapClaims)[issueAt] = time.Now().Add(+time.Minute).Unix()
 		})
 		authHeaders := map[string]string{"authorization": jwtToken}
 		ts.Run(t, test.TestCase{
-			Headers: authHeaders, Code: http.StatusOK,
+
+			Headers:   authHeaders,
+			Code:      http.StatusUnauthorized,
+			BodyMatch: "Key not authorized: token used before issued",
 		})
 	})
 
-	t.Run("IssueAt-After_now-no_skew--Invalid_jwt", func(t *testing.T) {
-		spec.JWTDisableIssuedAtValidation = false
-		spec.JWTIssuedAtValidationSkew = 0
-
+	t.Run("IssueAt-JWTDisableIssuedAtValidation--valid_jwt", func(t *testing.T) {
+		spec.JWTDisableIssuedAtValidation = true
+		spec.JWTIssuedAtValidationSkew = 1000 // This value doesn't matter since validation is disabled
 		loadAPI(spec)
 
 		jwtToken := createJWKToken(func(t *jwt.Token) {
 			t.Claims.(jwt.MapClaims)["policy_id"] = pID
 			t.Claims.(jwt.MapClaims)["user_id"] = "user123"
-			t.Claims.(jwt.MapClaims)[issueAt] = time.Now().Add(+time.Minute).Unix()
+			t.Claims.(jwt.MapClaims)[issueAt] = time.Now().Add(+time.Second).Unix()
 		})
 		authHeaders := map[string]string{"authorization": jwtToken}
 		ts.Run(t, test.TestCase{
-
-			Headers:   authHeaders,
-			Code:      http.StatusUnauthorized,
-			BodyMatch: "Key not authorized: token used before issued",
+			Headers: authHeaders, Code: http.StatusOK,
 		})
 	})
 
@@ -708,6 +711,7 @@ func TestJWTSessionNotBeforeValidationConfigs(t *testing.T) {
 
 	pID := createPolicy()
 
+	// This test is successful by definition
 	t.Run("NotBefore_Before_now-Valid_jwt", func(t *testing.T) {
 		spec.JWTDisableNotBeforeValidation = false
 		spec.JWTNotBeforeValidationSkew = 0