Add OidcTestSecurityAugmentor UserInfo unit test

quarkusio · Nov 29, 2024 · a576c24 · a576c24
1 parent 0f88242
commit a576c24
Showing 1 changed file with 28 additions and 0 deletions.
diff --git a/...dc/src/test/java/io/quarkus/test/security/oidc/OidcTestSecurityIdentityAugmentorTest.java b/...dc/src/test/java/io/quarkus/test/security/oidc/OidcTestSecurityIdentityAugmentorTest.java
@@ -1,6 +1,7 @@
 package io.quarkus.test.security.oidc;
 
 import static org.junit.jupiter.api.Assertions.assertEquals;
+import static org.junit.jupiter.api.Assertions.assertNotNull;
 import static org.junit.jupiter.api.Assertions.assertTrue;
 
 import java.lang.annotation.Annotation;
@@ -73,4 +74,31 @@ public String getName() {
         assertEquals("2", map.getString("b"));
     }
 
+    @Test
+    @OidcSecurity(userinfo = {
+            @UserInfo(key = "sub", value = "subject")
+    })
+    public void testUserInfo() throws Exception {
+        SecurityIdentity identity = QuarkusSecurityIdentity.builder()
+                .setPrincipal(new Principal() {
+                    @Override
+                    public String getName() {
+                        return "alice";
+                    }
+
+                })
+                .build();
+
+        OidcTestSecurityIdentityAugmentor augmentor = new OidcTestSecurityIdentityAugmentor(Optional.of("https://issuer.org"));
+
+        Annotation[] annotations = OidcTestSecurityIdentityAugmentorTest.class.getMethod("testUserInfo").getAnnotations();
+        SecurityIdentity augmentedIdentity = augmentor.augment(identity, annotations);
+        JsonWebToken jwt = (JsonWebToken) augmentedIdentity.getPrincipal();
+        assertEquals("alice", jwt.getName());
+
+        io.quarkus.oidc.UserInfo userInfo = augmentedIdentity.getAttribute("userinfo");
+        assertNotNull(userInfo);
+        assertEquals("subject", userInfo.getSubject());
+    }
+
 }