add schema test

googleapis · Jul 9, 2020 · aaa6b80 · aaa6b80
1 parent e73936b
commit aaa6b80
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/google-cloud-bigquery/src/test/java/com/google/cloud/bigquery/BigQueryImplTest.java b/google-cloud-bigquery/src/test/java/com/google/cloud/bigquery/BigQueryImplTest.java
@@ -2149,6 +2149,7 @@ public void testFastQuerySQLShouldRetry() throws Exception {
             .getService();
 
     TableResult response = bigquery.query(QUERY_JOB_CONFIGURATION_FOR_QUERY);
+    assertEquals(TABLE_SCHEMA, response.getSchema());
     assertEquals(1, response.getTotalRows());
     verify(bigqueryRpcMock, times(5)).fastQuery(PROJECT, requestPb);
   }
@@ -2184,6 +2185,7 @@ public void testFastQueryDMLShouldRetry() throws Exception {
             .getService();
 
     TableResult response = bigquery.query(QUERY_JOB_CONFIGURATION_FOR_DMLQUERY);
+    assertEquals(TABLE_SCHEMA, response.getSchema());
     assertEquals(1, response.getTotalRows());
     verify(bigqueryRpcMock, times(5)).fastQuery(PROJECT, requestPb);
   }
@@ -2218,6 +2220,7 @@ public void testFastQueryDDLShouldRetry() throws Exception {
             .getService();
 
     TableResult response = bigquery.query(QUERY_JOB_CONFIGURATION_FOR_DDLQUERY);
+    assertEquals(TABLE_SCHEMA, response.getSchema());
     assertEquals(0, response.getTotalRows());
     verify(bigqueryRpcMock, times(5)).fastQuery(PROJECT, requestPb);
   }