Add SubscriberTest
This commit is contained in:
@ -12,6 +12,7 @@ final class Subscriber
|
||||
{
|
||||
public const STOP = 0;
|
||||
public const CHANGE_TIMEOUT = 1;
|
||||
public const TIMEOUT = 2;
|
||||
|
||||
private Consumer $reader;
|
||||
|
||||
@ -21,7 +22,7 @@ final class Subscriber
|
||||
}
|
||||
|
||||
/**
|
||||
* @psalm-return Generator<int, Message|null, int|float|null, void>
|
||||
* @psalm-return Generator<int, Message|float|null, int|float|null, void>
|
||||
*/
|
||||
public function subscribe(string $topic, string $channel, float $timeout = 0): Generator
|
||||
{
|
||||
@ -45,6 +46,10 @@ final class Subscriber
|
||||
|
||||
$timeout = $newTimeout;
|
||||
}
|
||||
|
||||
if (self::TIMEOUT === $command) {
|
||||
yield $timeout;
|
||||
}
|
||||
}
|
||||
|
||||
$this->reader->disconnect();
|
||||
|
45
tests/SubscriberTest.php
Normal file
45
tests/SubscriberTest.php
Normal file
@ -0,0 +1,45 @@
|
||||
<?php
|
||||
|
||||
declare(strict_types=1);
|
||||
|
||||
use Nsq\Consumer;
|
||||
use Nsq\Subscriber;
|
||||
use PHPUnit\Framework\TestCase;
|
||||
|
||||
final class SubscriberTest extends TestCase
|
||||
{
|
||||
private Subscriber $subscriber;
|
||||
|
||||
protected function setUp(): void
|
||||
{
|
||||
$consumer = new Consumer(
|
||||
address: 'tcp://localhost:4150',
|
||||
);
|
||||
|
||||
$this->subscriber = new Subscriber($consumer);
|
||||
}
|
||||
|
||||
public function testChangeInterval(): void
|
||||
{
|
||||
$generator = $this->subscriber->subscribe(__FUNCTION__, __FUNCTION__, 0.1);
|
||||
|
||||
self::assertSame(0.1, $generator->send(Subscriber::TIMEOUT));
|
||||
$generator->next();
|
||||
|
||||
$generator->send(Subscriber::CHANGE_TIMEOUT);
|
||||
$generator->send(0.2);
|
||||
|
||||
self::assertSame(0.2, $generator->send(Subscriber::TIMEOUT));
|
||||
}
|
||||
|
||||
public function testInvalidChangeInterval(): void
|
||||
{
|
||||
$this->expectException(\Nsq\Exception::class);
|
||||
$this->expectExceptionMessage('Timeout must be float, "string" given.');
|
||||
|
||||
$generator = $this->subscriber->subscribe(__FUNCTION__, __FUNCTION__);
|
||||
$generator->send(Subscriber::CHANGE_TIMEOUT);
|
||||
// @phpstan-ignore-next-line
|
||||
$generator->send('bla');
|
||||
}
|
||||
}
|
Reference in New Issue
Block a user