test: Close resources after tests run (#3139)

* test: Close resources after tests run

* chore: generate libraries at Thu Dec 11 21:24:33 UTC 2025

* chore: Fix failing IT test case

---------

Co-authored-by: cloud-java-bot <[email protected]>

Author: lqiu96

google-cloud-bigquerystorage/src/test/java/com/google/cloud/bigquery/storage/v1/BigQueryReadClientTest.java

@@ -42,6 +42,7 @@
 import java.util.List;
 import java.util.UUID;
 import java.util.concurrent.ExecutionException;
+import java.util.concurrent.TimeUnit;
 import org.junit.After;
 import org.junit.AfterClass;
 import org.junit.Assert;
@@ -98,6 +99,7 @@ public void onRetryAttempt(Status prevStatus, Metadata prevMetadata) {
   @After
   public void tearDown() throws Exception {
     client.close();
+    client.awaitTermination(10, TimeUnit.SECONDS);
   }
 
   @Test

google-cloud-bigquerystorage/src/test/java/com/google/cloud/bigquery/storage/v1/ConnectionWorkerTest.java

@@ -47,6 +47,7 @@
 import java.util.List;
 import java.util.UUID;
 import java.util.concurrent.ExecutionException;
+import java.util.concurrent.TimeUnit;
 import java.util.logging.Logger;
 import org.apache.arrow.memory.RootAllocator;
 import org.apache.arrow.vector.VarCharVector;
@@ -60,6 +61,7 @@
 import org.apache.arrow.vector.types.pojo.Field;
 import org.apache.arrow.vector.types.pojo.FieldType;
 import org.apache.arrow.vector.types.pojo.Schema;
+import org.junit.After;
 import org.junit.Before;
 import org.junit.Test;
 import org.junit.runner.RunWith;
@@ -103,6 +105,14 @@ public void setUp() throws Exception {
                 .build());
   }
 
+  @After
+  public void cleanUp() throws InterruptedException {
+    serviceHelper.stop();
+
+    client.close();
+    client.awaitTermination(10, TimeUnit.SECONDS);
+  }
+
   @Test
   public void testMultiplexedAppendSuccess_NonNullTraceId() throws Exception {
     testMultiplexedIngestion(

google-cloud-bigquerystorage/src/test/java/com/google/cloud/bigquery/storage/v1/JsonStreamWriterTest.java

@@ -58,6 +58,7 @@
 import java.util.Map;
 import java.util.UUID;
 import java.util.concurrent.ExecutionException;
+import java.util.concurrent.TimeUnit;
 import org.json.JSONArray;
 import org.json.JSONObject;
 import org.junit.After;
@@ -158,6 +159,9 @@ public void setUp() throws Exception {
   @After
   public void tearDown() throws Exception {
     serviceHelper.stop();
+
+    client.close();
+    client.awaitTermination(10, TimeUnit.SECONDS);
   }
 
   private JsonStreamWriter.Builder getTestJsonStreamWriterBuilder(

google-cloud-bigquerystorage/src/test/java/com/google/cloud/bigquery/storage/v1/StreamWriterTest.java

@@ -189,9 +189,11 @@ public void setUp() throws Exception {
   @After
   public void tearDown() throws Exception {
     log.info("tearDown called");
-    client.close();
     serviceHelper.stop();
     StreamWriter.cleanUp();
+
+    client.close();
+    client.awaitTermination(10, TimeUnit.SECONDS);
   }
 
   private StreamWriter getMultiplexingTestStreamWriter() throws IOException {
@@ -1996,6 +1998,9 @@ public void testBuilderExplicitSetting() throws Exception {
         ((GoogleCredentialsProvider) writerSettings2.getCredentialsProvider())
             .getScopesToApply()
             .size());
+
+    client.close();
+    client.awaitTermination(10, TimeUnit.SECONDS);
   }
 
   @Test

google-cloud-bigquerystorage/src/test/java/com/google/cloud/bigquery/storage/v1/it/ITBigQueryBigDecimalByteStringEncoderTest.java

@@ -44,6 +44,7 @@
 import java.math.BigDecimal;
 import java.util.Iterator;
 import java.util.concurrent.ExecutionException;
+import java.util.concurrent.TimeUnit;
 import java.util.logging.Logger;
 import org.json.JSONArray;
 import org.json.JSONObject;
@@ -91,9 +92,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) {
       RemoteBigQueryHelper.forceDelete(bigquery, DATASET);

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

@@ -34,6 +34,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;
@@ -74,9 +75,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);
     }
   }
 

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

@@ -104,6 +104,7 @@
 import java.util.Iterator;
 import java.util.List;
 import java.util.Map;
+import java.util.concurrent.TimeUnit;
 import java.util.logging.Logger;
 import org.apache.avro.Conversions;
 import org.apache.avro.LogicalTypes;
@@ -516,9 +517,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) {
@@ -1436,7 +1438,8 @@ public void testStructAndArraySqlTypes() throws InterruptedException, IOExceptio
   }
 
   @Test
-  public void testSimpleReadWithBackgroundExecutorProvider() throws IOException {
+  public void testSimpleReadWithBackgroundExecutorProvider()
+      throws IOException, InterruptedException {
     BigQueryReadSettings bigQueryReadSettings =
         BigQueryReadSettings.newBuilder()
             .setBackgroundExecutorProvider(

google-cloud-bigquerystorage/src/test/java/com/google/cloud/bigquery/storage/v1/it/ITBigQueryTimeEncoderTest.java

@@ -44,6 +44,7 @@
 import java.time.LocalTime;
 import java.util.Iterator;
 import java.util.concurrent.ExecutionException;
+import java.util.concurrent.TimeUnit;
 import org.json.JSONArray;
 import org.json.JSONObject;
 import org.junit.AfterClass;
@@ -98,9 +99,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) {
       RemoteBigQueryHelper.forceDelete(bigquery, DATASET);

google-cloud-bigquerystorage/src/test/java/com/google/cloud/bigquery/storage/v1/it/ITBigQueryWriteManualClientTest.java

@@ -216,9 +216,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) {

google-cloud-bigquerystorage/src/test/java/com/google/cloud/bigquery/storage/v1/stub/ResourceHeaderTest.java

@@ -28,6 +28,7 @@
 import com.google.cloud.bigquery.storage.v1.ReadRowsRequest;
 import com.google.cloud.bigquery.storage.v1.ReadSession;
 import com.google.cloud.bigquery.storage.v1.SplitReadStreamRequest;
+import java.util.concurrent.TimeUnit;
 import java.util.regex.Pattern;
 import org.junit.After;
 import org.junit.AfterClass;
@@ -88,6 +89,7 @@ public void setUp() throws Exception {
   @After
   public void tearDown() throws Exception {
     client.close();
+    client.awaitTermination(10, TimeUnit.SECONDS);
   }
 
   @AfterClass