Fix the pagination bug for stateless queries

Author: Hectorhammett

BigQuery/src/BigQueryClient.php

@@ -438,13 +438,14 @@ public function runQuery(JobConfigurationInterface $query, array $options = [])
             $response = $this->connection->query($statelessArgs);
 
             if ($response['jobComplete'] ?? false) {
+                $jobId = $response['jobReference']['jobId'] ?? '';
                 return new QueryResults(
                     $this->connection,
-                    '',
+                    $jobId,
                     $this->projectId,
                     $response,
                     $this->mapper,
-                    $this->createJob([], ''), // create an empty job
+                    $this->createJob($response, $jobId),
                     $queryResultsOptions
                 );
             }

BigQuery/tests/Unit/BigQueryClientTest.php

@@ -182,6 +182,56 @@ public function testRunQueryStateless()
         $this->assertTrue($queryResults->isComplete());
     }
 
+    public function testRunQueryStatelessWithPagination()
+    {
+        $client = $this->getClient();
+        $query = $client->query(self::QUERY_STRING);
+
+        $this->connection->query(Argument::allOf(
+            Argument::withEntry('projectId', self::PROJECT_ID),
+            Argument::withEntry('query', self::QUERY_STRING),
+            Argument::withEntry('jobCreationMode', 'JOB_CREATION_OPTIONAL')
+        ))
+            ->willReturn([
+                'jobComplete' => true,
+                'jobReference' => [
+                    'jobId' => self::JOB_ID,
+                    'projectId' => self::PROJECT_ID
+                ],
+                'schema' => [
+                    'fields' => [
+                        ['name' => 'col1', 'type' => 'STRING']
+                    ]
+                ],
+                'rows' => [
+                    ['f' => [['v' => 'val1']]]
+                ],
+                'pageToken' => 'next-page-token'
+            ])
+            ->shouldBeCalledTimes(1);
+
+        $this->connection->getQueryResults(Argument::allOf(
+            Argument::withEntry('projectId', self::PROJECT_ID),
+            Argument::withEntry('jobId', self::JOB_ID),
+            Argument::withEntry('pageToken', 'next-page-token')
+        ))
+            ->willReturn([
+                'jobComplete' => true,
+                'rows' => [
+                    ['f' => [['v' => 'val2']]]
+                ]
+            ])
+            ->shouldBeCalledTimes(1);
+
+        $client->___setProperty('connection', $this->connection->reveal());
+        $results = $client->runQuery($query);
+        $rows = iterator_to_array($results);
+
+        $this->assertCount(2, $rows);
+        $this->assertEquals('val1', $rows[0]['col1']);
+        $this->assertEquals('val2', $rows[1]['col1']);
+    }
+
     public function testRunQueryJobQueryEndpointReturnsAJob()
     {
         $client = $this->getClient();