@@ -209,7 +209,7 @@ void onSuccess_withChangesToReport_notifyCallbackOnMessage() {
209209    requestService .sendRequest ();
210210
211211    // Await for onMessage call 
212-     await ().atMost (Duration .ofSeconds (1 )).until (() -> callbacks .onMessageCalls .get () == 1 );
212+     await ().atMost (Duration .ofSeconds (5 )).until (() -> callbacks .onMessageCalls .get () == 1 );
213213
214214    verify (callbacks ).onMessage (MessageData .builder ().setRemoteConfig (remoteConfig ).build ());
215215  }
@@ -267,7 +267,7 @@ void verifyRemoteConfigStatusSetter() {
267267  void  onConnectionSuccessful_notifyCallback () {
268268    initializeClient ();
269269
270-     await ().atMost (Duration .ofSeconds (1 )).until (() -> callbacks .onConnectCalls .get () == 1 );
270+     await ().atMost (Duration .ofSeconds (5 )).until (() -> callbacks .onConnectCalls .get () == 1 );
271271
272272    verify (callbacks ).onConnect ();
273273    verify (callbacks , never ()).onConnectFailed (any ());
@@ -311,7 +311,7 @@ void onFailedResponse_withServerErrorData_notifyCallback() {
311311    // Force request 
312312    requestService .sendRequest ();
313313
314-     await ().atMost (Duration .ofSeconds (1 )).until (() -> callbacks .onErrorResponseCalls .get () == 1 );
314+     await ().atMost (Duration .ofSeconds (5 )).until (() -> callbacks .onErrorResponseCalls .get () == 1 );
315315
316316    verify (callbacks ).onErrorResponse (errorResponse );
317317    verify (callbacks , never ()).onMessage (any ());
@@ -344,7 +344,7 @@ void whenServerProvidesNewInstanceUid_useIt() {
344344    enqueueServerToAgentResponse (response );
345345    requestService .sendRequest ();
346346
347-     await ().atMost (Duration .ofSeconds (1 )).until (() -> state .instanceUid .get () != initialUid );
347+     await ().atMost (Duration .ofSeconds (5 )).until (() -> state .instanceUid .get () != initialUid );
348348
349349    assertThat (state .instanceUid .get ()).isEqualTo (serverProvidedUid );
350350  }
@@ -359,7 +359,7 @@ private static AgentToServer getAgentToServerMessage(RecordedRequest request) {
359359
360360  private  RecordedRequest  takeRequest () {
361361    try  {
362-       return  server .takeRequest (1 , TimeUnit .SECONDS );
362+       return  server .takeRequest (5 , TimeUnit .SECONDS );
363363    } catch  (InterruptedException  e ) {
364364      throw  new  RuntimeException (e );
365365    }
0 commit comments