From 15c85303bd8e57493cfed2892f91c55f80c1664c Mon Sep 17 00:00:00 2001
From: Ryan Liang <jiallian@amazon.com>
Date: Wed, 23 Aug 2023 16:47:13 -0700
Subject: [PATCH] Change to assertThrows for obo authenticator tests

Signed-off-by: Ryan Liang <jiallian@amazon.com>
---
 .../http/OnBehalfOfAuthenticatorTest.java     | 48 ++++++++-----------
 1 file changed, 21 insertions(+), 27 deletions(-)

diff --git a/src/test/java/org/opensearch/security/http/OnBehalfOfAuthenticatorTest.java b/src/test/java/org/opensearch/security/http/OnBehalfOfAuthenticatorTest.java
index 12d0ddff5b..58b806c7ff 100644
--- a/src/test/java/org/opensearch/security/http/OnBehalfOfAuthenticatorTest.java
+++ b/src/test/java/org/opensearch/security/http/OnBehalfOfAuthenticatorTest.java
@@ -46,51 +46,45 @@ public class OnBehalfOfAuthenticatorTest {
     final static SecretKey secretKey = Keys.hmacShaKeyFor(signingKeyB64Encoded.getBytes(StandardCharsets.UTF_8));
 
     @Test
-    public void testNoKey() throws Exception {
-
-        try {
-            final AuthCredentials credentials = extractCredentialsFromJwtHeader(
+    public void testNoKey() {
+        Exception exception = Assert.assertThrows(
+            RuntimeException.class,
+            () -> extractCredentialsFromJwtHeader(
                 null,
                 claimsEncryptionKey,
                 Jwts.builder().setIssuer(clusterName).setSubject("Leonard McCoy"),
                 false
-            );
-            Assert.fail("Expected a RuntimeException");
-        } catch (RuntimeException e) {
-            Assert.assertTrue(e.getMessage().contains("Unable to find on behalf of authenticator signing key"));
-        }
+            )
+        );
+        Assert.assertTrue(exception.getMessage().contains("Unable to find on behalf of authenticator signing key"));
     }
 
     @Test
-    public void testEmptyKey() throws Exception {
-
-        try {
-            final AuthCredentials credentials = extractCredentialsFromJwtHeader(
+    public void testEmptyKey() {
+        Exception exception = Assert.assertThrows(
+            RuntimeException.class,
+            () -> extractCredentialsFromJwtHeader(
                 null,
                 claimsEncryptionKey,
                 Jwts.builder().setIssuer(clusterName).setSubject("Leonard McCoy"),
                 false
-            );
-            Assert.fail("Expected a RuntimeException");
-        } catch (RuntimeException e) {
-            Assert.assertTrue(e.getMessage().contains("Unable to find on behalf of authenticator signing key"));
-        }
+            )
+        );
+        Assert.assertTrue(exception.getMessage().contains("Unable to find on behalf of authenticator signing key"));
     }
 
     @Test
-    public void testBadKey() throws Exception {
-
-        try {
-            final AuthCredentials credentials = extractCredentialsFromJwtHeader(
+    public void testBadKey() {
+        Exception exception = Assert.assertThrows(
+            RuntimeException.class,
+            () -> extractCredentialsFromJwtHeader(
                 BaseEncoding.base64().encode(new byte[] { 1, 3, 3, 4, 3, 6, 7, 8, 3, 10 }),
                 claimsEncryptionKey,
                 Jwts.builder().setIssuer(clusterName).setSubject("Leonard McCoy"),
                 false
-            );
-            Assert.fail("Expected a WeakKeyException");
-        } catch (RuntimeException e) {
-            Assert.assertTrue(e.getMessage().contains("The specified key byte array is 80 bits"));
-        }
+            )
+        );
+        Assert.assertTrue(exception.getMessage().contains("The specified key byte array is 80 bits"));
     }
 
     @Test