public function testClass()
{
$this->connection->runQuery(Argument::any())->shouldBeCalled()->willReturn(['batch' => ['entityResults' => [['entity' => ['key' => ['path' => []], 'properties' => ['companyName' => ['stringValue' => 'Google']]]]], 'moreResults' => 'no']]);
$this->operation->setConnection($this->connection->reveal());
$snippet = $this->snippetFromClass(Query::class);
$snippet->addLocal('operation', $this->operation);
$snippet->insertAfterLine(3, '$reflection = new \\ReflectionClass($datastore);
$property = $reflection->getProperty(\'operation\');
$property->setAccessible(true);
$property->setValue($datastore, $operation);
$property->setAccessible(false);');
$res = $snippet->invoke('res');
$this->assertEquals('Google', $res->output());
$this->assertInstanceOf(\Generator::class, $res->returnVal());
}