test(bigquery): use queryWithTimeout on TABLE_ID_LARGE to guarantee pagination and avoid flaky sleep loop

b/467064659

Author: jinseopkim0

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

@@ -4831,30 +4831,30 @@ void testLocationFastSQLQueryWithJobId() throws InterruptedException {
 
   /* TODO(prasmish): replicate the entire test case for executeSelect */
   @Test
-  void testFastSQLQueryMultiPage() throws InterruptedException {
+  void testFastSQLQueryMultiPage() throws Exception {
     String query =
         "SELECT date, county, state_name, county_fips_code, confirmed_cases, deaths FROM "
             + TABLE_ID_LARGE.getTable();
     QueryJobConfiguration config =
         QueryJobConfiguration.newBuilder(query).setDefaultDataset(DatasetId.of(DATASET)).build();
-    TableResult result = bigquery.query(config);
+    TableResult result = (TableResult) bigquery.queryWithTimeout(config, null, 60000L);
     assertNotNull(result.getJobId());
     assertEquals(LARGE_TABLE_SCHEMA, result.getSchema());
     assertEquals(313348, result.getTotalRows());
     assertNotNull(result.getNextPage());
     assertNotNull(result.getNextPageToken());
     assertTrue(result.hasNextPage());
 
-    TableResult result1 = bigquery.query(config);
+    TableResult result1 = (TableResult) bigquery.queryWithTimeout(config, null, 60000L);
     assertNotNull(result1.getJobId());
     assertEquals(LARGE_TABLE_SCHEMA, result.getSchema());
     assertEquals(313348, result.getTotalRows());
     assertNotNull(result1.getNextPage());
     assertNotNull(result1.getNextPageToken());
     assertTrue(result1.hasNextPage());
 
-    config.toBuilder().setQuery(query).build();
-    TableResult result2 = bigquery.query(config);
+    config = config.toBuilder().setQuery(query).build();
+    TableResult result2 = (TableResult) bigquery.queryWithTimeout(config, null, 60000L);
     assertNotNull(result2.getJobId());
     assertEquals(LARGE_TABLE_SCHEMA, result2.getSchema());
     assertEquals(313348, result2.getTotalRows());