fix:add more diagnostic info for test awaitTermination (#290)

yirutang · web-flow · commit a5b7c3922bee · 2020-05-18T16:18:08.000-04:00
modified:   google-cloud-bigquerystorage/src/test/java/com/google/cloud/bigquery/storage/v1alpha2/StreamWriterTest.java
diff --git a/google-cloud-bigquerystorage/src/test/java/com/google/cloud/bigquery/storage/v1alpha2/StreamWriterTest.java b/google-cloud-bigquerystorage/src/test/java/com/google/cloud/bigquery/storage/v1alpha2/StreamWriterTest.java
@@ -94,13 +94,17 @@ public void tearDown() throws Exception {
     serviceHelper.stop();
   }
 
-  private StreamWriter.Builder getTestStreamWriterBuilder() {
-    return StreamWriter.newBuilder(TEST_STREAM)
+  private StreamWriter.Builder getTestStreamWriterBuilder(String testStream) {
+    return StreamWriter.newBuilder(testStream)
         .setChannelProvider(channelProvider)
         .setExecutorProvider(SINGLE_THREAD_EXECUTOR)
         .setCredentialsProvider(NoCredentialsProvider.create());
   }
 
+  private StreamWriter.Builder getTestStreamWriterBuilder() {
+    return getTestStreamWriterBuilder(TEST_STREAM);
+  }
+
   private AppendRowsRequest createAppendRequest(String[] messages, long offset) {
     AppendRowsRequest.Builder requestBuilder = AppendRowsRequest.newBuilder();
     AppendRowsRequest.ProtoData.Builder dataBuilder = AppendRowsRequest.ProtoData.newBuilder();
@@ -821,9 +825,10 @@ public void testBuilderInvalidArguments() {
 
   @Test
   public void testAwaitTermination() throws Exception {
-    StreamWriter writer = getTestStreamWriterBuilder().build();
+    StreamWriter writer =
+        getTestStreamWriterBuilder("projects/p/datasets/d/tables/t/streams/await").build();
     testBigQueryWrite.addResponse(AppendRowsResponse.newBuilder().build());
-    ApiFuture<AppendRowsResponse> appendFuture1 = sendTestMessage(writer, new String[] {"A"});
+    ApiFuture<AppendRowsResponse> appendFuture1 = sendTestMessage(writer, new String[] {"AWAIT"});
     writer.shutdown();
     assertTrue(writer.awaitTermination(1, TimeUnit.MINUTES));
   }
