packages/ztd-query-pdo-adapter/tests/Integration/PostgreSql/DoBlockTest.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 DoBlockTest extends TestCase
23{
24    public function testDoBlockPassesThroughAndLaterShadowDmlStillWorks(): 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 items (id INTEGER PRIMARY KEY, name TEXT NOT NULL)');
42            $ztdPdo = ZtdPdo::fromPdo($pdo);
43
44            self::assertSame(0, $ztdPdo->exec(
45                "DO \$block\$ BEGIN INSERT INTO items VALUES (1, 'physical'); END \$block\$",
46            ));
47            self::assertSame(1, $ztdPdo->exec("INSERT INTO items VALUES (2, 'shadow')"));
48
49            $shadow = $ztdPdo->query('SELECT id, name FROM items ORDER BY id');
50            self::assertNotFalse($shadow);
51            self::assertSame([['id' => 2, 'name' => 'shadow']], $shadow->fetchAll());
52
53            $physical = $pdo->query('SELECT id, name FROM items ORDER BY id');
54            self::assertNotFalse($physical);
55            self::assertSame([['id' => 1, 'name' => 'physical']], $physical->fetchAll());
56        } finally {
57            $pdo->exec(sprintf('DROP SCHEMA IF EXISTS "%s" CASCADE', $schemaName));
58        }
59    }
60}
61