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

Author: jinseopkim0

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

@@ -4826,32 +4826,39 @@ void testLocationFastSQLQueryWithJobId() throws InterruptedException {
   /* TODO(prasmish): replicate the entire test case for executeSelect */
   @Test
   void testFastSQLQueryMultiPage() throws InterruptedException {
-    String query =
-        "SELECT date, county, state_name, county_fips_code, confirmed_cases, deaths FROM "
-            + TABLE_ID_LARGE.getTable();
+    String query = "SELECT * FROM UNNEST(GENERATE_ARRAY(1, 1005)) as num";
     QueryJobConfiguration config =
-        QueryJobConfiguration.newBuilder(query).setDefaultDataset(DatasetId.of(DATASET)).build();
+        QueryJobConfiguration.newBuilder(query)
+            .setDefaultDataset(DatasetId.of(DATASET))
+            .setMaxResults(1000L)
+            .build();
     TableResult result = bigquery.query(config);
     assertNotNull(result.getJobId());
-    assertEquals(LARGE_TABLE_SCHEMA, result.getSchema());
-    assertEquals(313348, result.getTotalRows());
+    assertEquals(1, result.getSchema().getFields().size());
+    assertEquals("num", result.getSchema().getFields().get(0).getName());
+    assertEquals(LegacySQLTypeName.INTEGER, result.getSchema().getFields().get(0).getType());
+    assertEquals(1005, result.getTotalRows());
     assertNotNull(result.getNextPage());
     assertNotNull(result.getNextPageToken());
     assertTrue(result.hasNextPage());
 
     TableResult result1 = bigquery.query(config);
     assertNotNull(result1.getJobId());
-    assertEquals(LARGE_TABLE_SCHEMA, result.getSchema());
-    assertEquals(313348, result.getTotalRows());
+    assertEquals(1, result1.getSchema().getFields().size());
+    assertEquals("num", result1.getSchema().getFields().get(0).getName());
+    assertEquals(LegacySQLTypeName.INTEGER, result1.getSchema().getFields().get(0).getType());
+    assertEquals(1005, result1.getTotalRows());
     assertNotNull(result1.getNextPage());
     assertNotNull(result1.getNextPageToken());
     assertTrue(result1.hasNextPage());
 
-    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());
+    assertEquals(1, result2.getSchema().getFields().size());
+    assertEquals("num", result2.getSchema().getFields().get(0).getName());
+    assertEquals(LegacySQLTypeName.INTEGER, result2.getSchema().getFields().get(0).getType());
+    assertEquals(1005, result2.getTotalRows());
     assertNotNull(result2.getNextPage());
     assertNotNull(result2.getNextPageToken());
     assertTrue(result2.hasNextPage());