---

pongad · garrettjonesgoogle · commit 219f842705dc · 2016-12-08T08:10:07.000-08:00
yaml --- r: 5267 b: refs/heads/master c: e99857e h: refs/heads/master i: 5265: 64661b1 5263: fb4adaa
diff --git a/[refs] b/[refs]
@@ -1,5 +1,5 @@
 ---
-refs/heads/master: 4567c49c99a134758dc12c208a9a034cae63f7c1
+refs/heads/master: e99857e80771ede2cb795dd249f2ac3ee14cf3f9
 refs/heads/travis: dae77e558b884bc1b165155482d76c8e40b0fca4
 refs/heads/gh-pages: 229631582f8957646f81e92ae5a326504f48ee5b
 refs/tags/0.0.9: 22f1839238f66c39e67ed4dfdcd273b1ae2e8444
diff --git a/trunk/google-cloud-bigquery/src/test/java/com/google/cloud/bigquery/it/ITBigQueryTest.java b/trunk/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));
   }
 
