wip

integration/test/Performance/BigConversation.hs

@@ -38,17 +38,19 @@ batchForSize :: ConversationSize -> Word
 batchForSize Tiny = 10
 batchForSize Small = 20
 batchForSize Medium = 100
-batchForSize Big = 100
-batchForSize Large = 250
+batchForSize Big = 250
+batchForSize Large = 500
 batchForSize VeryLarge = 500
 
-testCreateBigMLSConversation :: ConversationSize -> App ()
-testCreateBigMLSConversation convSize = do
-  let teamSize = sizeToNumber convSize
-  let batchSize = fromIntegral . batchForSize $ convSize
+testCreateBigMLSConversation :: App ()
+testCreateBigMLSConversation = do
+  let teamSize = 200
+  let batchSize = 50
+  putStrLn $ "Creating a team with " <> show teamSize <> " members"
+  (_, ownerClient, _, members, _) <- createTeamAndClients teamSize
+  putStrLn $ "Creating a conversation with " <> show teamSize <> " members in batches of " <> show batchSize
   totalTime <-
-    fmap snd $ timeIt do
-      (_, ownerClient, _, members, _) <- createTeamAndClients . fromIntegral $ teamSize
+    snd <$> timeIt do
       convId <- createNewGroup def ownerClient
       let memberChunks = chunksOf batchSize members
       for_ memberChunks $ \chunk -> do
@@ -87,7 +89,8 @@ createTeamAndClients teamSize = do
                 { clientArgs =
                     def
                       { prekeys = Just [firstPrekey],
-                        lastPrekey = Just lastPrekey
+                        lastPrekey = Just lastPrekey,
+                        acapabilities = Just ["consumable-notifications"]
                       }
                 }
         createMLSClient def mlsClientOpts user