packages/ztd-query-pdo-adapter/tests/Integration/PostgreSql/PartitionChildTest.php

1<?php
2
3declare(strict_types=1);
4
5namespace Tests\Integration\PostgreSql;
6
7use Container\Endpoint;
8use Container\PostgreSqlRelease;
9use PDO;
10use PHPUnit\Framework\Attributes\CoversNothing;
11use PHPUnit\Framework\Attributes\Large;
12use PHPUnit\Framework\TestCase;
13use ZtdQuery\Adapter\Pdo\ZtdPdo;
14
15/**
16 * @requires extension pdo_pgsql
17 * @group integration
18 * @group postgres
19 */
20#[CoversNothing]
21#[Large]
22final class PartitionChildTest extends TestCase
23{
24    public function testParentAndChildDmlSharePartitionedShadowRows(): void
25    {
26        $endpoint = \Testcontainers\Testcontainers::run(PostgreSqlRelease::fromEnvironment())->getData(Endpoint::class);
27        /** @var PDO $pdo */
28        $pdo = new PDO(
29            $endpoint->dsn(),
30            $endpoint->username,
31            $endpoint->password,
32            [PDO::ATTR_ERRMODE => PDO::ERRMODE_EXCEPTION, PDO::ATTR_DEFAULT_FETCH_MODE => PDO::FETCH_ASSOC],
33        );
34
35        $schemaName = 'ztd_' . bin2hex(random_bytes(8));
36        $pdo->exec(sprintf('CREATE SCHEMA "%s"', $schemaName));
37        $pdo->exec(sprintf('SET search_path TO "%s"', $schemaName));
38
39
40        try {
41            $pdo->exec('CREATE TABLE logs (id INTEGER NOT NULL, log_date DATE NOT NULL, level TEXT NOT NULL, '
42                . 'PRIMARY KEY (id, log_date)) PARTITION BY RANGE (log_date)');
43            $pdo->exec("CREATE TABLE logs_2024 PARTITION OF logs FOR VALUES FROM ('2024-01-01') TO ('2025-01-01')");
44            $pdo->exec("CREATE TABLE logs_2025 PARTITION OF logs FOR VALUES FROM ('2025-01-01') TO ('2026-01-01')");
45            $ztdPdo = ZtdPdo::fromPdo($pdo);
46
47            self::assertSame(2, $ztdPdo->exec(
48                "INSERT INTO logs VALUES (1, '2024-05-01', 'INFO'), (2, '2025-05-01', 'WARN')",
49            ));
50
51            $parent = $ztdPdo->query('SELECT id FROM logs ORDER BY id');
52            self::assertNotFalse($parent);
53            self::assertSame([1, 2], $parent->fetchAll(PDO::FETCH_COLUMN));
54            $child2024 = $ztdPdo->query('SELECT id FROM logs_2024 ORDER BY id');
55            self::assertNotFalse($child2024);
56            self::assertSame([1], $child2024->fetchAll(PDO::FETCH_COLUMN));
57            $child2025 = $ztdPdo->query('SELECT id FROM logs_2025 ORDER BY id');
58            self::assertNotFalse($child2025);
59            self::assertSame([2], $child2025->fetchAll(PDO::FETCH_COLUMN));
60
61            self::assertSame(1, $ztdPdo->exec("INSERT INTO logs_2024 VALUES (3, '2024-09-01', 'DEBUG')"));
62            self::assertSame(1, $ztdPdo->exec("UPDATE logs_2024 SET level = 'NOTICE' WHERE id = 1"));
63            self::assertSame(1, $ztdPdo->exec('DELETE FROM logs_2025 WHERE id = 2'));
64
65            $updated = $ztdPdo->query('SELECT id, level FROM logs ORDER BY id');
66            self::assertNotFalse($updated);
67            self::assertSame(
68                [['id' => 1, 'level' => 'NOTICE'], ['id' => 3, 'level' => 'DEBUG']],
69                $updated->fetchAll(),
70            );
71            $physical = $pdo->query('SELECT COUNT(*) FROM logs');
72            self::assertNotFalse($physical);
73            self::assertSame(0, (int) $physical->fetchColumn());
74        } finally {
75            $pdo->exec(sprintf('DROP SCHEMA IF EXISTS "%s" CASCADE', $schemaName));
76        }
77    }
78}
79