fix: creates quickstart db for test

thiagotnunes · thiagotnunes · commit adf1b53a50de · 2020-11-11T11:04:53.000+11:00
Creates the quickstart database for the integration test.
diff --git a/samples/install-without-bom/pom.xml b/samples/install-without-bom/pom.xml
@@ -144,7 +144,7 @@
           <systemPropertyVariables>
             <spanner.test.instance>spanner-testing</spanner.test.instance>
             <spanner.sample.database>mysample</spanner.sample.database>
-            <spanner.quickstart.database>quickstart-db</spanner.quickstart.database>
+            <spanner.quickstart.database>quick-db</spanner.quickstart.database>
           </systemPropertyVariables>
         </configuration>
       </plugin>
diff --git a/samples/snapshot/pom.xml b/samples/snapshot/pom.xml
@@ -143,7 +143,7 @@
           <systemPropertyVariables>
             <spanner.test.instance>spanner-testing</spanner.test.instance>
             <spanner.sample.database>mysample</spanner.sample.database>
-            <spanner.quickstart.database>quickstart-db</spanner.quickstart.database>
+            <spanner.quickstart.database>quick-db</spanner.quickstart.database>
           </systemPropertyVariables>
         </configuration>
       </plugin>
diff --git a/samples/snippets/pom.xml b/samples/snippets/pom.xml
@@ -148,7 +148,7 @@
           <systemPropertyVariables>
             <spanner.test.instance>spanner-testing</spanner.test.instance>
             <spanner.sample.database>mysample</spanner.sample.database>
-            <spanner.quickstart.database>quickstart-db</spanner.quickstart.database>
+            <spanner.quickstart.database>quick-db</spanner.quickstart.database>
           </systemPropertyVariables>
         </configuration>
       </plugin>
diff --git a/samples/snippets/src/test/java/com/example/spanner/QuickstartSampleIT.java b/samples/snippets/src/test/java/com/example/spanner/QuickstartSampleIT.java
@@ -18,10 +18,19 @@
 
 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;
+import org.junit.AfterClass;
 import org.junit.Before;
+import org.junit.BeforeClass;
 import org.junit.Test;
 import org.junit.runner.RunWith;
 import org.junit.runners.JUnit4;
@@ -32,13 +41,27 @@
 @RunWith(JUnit4.class)
 @SuppressWarnings("checkstyle:abbreviationaswordinname")
 public class QuickstartSampleIT {
-  private String instanceId = System.getProperty("spanner.test.instance");
-  // This database needs to exist for test to pass.
-  private String dbId = System.getProperty("spanner.quickstart.database");
+  private static String instanceId = System.getProperty("spanner.test.instance");
+  private static String dbId = formatForTest(System.getProperty("spanner.quickstart.database"));
+  private static DatabaseAdminClient dbClient;
+
   private ByteArrayOutputStream bout;
   private PrintStream stdOut = System.out;
   private PrintStream out;
 
+  @BeforeClass
+  public static void createDatabase() {
+    final SpannerOptions options = SpannerOptions.newBuilder().build();
+    final Spanner spanner = options.getService();
+    dbClient = spanner.getDatabaseAdminClient();
+    dbClient.createDatabase(instanceId, dbId, Collections.emptyList());
+  }
+
+  @AfterClass
+  public static void dropDatabase() {
+    dbClient.dropDatabase(instanceId, dbId);
+  }
+
   @Before
   public void setUp() {
     bout = new ByteArrayOutputStream();
@@ -59,4 +82,8 @@ public void testQuickstart() throws Exception {
     String got = bout.toString();
     assertThat(got).contains("1");
   }
+
+  private static String formatForTest(String name) {
+    return name + "-" + UUID.randomUUID().toString().substring(0, 20);
+  }
 }
