76 lines
1.9 KiB
PHP
76 lines
1.9 KiB
PHP
<?php
|
|
|
|
declare(strict_types=1);
|
|
|
|
use Amp\Success;
|
|
use Nsq\Consumer;
|
|
use Nsq\Exception\MessageAlreadyFinished;
|
|
use Nsq\Protocol\Message;
|
|
use PHPUnit\Framework\TestCase;
|
|
use function Amp\Promise\wait;
|
|
|
|
final class MessageTest extends TestCase
|
|
{
|
|
/**
|
|
* @dataProvider messages
|
|
*/
|
|
public function testDoubleFinish(Message $message): void
|
|
{
|
|
self::assertFalse($message->isFinished());
|
|
|
|
wait($message->finish());
|
|
|
|
self::assertTrue($message->isFinished());
|
|
|
|
$this->expectException(MessageAlreadyFinished::class);
|
|
$this->expectExceptionMessage('Can\'t finish message as it already finished.');
|
|
|
|
wait($message->finish());
|
|
}
|
|
|
|
/**
|
|
* @dataProvider messages
|
|
*/
|
|
public function testDoubleRequeue(Message $message): void
|
|
{
|
|
self::assertFalse($message->isFinished());
|
|
|
|
wait($message->requeue(1));
|
|
|
|
self::assertTrue($message->isFinished());
|
|
|
|
$this->expectException(MessageAlreadyFinished::class);
|
|
$this->expectExceptionMessage('Can\'t requeue message as it already finished.');
|
|
|
|
wait($message->requeue(5));
|
|
}
|
|
|
|
/**
|
|
* @dataProvider messages
|
|
*/
|
|
public function testTouchAfterFinish(Message $message): void
|
|
{
|
|
self::assertFalse($message->isFinished());
|
|
|
|
wait($message->finish());
|
|
|
|
$this->expectException(MessageAlreadyFinished::class);
|
|
$this->expectExceptionMessage('Can\'t touch message as it already finished.');
|
|
|
|
wait($message->touch());
|
|
}
|
|
|
|
/**
|
|
* @return Generator<int, array{0: Message}>
|
|
*/
|
|
public function messages(): Generator
|
|
{
|
|
$consumer = $this->createMock(Consumer::class);
|
|
$consumer->method('fin')->willReturn(new Success());
|
|
$consumer->method('touch')->willReturn(new Success());
|
|
$consumer->method('req')->willReturn(new Success());
|
|
|
|
yield [new Message(0, 0, 'id', 'body', $consumer)];
|
|
}
|
|
}
|