fix: add wait in retry test (#41)

Adding a wait in the retry test to ensure that the scheduled retry completes.
googleapis · May 18, 2020 · 2e211ce · 2e211ce
1 parent 5be8343
commit 2e211ce
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/...pubsublite/src/test/java/com/google/cloud/pubsublite/internal/wire/PublisherImplTest.java b/...pubsublite/src/test/java/com/google/cloud/pubsublite/internal/wire/PublisherImplTest.java
@@ -247,6 +247,10 @@ public void retryableError_RecreatesAndRetriesAll() throws Exception {
         .when(mockPublisherFactory)
         .New(any(), any(), eq(INITIAL_PUBLISH_REQUEST));
     leakedOffsetStream.onError(Status.UNKNOWN.asRuntimeException());
+
+    // wait for retry to complete
+    Thread.sleep(500);
+
     verify(mockBatchPublisher).close();
     verifyNoMoreInteractions(mockBatchPublisher);
     verify(mockPublisherFactory, times(2)).New(any(), any(), eq(INITIAL_PUBLISH_REQUEST));