Skip to content
This repository was archived by the owner on Feb 24, 2026. It is now read-only.

Commit edd650d

Browse files
VeronicaWassonVeronicaWasson
committed
docs(samples): Create test resouces
Create temporary dataset and table for sample integration tests
1 parent ec2ea61 commit edd650d

3 files changed

Lines changed: 93 additions & 42 deletions

File tree

‎samples/snippets/src/test/java/com/example/bigquerystorage/WriteCommittedStreamIT.java‎

Lines changed: 31 additions & 14 deletions
Original file line numberDiff line numberDiff line change
@@ -19,10 +19,20 @@
1919
import static com.google.common.truth.Truth.assertThat;
2020
import static junit.framework.TestCase.assertNotNull;
2121

22+
import com.google.cloud.bigquery.BigQuery;
23+
import com.google.cloud.bigquery.BigQuery.DatasetDeleteOption;
24+
import com.google.cloud.bigquery.BigQueryOptions;
25+
import com.google.cloud.bigquery.DatasetId;
26+
import com.google.cloud.bigquery.DatasetInfo;
27+
import com.google.cloud.bigquery.Field;
28+
import com.google.cloud.bigquery.Schema;
29+
import com.google.cloud.bigquery.StandardSQLTypeName;
30+
import com.google.cloud.bigquery.StandardTableDefinition;
31+
import com.google.cloud.bigquery.TableId;
32+
import com.google.cloud.bigquery.TableInfo;
2233
import java.io.ByteArrayOutputStream;
2334
import java.io.PrintStream;
24-
import java.util.logging.Level;
25-
import java.util.logging.Logger;
35+
import java.util.UUID;
2636
import org.junit.After;
2737
import org.junit.Before;
2838
import org.junit.BeforeClass;
@@ -34,13 +44,12 @@
3444
public class WriteCommittedStreamIT {
3545

3646
private static final String GOOGLE_CLOUD_PROJECT = System.getenv("GOOGLE_CLOUD_PROJECT");
37-
private static final String BIGQUERY_DATASET_NAME = System.getenv("BIGQUERY_DATASET_NAME");
38-
private static final String BIGQUERY_TABLE_NAME = System.getenv("BIGQUERY_TABLE_NAME");
3947

40-
private final Logger log = Logger.getLogger(this.getClass().getName());
4148
private ByteArrayOutputStream bout;
4249
private PrintStream out;
43-
private PrintStream originalPrintStream;
50+
private BigQuery bigquery;
51+
private String datasetName;
52+
private String tableName;
4453

4554
private static void requireEnvVar(String varName) {
4655
assertNotNull(
@@ -51,29 +60,37 @@ private static void requireEnvVar(String varName) {
5160
@BeforeClass
5261
public static void checkRequirements() {
5362
requireEnvVar("GOOGLE_CLOUD_PROJECT");
54-
requireEnvVar("BIGQUERY_DATASET_NAME");
55-
requireEnvVar("BIGQUERY_TABLE_NAME");
5663
}
5764

5865
@Before
5966
public void setUp() {
6067
bout = new ByteArrayOutputStream();
6168
out = new PrintStream(bout);
62-
originalPrintStream = System.out;
6369
System.setOut(out);
70+
71+
bigquery = BigQueryOptions.getDefaultInstance().getService();
72+
73+
// Create a new dataset and table for each test.
74+
datasetName = "WRITE_STREAM_TEST" + UUID.randomUUID().toString().substring(0, 8);
75+
tableName = "COMMITTED_STREAM_TEST" + UUID.randomUUID().toString().substring(0, 8);
76+
Schema schema = Schema.of(Field.of("col1", StandardSQLTypeName.STRING));
77+
bigquery.create(DatasetInfo.newBuilder(datasetName).build());
78+
TableInfo tableInfo =
79+
TableInfo.newBuilder(TableId.of(datasetName, tableName), StandardTableDefinition.of(schema))
80+
.build();
81+
bigquery.create(tableInfo);
6482
}
6583

6684
@After
6785
public void tearDown() {
68-
System.out.flush();
69-
System.setOut(originalPrintStream);
70-
log.log(Level.INFO, "\n" + bout.toString());
86+
bigquery.delete(
87+
DatasetId.of(GOOGLE_CLOUD_PROJECT, datasetName), DatasetDeleteOption.deleteContents());
88+
System.setOut(null);
7189
}
7290

7391
@Test
7492
public void testWriteCommittedStream() throws Exception {
75-
WriteCommittedStream.writeCommittedStream(
76-
GOOGLE_CLOUD_PROJECT, BIGQUERY_DATASET_NAME, BIGQUERY_TABLE_NAME);
93+
WriteCommittedStream.writeCommittedStream(GOOGLE_CLOUD_PROJECT, datasetName, tableName);
7794
assertThat(bout.toString()).contains("Appended records successfully.");
7895
}
7996
}

‎samples/snippets/src/test/java/com/example/bigquerystorage/WritePendingStreamIT.java‎

Lines changed: 31 additions & 14 deletions
Original file line numberDiff line numberDiff line change
@@ -19,10 +19,20 @@
1919
import static com.google.common.truth.Truth.assertThat;
2020
import static junit.framework.TestCase.assertNotNull;
2121

22+
import com.google.cloud.bigquery.BigQuery;
23+
import com.google.cloud.bigquery.BigQuery.DatasetDeleteOption;
24+
import com.google.cloud.bigquery.BigQueryOptions;
25+
import com.google.cloud.bigquery.DatasetId;
26+
import com.google.cloud.bigquery.DatasetInfo;
27+
import com.google.cloud.bigquery.Field;
28+
import com.google.cloud.bigquery.Schema;
29+
import com.google.cloud.bigquery.StandardSQLTypeName;
30+
import com.google.cloud.bigquery.StandardTableDefinition;
31+
import com.google.cloud.bigquery.TableId;
32+
import com.google.cloud.bigquery.TableInfo;
2233
import java.io.ByteArrayOutputStream;
2334
import java.io.PrintStream;
24-
import java.util.logging.Level;
25-
import java.util.logging.Logger;
35+
import java.util.UUID;
2636
import org.junit.After;
2737
import org.junit.Before;
2838
import org.junit.BeforeClass;
@@ -34,13 +44,12 @@
3444
public class WritePendingStreamIT {
3545

3646
private static final String GOOGLE_CLOUD_PROJECT = System.getenv("GOOGLE_CLOUD_PROJECT");
37-
private static final String BIGQUERY_DATASET_NAME = System.getenv("BIGQUERY_DATASET_NAME");
38-
private static final String BIGQUERY_TABLE_NAME = System.getenv("BIGQUERY_TABLE_NAME");
3947

40-
private final Logger log = Logger.getLogger(this.getClass().getName());
4148
private ByteArrayOutputStream bout;
4249
private PrintStream out;
43-
private PrintStream originalPrintStream;
50+
private BigQuery bigquery;
51+
private String datasetName;
52+
private String tableName;
4453

4554
private static void requireEnvVar(String varName) {
4655
assertNotNull(
@@ -51,29 +60,37 @@ private static void requireEnvVar(String varName) {
5160
@BeforeClass
5261
public static void checkRequirements() {
5362
requireEnvVar("GOOGLE_CLOUD_PROJECT");
54-
requireEnvVar("BIGQUERY_DATASET_NAME");
55-
requireEnvVar("BIGQUERY_TABLE_NAME");
5663
}
5764

5865
@Before
5966
public void setUp() {
6067
bout = new ByteArrayOutputStream();
6168
out = new PrintStream(bout);
62-
originalPrintStream = System.out;
6369
System.setOut(out);
70+
71+
bigquery = BigQueryOptions.getDefaultInstance().getService();
72+
73+
// Create a new dataset and table for each test.
74+
datasetName = "WRITE_STREAM_TEST" + UUID.randomUUID().toString().substring(0, 8);
75+
tableName = "PENDING_STREAM_TEST" + UUID.randomUUID().toString().substring(0, 8);
76+
Schema schema = Schema.of(Field.of("col1", StandardSQLTypeName.STRING));
77+
bigquery.create(DatasetInfo.newBuilder(datasetName).build());
78+
TableInfo tableInfo =
79+
TableInfo.newBuilder(TableId.of(datasetName, tableName), StandardTableDefinition.of(schema))
80+
.build();
81+
bigquery.create(tableInfo);
6482
}
6583

6684
@After
6785
public void tearDown() {
68-
System.out.flush();
69-
System.setOut(originalPrintStream);
70-
log.log(Level.INFO, "\n" + bout.toString());
86+
bigquery.delete(
87+
DatasetId.of(GOOGLE_CLOUD_PROJECT, datasetName), DatasetDeleteOption.deleteContents());
88+
System.setOut(null);
7189
}
7290

7391
@Test
7492
public void testWritePendingStream() throws Exception {
75-
WritePendingStream.writePendingStream(
76-
GOOGLE_CLOUD_PROJECT, BIGQUERY_DATASET_NAME, BIGQUERY_TABLE_NAME);
93+
WritePendingStream.writePendingStream(GOOGLE_CLOUD_PROJECT, datasetName, tableName);
7794
assertThat(bout.toString()).contains("Appended and committed records successfully.");
7895
}
7996
}

‎samples/snippets/src/test/java/com/example/bigquerystorage/WriteToDefaultStreamIT.java‎

Lines changed: 31 additions & 14 deletions
Original file line numberDiff line numberDiff line change
@@ -19,10 +19,20 @@
1919
import static com.google.common.truth.Truth.assertThat;
2020
import static junit.framework.TestCase.assertNotNull;
2121

22+
import com.google.cloud.bigquery.BigQuery;
23+
import com.google.cloud.bigquery.BigQuery.DatasetDeleteOption;
24+
import com.google.cloud.bigquery.BigQueryOptions;
25+
import com.google.cloud.bigquery.DatasetId;
26+
import com.google.cloud.bigquery.DatasetInfo;
27+
import com.google.cloud.bigquery.Field;
28+
import com.google.cloud.bigquery.Schema;
29+
import com.google.cloud.bigquery.StandardSQLTypeName;
30+
import com.google.cloud.bigquery.StandardTableDefinition;
31+
import com.google.cloud.bigquery.TableId;
32+
import com.google.cloud.bigquery.TableInfo;
2233
import java.io.ByteArrayOutputStream;
2334
import java.io.PrintStream;
24-
import java.util.logging.Level;
25-
import java.util.logging.Logger;
35+
import java.util.UUID;
2636
import org.junit.After;
2737
import org.junit.Before;
2838
import org.junit.BeforeClass;
@@ -34,13 +44,12 @@
3444
public class WriteToDefaultStreamIT {
3545

3646
private static final String GOOGLE_CLOUD_PROJECT = System.getenv("GOOGLE_CLOUD_PROJECT");
37-
private static final String BIGQUERY_DATASET_NAME = System.getenv("BIGQUERY_DATASET_NAME");
38-
private static final String BIGQUERY_TABLE_NAME = System.getenv("BIGQUERY_TABLE_NAME");
3947

40-
private final Logger log = Logger.getLogger(this.getClass().getName());
4148
private ByteArrayOutputStream bout;
4249
private PrintStream out;
43-
private PrintStream originalPrintStream;
50+
private BigQuery bigquery;
51+
private String datasetName;
52+
private String tableName;
4453

4554
private static void requireEnvVar(String varName) {
4655
assertNotNull(
@@ -51,29 +60,37 @@ private static void requireEnvVar(String varName) {
5160
@BeforeClass
5261
public static void checkRequirements() {
5362
requireEnvVar("GOOGLE_CLOUD_PROJECT");
54-
requireEnvVar("BIGQUERY_DATASET_NAME");
55-
requireEnvVar("BIGQUERY_TABLE_NAME");
5663
}
5764

5865
@Before
5966
public void setUp() {
6067
bout = new ByteArrayOutputStream();
6168
out = new PrintStream(bout);
62-
originalPrintStream = System.out;
6369
System.setOut(out);
70+
71+
bigquery = BigQueryOptions.getDefaultInstance().getService();
72+
73+
// Create a new dataset and table for each test.
74+
datasetName = "WRITE_STREAM_TEST" + UUID.randomUUID().toString().substring(0, 8);
75+
tableName = "DEFAULT_STREAM_TEST" + UUID.randomUUID().toString().substring(0, 8);
76+
Schema schema = Schema.of(Field.of("col1", StandardSQLTypeName.STRING));
77+
bigquery.create(DatasetInfo.newBuilder(datasetName).build());
78+
TableInfo tableInfo =
79+
TableInfo.newBuilder(TableId.of(datasetName, tableName), StandardTableDefinition.of(schema))
80+
.build();
81+
bigquery.create(tableInfo);
6482
}
6583

6684
@After
6785
public void tearDown() {
68-
System.out.flush();
69-
System.setOut(originalPrintStream);
70-
log.log(Level.INFO, "\n" + bout.toString());
86+
bigquery.delete(
87+
DatasetId.of(GOOGLE_CLOUD_PROJECT, datasetName), DatasetDeleteOption.deleteContents());
88+
System.setOut(null);
7189
}
7290

7391
@Test
7492
public void testWriteToDefaultStream() throws Exception {
75-
WriteToDefaultStream.writeToDefaultStream(
76-
GOOGLE_CLOUD_PROJECT, BIGQUERY_DATASET_NAME, BIGQUERY_TABLE_NAME);
93+
WriteToDefaultStream.writeToDefaultStream(GOOGLE_CLOUD_PROJECT, datasetName, tableName);
7794
assertThat(bout.toString()).contains("Appended records successfully.");
7895
}
7996
}

0 commit comments

Comments
 (0)