[feat] add the timeout to the global and local environment

- have the functions that want a timeout `ask` it from the environment
- adjust all usages to not take a timeout explicitly

.envrc

@@ -57,3 +57,6 @@ export INTEGRATION_DYNAMIC_BACKENDS_POOLSIZE=3
 export AWS_REGION="eu-west-1"
 export AWS_ACCESS_KEY_ID="dummykey"
 export AWS_SECRET_ACCESS_KEY="dummysecret"
+
+# integration test suite timeout 
+export TEST_TIMEOUT_SECONDS=2

integration/test/MLS/Util.hs

@@ -544,7 +544,7 @@ consumingMessages mp = Codensity $ \k -> do
     -- each new user and wait for its join event
     when (mls.protocol == MLSProtocolMLS) $
       traverse_
-        (awaitMatch 10 isMemberJoinNotif)
+        (awaitMatch isMemberJoinNotif)
         ( flip Map.restrictKeys newUsers
             . Map.mapKeys ((.user) . fst)
             . Map.fromList
@@ -564,7 +564,7 @@ consumingMessages mp = Codensity $ \k -> do
 consumeMessage :: HasCallStack => ClientIdentity -> Maybe MessagePackage -> WebSocket -> App Value
 consumeMessage cid mmp ws = do
   mls <- getMLSState
-  notif <- awaitMatch 10 isNewMLSMessageNotif ws
+  notif <- awaitMatch isNewMLSMessageNotif ws
   event <- notif %. "payload.0"
 
   for_ mmp $ \mp -> do
@@ -623,7 +623,7 @@ sendAndConsumeCommitBundle mp = do
 consumeWelcome :: HasCallStack => ClientIdentity -> MessagePackage -> WebSocket -> App ()
 consumeWelcome cid mp ws = do
   mls <- getMLSState
-  notif <- awaitMatch 10 isWelcomeNotif ws
+  notif <- awaitMatch isWelcomeNotif ws
   event <- notif %. "payload.0"
 
   shouldMatch (eventSubConv event) (fromMaybe A.Null mls.convId)

integration/test/Test/AccessUpdate.hs

@@ -115,7 +115,7 @@ testAccessUpdateWithRemotes = do
   withWebSockets [alice, bob, charlie] $ \wss -> do
     void $ updateAccess alice conv update >>= getJSON 200
     for_ wss $ \ws -> do
-      notif <- awaitMatch 10 isConvAccessUpdateNotif ws
+      notif <- awaitMatch isConvAccessUpdateNotif ws
       notif %. "payload.0.qualified_conversation" `shouldMatch` objQidObject conv
       notif %. "payload.0.qualified_from" `shouldMatch` objQidObject alice
       notif %. "payload.0.data.access" `shouldMatch` update_access_value

integration/test/Test/Conversation.hs

@@ -326,7 +326,7 @@ testAddUnreachableUserFromFederatingBackend = do
         conv <-
           postConversation alice (defProteus {qualifiedUsers = [bob, charlie]})
             >>= getJSON 201
-        forM_ wss $ awaitMatch 5 isMemberJoinNotif
+        forM_ wss $ awaitMatch isMemberJoinNotif
         pure conv
       chadId <- chad %. "qualified_id"
       pure (alice, chadId, conv)
@@ -521,7 +521,7 @@ testConvRenaming = do
   withWebSockets [alice, bob] $ \wss -> do
     for_ wss $ \ws -> do
       void $ changeConversationName alice conv newConvName >>= getBody 200
-      nameNotif <- awaitMatch 10 isConvNameChangeNotif ws
+      nameNotif <- awaitMatch isConvNameChangeNotif ws
       nameNotif %. "payload.0.data.name" `shouldMatch` newConvName
       nameNotif %. "payload.0.qualified_conversation" `shouldMatch` objQidObject conv
 
@@ -534,7 +534,7 @@ testReceiptModeWithRemotesOk = do
   withWebSockets [alice, bob] $ \wss -> do
     void $ updateReceiptMode alice conv (43 :: Int) >>= getBody 200
     for_ wss $ \ws -> do
-      notif <- awaitMatch 10 isReceiptModeUpdateNotif ws
+      notif <- awaitMatch isReceiptModeUpdateNotif ws
       notif %. "payload.0.qualified_conversation" `shouldMatch` objQidObject conv
       notif %. "payload.0.qualified_from" `shouldMatch` objQidObject alice
       notif %. "payload.0.data.receipt_mode" `shouldMatchInt` 43
@@ -550,7 +550,7 @@ testReceiptModeWithRemotesUnreachable = do
       >>= getJSON 201
   withWebSocket alice $ \ws -> do
     void $ updateReceiptMode alice conv (43 :: Int) >>= getBody 200
-    notif <- awaitMatch 10 isReceiptModeUpdateNotif ws
+    notif <- awaitMatch isReceiptModeUpdateNotif ws
     notif %. "payload.0.qualified_conversation" `shouldMatch` objQidObject conv
     notif %. "payload.0.qualified_from" `shouldMatch` objQidObject alice
     notif %. "payload.0.data.receipt_mode" `shouldMatchInt` 43
@@ -608,7 +608,7 @@ testDeleteRemoteMemberRemoteUnreachable = do
   void $ withWebSockets [alice, bob] $ \wss -> do
     void $ removeMember alice conv bob >>= getBody 200
     for wss $ \ws -> do
-      leaveNotif <- awaitMatch 10 isConvLeaveNotif ws
+      leaveNotif <- awaitMatch isConvLeaveNotif ws
       leaveNotif %. "payload.0.qualified_conversation" `shouldMatch` objQidObject conv
       leaveNotif %. "payload.0.qualified_from" `shouldMatch` objQidObject alice
       leaveNotif %. "payload.0.data.qualified_user_ids.0" `shouldMatch` objQidObject bob
@@ -629,7 +629,7 @@ testDeleteTeamConversationWithRemoteMembers = do
   void $ withWebSockets [alice, bob] $ \wss -> do
     void $ deleteTeamConversation team conv alice >>= getBody 200
     for wss $ \ws -> do
-      notif <- awaitMatch 10 isConvDeleteNotif ws
+      notif <- awaitMatch isConvDeleteNotif ws
       notif %. "payload.0.qualified_conversation" `shouldMatch` objQidObject conv
       notif %. "payload.0.qualified_from" `shouldMatch` objQidObject alice
 
@@ -654,7 +654,7 @@ testDeleteTeamConversationWithUnreachableRemoteMembers = do
       pure (bob, bobClient)
     withWebSocket alice $ \ws -> do
       void $ deleteTeamConversation team conv alice >>= getBody 200
-      notif <- awaitMatch 10 isConvDeleteNotif ws
+      notif <- awaitMatch isConvDeleteNotif ws
       assertNotification notif
     void $ runCodensity (startDynamicBackend dynBackend mempty) $ \_ -> do
       notif <- awaitNotification bob bobClient noValue 2 isConvDeleteNotif
@@ -705,7 +705,7 @@ testOnUserDeletedConversations = do
 
     void $ withWebSocket alex $ \ws -> do
       void $ deleteUser bob >>= getBody 200
-      n <- awaitMatch 10 isConvLeaveNotif ws
+      n <- awaitMatch isConvLeaveNotif ws
       n %. "payload.0.qualified_from" `shouldMatch` bobId
       n %. "payload.0.qualified_conversation" `shouldMatch` (mainConvBefore %. "qualified_id")
 
@@ -732,7 +732,7 @@ testUpdateConversationByRemoteAdmin = do
   void $ updateRole alice bob "wire_admin" (conv %. "qualified_id") >>= getBody 200
   void $ withWebSockets [alice, bob, charlie] $ \wss -> do
     void $ updateReceiptMode bob conv (41 :: Int) >>= getBody 200
-    for_ wss $ \ws -> awaitMatch 10 isReceiptModeUpdateNotif ws
+    for_ wss $ \ws -> awaitMatch isReceiptModeUpdateNotif ws
 
 testGuestCreatesConversation :: HasCallStack => App ()
 testGuestCreatesConversation = do

integration/test/Test/Demo.hs

@@ -185,7 +185,7 @@ testWebSockets = do
   user <- randomUser OwnDomain def
   withWebSocket user $ \ws -> do
     client <- BrigP.addClient user def >>= getJSON 201
-    n <- awaitMatch 3 (\n -> nPayload n %. "type" `isEqual` "user.client-add") ws
+    n <- awaitMatch (\n -> nPayload n %. "type" `isEqual` "user.client-add") ws
     nPayload n %. "client.id" `shouldMatch` (client %. "id")
 
 testMultipleBackends :: App ()

integration/test/Test/Federation.hs

@@ -103,14 +103,14 @@ testNotificationsForOfflineBackends = do
           isDelUserLeaveUpConvNotif = allPreds [isConvLeaveNotif, isNotifConv upBackendConv, isNotifForUser delUser]
 
       do
-        newMsgNotif <- awaitMatch 10 isNewMessageNotif ws
+        newMsgNotif <- awaitMatch isNewMessageNotif ws
         newMsgNotif %. "payload.0.qualified_conversation" `shouldMatch` objQidObject upBackendConv
         newMsgNotif %. "payload.0.data.text" `shouldMatchBase64` "success message for other user"
 
-        void $ awaitMatch 10 isOtherUser2LeaveUpConvNotif ws
-        void $ awaitMatch 10 isDelUserLeaveUpConvNotif ws
+        void $ awaitMatch isOtherUser2LeaveUpConvNotif ws
+        void $ awaitMatch isDelUserLeaveUpConvNotif ws
 
-        delUserDeletedNotif <- nPayload $ awaitMatch 10 isDeleteUserNotif ws
+        delUserDeletedNotif <- nPayload $ awaitMatch isDeleteUserNotif ws
         objQid delUserDeletedNotif `shouldMatch` objQid delUser
 
     runCodensity (startDynamicBackend downBackend mempty) $ \_ -> do

integration/test/Test/MLS.hs

@@ -27,11 +27,10 @@ testSendMessageNoReturnToSender = do
     void . bindResponse (postMLSMessage mp.sender mp.message) $ \resp -> do
       resp.status `shouldMatchInt` 201
     for_ wss $ \ws -> do
-      n <- awaitMatch 3 (\n -> nPayload n %. "type" `isEqual` "conversation.mls-message-add") ws
+      n <- awaitMatch (\n -> nPayload n %. "type" `isEqual` "conversation.mls-message-add") ws
       nPayload n %. "data" `shouldMatch` T.decodeUtf8 (Base64.encode mp.message)
     expectFailure (const $ pure ()) $
       awaitMatch
-        3
         ( \n ->
             liftM2
               (&&)
@@ -93,7 +92,7 @@ testMixedProtocolUpgrade secondDomain = do
     modifyMLSState $ \mls -> mls {protocol = MLSProtocolMixed}
 
     for_ websockets $ \ws -> do
-      n <- awaitMatch 3 (\value -> nPayload value %. "type" `isEqual` "conversation.protocol-update") ws
+      n <- awaitMatch (\value -> nPayload value %. "type" `isEqual` "conversation.protocol-update") ws
       nPayload n %. "data.protocol" `shouldMatch` "mixed"
 
   bindResponse (getConversation alice qcnv) $ \resp -> do
@@ -145,7 +144,7 @@ testMixedProtocolAddUsers secondDomain = do
     mp <- createAddCommit alice1 [bob]
     welcome <- assertJust "should have welcome" mp.welcome
     void $ sendAndConsumeCommitBundle mp
-    n <- awaitMatch 3 (\n -> nPayload n %. "type" `isEqual` "conversation.mls-welcome") ws
+    n <- awaitMatch (\n -> nPayload n %. "type" `isEqual` "conversation.mls-welcome") ws
     nPayload n %. "data" `shouldMatch` T.decodeUtf8 (Base64.encode welcome)
 
 testMixedProtocolUserLeaves :: HasCallStack => Domain -> App ()
@@ -177,7 +176,7 @@ testMixedProtocolUserLeaves secondDomain = do
     bindResponse (removeConversationMember bob qcnv) $ \resp ->
       resp.status `shouldMatchInt` 200
 
-    n <- awaitMatch 3 (\n -> nPayload n %. "type" `isEqual` "conversation.mls-message-add") ws
+    n <- awaitMatch (\n -> nPayload n %. "type" `isEqual` "conversation.mls-message-add") ws
 
     msg <- asByteString (nPayload n %. "data") >>= showMessage alice1
     let leafIndexBob = 1
@@ -293,7 +292,7 @@ testMLSProtocolUpgrade secondDomain = do
     -- charlie is added to the group
     void $ uploadNewKeyPackage charlie1
     void $ createAddCommit alice1 [charlie] >>= sendAndConsumeCommitBundle
-    awaitMatch 10 isNewMLSMessageNotif ws
+    awaitMatch isNewMLSMessageNotif ws
 
   supportMLS alice
   bindResponse (putConversationProtocol bob conv "mls") $ \resp -> do
@@ -310,7 +309,7 @@ testMLSProtocolUpgrade secondDomain = do
       resp.status `shouldMatchInt` 200
     modifyMLSState $ \mls -> mls {protocol = MLSProtocolMLS}
     for_ wss $ \ws -> do
-      n <- awaitMatch 3 isNewMLSMessageNotif ws
+      n <- awaitMatch isNewMLSMessageNotif ws
       msg <- asByteString (nPayload n %. "data") >>= showMessage alice1
       let leafIndexCharlie = 2
       msg %. "message.content.body.Proposal.Remove.removed" `shouldMatchInt` leafIndexCharlie
@@ -377,7 +376,7 @@ testRemoteRemoveClient = do
   withWebSocket alice $ \wsAlice -> do
     void $ deleteClient bob bob1.client >>= getBody 200
     let predicate n = nPayload n %. "type" `isEqual` "conversation.mls-message-add"
-    n <- awaitMatch 5 predicate wsAlice
+    n <- awaitMatch predicate wsAlice
     shouldMatch (nPayload n %. "conversation") (objId conv)
     shouldMatch (nPayload n %. "from") (objId bob)
 
@@ -527,7 +526,7 @@ testLocalWelcome = do
     es <- sendAndConsumeCommitBundle commit
     let isWelcome n = nPayload n %. "type" `isEqual` "conversation.mls-welcome"
 
-    n <- awaitMatch 5 isWelcome wsBob
+    n <- awaitMatch isWelcome wsBob
 
     shouldMatch (nPayload n %. "conversation") (objId qcnv)
     shouldMatch (nPayload n %. "from") (objId alice)

integration/test/Test/MLS/Message.hs

@@ -45,15 +45,15 @@ testApplicationMessage = do
   withWebSockets [alice, alex, bob, betty] $ \wss -> do
     -- alice adds all other users (including her own client)
     void $ createAddCommit alice1 [alice, alex, bob, betty] >>= sendAndConsumeCommitBundle
-    traverse_ (awaitMatch 10 isMemberJoinNotif) wss
+    traverse_ (awaitMatch isMemberJoinNotif) wss
 
     -- alex sends a message
     void $ createApplicationMessage alex1 "hello" >>= sendAndConsumeMessage
-    traverse_ (awaitMatch 10 isNewMLSMessageNotif) wss
+    traverse_ (awaitMatch isNewMLSMessageNotif) wss
 
     -- bob sends a message
     void $ createApplicationMessage bob1 "hey" >>= sendAndConsumeMessage
-    traverse_ (awaitMatch 10 isNewMLSMessageNotif) wss
+    traverse_ (awaitMatch isNewMLSMessageNotif) wss
 
 testAppMessageSomeReachable :: HasCallStack => App ()
 testAppMessageSomeReachable = do
@@ -67,7 +67,7 @@ testAppMessageSomeReachable = do
     void $ createNewGroup alice1
     void $ withWebSocket charlie $ \ws -> do
       void $ createAddCommit alice1 [bob, charlie] >>= sendAndConsumeCommitBundle
-      awaitMatch 10 isMemberJoinNotif ws
+      awaitMatch isMemberJoinNotif ws
     pure alice1
 
   -- charlie isn't able to receive this message, so we make sure we can post it
@@ -88,7 +88,7 @@ testMessageNotifications bobDomain = do
 
   void $ withWebSocket bob $ \ws -> do
     void $ createAddCommit alice1 [alice, bob] >>= sendAndConsumeCommitBundle
-    awaitMatch 10 isMemberJoinNotif ws
+    awaitMatch isMemberJoinNotif ws
 
   let get (opts :: GetNotifications) = do
         notifs <- getNotifications bob opts {size = Just 10000} >>= getJSON 200
@@ -100,7 +100,7 @@ testMessageNotifications bobDomain = do
 
   void $ withWebSocket bob $ \ws -> do
     void $ createApplicationMessage alice1 "hi bob" >>= sendAndConsumeMessage
-    awaitMatch 10 isNewMLSMessageNotif ws
+    awaitMatch isNewMLSMessageNotif ws
 
   get def `shouldMatchInt` (numNotifs + 1)
   get def {client = Just bobClient} `shouldMatchInt` (numNotifsClient + 1)
@@ -114,10 +114,10 @@ testMultipleMessages = do
 
   withWebSockets [bob] $ \wss -> do
     void $ createAddCommit alice1 [bob] >>= sendAndConsumeCommitBundle
-    traverse_ (awaitMatch 10 isMemberJoinNotif) wss
+    traverse_ (awaitMatch isMemberJoinNotif) wss
 
     void $ createApplicationMessage alice1 "hello" >>= sendAndConsumeMessage
-    traverse_ (awaitMatch 10 isNewMLSMessageNotif) wss
+    traverse_ (awaitMatch isNewMLSMessageNotif) wss
 
     void $ createApplicationMessage alice1 "world" >>= sendAndConsumeMessage
-    traverse_ (awaitMatch 10 isNewMLSMessageNotif) wss
+    traverse_ (awaitMatch isNewMLSMessageNotif) wss

integration/test/Test/MLS/One2One.hs

@@ -100,14 +100,14 @@ testMLSOne2One scenario = do
     void $ sendAndConsumeCommitBundle commit
 
     let isWelcome n = nPayload n %. "type" `isEqual` "conversation.mls-welcome"
-    n <- awaitMatch 3 isWelcome ws
+    n <- awaitMatch isWelcome ws
     nPayload n %. "data" `shouldMatch` B8.unpack (Base64.encode (fold commit.welcome))
 
-    void $ awaitMatch 3 isMemberJoinNotif ws
+    void $ awaitMatch isMemberJoinNotif ws
 
   withWebSocket bob1 $ \ws -> do
     mp <- createApplicationMessage alice1 "hello, world"
     void $ sendAndConsumeMessage mp
     let isMessage n = nPayload n %. "type" `isEqual` "conversation.mls-message-add"
-    n <- awaitMatch 3 isMessage ws
+    n <- awaitMatch isMessage ws
     nPayload n %. "data" `shouldMatch` B8.unpack (Base64.encode mp.message)

integration/test/Test/MLS/SubConversation.hs

@@ -60,7 +60,7 @@ testDeleteParentOfSubConv secondDomain = do
   withWebSocket bob $ \ws -> do
     void . bindResponse (deleteTeamConv tid qcnv alice) $ \resp -> do
       resp.status `shouldMatchInt` 200
-    void $ awaitMatch 3 isConvDeleteNotif ws
+    void $ awaitMatch isConvDeleteNotif ws
 
   -- bob fails to send a message to the subconversation
   do
@@ -115,7 +115,7 @@ testLeaveSubConv variant = do
 
   withWebSockets [bob, charlie] $ \wss -> do
     void $ createAddCommit alice1 [bob, charlie] >>= sendAndConsumeCommitBundle
-    traverse_ (awaitMatch 10 isMemberJoinNotif) wss
+    traverse_ (awaitMatch isMemberJoinNotif) wss
 
   createSubConv bob1 "conference"
   void $ createExternalCommit alice1 Nothing >>= sendAndConsumeCommitBundle
@@ -140,11 +140,11 @@ testLeaveSubConv variant = do
   withWebSockets (tail others) $ \wss -> do
     -- a member commits the pending proposal
     void $ createPendingProposalCommit (head others) >>= sendAndConsumeCommitBundle
-    traverse_ (awaitMatch 10 isNewMLSMessageNotif) wss
+    traverse_ (awaitMatch isNewMLSMessageNotif) wss
 
     -- send an application message
     void $ createApplicationMessage (head others) "good riddance" >>= sendAndConsumeMessage
-    traverse_ (awaitMatch 10 isNewMLSMessageNotif) wss
+    traverse_ (awaitMatch isNewMLSMessageNotif) wss
 
   -- check that only 3 clients are left in the subconv
   do

integration/test/Test/MLS/Unreachable.hs

@@ -38,7 +38,7 @@ testAddUsersSomeReachable = do
     void $ createNewGroup alice1
     void $ withWebSocket bob $ \ws -> do
       void $ createAddCommit alice1 [bob] >>= sendAndConsumeCommitBundle
-      awaitMatch 10 isMemberJoinNotif ws
+      awaitMatch isMemberJoinNotif ws
     mp <- createAddCommit alice1 [charlie]
     pure (mp, thirdDomain)
 
@@ -62,7 +62,7 @@ testAddUserWithUnreachableRemoteUsers = do
       void $ createNewGroup alice1
       void $ withWebSocket charlie $ \ws -> do
         void $ createAddCommit alice1 [charlie] >>= sendAndConsumeCommitBundle
-        awaitMatch 10 isMemberJoinNotif ws
+        awaitMatch isMemberJoinNotif ws
       pure (alice1, bob, brad, chris)
 
     [bob1, brad1] <- traverse (createMLSClient def) [bob, brad]
@@ -103,7 +103,7 @@ testAddUnreachableUserFromFederatingBackend = do
       void $ createNewGroup alice1
       withWebSockets [bob, charlie] $ \wss -> do
         void $ createAddCommit alice1 [bob, charlie] >>= sendAndConsumeCommitBundle
-        forM_ wss $ awaitMatch 5 isMemberJoinNotif
+        forM_ wss $ awaitMatch isMemberJoinNotif
       createAddCommit alice1 [chad]
 
     bindResponse (postMLSCommitBundle mp.sender (mkBundle mp)) $ \resp -> do

integration/test/Test/MessageTimer.hs

@@ -33,7 +33,7 @@ testMessageTimerChangeWithRemotes = do
   withWebSockets [alice, bob] $ \wss -> do
     void $ updateMessageTimer alice conv 1000 >>= getBody 200
     for_ wss $ \ws -> do
-      notif <- awaitMatch 10 isConvMsgTimerUpdateNotif ws
+      notif <- awaitMatch isConvMsgTimerUpdateNotif ws
       notif %. "payload.0.qualified_conversation" `shouldMatch` objQidObject conv
       notif %. "payload.0.qualified_from" `shouldMatch` objQidObject alice
 
@@ -48,6 +48,6 @@ testMessageTimerChangeWithUnreachableRemotes = do
       postConversation alice (defProteus {qualifiedUsers = [bob]}) >>= getJSON 201
   withWebSocket alice $ \ws -> do
     void $ updateMessageTimer alice conv 1000 >>= getBody 200
-    notif <- awaitMatch 10 isConvMsgTimerUpdateNotif ws
+    notif <- awaitMatch isConvMsgTimerUpdateNotif ws
     notif %. "payload.0.qualified_conversation" `shouldMatch` objQidObject conv
     notif %. "payload.0.qualified_from" `shouldMatch` objQidObject alice

integration/test/Test/Roles.hs

@@ -41,7 +41,7 @@ testRoleUpdateWithRemotesOk = do
       resp.json %. "members.others.0.qualified_id" `shouldMatch` objQidObject charlie
       resp.json %. "members.others.0.conversation_role" `shouldMatch` "wire_admin"
     for_ wss $ \ws -> do
-      notif <- awaitMatch 10 isMemberUpdateNotif ws
+      notif <- awaitMatch isMemberUpdateNotif ws
       notif %. "payload.0.qualified_conversation" `shouldMatch` objQidObject conv
       notif %. "payload.0.qualified_from" `shouldMatch` objQidObject bob
 
@@ -61,6 +61,6 @@ testRoleUpdateWithRemotesUnreachable = do
       void $ updateRole bob charlie adminRole conv >>= getBody 200
 
       for_ wss $ \ws -> do
-        notif <- awaitMatch 10 isMemberUpdateNotif ws
+        notif <- awaitMatch isMemberUpdateNotif ws
         notif %. "payload.0.qualified_conversation" `shouldMatch` objQidObject conv
         notif %. "payload.0.qualified_from" `shouldMatch` objQidObject bob