.

yirutang · yirutang · commit 31af7be52e5a · 2021-02-26T10:29:02.000-08:00
diff --git a/google-cloud-bigquerystorage/src/test/java/com/google/cloud/bigquery/storage/v1beta2/StreamWriterTest.java b/google-cloud-bigquerystorage/src/test/java/com/google/cloud/bigquery/storage/v1beta2/StreamWriterTest.java
@@ -637,6 +637,10 @@ public Throwable call() {
                 }
               }
               return null;
+            } catch (IllegalStateException ex) {
+              // Sometimes the append will happen after the stream is shutdown.
+              ex.printStackTrace();
+              return null;
             } catch (Exception e) {
               return e;
             }
@@ -650,6 +654,7 @@ public Throwable call() {
     fakeExecutor.advanceTime(Duration.ofSeconds(10));
     // The first requests gets back while the second one is blocked.
     assertEquals(2L, appendFuture1.get().getAppendResult().getOffset().getValue());
+    Thread.sleep(500);
     // When close is called, there should be one inflight request waiting.
     writer.close();
     if (future.get() != null) {

Original file line number	Diff line number	Diff line change
`@@ -637,6 +637,10 @@ public Throwable call() {`
`637`	`637`	`}`
`638`	`638`	`}`
`639`	`639`	`return null;`
	`640`	`+ } catch (IllegalStateException ex) {`
	`641`	`+ // Sometimes the append will happen after the stream is shutdown.`
	`642`	`+ ex.printStackTrace();`
	`643`	`+ return null;`
`640`	`644`	`} catch (Exception e) {`
`641`	`645`	`return e;`
`642`	`646`	`}`
`@@ -650,6 +654,7 @@ public Throwable call() {`
`650`	`654`	`fakeExecutor.advanceTime(Duration.ofSeconds(10));`
`651`	`655`	`// The first requests gets back while the second one is blocked.`
`652`	`656`	`assertEquals(2L, appendFuture1.get().getAppendResult().getOffset().getValue());`
	`657`	`+ Thread.sleep(500);`
`653`	`658`	`// When close is called, there should be one inflight request waiting.`
`654`	`659`	`writer.close();`
`655`	`660`	`if (future.get() != null) {`