test(spanner): clean up stale backup schedules older than 3 hours

java-spanner/samples/snippets/pom.xml

@@ -25,6 +25,7 @@
     <maven.compiler.source>1.8</maven.compiler.source>
     <project.build.sourceEncoding>UTF-8</project.build.sourceEncoding>
     <opencensus.version>0.31.1</opencensus.version>
+    <spanner.sample.database>mysample</spanner.sample.database>
   </properties>
 
 
@@ -137,7 +138,7 @@
                 <spanner.test.key.location>us-east1</spanner.test.key.location>
                 <spanner.test.key.ring>cmek-test-key-ring</spanner.test.key.ring>
                 <spanner.test.key.name>cmek-test-key</spanner.test.key.name>
-                <spanner.sample.database>mysample</spanner.sample.database>
+                <spanner.sample.database>${spanner.sample.database}</spanner.sample.database>
                 <spanner.quickstart.database>quick-db</spanner.quickstart.database>
               </systemPropertyVariables>
               <excludes>
@@ -166,7 +167,7 @@
                 <spanner.test.key.location>us-east1</spanner.test.key.location>
                 <spanner.test.key.ring>cmek-test-key-ring</spanner.test.key.ring>
                 <spanner.test.key.name>cmek-test-key</spanner.test.key.name>
-                <spanner.sample.database>mysample</spanner.sample.database>
+                <spanner.sample.database>${spanner.sample.database}</spanner.sample.database>
                 <spanner.quickstart.database>quick-db</spanner.quickstart.database>
               </systemPropertyVariables>
               <includes>

java-spanner/samples/snippets/src/test/java/com/example/spanner/CreateFullBackupScheduleSampleIT.java

@@ -20,14 +20,20 @@
 
 import com.google.spanner.admin.database.v1.BackupScheduleName;
 import java.util.UUID;
+import org.junit.Before;
 import org.junit.Test;
 import org.junit.runner.RunWith;
 import org.junit.runners.JUnit4;
 
 @RunWith(JUnit4.class)
 public class CreateFullBackupScheduleSampleIT extends SampleTestBaseV2 {
   // Default instance and given db should exist for tests to pass.
-  private static String databaseId = System.getProperty("spanner.sample.database", "mysample");
+  private static String databaseId = getSampleDatabase();
+
+  @Before
+  public void cleanUpPreExistingBackupSchedules() {
+    cleanUpStaleBackupSchedules(instanceId, databaseId);
+  }
 
   @Test
   public void testCreateFullBackupScheduleSample() throws Exception {

java-spanner/samples/snippets/src/test/java/com/example/spanner/CreateIncrementalBackupScheduleSampleIT.java

@@ -20,14 +20,20 @@
 
 import com.google.spanner.admin.database.v1.BackupScheduleName;
 import java.util.UUID;
+import org.junit.Before;
 import org.junit.Test;
 import org.junit.runner.RunWith;
 import org.junit.runners.JUnit4;
 
 @RunWith(JUnit4.class)
 public class CreateIncrementalBackupScheduleSampleIT extends SampleTestBaseV2 {
   // Default instance and given db should exist for tests to pass.
-  private static String databaseId = System.getProperty("spanner.sample.database", "mysample");
+  private static String databaseId = getSampleDatabase();
+
+  @Before
+  public void cleanUpPreExistingBackupSchedules() {
+    cleanUpStaleBackupSchedules(multiRegionalInstanceId, databaseId);
+  }
 
   @Test
   public void testCreateIncrementalBackupScheduleSample() throws Exception {

java-spanner/samples/snippets/src/test/java/com/example/spanner/DeleteBackupScheduleSampleIT.java

@@ -20,14 +20,20 @@
 
 import com.google.spanner.admin.database.v1.BackupScheduleName;
 import java.util.UUID;
+import org.junit.Before;
 import org.junit.Test;
 import org.junit.runner.RunWith;
 import org.junit.runners.JUnit4;
 
 @RunWith(JUnit4.class)
 public class DeleteBackupScheduleSampleIT extends SampleTestBaseV2 {
   // Default instance and given db should exist for tests to pass.
-  private static String databaseId = System.getProperty("spanner.sample.database", "mysample");
+  private static String databaseId = getSampleDatabase();
+
+  @Before
+  public void cleanUpPreExistingBackupSchedules() {
+    cleanUpStaleBackupSchedules(instanceId, databaseId);
+  }
 
   @Test
   public void testDeleteBackupScheduleSample() throws Exception {

java-spanner/samples/snippets/src/test/java/com/example/spanner/GetBackupScheduleSampleIT.java

@@ -20,14 +20,20 @@
 
 import com.google.spanner.admin.database.v1.BackupScheduleName;
 import java.util.UUID;
+import org.junit.Before;
 import org.junit.Test;
 import org.junit.runner.RunWith;
 import org.junit.runners.JUnit4;
 
 @RunWith(JUnit4.class)
 public class GetBackupScheduleSampleIT extends SampleTestBaseV2 {
   // Default instance and given db should exist for tests to pass.
-  private static String databaseId = System.getProperty("spanner.sample.database", "mysample");
+  private static String databaseId = getSampleDatabase();
+
+  @Before
+  public void cleanUpPreExistingBackupSchedules() {
+    cleanUpStaleBackupSchedules(instanceId, databaseId);
+  }
 
   @Test
   public void testGetBackupScheduleSample() throws Exception {

java-spanner/samples/snippets/src/test/java/com/example/spanner/ListBackupSchedulesSampleIT.java

@@ -20,14 +20,20 @@
 
 import com.google.spanner.admin.database.v1.BackupScheduleName;
 import java.util.UUID;
+import org.junit.Before;
 import org.junit.Test;
 import org.junit.runner.RunWith;
 import org.junit.runners.JUnit4;
 
 @RunWith(JUnit4.class)
 public class ListBackupSchedulesSampleIT extends SampleTestBaseV2 {
   // Default instance and given db should exist for tests to pass.
-  private static String databaseId = System.getProperty("spanner.sample.database", "mysample");
+  private static String databaseId = getSampleDatabase();
+
+  @Before
+  public void cleanUpPreExistingBackupSchedules() {
+    cleanUpStaleBackupSchedules(instanceId, databaseId);
+  }
 
   @Test
   public void testListBackupSchedulesSample() throws Exception {

java-spanner/samples/snippets/src/test/java/com/example/spanner/SampleTestBaseV2.java

@@ -23,8 +23,10 @@
 import com.google.cloud.spanner.admin.database.v1.DatabaseAdminSettings;
 import com.google.cloud.spanner.admin.instance.v1.InstanceAdminClient;
 import com.google.cloud.spanner.admin.instance.v1.InstanceAdminSettings;
+import com.google.spanner.admin.database.v1.BackupSchedule;
 import com.google.spanner.admin.database.v1.DatabaseDialect;
 import java.io.IOException;
+import java.time.Instant;
 import java.util.concurrent.TimeUnit;
 import org.junit.AfterClass;
 import org.junit.BeforeClass;
@@ -37,12 +39,45 @@ public class SampleTestBaseV2 {
   private static final String BASE_INSTANCE_ID =
       System.getProperty("spanner.sample.instance", "mysample-instance");
 
-  private static final String BASE_DATABASE_ID =
-      System.getProperty("spanner.sample.database", "sampledb");
+  private static final String BASE_DATABASE_ID = getSampleDatabase();
   private static final String BASE_BACKUP_ID = "samplebk";
   private static final String BASE_INSTANCE_CONFIG_ID = "sampleconfig";
   private static final int AWAIT_TERMINATION_SECONDS = 10;
 
+  static String resolveSampleDatabase(String sysProp, String envDb, String kokoroJob) {
+    if (sysProp != null && !sysProp.trim().isEmpty() && !sysProp.equals("mysample")) {
+      return sysProp.trim();
+    }
+    if (envDb != null && !envDb.trim().isEmpty()) {
+      return envDb.trim();
+    }
+    if (kokoroJob != null) {
+      if (kokoroJob.contains("java8-samples")) {
+        return "mysample_java8";
+      } else if (kokoroJob.contains("java11-samples")) {
+        return "mysample_java11";
+      }
+    }
+    if (sysProp != null && !sysProp.trim().isEmpty()) {
+      return sysProp.trim();
+    }
+    return "mysample";
+  }
+
+  public static String getSampleDatabase() {
+    String kokoroJob = System.getenv("KOKORO_JOB_NAME");
+    if (kokoroJob == null || kokoroJob.isEmpty()) {
+      kokoroJob = System.getenv("JOB_NAME");
+    }
+    if (kokoroJob == null || kokoroJob.isEmpty()) {
+      kokoroJob = System.getenv("JOB_TYPE");
+    }
+    return resolveSampleDatabase(
+        System.getProperty("spanner.sample.database"),
+        System.getenv("SPANNER_SAMPLE_DATABASE"),
+        kokoroJob);
+  }
+
   protected static String projectId;
   protected static final String instanceId = System.getProperty("spanner.test.instance");
   protected static DatabaseAdminClient databaseAdminClient;
@@ -148,6 +183,40 @@ public static void afterClass() throws InterruptedException {
     instanceAdminClient.awaitTermination(AWAIT_TERMINATION_SECONDS, TimeUnit.SECONDS);
   }
 
+  protected static void cleanUpStaleBackupSchedules(
+      final String targetInstanceId, final String targetDatabaseId) {
+    try {
+      final long nowSeconds = Instant.now().getEpochSecond();
+      final long staleThresholdSeconds = TimeUnit.HOURS.toSeconds(3);
+
+      for (BackupSchedule schedule :
+          databaseAdminClient
+              .listBackupSchedules(getDatabaseName(projectId, targetInstanceId, targetDatabaseId))
+              .iterateAll()) {
+        if (schedule.hasUpdateTime()) {
+          long ageSeconds = nowSeconds - schedule.getUpdateTime().getSeconds();
+          if (ageSeconds > staleThresholdSeconds) {
+            try {
+              databaseAdminClient.deleteBackupSchedule(schedule.getName());
+            } catch (Exception e) {
+              System.out.println(
+                  "Failed to delete stale backup schedule "
+                      + schedule.getName()
+                      + " due to "
+                      + e.getMessage()
+                      + ", skipping...");
+            }
+          }
+        }
+      }
+    } catch (Exception e) {
+      System.out.println(
+          "Failed to list or clean up stale backup schedules due to "
+              + e.getMessage()
+              + ", skipping...");
+    }
+  }
+
   static String getDatabaseName(final String projectId,
       final String instanceId, final String databaseId) {
     return String.format(

java-spanner/samples/snippets/src/test/java/com/example/spanner/SampleTestBaseV2Test.java

@@ -0,0 +1,78 @@
+/*
+ * Copyright 2026 Google LLC
+ *
+ * Licensed under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License.
+ * You may obtain a copy of the License at
+ *
+ *       http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+
+package com.example.spanner;
+
+import static com.google.common.truth.Truth.assertThat;
+
+import org.junit.Test;
+import org.junit.runner.RunWith;
+import org.junit.runners.JUnit4;
+
+@RunWith(JUnit4.class)
+public class SampleTestBaseV2Test {
+
+  @Test
+  public void testDefaultFallbackToMySample() {
+    assertThat(SampleTestBaseV2.resolveSampleDatabase(null, null, null)).isEqualTo("mysample");
+    assertThat(SampleTestBaseV2.resolveSampleDatabase("", "", "")).isEqualTo("mysample");
+  }
+
+  @Test
+  public void testDefaultPropertyReturnsMySample() {
+    assertThat(SampleTestBaseV2.resolveSampleDatabase("mysample", null, null)).isEqualTo("mysample");
+  }
+
+  @Test
+  public void testKokoroJava8SamplesAutoDetection() {
+    assertThat(
+            SampleTestBaseV2.resolveSampleDatabase(
+                "mysample", null, "cloud-devrel/client-libraries/java/java-spanner/nightly/java8-samples"))
+        .isEqualTo("mysample_java8");
+  }
+
+  @Test
+  public void testKokoroJava11SamplesAutoDetection() {
+    assertThat(
+            SampleTestBaseV2.resolveSampleDatabase(
+                "mysample", null, "cloud-devrel/client-libraries/java/java-spanner/nightly/java11-samples"))
+        .isEqualTo("mysample_java11");
+  }
+
+  @Test
+  public void testKokoroSamplesJobFallback() {
+    assertThat(
+            SampleTestBaseV2.resolveSampleDatabase(
+                "mysample", null, "cloud-devrel/client-libraries/java/java-spanner/nightly/samples"))
+        .isEqualTo("mysample");
+  }
+
+  @Test
+  public void testExplicitSystemPropertyOverrideTakesPrecedence() {
+    assertThat(
+            SampleTestBaseV2.resolveSampleDatabase(
+                "custom-db", null, "cloud-devrel/client-libraries/java/java-spanner/nightly/java8-samples"))
+        .isEqualTo("custom-db");
+  }
+
+  @Test
+  public void testExplicitEnvVarOverrideTakesPrecedence() {
+    assertThat(
+            SampleTestBaseV2.resolveSampleDatabase(
+                "mysample", "env-custom-db", "cloud-devrel/client-libraries/java/java-spanner/nightly/java8-samples"))
+        .isEqualTo("env-custom-db");
+  }
+}

java-spanner/samples/snippets/src/test/java/com/example/spanner/UpdateBackupScheduleSampleIT.java

@@ -20,14 +20,20 @@
 
 import com.google.spanner.admin.database.v1.BackupScheduleName;
 import java.util.UUID;
+import org.junit.Before;
 import org.junit.Test;
 import org.junit.runner.RunWith;
 import org.junit.runners.JUnit4;
 
 @RunWith(JUnit4.class)
 public class UpdateBackupScheduleSampleIT extends SampleTestBaseV2 {
   // Default instance and given db should exist for tests to pass.
-  private static String databaseId = System.getProperty("spanner.sample.database", "mysample");
+  private static String databaseId = getSampleDatabase();
+
+  @Before
+  public void cleanUpPreExistingBackupSchedules() {
+    cleanUpStaleBackupSchedules(instanceId, databaseId);
+  }
 
   @Test
   public void testUpdateBackupScheduleSample() throws Exception {