public function testRunQuery()
{
$snippet = $this->snippetFromMethod(BigQueryClient::class, 'runQuery');
$snippet->addLocal('bigQuery', $this->client);
$snippet->replace('sleep(1);', '');
$this->connection->query(Argument::any())->shouldBeCalled()->willReturn(['jobComplete' => false, 'jobReference' => ['jobId' => 'job']]);
$this->connection->getQueryResults(Argument::any())->shouldBeCalledTimes(1)->willReturn($this->result);
$this->client->setConnection($this->connection->reveal());
$res = $snippet->invoke('queryResults');
$this->assertInstanceOf(QueryResults::class, $res->returnVal());
$this->assertEquals('abcd', $res->output());
}