test: reduce administrative calls in samples (#717)

Author: olavloite

samples/snippets/src/main/java/com/example/spanner/SpannerSample.java

@@ -1671,12 +1671,16 @@ static void cancelCreateBackup(
   static void listBackupOperations(InstanceAdminClient instanceAdminClient, DatabaseId databaseId) {
     Instance instance = instanceAdminClient.getInstance(databaseId.getInstanceId().getInstance());
     // Get create backup operations for the sample database.
+    Timestamp last24Hours = Timestamp.ofTimeSecondsAndNanos(TimeUnit.SECONDS.convert(
+        TimeUnit.HOURS.convert(Timestamp.now().getSeconds(), TimeUnit.SECONDS) - 24,
+        TimeUnit.HOURS), 0);
     String filter =
         String.format(
             "(metadata.database:%s) AND "
                 + "(metadata.@type:type.googleapis.com/"
-                + "google.spanner.admin.database.v1.CreateBackupMetadata)",
-            databaseId.getName());
+                + "google.spanner.admin.database.v1.CreateBackupMetadata) AND "
+                + "(metadata.progress.start_time > \"%s\")",
+            databaseId.getName(), last24Hours);
     Page<Operation> operations = instance.listBackupOperations(Options.filter(filter));
     for (Operation op : operations.iterateAll()) {
       try {
@@ -1702,8 +1706,12 @@ static void listDatabaseOperations(
       InstanceId instanceId) {
     Instance instance = instanceAdminClient.getInstance(instanceId.getInstance());
     // Get optimize restored database operations.
-    String filter = "(metadata.@type:type.googleapis.com/"
-                    + "google.spanner.admin.database.v1.OptimizeRestoredDatabaseMetadata)";
+    Timestamp last24Hours = Timestamp.ofTimeSecondsAndNanos(TimeUnit.SECONDS.convert(
+        TimeUnit.HOURS.convert(Timestamp.now().getSeconds(), TimeUnit.SECONDS) - 24,
+        TimeUnit.HOURS), 0);
+    String filter = String.format("(metadata.@type:type.googleapis.com/"
+                    + "google.spanner.admin.database.v1.OptimizeRestoredDatabaseMetadata) AND "
+                    + "(metadata.progress.start_time > \"%s\")", last24Hours);
     for (Operation op : instance.listDatabaseOperations(Options.filter(filter)).iterateAll()) {
       try {
         OptimizeRestoredDatabaseMetadata metadata =

samples/snippets/src/test/java/com/example/spanner/AsyncExamplesIT.java

@@ -72,7 +72,8 @@ private String runExample(AsyncRunnable example)
 
   @BeforeClass
   public static void createTestDatabase() throws Exception {
-    SpannerOptions options = SpannerOptions.newBuilder().build();
+    SpannerOptions options =
+        SpannerOptions.newBuilder().setAutoThrottleAdministrativeRequests().build();
     spanner = options.getService();
     dbClient = spanner.getDatabaseAdminClient();
     if (instanceId == null) {

samples/snippets/src/test/java/com/example/spanner/QuickstartSampleIT.java

@@ -19,12 +19,10 @@
 import static com.google.common.truth.Truth.assertThat;
 
 import com.google.cloud.spanner.DatabaseAdminClient;
-import com.google.cloud.spanner.DatabaseNotFoundException;
 import com.google.cloud.spanner.Spanner;
 import com.google.cloud.spanner.SpannerOptions;
 import java.io.ByteArrayOutputStream;
 import java.io.PrintStream;
-import java.util.Arrays;
 import java.util.Collections;
 import java.util.UUID;
 import org.junit.After;
@@ -43,6 +41,7 @@
 public class QuickstartSampleIT {
   private static String instanceId = System.getProperty("spanner.test.instance");
   private static String dbId = formatForTest(System.getProperty("spanner.quickstart.database"));
+  private static Spanner spanner;
   private static DatabaseAdminClient dbClient;
 
   private ByteArrayOutputStream bout;
@@ -51,15 +50,17 @@ public class QuickstartSampleIT {
 
   @BeforeClass
   public static void createDatabase() {
-    final SpannerOptions options = SpannerOptions.newBuilder().build();
-    final Spanner spanner = options.getService();
+    final SpannerOptions options =
+        SpannerOptions.newBuilder().setAutoThrottleAdministrativeRequests().build();
+    spanner = options.getService();
     dbClient = spanner.getDatabaseAdminClient();
     dbClient.createDatabase(instanceId, dbId, Collections.emptyList());
   }
 
   @AfterClass
   public static void dropDatabase() {
     dbClient.dropDatabase(instanceId, dbId);
+    spanner.close();
   }
 
   @Before

samples/snippets/src/test/java/com/example/spanner/SpannerSampleIT.java

@@ -68,7 +68,8 @@ private String runSample(String command) throws Exception {
 
   @BeforeClass
   public static void setUp() throws Exception {
-    SpannerOptions options = SpannerOptions.newBuilder().build();
+    SpannerOptions options =
+        SpannerOptions.newBuilder().setAutoThrottleAdministrativeRequests().build();
     spanner = options.getService();
     dbClient = spanner.getDatabaseAdminClient();
     dbId = DatabaseId.of(options.getProjectId(), instanceId, databaseId);

samples/snippets/src/test/java/com/example/spanner/SpannerStandaloneExamplesIT.java

@@ -66,7 +66,8 @@ private String runExample(Runnable example) {
 
   @BeforeClass
   public static void createTestDatabase() throws Exception {
-    SpannerOptions options = SpannerOptions.newBuilder().build();
+    SpannerOptions options =
+        SpannerOptions.newBuilder().setAutoThrottleAdministrativeRequests().build();
     spanner = options.getService();
     dbClient = spanner.getDatabaseAdminClient();
     if (instanceId == null) {