Google\Cloud\Tests\Snippets\Datastore\Query\QueryTest::testClass PHP Method

    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());
    }