Add generic update methods for builders (#40312)

Closes #40311 Signed-off-by: stianst <stianst@gmail.com>
2025-12-30 11:29:57 -06:00 · 2025-07-11 09:03:28 +02:00
parent 5219a331b9
commit 8a694a585b
4 changed files with 143 additions and 0 deletions
--- a/test-framework/examples/tests/src/test/java/org/keycloak/test/examples/CustomConfigBuilderTest.java
+++ b/test-framework/examples/tests/src/test/java/org/keycloak/test/examples/CustomConfigBuilderTest.java
@@ -0,0 +1,81 @@
+package org.keycloak.test.examples;
+
+import org.junit.jupiter.api.Assertions;
+import org.junit.jupiter.api.Test;
+import org.keycloak.representations.idm.GroupRepresentation;
+import org.keycloak.testframework.annotations.InjectClient;
+import org.keycloak.testframework.annotations.InjectRealm;
+import org.keycloak.testframework.annotations.InjectUser;
+import org.keycloak.testframework.annotations.KeycloakIntegrationTest;
+import org.keycloak.testframework.realm.ClientConfig;
+import org.keycloak.testframework.realm.ClientConfigBuilder;
+import org.keycloak.testframework.realm.ManagedClient;
+import org.keycloak.testframework.realm.ManagedRealm;
+import org.keycloak.testframework.realm.ManagedUser;
+import org.keycloak.testframework.realm.RealmConfig;
+import org.keycloak.testframework.realm.RealmConfigBuilder;
+import org.keycloak.testframework.realm.UserConfig;
+import org.keycloak.testframework.realm.UserConfigBuilder;
+
+import java.util.LinkedList;
+
+@KeycloakIntegrationTest
+public class CustomConfigBuilderTest {
+
+    @InjectRealm(config = CustomRealmConfig.class)
+    ManagedRealm realm;
+
+    @InjectClient(config = CustomClientConfig.class)
+    ManagedClient client;
+
+    @InjectUser(config = CustomUserConfig.class)
+    ManagedUser user;
+
+    @Test
+    public void testRealm() {
+        Assertions.assertEquals(1, realm.admin().groups().query("mygroup").size());
+    }
+
+    @Test
+    public void testClient() {
+        Assertions.assertTrue(client.admin().toRepresentation().isBearerOnly());
+    }
+
+    @Test
+    public void testUser() {
+        Assertions.assertFalse(user.admin().toRepresentation().isEnabled());
+    }
+
+    public static class CustomRealmConfig implements RealmConfig {
+
+        @Override
+        public RealmConfigBuilder configure(RealmConfigBuilder realm) {
+            return realm.update(r -> {
+                if (r.getGroups() == null) {
+                    r.setGroups(new LinkedList<>());
+                }
+                GroupRepresentation group = new GroupRepresentation();
+                group.setName("mygroup");
+                group.setPath("/mygroup");
+                r.getGroups().add(group);
+            });
+        }
+    }
+
+    public static class CustomClientConfig implements ClientConfig {
+
+        @Override
+        public ClientConfigBuilder configure(ClientConfigBuilder client) {
+            return client.update(u -> u.setBearerOnly(true));
+        }
+    }
+
+    public static class CustomUserConfig implements UserConfig {
+
+        @Override
+        public UserConfigBuilder configure(UserConfigBuilder user) {
+            return user.update(u -> u.setEnabled(false));
+        }
+    }
+
+}