test(spanner): unflake ITBatchReadTest by skipping DataBoost tests on… (#14473)

… cloud-devel and cloud-staging

Skip dataBoostRead and dataBoostQuery in ITBatchReadTest when running
against cloud-devel (staging-wrenchworks.sandbox.googleapis.com) or
cloud-staging (preprod-spanner.sandbox.googleapis.com) while DataBoost
backend capacity adjustments roll out.

Author: sakthivelmanii

java-spanner/google-cloud-spanner/src/test/java/com/google/cloud/spanner/IntegrationTestEnv.java

@@ -123,12 +123,22 @@ protected void initializeConfig()
     config = configClass.newInstance();
   }
 
-  boolean isCloudDevel() {
+  public boolean isCloudDevel() {
     return Objects.equals(
         System.getProperty("spanner.gce.config.server_url"),
         "https://staging-wrenchworks.sandbox.googleapis.com");
   }
 
+  public boolean isCloudStaging() {
+    return Objects.equals(
+        System.getProperty("spanner.gce.config.server_url"),
+        "https://preprod-spanner.sandbox.googleapis.com");
+  }
+
+  public boolean isCloudDevelOrStaging() {
+    return isCloudDevel() || isCloudStaging();
+  }
+
   @Override
   protected void before() throws Throwable {
     this.initializeConfig();

java-spanner/google-cloud-spanner/src/test/java/com/google/cloud/spanner/IntegrationTestEnvTest.java

@@ -58,4 +58,39 @@ public void testIsRetryableResourceExhaustedException() {
                 "Could not create instance. Quota exceeded for quota metric 'Instance create"
                     + " requests' and limit 'Instance create requests per minute'")));
   }
+
+  @Test
+  public void testCloudDevelAndStagingDetection() {
+    IntegrationTestEnv env = new IntegrationTestEnv();
+    String originalServerUrl = System.getProperty("spanner.gce.config.server_url");
+    try {
+      System.clearProperty("spanner.gce.config.server_url");
+      assertFalse(env.isCloudDevel());
+      assertFalse(env.isCloudStaging());
+      assertFalse(env.isCloudDevelOrStaging());
+
+      System.setProperty(
+          "spanner.gce.config.server_url", "https://staging-wrenchworks.sandbox.googleapis.com");
+      assertTrue(env.isCloudDevel());
+      assertFalse(env.isCloudStaging());
+      assertTrue(env.isCloudDevelOrStaging());
+
+      System.setProperty(
+          "spanner.gce.config.server_url", "https://preprod-spanner.sandbox.googleapis.com");
+      assertFalse(env.isCloudDevel());
+      assertTrue(env.isCloudStaging());
+      assertTrue(env.isCloudDevelOrStaging());
+
+      System.setProperty("spanner.gce.config.server_url", "https://spanner.googleapis.com");
+      assertFalse(env.isCloudDevel());
+      assertFalse(env.isCloudStaging());
+      assertFalse(env.isCloudDevelOrStaging());
+    } finally {
+      if (originalServerUrl == null) {
+        System.clearProperty("spanner.gce.config.server_url");
+      } else {
+        System.setProperty("spanner.gce.config.server_url", originalServerUrl);
+      }
+    }
+  }
 }

java-spanner/google-cloud-spanner/src/test/java/com/google/cloud/spanner/it/ITBatchReadTest.java

@@ -250,6 +250,7 @@ public void readUsingIndex() {
   @Test
   public void dataBoostRead() {
     assumeFalse("data boost is not supported on Spanner Omni yet", isSpannerOmni());
+    assumeFalse("Skipping the test due to a known bug b/557058207", env.isCloudDevelOrStaging());
     BitSet seenRows = new BitSet(numRows);
     TimestampBound bound = getRandomBound();
     PartitionOptions partitionParams = getRandomPartitionOptions();
@@ -303,6 +304,7 @@ private PartitionOptions getRandomPartitionOptions() {
   @Test
   public void dataBoostQuery() {
     assumeFalse("data boost is not supported on Spanner Omni yet", isSpannerOmni());
+    assumeFalse("Skipping the test due to a known bug b/557058207", env.isCloudDevelOrStaging());
     BitSet seenRows = new BitSet(numRows);
     TimestampBound bound = getRandomBound();
     PartitionOptions partitionParams = getRandomPartitionOptions();