chore: Remove the deprecated v1beta2 Write API tests

Author: lqiu96

google-cloud-bigquerystorage/src/test/java/com/google/cloud/bigquery/storage/v1beta1/it/ITBigQueryStorageLongRunningTest.java

@@ -36,6 +36,7 @@
 import java.util.concurrent.ExecutorService;
 import java.util.concurrent.Executors;
 import java.util.concurrent.Future;
+import java.util.concurrent.TimeUnit;
 import java.util.logging.Logger;
 import org.junit.AfterClass;
 import org.junit.Assume;
@@ -78,9 +79,10 @@ public static void beforeClass() throws IOException {
   }
 
   @AfterClass
-  public static void afterClass() {
+  public static void afterClass() throws InterruptedException {
     if (client != null) {
       client.close();
+      client.awaitTermination(10, TimeUnit.SECONDS);
     }
   }
 
@@ -111,17 +113,12 @@ public void testLongRunningReadSession() throws InterruptedException, ExecutionE
 
     List<Callable<Long>> tasks = new ArrayList<>(session.getStreamsCount());
     for (final Stream stream : session.getStreamsList()) {
-      tasks.add(
-          new Callable<Long>() {
-            @Override
-            public Long call() throws Exception {
-              return readAllRowsFromStream(stream);
-            }
-          });
+      tasks.add(() -> readAllRowsFromStream(stream));
     }
 
     ExecutorService executor = Executors.newFixedThreadPool(tasks.size());
     List<Future<Long>> results = executor.invokeAll(tasks);
+    executor.shutdown();
 
     long rowCount = 0;
     for (Future<Long> result : results) {

google-cloud-bigquerystorage/src/test/java/com/google/cloud/bigquery/storage/v1beta1/it/ITBigQueryStorageTest.java

@@ -78,6 +78,7 @@
 import java.util.Collections;
 import java.util.Iterator;
 import java.util.List;
+import java.util.concurrent.TimeUnit;
 import java.util.logging.Logger;
 import org.apache.avro.Conversions;
 import org.apache.avro.LogicalTypes;
@@ -199,9 +200,10 @@ public static void beforeClass() throws IOException {
   }
 
   @AfterClass
-  public static void afterClass() {
+  public static void afterClass() throws InterruptedException {
     if (client != null) {
       client.close();
+      client.awaitTermination(10, TimeUnit.SECONDS);
     }
 
     if (bigquery != null) {
@@ -1318,8 +1320,7 @@ private Job RunQueryJobAndExpectSuccess(QueryJobConfiguration configuration)
   }
 
   static ServiceAccountCredentials loadCredentials(String credentialFile) {
-    try {
-      InputStream keyStream = new ByteArrayInputStream(credentialFile.getBytes());
+    try (InputStream keyStream = new ByteArrayInputStream(credentialFile.getBytes())) {
       return ServiceAccountCredentials.fromStream(keyStream);
     } catch (IOException e) {
       fail("Couldn't create fake JSON credentials.");