Exception -> IllegalArgumentException

Signed-off-by: Maciej Mierzwa <[email protected]>
MaciejMierzwa · Oct 5, 2023 · 3002f11 · 3002f11
1 parent a245f58
commit 3002f11
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/src/test/java/org/opensearch/security/authtoken/jwt/JwtVendorTest.java b/src/test/java/org/opensearch/security/authtoken/jwt/JwtVendorTest.java
@@ -116,6 +116,7 @@ public void testCreateJwtWithRoleSecurityMode() throws Exception {
         assertThat(signedJWT.getJWTClaimsSet().getClaims().get("exp"), is(notNullValue()));
         assertThat(signedJWT.getJWTClaimsSet().getClaims().get("br"), is(notNullValue()));
         assertThat(signedJWT.getJWTClaimsSet().getClaims().get("br").toString(), equalTo(expectedBackendRoles));
+
         EncryptionDecryptionUtil encryptionUtil = new EncryptionDecryptionUtil(claimsEncryptionKey);
         assertThat(encryptionUtil.decrypt(signedJWT.getJWTClaimsSet().getClaims().get("er").toString()), equalTo(expectedRoles));
     }
@@ -138,7 +139,7 @@ public void testCreateJwtWithBadExpiry() {
                 throw new RuntimeException(e);
             }
         });
-        Assert.assertEquals("java.lang.Exception: The expiration time should be a positive integer", exception.getMessage());
+        Assert.assertEquals("java.lang.IllegalArgumentException: The expiration time should be a positive integer", exception.getMessage());
     }
 
     @Test
@@ -179,6 +180,6 @@ public void testCreateJwtWithBadRoles() {
                 throw new RuntimeException(e);
             }
         });
-        Assert.assertEquals("java.lang.Exception: Roles cannot be null", exception.getMessage());
+        Assert.assertEquals("java.lang.IllegalArgumentException: Roles cannot be null", exception.getMessage());
     }
 }