mirror of
https://github.com/keycloak/keycloak.git
synced 2025-12-21 06:20:05 -06:00
Move ConcurrencyTest.java, AbstractConcurrencyTest.java to the new testsuite
Part of: #34494 Signed-off-by: Lukas Hanusovsky <lhanusov@redhat.com>
This commit is contained in:
Lukas Hanusovsky
committed by
Stian Thorgersen
Stian Thorgersen
parent
adae1bbcb1
commit
cabd7cd474
@@ -41,6 +41,7 @@ import java.util.concurrent.atomic.AtomicInteger;
|
||||
/**
|
||||
* @author <a href="mailto:sthorger@redhat.com">Stian Thorgersen</a>
|
||||
*/
|
||||
@Deprecated(forRemoval = true)
|
||||
public abstract class AbstractConcurrencyTest extends AbstractTestRealmKeycloakTest {
|
||||
|
||||
private static final int DEFAULT_THREADS = 4;
|
||||
|
||||
@@ -1,302 +0,0 @@
|
||||
/*
|
||||
* Copyright 2016 Red Hat, Inc. and/or its affiliates
|
||||
* and other contributors as indicated by the @author tags.
|
||||
*
|
||||
* Licensed under the Apache License, Version 2.0 (the "License");
|
||||
* you may not use this file except in compliance with the License.
|
||||
* You may obtain a copy of the License at
|
||||
*
|
||||
* http://www.apache.org/licenses/LICENSE-2.0
|
||||
*
|
||||
* Unless required by applicable law or agreed to in writing, software
|
||||
* distributed under the License is distributed on an "AS IS" BASIS,
|
||||
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
||||
* See the License for the specific language governing permissions and
|
||||
* limitations under the License.
|
||||
*/
|
||||
|
||||
package org.keycloak.testsuite.admin.concurrency;
|
||||
|
||||
import java.util.stream.Collectors;
|
||||
import org.junit.Test;
|
||||
import org.keycloak.admin.client.Keycloak;
|
||||
import org.keycloak.admin.client.resource.ClientResource;
|
||||
import org.keycloak.admin.client.resource.ClientsResource;
|
||||
import org.keycloak.admin.client.resource.RealmResource;
|
||||
import org.keycloak.admin.client.resource.RolesResource;
|
||||
import org.keycloak.admin.client.resource.UserResource;
|
||||
import org.keycloak.admin.client.resource.UsersResource;
|
||||
import org.keycloak.representations.idm.ClientRepresentation;
|
||||
import org.keycloak.representations.idm.GroupRepresentation;
|
||||
import org.keycloak.representations.idm.RoleRepresentation;
|
||||
|
||||
import jakarta.ws.rs.NotFoundException;
|
||||
import jakarta.ws.rs.core.Response;
|
||||
|
||||
import org.keycloak.representations.idm.UserRepresentation;
|
||||
import org.keycloak.testsuite.admin.ApiUtil;
|
||||
import org.keycloak.testsuite.model.StoreProvider;
|
||||
import org.keycloak.testsuite.util.UserBuilder;
|
||||
|
||||
import java.util.List;
|
||||
import java.util.Map;
|
||||
import java.util.Objects;
|
||||
import java.util.concurrent.atomic.AtomicInteger;
|
||||
import java.util.stream.IntStream;
|
||||
|
||||
import org.junit.Ignore;
|
||||
|
||||
import static org.junit.Assert.assertEquals;
|
||||
import static org.junit.Assert.assertFalse;
|
||||
import static org.junit.Assert.assertNotNull;
|
||||
import static org.junit.Assert.assertTrue;
|
||||
import static org.junit.Assert.fail;
|
||||
|
||||
/**
|
||||
* @author <a href="mailto:sthorger@redhat.com">Stian Thorgersen</a>
|
||||
*/
|
||||
public class ConcurrencyTest extends AbstractConcurrencyTest {
|
||||
|
||||
public void concurrentTest(KeycloakRunnable... tasks) throws Throwable {
|
||||
System.out.println("***************************");
|
||||
long start = System.currentTimeMillis();
|
||||
run(tasks);
|
||||
long end = System.currentTimeMillis() - start;
|
||||
System.out.println("took " + end + " ms");
|
||||
}
|
||||
|
||||
// KEYCLOAK-8141 Verify that no attribute values are duplicated, and there are no locking exceptions when adding attributes in parallell
|
||||
@Test
|
||||
@Ignore
|
||||
public void createUserAttributes() throws Throwable {
|
||||
AtomicInteger c = new AtomicInteger();
|
||||
|
||||
UsersResource users = testRealm().users();
|
||||
|
||||
UserRepresentation u = UserBuilder.create().username("attributes").build();
|
||||
Response response = users.create(u);
|
||||
String userId = ApiUtil.getCreatedId(response);
|
||||
response.close();
|
||||
|
||||
UserResource user = users.get(userId);
|
||||
|
||||
concurrentTest((threadIndex, keycloak, realm) -> {
|
||||
UserRepresentation rep = user.toRepresentation();
|
||||
rep.singleAttribute("a-" + c.getAndIncrement(), "value");
|
||||
user.update(rep);
|
||||
});
|
||||
|
||||
UserRepresentation rep = user.toRepresentation();
|
||||
|
||||
// Number of attributes should be equal to created attributes, or less (concurrent requests may drop attributes added by other threads)
|
||||
assertTrue(rep.getAttributes().size() <= c.get());
|
||||
|
||||
// All attributes should have a single value
|
||||
for (Map.Entry<String, List<String>> e : rep.getAttributes().entrySet()) {
|
||||
assertEquals(1, e.getValue().size());
|
||||
}
|
||||
}
|
||||
|
||||
|
||||
@Test
|
||||
public void testAllConcurrently() throws Throwable {
|
||||
AtomicInteger uniqueCounter = new AtomicInteger(100000);
|
||||
concurrentTest(
|
||||
new CreateClient(uniqueCounter),
|
||||
new CreateRemoveClient(uniqueCounter),
|
||||
new CreateGroup(uniqueCounter),
|
||||
new CreateRole(uniqueCounter)
|
||||
);
|
||||
}
|
||||
|
||||
@Test
|
||||
public void createClient() throws Throwable {
|
||||
AtomicInteger uniqueCounter = new AtomicInteger();
|
||||
concurrentTest(new CreateClient(uniqueCounter));
|
||||
}
|
||||
|
||||
@Test
|
||||
public void createGroup() throws Throwable {
|
||||
AtomicInteger uniqueCounter = new AtomicInteger();
|
||||
concurrentTest(new CreateGroup(uniqueCounter));
|
||||
}
|
||||
|
||||
@Test
|
||||
public void createRemoveClient() throws Throwable {
|
||||
// FYI< this will fail as HSQL seems to be trying to perform table locks.
|
||||
AtomicInteger uniqueCounter = new AtomicInteger();
|
||||
concurrentTest(new CreateRemoveClient(uniqueCounter));
|
||||
}
|
||||
|
||||
@Test
|
||||
public void createClientRole() throws Throwable {
|
||||
ClientRepresentation c = new ClientRepresentation();
|
||||
c.setClientId("client");
|
||||
Response response = adminClient.realm(REALM_NAME).clients().create(c);
|
||||
final String clientId = ApiUtil.getCreatedId(response);
|
||||
response.close();
|
||||
|
||||
AtomicInteger uniqueCounter = new AtomicInteger();
|
||||
concurrentTest(new CreateClientRole(uniqueCounter, clientId));
|
||||
}
|
||||
|
||||
@Test
|
||||
public void createRole() throws Throwable {
|
||||
AtomicInteger uniqueCounter = new AtomicInteger();
|
||||
run(new CreateRole(uniqueCounter));
|
||||
}
|
||||
|
||||
private class CreateClient implements KeycloakRunnable {
|
||||
|
||||
private final AtomicInteger clientIndex;
|
||||
|
||||
public CreateClient(AtomicInteger clientIndex) {
|
||||
this.clientIndex = clientIndex;
|
||||
}
|
||||
|
||||
@Override
|
||||
public void run(int threadIndex, Keycloak keycloak, RealmResource realm) throws Throwable {
|
||||
String name = "c-" + clientIndex.getAndIncrement();
|
||||
ClientRepresentation c = new ClientRepresentation();
|
||||
c.setClientId(name);
|
||||
Response response = realm.clients().create(c);
|
||||
String id = ApiUtil.getCreatedId(response);
|
||||
response.close();
|
||||
|
||||
c = realm.clients().get(id).toRepresentation();
|
||||
assertNotNull(c);
|
||||
|
||||
int findAttempts = 1;
|
||||
if (StoreProvider.getCurrentProvider().equals(StoreProvider.DEFAULT)) {
|
||||
findAttempts = 5;
|
||||
}
|
||||
boolean clientFound = IntStream.range(0, findAttempts)
|
||||
.anyMatch(i -> realm.clients().findAll().stream()
|
||||
.map(ClientRepresentation::getClientId)
|
||||
.filter(Objects::nonNull)
|
||||
.anyMatch(name::equals));
|
||||
|
||||
assertTrue("Client " + name + " not found in client list after " + findAttempts + " attempts", clientFound);
|
||||
}
|
||||
}
|
||||
|
||||
private class CreateRemoveClient implements KeycloakRunnable {
|
||||
|
||||
private final AtomicInteger clientIndex;
|
||||
|
||||
public CreateRemoveClient(AtomicInteger clientIndex) {
|
||||
this.clientIndex = clientIndex;
|
||||
}
|
||||
|
||||
@Override
|
||||
public void run(int threadIndex, Keycloak keycloak, RealmResource realm) throws Throwable {
|
||||
String name = "c-" + clientIndex.getAndIncrement();
|
||||
ClientRepresentation c = new ClientRepresentation();
|
||||
c.setClientId(name);
|
||||
final ClientsResource clients = realm.clients();
|
||||
|
||||
Response response = clients.create(c);
|
||||
String id = ApiUtil.getCreatedId(response);
|
||||
response.close();
|
||||
final ClientResource client = clients.get(id);
|
||||
|
||||
c = client.toRepresentation();
|
||||
assertNotNull(c);
|
||||
|
||||
int findAttempts = 1;
|
||||
if (StoreProvider.getCurrentProvider().equals(StoreProvider.DEFAULT)) {
|
||||
findAttempts = 5;
|
||||
}
|
||||
boolean clientFound = IntStream.range(0, findAttempts)
|
||||
.anyMatch(i -> clients.findAll().stream()
|
||||
.map(ClientRepresentation::getClientId)
|
||||
.filter(Objects::nonNull)
|
||||
.anyMatch(name::equals));
|
||||
|
||||
assertTrue("Client " + name + " not found in client list after " + findAttempts + " attempts", clientFound);
|
||||
|
||||
client.remove();
|
||||
try {
|
||||
client.toRepresentation();
|
||||
fail("Client " + name + " should not be found. Should throw a 404");
|
||||
} catch (NotFoundException e) {
|
||||
|
||||
}
|
||||
|
||||
assertFalse("Client " + name + " should now not present in client list",
|
||||
clients.findAll().stream()
|
||||
.map(ClientRepresentation::getClientId)
|
||||
.filter(Objects::nonNull)
|
||||
.anyMatch(name::equals));
|
||||
}
|
||||
}
|
||||
|
||||
private class CreateGroup implements KeycloakRunnable {
|
||||
|
||||
private final AtomicInteger uniqueIndex;
|
||||
|
||||
public CreateGroup(AtomicInteger uniqueIndex) {
|
||||
this.uniqueIndex = uniqueIndex;
|
||||
}
|
||||
|
||||
@Override
|
||||
public void run(int threadIndex, Keycloak keycloak, RealmResource realm) throws Throwable {
|
||||
String name = "g-" + uniqueIndex.getAndIncrement();
|
||||
GroupRepresentation c = new GroupRepresentation();
|
||||
c.setName(name);
|
||||
Response response = realm.groups().add(c);
|
||||
String id = ApiUtil.getCreatedId(response);
|
||||
response.close();
|
||||
|
||||
c = realm.groups().group(id).toRepresentation();
|
||||
assertNotNull(c);
|
||||
|
||||
assertTrue("Group " + name + " [" + id + "] " + " not found in group list",
|
||||
realm.groups().groups().stream()
|
||||
.map(GroupRepresentation::getName)
|
||||
.filter(Objects::nonNull)
|
||||
.anyMatch(name::equals));
|
||||
}
|
||||
}
|
||||
|
||||
private class CreateClientRole implements KeycloakRunnable {
|
||||
|
||||
private final AtomicInteger uniqueCounter;
|
||||
private final String clientId;
|
||||
|
||||
public CreateClientRole(AtomicInteger uniqueCounter, String clientId) {
|
||||
this.uniqueCounter = uniqueCounter;
|
||||
this.clientId = clientId;
|
||||
}
|
||||
|
||||
@Override
|
||||
public void run(int threadIndex, Keycloak keycloak, RealmResource realm) throws Throwable {
|
||||
String name = "cr-" + uniqueCounter.getAndIncrement();
|
||||
RoleRepresentation r = new RoleRepresentation(name, null, false);
|
||||
|
||||
final RolesResource roles = realm.clients().get(clientId).roles();
|
||||
roles.create(r);
|
||||
assertNotNull(roles.get(name).toRepresentation());
|
||||
}
|
||||
}
|
||||
|
||||
private class CreateRole implements KeycloakRunnable {
|
||||
|
||||
private final AtomicInteger uniqueCounter;
|
||||
|
||||
public CreateRole(AtomicInteger uniqueCounter) {
|
||||
this.uniqueCounter = uniqueCounter;
|
||||
}
|
||||
|
||||
@Override
|
||||
public void run(int threadIndex, Keycloak keycloak, RealmResource realm) throws Throwable {
|
||||
String name = "r-" + uniqueCounter.getAndIncrement();
|
||||
RoleRepresentation r = new RoleRepresentation(name, null, false);
|
||||
|
||||
final RolesResource roles = realm.roles();
|
||||
roles.create(r);
|
||||
assertNotNull(roles.get(name).toRepresentation());
|
||||
}
|
||||
}
|
||||
|
||||
}
|
||||
Reference in New Issue
Block a user