Merge pull request #2334 from mposolda/dblock2

Fix testsuite with Mongo

testsuite/integration/src/test/java/org/keycloak/testsuite/broker/AbstractKeycloakIdentityProviderTest.java

@@ -547,10 +547,11 @@ public abstract class AbstractKeycloakIdentityProviderTest extends AbstractIdent
 
     @Test
     public void testWithLinkedFederationProvider() throws Exception {
+        setUpdateProfileFirstLogin(IdentityProviderRepresentation.UPFLM_OFF);
+
         // Add federationProvider to realm. It's configured with sync registrations
         RealmModel realm = getRealm();
         UserFederationProviderModel dummyModel = realm.addUserFederationProvider(DummyUserFederationProviderFactory.PROVIDER_NAME, new HashMap<String, String>(), 1, "test-dummy", -1, -1, 0);
-        setUpdateProfileFirstLogin(IdentityProviderRepresentation.UPFLM_OFF);
 
         brokerServerRule.stopSession(session, true);
         session = brokerServerRule.startSession();

testsuite/integration/src/test/java/org/keycloak/testsuite/util/cli/AbstractOfflineCacheCommand.java

@@ -103,6 +103,34 @@ public abstract class AbstractOfflineCacheCommand extends AbstractCommand {
         }
     }
 
+    // Just to check performance of multiple get calls. And comparing what's the change between the case when item is available locally or not.
+    public static class GetMultipleCommand extends AbstractOfflineCacheCommand {
+
+        @Override
+        public String getName() {
+            return "getMulti";
+        }
+
+        @Override
+        protected void doRunCacheCommand(KeycloakSession session, Cache<String, SessionEntity> cache) {
+            String id = getArg(0);
+            int count = getIntArg(1);
+
+            long start = System.currentTimeMillis();
+            for (int i=0 ; i<count ; i++) {
+                UserSessionEntity userSession = (UserSessionEntity) cache.get(id);
+                //printSession(id, userSession);
+            }
+            long took = System.currentTimeMillis() - start;
+            log.infof("Took %d milliseconds", took);
+        }
+
+        @Override
+        public String printUsage() {
+            return getName() + " <user-session-id> <count-of-gets>";
+        }
+    }
+
 
     public static class RemoveCommand extends AbstractOfflineCacheCommand {
 

testsuite/integration/src/test/java/org/keycloak/testsuite/util/cli/TestsuiteCLI.java

@@ -45,6 +45,7 @@ public class TestsuiteCLI {
             HelpCommand.class,
             AbstractOfflineCacheCommand.PutCommand.class,
             AbstractOfflineCacheCommand.GetCommand.class,
+            AbstractOfflineCacheCommand.GetMultipleCommand.class,
             AbstractOfflineCacheCommand.GetLocalCommand.class,
             AbstractOfflineCacheCommand.RemoveCommand.class,
             AbstractOfflineCacheCommand.SizeCommand.class,