diff --git a/google-cloud-pubsublite/src/test/java/com/google/cloud/pubsublite/internal/wire/PublisherImplTest.java b/google-cloud-pubsublite/src/test/java/com/google/cloud/pubsublite/internal/wire/PublisherImplTest.java index 1af6592aa..fe14542dd 100755 --- a/google-cloud-pubsublite/src/test/java/com/google/cloud/pubsublite/internal/wire/PublisherImplTest.java +++ b/google-cloud-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));