enable in memory db again

wireapp · Dec 5, 2023 · 429471c · 429471c
1 parent 8d9e781
commit 429471c
Show file tree

Hide file tree

Showing 2 changed files with 11 additions and 6 deletions.
diff --git a/monkeys/src/main/kotlin/com/wire/kalium/monkeys/homeDirectory.kt b/monkeys/src/main/kotlin/com/wire/kalium/monkeys/homeDirectory.kt
@@ -34,7 +34,7 @@ fun coreLogic(
             encryptProteusStorage = true,
             isMLSSupportEnabled = true,
             wipeOnDeviceRemoval = true,
-        ), userAgent = "Wire Infinite Monkeys", useInMemoryStorage = false
+        ), userAgent = "Wire Infinite Monkeys", useInMemoryStorage = true
     )
     coreLogic.updateApiVersionsScheduler.scheduleImmediateApiVersionUpdate()
     return coreLogic

diff --git a/monkeys/src/main/kotlin/com/wire/kalium/monkeys/pool/MonkeyPool.kt b/monkeys/src/main/kotlin/com/wire/kalium/monkeys/pool/MonkeyPool.kt
@@ -26,6 +26,9 @@ import com.wire.kalium.monkeys.importer.Team
 import com.wire.kalium.monkeys.importer.UserCount
 import com.wire.kalium.monkeys.importer.UserData
 import io.micrometer.core.instrument.Tag
+import kotlinx.coroutines.async
+import kotlinx.coroutines.awaitAll
+import kotlinx.coroutines.coroutineScope
 import java.util.concurrent.ConcurrentHashMap
 import kotlin.math.roundToInt
 
@@ -67,12 +70,14 @@ class MonkeyPool(users: List<UserData>, testCase: String) {
         }
     }
 
-    suspend fun warmUp(core: CoreLogic) {
+    suspend fun warmUp(core: CoreLogic) = coroutineScope {
         // this is needed to create key packages for clients at least once
-        poolById.values.forEach {
-            it.login(core) {}
-            it.logout {}
-        }
+        poolById.values.map {
+            async {
+                it.login(core) {}
+                it.logout {}
+            }
+        }.awaitAll()
     }
 
     fun randomMonkeysFromTeam(team: String, userCount: UserCount): List<Monkey> {