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