Skip to content

Commit

Permalink
Add integration tests to BigQuery quickstart sample.
Browse files Browse the repository at this point in the history
  • Loading branch information
tswast committed Oct 24, 2016
1 parent 16cdd3f commit 04ac9da
Show file tree
Hide file tree
Showing 3 changed files with 78 additions and 1 deletion.
4 changes: 4 additions & 0 deletions bigquery/cloud-client/README.md
Original file line number Diff line number Diff line change
Expand Up @@ -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 \
Expand Down
Original file line number Diff line number Diff line change
@@ -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.");
}
}
Original file line number Diff line number Diff line change
Expand Up @@ -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;

Expand Down

0 comments on commit 04ac9da

Please sign in to comment.