From 682379dd7ed049ce5e2b8505fa0987c46670c95a Mon Sep 17 00:00:00 2001 From: Ryan Liang Date: Wed, 23 Aug 2023 17:21:11 -0700 Subject: [PATCH] Switch to assertThat in obo authenticator test Signed-off-by: Ryan Liang --- .../security/http/OnBehalfOfAuthenticatorTest.java | 8 +++++++- 1 file changed, 7 insertions(+), 1 deletion(-) diff --git a/src/test/java/org/opensearch/security/http/OnBehalfOfAuthenticatorTest.java b/src/test/java/org/opensearch/security/http/OnBehalfOfAuthenticatorTest.java index 58b806c7ff..fbb03bf7c3 100644 --- a/src/test/java/org/opensearch/security/http/OnBehalfOfAuthenticatorTest.java +++ b/src/test/java/org/opensearch/security/http/OnBehalfOfAuthenticatorTest.java @@ -34,6 +34,8 @@ import org.opensearch.security.user.AuthCredentials; import org.opensearch.security.util.FakeRestRequest; +import static org.hamcrest.Matchers.equalTo; + public class OnBehalfOfAuthenticatorTest { final static String clusterName = "cluster_0"; final static String enableOBO = "true"; @@ -147,6 +149,10 @@ public void testNonSpecifyOBOSetting() throws Exception { @Test public void testBearer() throws Exception { + Map expectedAttributes = new HashMap<>(); + expectedAttributes.put("attr.jwt.iss", "cluster_0"); + expectedAttributes.put("attr.jwt.sub", "Leonard McCoy"); + expectedAttributes.put("attr.jwt.aud", "ext_0"); String jwsToken = Jwts.builder() .setIssuer(clusterName) @@ -165,7 +171,7 @@ public void testBearer() throws Exception { Assert.assertEquals("Leonard McCoy", credentials.getUsername()); Assert.assertEquals(0, credentials.getSecurityRoles().size()); Assert.assertEquals(0, credentials.getBackendRoles().size()); - Assert.assertEquals(3, credentials.getAttributes().size()); + Assert.assertThat(credentials.getAttributes(), equalTo(expectedAttributes)); } @Test