SCB-164 added waiting for async update to fix random test failures

Signed-off-by: seanyinx <sean.yin@huawei.com>
apache · Jan 10, 2018 · b349c19 · b349c19
1 parent 88cebbd
commit b349c19
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/...a-server/src/test/java/org/apache/servicecomb/saga/alpha/server/AlphaIntegrationTest.java b/...a-server/src/test/java/org/apache/servicecomb/saga/alpha/server/AlphaIntegrationTest.java
@@ -149,7 +149,7 @@ public void closeStreamOnDisconnected() {
         omegaCallbacks.get(serviceConfig.getServiceName()).containsKey(serviceConfig.getInstanceId()),
         is(false));
 
-    assertThat(compensateResponseObserver.isCompleted(), is(true));
+    await().atMost(1, SECONDS).until(compensateResponseObserver::isCompleted);
   }
 
   @Test

diff --git a/...rg/apache/servicecomb/saga/omega/connector/grpc/LoadBalancedClusterMessageSenderTest.java b/...rg/apache/servicecomb/saga/omega/connector/grpc/LoadBalancedClusterMessageSenderTest.java
@@ -250,7 +250,7 @@ public void considerFasterServerFirst() throws Exception {
     messageSender.send(event);
     messageSender.send(event);
 
-    assertThat(eventsMap.get(8080).size(), is(3));
+    await().atMost(1, SECONDS).until(() -> eventsMap.get(8080).size() == 3);
     assertThat(eventsMap.get(8090).size(), is(1));
   }