Refactor the encryptiondecryptionutilstests

RyanL1997 · RyanL1997 · commit 878a1077e0a8 · 2023-08-22T09:26:50.000-07:00
Signed-off-by: Ryan Liang &lt;jiallian@amazon.com&gt;
diff --git a/src/test/java/org/opensearch/security/authtoken/jwt/EncryptionDecryptionUtilsTest.java b/src/test/java/org/opensearch/security/authtoken/jwt/EncryptionDecryptionUtilsTest.java
@@ -15,7 +15,7 @@
 import org.junit.Test;
 import java.util.Base64;
 
-public class EncryptionDecryptionUtilTest {
+public class EncryptionDecryptionUtilsTest {
 
     @Test
     public void testEncryptDecrypt() {
@@ -36,25 +36,22 @@ public void testDecryptingWithWrongKey() {
 
         String encryptedString = EncryptionDecryptionUtil.encrypt(secret1, data);
 
-        try {
-            EncryptionDecryptionUtil.decrypt(secret2, encryptedString);
-            Assert.fail("Should have thrown an exception when decrypting with a wrong key");
-        } catch (RuntimeException e) {
-            // Expected exception
-        }
+        RuntimeException ex = Assert.assertThrows(RuntimeException.class, () -> EncryptionDecryptionUtil.decrypt(secret2, encryptedString));
+
+        Assert.assertEquals("The cipher was unable to perform pass over data", ex.getMessage());
     }
 
     @Test
     public void testDecryptingCorruptedData() {
         String secret = Base64.getEncoder().encodeToString("mySecretKey12345".getBytes());
         String corruptedEncryptedString = "corruptedData";
 
-        try {
-            EncryptionDecryptionUtil.decrypt(secret, corruptedEncryptedString);
-            Assert.fail("Should have thrown an exception when trying to decrypt corrupted data");
-        } catch (RuntimeException e) {
-            // Expected exception
-        }
+        RuntimeException ex = Assert.assertThrows(
+            RuntimeException.class,
+            () -> EncryptionDecryptionUtil.decrypt(secret, corruptedEncryptedString)
+        );
+
+        Assert.assertEquals("Last unit does not have enough valid bits", ex.getMessage());
     }
 
     @Test
@@ -69,7 +66,7 @@ public void testEncryptDecryptEmptyString() {
     }
 
     @Test(expected = NullPointerException.class)
-    public void testEncryptDecryptNullValue() {
+    public void testEncryptNullValue() {
         String secret = Base64.getEncoder().encodeToString("mySecretKey12345".getBytes());
         String data = null;
 
