Add integration tests to BigQuery quickstart sample.

tswast · tswast · commit 7b4389bc3a8c · 2016-10-25T16:05:42.000-07:00
diff --git a/bigquery/cloud-client/README.md b/bigquery/cloud-client/README.md
@@ -22,6 +22,10 @@ You can then run a given `ClassName` via:
 	    -DpropertyName=propertyValue \
 		-Dexec.args="any arguments to the app"
 
+### Creating a new dataset (using the quickstart sample)
+
+    mvn exec:java -Dexec.mainClass=com.example.bigquery.QuickstartSample
+
 ### Running a synchronous query
 
     mvn exec:java -Dexec.mainClass=com.example.bigquery.SyncQuerySample \
diff --git a/bigquery/cloud-client/src/test/java/com/example/bigquery/QuickstartSampleIT.java b/bigquery/cloud-client/src/test/java/com/example/bigquery/QuickstartSampleIT.java
@@ -0,0 +1,72 @@
+/*
+  Copyright 2016, Google, Inc.
+
+  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.bigquery;
+
+import static com.google.common.truth.Truth.assertThat;
+
+import com.google.cloud.bigquery.BigQuery;
+import com.google.cloud.bigquery.BigQuery.DatasetDeleteOption;
+import com.google.cloud.bigquery.BigQueryOptions;
+import com.google.cloud.bigquery.DatasetId;
+import org.junit.After;
+import org.junit.Before;
+import org.junit.Test;
+import org.junit.runner.RunWith;
+import org.junit.runners.JUnit4;
+
+import java.io.ByteArrayOutputStream;
+import java.io.PrintStream;
+
+/**
+ * Tests for quickstart sample.
+ */
+@RunWith(JUnit4.class)
+@SuppressWarnings("checkstyle:abbreviationaswordinname")
+public class QuickstartSampleIT {
+  private ByteArrayOutputStream bout;
+  private PrintStream out;
+
+  private static final void deleteMyNewDataset() {
+    BigQuery bigquery = BigQueryOptions.defaultInstance().service();
+    String datasetName = "my_new_dataset";
+    DatasetId datasetId = DatasetId.of(datasetName);
+    DatasetDeleteOption deleteContents = DatasetDeleteOption.deleteContents();
+    bigquery.delete(datasetId, deleteContents);
+  }
+
+  @Before
+  public void setUp() {
+    deleteMyNewDataset();
+
+    bout = new ByteArrayOutputStream();
+    out = new PrintStream(bout);
+    System.setOut(out);
+  }
+
+  @After
+  public void tearDown() {
+    System.setOut(null);
+    deleteMyNewDataset();
+  }
+
+  @Test
+  public void testQuickstart() throws Exception {
+    QuickstartSample.main();
+    String got = bout.toString();
+    assertThat(got).contains("Dataset my_new_dataset created.");
+  }
+}
diff --git a/bigquery/cloud-client/src/test/java/com/example/bigquery/SyncQuerySampleIT.java b/bigquery/cloud-client/src/test/java/com/example/bigquery/SyncQuerySampleIT.java
@@ -31,7 +31,8 @@
  * Tests for synchronous query sample.
  */
 @RunWith(JUnit4.class)
-public class SyncQuerySampleTest {
+@SuppressWarnings("checkstyle:abbreviationaswordinname")
+public class SyncQuerySampleIT {
   private ByteArrayOutputStream bout;
   private PrintStream out;
 
