queue/tests/Unit/Middleware/Push/Implementation/IdMiddlewareTest.php at ea8387fc8fe60e27491818ab207e155d7744127a · yiisoft/queue · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
<?php

declare(strict_types=1);

namespace Yiisoft\Queue\Tests\Unit\Middleware\Push\Implementation;

use PHPUnit\Framework\TestCase;
use Yiisoft\Queue\Message\IdEnvelope;
use Yiisoft\Queue\Message\Message;
use Yiisoft\Queue\Middleware\Push\Implementation\IdMiddleware;
use Yiisoft\Queue\Middleware\Push\MessageHandlerPushInterface;
use Yiisoft\Queue\Middleware\Push\PushRequest;

final class IdMiddlewareTest extends TestCase
{
    public function testWithId(): void
    {
        $message = new Message('test', null, [IdEnvelope::MESSAGE_ID_KEY => 'test-id']);
        $originalRequest = new PushRequest($message, null);
        $handler = $this->createMock(MessageHandlerPushInterface::class);

        $handler->expects($this->once())
            ->method('handlePush')
            ->willReturnArgument(0);

        $middleware = new IdMiddleware();
        $finalRequest = $middleware->processPush($originalRequest, $handler);

        $this->assertSame($originalRequest, $finalRequest);
        $this->assertNotInstanceOf(IdEnvelope::class, $finalRequest->getMessage());
        $this->assertEquals('test-id', $finalRequest->getMessage()->getMetadata()[IdEnvelope::MESSAGE_ID_KEY]);
        $this->assertSame($originalRequest->getMessage()->getData(), $finalRequest->getMessage()->getData());
        $this->assertSame($originalRequest->getMessage()->getHandlerName(), $finalRequest->getMessage()->getHandlerName());
    }

    public function testWithoutId(): void
    {
        $message = new Message('test', null);
        $originalRequest = new PushRequest($message, null);
        $handler = $this->createMock(MessageHandlerPushInterface::class);

        $handler->expects($this->once())
            ->method('handlePush')
            ->willReturnArgument(0);

        $middleware = new IdMiddleware();
        $finalRequest = $middleware->processPush($originalRequest, $handler);

        $this->assertInstanceOf(IdEnvelope::class, $finalRequest->getMessage());
        $this->assertNotSame($originalRequest, $finalRequest);
        $this->assertNotEmpty($finalRequest->getMessage()->getMetadata()[IdEnvelope::MESSAGE_ID_KEY] ?? null);
        $this->assertSame($originalRequest->getMessage()->getData(), $finalRequest->getMessage()->getData());
        $this->assertSame($originalRequest->getMessage()->getHandlerName(), $finalRequest->getMessage()->getHandlerName());
    }
}