test(bigquery): support us-east7 regional endpoints in integration tests

Author: jinseopkim0

.kokoro/presubmit/bigquery-graalvm-native-presubmit.cfg

@@ -22,6 +22,16 @@ env_vars: {
   value: "gcloud-devel"
 }
 
+env_vars: {
+  key: "BIGQUERY_ENDPOINT"
+  value: "https://us-east7-bigquery.googleapis.com"
+}
+
+env_vars: {
+  key: "BIGQUERY_STORAGE_ENDPOINT"
+  value: "us-east7-bigquerystorage.googleapis.com:443"
+}
+
 env_vars: {
   key: "GOOGLE_APPLICATION_CREDENTIALS"
   value: "secret_manager/java-it-service-account"

.kokoro/presubmit/bigquery-integration.cfg

@@ -22,6 +22,11 @@ env_vars: {
   value: "gcloud-devel"
 }
 
+env_vars: {
+  key: "INTEGRATION_TEST_ARGS"
+  value: "-Dbigquery.endpoint=https://us-east7-bigquery.googleapis.com -Dbigquery.storage.endpoint=us-east7-bigquerystorage.googleapis.com:443"
+}
+
 env_vars: {
   key: "GOOGLE_APPLICATION_CREDENTIALS"
   value: "secret_manager/java-it-service-account"

.kokoro/presubmit/bigquerystorage-graalvm-native-presubmit.cfg

@@ -22,6 +22,8 @@ env_vars: {
   value: "gcloud-devel"
 }
 
+
+
 env_vars: {
   key: "GOOGLE_APPLICATION_CREDENTIALS"
   value: "secret_manager/java-it-service-account"
@@ -46,3 +48,13 @@ env_vars: {
   key: "INTEGRATION_TEST_ARGS"
   value: "-Dit.test=!ITBigQueryWrite*RetryTest -Dsurefire.failIfNoSpecifiedTests=false -Dfailsafe.failIfNoSpecifiedTests=false"
 }
+
+env_vars: {
+  key: "BIGQUERY_ENDPOINT"
+  value: "https://us-east7-bigquery.googleapis.com"
+}
+
+env_vars: {
+  key: "BIGQUERY_STORAGE_ENDPOINT"
+  value: "us-east7-bigquerystorage.googleapis.com:443"
+}

.kokoro/presubmit/bigquerystorage-integration.cfg

@@ -22,6 +22,8 @@ env_vars: {
   value: "gcloud-devel"
 }
 
+
+
 env_vars: {
   key: "GOOGLE_APPLICATION_CREDENTIALS"
   value: "secret_manager/java-it-service-account"
@@ -39,5 +41,5 @@ env_vars: {
 
 env_vars: {
   key: "INTEGRATION_TEST_ARGS"
-  value: "-Dit.test=!ITBigQueryWrite*RetryTest -Dsurefire.failIfNoSpecifiedTests=false -Dfailsafe.failIfNoSpecifiedTests=false"
+  value: "-Dit.test=!ITBigQueryWrite*RetryTest -Dsurefire.failIfNoSpecifiedTests=false -Dfailsafe.failIfNoSpecifiedTests=false -Dbigquery.storage.endpoint=us-east7-bigquerystorage.googleapis.com:443 -Dbigquery.endpoint=https://us-east7-bigquery.googleapis.com"
 }

bigquery_job.log

@@ -0,0 +1 @@
+Subproject commit 122ac0f95d6ee5ac75e06b6a8bb90b576179c119

bigquery_job_api.log

@@ -0,0 +1 @@
+Subproject commit 0b5eedd3dd989e77c18e161a1ff41ce1ef224732

client-library-releases

@@ -0,0 +1 @@
+Subproject commit 941c36024d6b88eb3a610fd2eed3bece551dcb76

discovery-artifact-manager

@@ -25,6 +25,7 @@
 import java.io.IOException;
 import java.io.InputStream;
 import java.time.Duration;
+// Dummy comment to trigger Kokoro presubmit for testing regional endpoints.
 import java.util.UUID;
 import java.util.logging.Level;
 import java.util.logging.Logger;
@@ -105,7 +106,7 @@ public static RemoteBigQueryHelper create(String projectId, InputStream keyStrea
               .setProjectId(projectId)
               .setRetrySettings(retrySettings())
               .setTransportOptions(transportOptions);
-      String endpoint = System.getenv("BIGQUERY_ENDPOINT");
+      String endpoint = System.getProperty("bigquery.endpoint", System.getenv("BIGQUERY_ENDPOINT"));
       if (endpoint != null) {
         builder.setHost(endpoint);
       }
@@ -143,7 +144,7 @@ public static RemoteBigQueryHelper create(BigQueryOptions.Builder bigqueryOption
         bigqueryOptionsBuilder
             .setRetrySettings(retrySettings())
             .setTransportOptions(transportOptions);
-    String endpoint = System.getenv("BIGQUERY_ENDPOINT");
+    String endpoint = System.getProperty("bigquery.endpoint", System.getenv("BIGQUERY_ENDPOINT"));
     if (endpoint != null) {
       builder.setHost(endpoint);
     }
@@ -184,4 +185,15 @@ public static BigQueryHelperException translate(Exception ex) {
       return new BigQueryHelperException(ex.getMessage(), ex);
     }
   }
+
+  /**
+   * Helper to check if the provided BigQuery client is configured to target a regional endpoint.
+   */
+  public static boolean isRegionalEndpoint(BigQuery bigquery) {
+    if (bigquery == null || bigquery.getOptions() == null) {
+      return false;
+    }
+    String host = bigquery.getOptions().getHost();
+    return host != null && (host.contains("-bigquery.googleapis.com") || host.contains("us-east7"));
+  }
 }