event-sourcing/tests/Unit/Subscription/Cleanup/Dbal/DbalCleanupTaskHandlerTest.php at b05a448cde2bb4c17ccf09f892dd4944d26ad13e · patchlevel/event-sourcing · 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
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
<?php

declare(strict_types=1);

namespace Patchlevel\EventSourcing\Tests\Unit\Subscription\Cleanup\Dbal;

use Doctrine\DBAL\Connection;
use Doctrine\DBAL\Schema\AbstractSchemaManager;
use Doctrine\Persistence\ConnectionRegistry;
use Patchlevel\EventSourcing\Subscription\Cleanup\CleanupTaskNotSupported;
use Patchlevel\EventSourcing\Subscription\Cleanup\Dbal\ConnectionNameNotSupported;
use Patchlevel\EventSourcing\Subscription\Cleanup\Dbal\DbalCleanupTaskHandler;
use Patchlevel\EventSourcing\Subscription\Cleanup\Dbal\DropIndexTask;
use Patchlevel\EventSourcing\Subscription\Cleanup\Dbal\DropTableTask;
use Patchlevel\EventSourcing\Subscription\Cleanup\Dbal\UnexpectedConnectionType;
use PHPUnit\Framework\Attributes\CoversClass;
use PHPUnit\Framework\TestCase;
use stdClass;

#[CoversClass(DbalCleanupTaskHandler::class)]
final class DbalCleanupTaskHandlerTest extends TestCase
{
    public function testNoSupport(): void
    {
        $handler = new DbalCleanupTaskHandler(
            $this->createMock(Connection::class),
        );

        self::assertFalse($handler->supports(new stdClass()));
    }

    public function testSupports(): void
    {
        $handler = new DbalCleanupTaskHandler(
            $this->createMock(Connection::class),
        );

        self::assertTrue($handler->supports(new DropTableTask('test')));
        self::assertTrue($handler->supports(new DropIndexTask('test', 'test')));
    }

    public function testHandleNoSupportedTask(): void
    {
        $handler = new DbalCleanupTaskHandler(
            $this->createMock(Connection::class),
        );

        $this->expectException(CleanupTaskNotSupported::class);

        $handler(new stdClass());
    }

    public function testHandleDropTable(): void
    {
        $schemaManager = $this->createMock(AbstractSchemaManager::class);
        $schemaManager->expects($this->once())->method('dropTable')->with('test');

        $connection = $this->createMock(Connection::class);
        $connection
            ->expects($this->once())
            ->method('createSchemaManager')
            ->willReturn($schemaManager);

        $handler = new DbalCleanupTaskHandler($connection);

        $handler(new DropTableTask('test'));
    }

    public function testHandleDropIndex(): void
    {
        $schemaManager = $this->createMock(AbstractSchemaManager::class);
        $schemaManager->expects($this->once())->method('dropIndex')->with('foo', 'bar');

        $connection = $this->createMock(Connection::class);
        $connection
            ->expects($this->once())
            ->method('createSchemaManager')
            ->willReturn($schemaManager);

        $handler = new DbalCleanupTaskHandler($connection);

        $handler(new DropIndexTask('foo', 'bar'));
    }

    public function testHandleWithConnectionRegistry(): void
    {
        $schemaManager = $this->createMock(AbstractSchemaManager::class);
        $schemaManager->expects($this->once())->method('dropTable')->with('test');

        $connection = $this->createMock(Connection::class);
        $connection
            ->expects($this->once())
            ->method('createSchemaManager')
            ->willReturn($schemaManager);

        $registry = $this->createMock(ConnectionRegistry::class);
        $registry
            ->expects($this->once())
            ->method('getConnection')
            ->with('foo')
            ->willReturn($connection);

        $handler = new DbalCleanupTaskHandler($registry);

        $handler(new DropTableTask('test', 'foo'));
    }

    public function testHandleWithConnectionRegistryAndUnexpectedType(): void
    {
        $registry = $this->createMock(ConnectionRegistry::class);
        $registry
            ->expects($this->once())
            ->method('getConnection')
            ->with('foo')
            ->willReturn(new stdClass());

        $handler = new DbalCleanupTaskHandler($registry);

        $this->expectException(UnexpectedConnectionType::class);
        $handler(new DropTableTask('test', 'foo'));
    }

    public function testHandleWithConnectionAndConnectionNameNotSupported(): void
    {
        $connection = $this->createMock(Connection::class);
        $handler = new DbalCleanupTaskHandler($connection);

        $this->expectException(ConnectionNameNotSupported::class);
        $handler(new DropTableTask('test', 'foo'));
    }
}