public function testClass()
{
$connection = $this->prophesize(ConnectionInterface::class);
$connection->pull(Argument::withEntry('subscription', self::SUBSCRIPTION))->shouldBeCalled()->willReturn(['receivedMessages' => [['message' => ['data' => base64_encode('hello world')]]]]);
$snippet = $this->snippetFromClass(Message::class);
$snippet->addLocal('connectionStub', $connection->reveal());
$snippet->insertAfterLine(3, '$reflection = new \\ReflectionClass($pubsub);
$property = $reflection->getProperty(\'connection\');
$property->setAccessible(true);
$property->setValue($pubsub, $connectionStub);
$property->setAccessible(false);
$property = $reflection->getProperty(\'encode\');
$property->setAccessible(true);
$property->setValue($pubsub, \'false\');
$property->setAccessible(false);');
$res = $snippet->invoke('messages');
$this->assertInstanceOf(\Generator::class, $res->returnVal());
$this->assertEquals('hello world', $res->output());
}