Adding BigQuery integration test for query dry runs

garrettjonesgoogle · garrettjonesgoogle · commit 2c1017820391 · 2017-08-10T11:50:20.000-07:00
diff --git a/google-cloud-bigquery/src/test/java/com/google/cloud/bigquery/it/ITBigQueryTest.java b/google-cloud-bigquery/src/test/java/com/google/cloud/bigquery/it/ITBigQueryTest.java
@@ -16,6 +16,7 @@
 
 package com.google.cloud.bigquery.it;
 
+import static com.google.cloud.bigquery.JobStatus.State.DONE;
 import static org.junit.Assert.assertArrayEquals;
 import static org.junit.Assert.assertEquals;
 import static org.junit.Assert.assertFalse;
@@ -1067,6 +1068,26 @@ public void testQueryJob() throws InterruptedException, TimeoutException {
     assertNotNull(statistics.getQueryPlan());
   }
 
+  @Test
+  public void testQueryJobWithDryRun() throws InterruptedException, TimeoutException {
+    String tableName = "test_query_job_table";
+    String query = new StringBuilder()
+        .append("SELECT TimestampField, StringField, BooleanField FROM ")
+        .append(TABLE_ID.getTable())
+        .toString();
+    TableId destinationTable = TableId.of(DATASET, tableName);
+    QueryJobConfiguration configuration = QueryJobConfiguration.newBuilder(query)
+        .setDefaultDataset(DatasetId.of(DATASET))
+        .setDestinationTable(destinationTable)
+        .setDryRun(true)
+        .build();
+    Job remoteJob = bigquery.create(JobInfo.of(configuration));
+    System.out.println("job (dryrun): " + remoteJob);
+    assertNull(remoteJob.getJobId().getJob());
+    assertEquals(DONE, remoteJob.getStatus().getState());
+    assertNotNull(remoteJob.getConfiguration());
+  }
+
   @Test
   public void testExtractJob() throws InterruptedException, TimeoutException {
     String tableName = "test_export_job_table";
