test(bigquery): use TABLE_ID_FAST_QUERY and maxResults to make testFastSQLQueryMultiPage hermetic and fast

b/467064659

Author: jinseopkim0

java-bigquery/google-cloud-bigquery/src/test/java/com/google/cloud/bigquery/it/ITBigQueryTest.java

@@ -4827,34 +4827,32 @@ void testLocationFastSQLQueryWithJobId() throws InterruptedException {
   @Test
   void testFastSQLQueryMultiPage() throws InterruptedException {
     String query =
-        "SELECT date, county, state_name, county_fips_code, confirmed_cases, deaths FROM "
-            + TABLE_ID_LARGE.getTable();
+        "SELECT TimestampField, StringField, BooleanField FROM " + TABLE_ID_FAST_QUERY.getTable();
     QueryJobConfiguration config =
-        QueryJobConfiguration.newBuilder(query).setDefaultDataset(DatasetId.of(DATASET)).build();
+        QueryJobConfiguration.newBuilder(query)
+            .setDefaultDataset(DatasetId.of(DATASET))
+            .setMaxResults(1L)
+            .build();
+
+    java.util.function.Consumer<TableResult> assertResult =
+        r -> {
+          assertNotNull(r.getJobId());
+          assertEquals(QUERY_RESULT_SCHEMA, r.getSchema());
+          assertEquals(2, r.getTotalRows());
+          assertNotNull(r.getNextPage());
+          assertNotNull(r.getNextPageToken());
+          assertTrue(r.hasNextPage());
+        };
+
     TableResult result = bigquery.query(config);
-    assertNotNull(result.getJobId());
-    assertEquals(LARGE_TABLE_SCHEMA, result.getSchema());
-    assertEquals(313348, result.getTotalRows());
-    assertNotNull(result.getNextPage());
-    assertNotNull(result.getNextPageToken());
-    assertTrue(result.hasNextPage());
+    assertResult.accept(result);
 
     TableResult result1 = bigquery.query(config);
-    assertNotNull(result1.getJobId());
-    assertEquals(LARGE_TABLE_SCHEMA, result.getSchema());
-    assertEquals(313348, result.getTotalRows());
-    assertNotNull(result1.getNextPage());
-    assertNotNull(result1.getNextPageToken());
-    assertTrue(result1.hasNextPage());
+    assertResult.accept(result1);
 
-    config.toBuilder().setQuery(query).build();
+    config = config.toBuilder().setQuery(query).build();
     TableResult result2 = bigquery.query(config);
-    assertNotNull(result2.getJobId());
-    assertEquals(LARGE_TABLE_SCHEMA, result2.getSchema());
-    assertEquals(313348, result2.getTotalRows());
-    assertNotNull(result2.getNextPage());
-    assertNotNull(result2.getNextPageToken());
-    assertTrue(result2.hasNextPage());
+    assertResult.accept(result2);
   }
 
   @Test