---

pongad · garrettjonesgoogle · commit 9a7e921517c5 · 2016-12-08T08:10:07.000-08:00
yaml --- r: 7715 b: refs/heads/tswast-patch-1 c: e99857e h: refs/heads/master i: 7713: 9477fbf 7711: 48dd6dc
diff --git a/[refs] b/[refs]
@@ -57,5 +57,5 @@ refs/tags/v0.18.0: 9d193c4c4b9d1c6f21515dd8e50836b9194ec9bb
 refs/tags/v0.19.0: e67b56e4d8dad5f9a7b38c9b2107c23c828f2ed5
 refs/tags/v0.20.0: 839f7fb7156535146aa1cb2c5aadd8d375d854e8
 refs/tags/v0.20.1: 370471f437f1f4f68a11e068df5cd6bf39edb1fa
-refs/heads/tswast-patch-1: 4567c49c99a134758dc12c208a9a034cae63f7c1
+refs/heads/tswast-patch-1: e99857e80771ede2cb795dd249f2ac3ee14cf3f9
 refs/heads/pubsub-streaming-pull: 19262b752ee874eb2ca3b950eb2aef44d5a5267b
diff --git a/branches/tswast-patch-1/google-cloud-bigquery/src/test/java/com/google/cloud/bigquery/it/ITBigQueryTest.java b/branches/tswast-patch-1/google-cloud-bigquery/src/test/java/com/google/cloud/bigquery/it/ITBigQueryTest.java
@@ -25,6 +25,7 @@
 import static org.junit.Assert.fail;
 
 import com.google.cloud.Page;
+import com.google.cloud.WaitForOption;
 import com.google.cloud.bigquery.BigQuery;
 import com.google.cloud.bigquery.BigQuery.DatasetField;
 import com.google.cloud.bigquery.BigQuery.DatasetOption;
@@ -899,7 +900,7 @@ public void testListJobsWithSelectedFields() {
   }
 
   @Test
-  public void testCreateAndGetJob() {
+  public void testCreateAndGetJob() throws InterruptedException, TimeoutException {
     String sourceTableName = "test_create_and_get_job_source_table";
     String destinationTableName = "test_create_and_get_job_destination_table";
     TableId sourceTable = TableId.of(DATASET, sourceTableName);
@@ -930,11 +931,18 @@ public void testCreateAndGetJob() {
     assertEquals(createdJob.getSelfLink(), remoteJob.getSelfLink());
     assertEquals(createdJob.getUserEmail(), remoteJob.getUserEmail());
     assertTrue(createdTable.delete());
+
+    Job completedJob =
+        remoteJob.waitFor(
+            WaitForOption.checkEvery(1, TimeUnit.SECONDS),
+            WaitForOption.timeout(1, TimeUnit.MINUTES));
+    assertNotNull(completedJob);
+    assertNull(completedJob.getStatus().getError());
     assertTrue(bigquery.delete(DATASET, destinationTableName));
   }
 
   @Test
-  public void testCreateAndGetJobWithSelectedFields() {
+  public void testCreateAndGetJobWithSelectedFields() throws InterruptedException, TimeoutException {
     String sourceTableName = "test_create_and_get_job_with_selected_fields_source_table";
     String destinationTableName = "test_create_and_get_job_with_selected_fields_destination_table";
     TableId sourceTable = TableId.of(DATASET, sourceTableName);
@@ -972,6 +980,13 @@ public void testCreateAndGetJobWithSelectedFields() {
     assertNull(remoteJob.getSelfLink());
     assertNull(remoteJob.getUserEmail());
     assertTrue(createdTable.delete());
+
+    Job completedJob =
+        remoteJob.waitFor(
+            WaitForOption.checkEvery(1, TimeUnit.SECONDS),
+            WaitForOption.timeout(1, TimeUnit.MINUTES));
+    assertNotNull(completedJob);
+    assertNull(completedJob.getStatus().getError());
     assertTrue(bigquery.delete(DATASET, destinationTableName));
   }
 
