packages/ztd-query-pdo-adapter/tests/Integration/PostgreSql/TruncateTest.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 * @phpstan-type Row array<string, mixed>
21 */
22#[CoversNothing]
23#[Large]
24final class TruncateTest extends TestCase
25{
26 public function testTruncateTable(): void
27 {
28 $endpoint = \Testcontainers\Testcontainers::run(PostgreSqlRelease::fromEnvironment())->getData(Endpoint::class);
29 /** @var PDO $rawPdo */
30 $rawPdo = new PDO(
31 $endpoint->dsn(),
32 $endpoint->username,
33 $endpoint->password,
34 [PDO::ATTR_ERRMODE => PDO::ERRMODE_EXCEPTION, PDO::ATTR_DEFAULT_FETCH_MODE => PDO::FETCH_ASSOC],
35 );
36
37 $schemaName = 'ztd_' . bin2hex(random_bytes(8));
38 $rawPdo->exec(sprintf('CREATE SCHEMA "%s"', $schemaName));
39 $rawPdo->exec(sprintf('SET search_path TO "%s"', $schemaName));
40
41 $table = 'prefix_' . bin2hex(random_bytes(8));
42
43 try {
44 $rawPdo->exec("CREATE TABLE {$table} (id INTEGER PRIMARY KEY, name TEXT NOT NULL, age INTEGER NOT NULL)");
45 $rawPdo->exec("INSERT INTO {$table} (id, name, age) VALUES (1, 'Alice', 30), (2, 'Bob', 25), (3, 'Charlie', 35)");
46
47 $ztdPdo = ZtdPdo::fromPdo($rawPdo);
48
49 $ztdPdo->exec("INSERT INTO {$table} (id, name, age) VALUES (1, 'Alice', 30), (2, 'Bob', 25), (3, 'Charlie', 35)");
50
51 $rawPdo->exec("TRUNCATE TABLE {$table}");
52 $ztdPdo->exec("TRUNCATE TABLE {$table}");
53
54 $stmt = $rawPdo->query("SELECT * FROM {$table}");
55 self::assertNotFalse($stmt);
56 /** @var list<Row> */
57 $rawRows = $stmt->fetchAll();
58
59 $stmt = $ztdPdo->query("SELECT * FROM {$table}");
60 self::assertNotFalse($stmt);
61 /** @var list<Row> */
62 $ztdRows = $stmt->fetchAll();
63
64 self::assertSame($rawRows, $ztdRows);
65 } finally {
66 $rawPdo->exec(sprintf('DROP SCHEMA IF EXISTS "%s" CASCADE', $schemaName));
67 }
68 }
69
70 public function testTruncateWithoutTableKeyword(): void
71 {
72 $endpoint = \Testcontainers\Testcontainers::run(PostgreSqlRelease::fromEnvironment())->getData(Endpoint::class);
73 /** @var PDO $rawPdo */
74 $rawPdo = new PDO(
75 $endpoint->dsn(),
76 $endpoint->username,
77 $endpoint->password,
78 [PDO::ATTR_ERRMODE => PDO::ERRMODE_EXCEPTION, PDO::ATTR_DEFAULT_FETCH_MODE => PDO::FETCH_ASSOC],
79 );
80
81 $schemaName = 'ztd_' . bin2hex(random_bytes(8));
82 $rawPdo->exec(sprintf('CREATE SCHEMA "%s"', $schemaName));
83 $rawPdo->exec(sprintf('SET search_path TO "%s"', $schemaName));
84
85 $table = 'prefix_' . bin2hex(random_bytes(8));
86
87 try {
88 $rawPdo->exec("CREATE TABLE {$table} (id INTEGER PRIMARY KEY, name TEXT NOT NULL, age INTEGER NOT NULL)");
89 $rawPdo->exec("INSERT INTO {$table} (id, name, age) VALUES (1, 'Alice', 30), (2, 'Bob', 25), (3, 'Charlie', 35)");
90
91 $ztdPdo = ZtdPdo::fromPdo($rawPdo);
92
93 $ztdPdo->exec("INSERT INTO {$table} (id, name, age) VALUES (1, 'Alice', 30), (2, 'Bob', 25), (3, 'Charlie', 35)");
94
95 $rawPdo->exec("TRUNCATE {$table}");
96 $ztdPdo->exec("TRUNCATE {$table}");
97
98 $stmt = $rawPdo->query("SELECT * FROM {$table}");
99 self::assertNotFalse($stmt);
100 /** @var list<Row> */
101 $rawRows = $stmt->fetchAll();
102
103 $stmt = $ztdPdo->query("SELECT * FROM {$table}");
104 self::assertNotFalse($stmt);
105 /** @var list<Row> */
106 $ztdRows = $stmt->fetchAll();
107
108 self::assertSame($rawRows, $ztdRows);
109 } finally {
110 $rawPdo->exec(sprintf('DROP SCHEMA IF EXISTS "%s" CASCADE', $schemaName));
111 }
112 }
113
114 public function testTruncateDoesNotModifyPhysicalDatabase(): void
115 {
116 $endpoint = \Testcontainers\Testcontainers::run(PostgreSqlRelease::fromEnvironment())->getData(Endpoint::class);
117 /** @var PDO $rawPdo */
118 $rawPdo = new PDO(
119 $endpoint->dsn(),
120 $endpoint->username,
121 $endpoint->password,
122 [PDO::ATTR_ERRMODE => PDO::ERRMODE_EXCEPTION, PDO::ATTR_DEFAULT_FETCH_MODE => PDO::FETCH_ASSOC],
123 );
124
125 $schemaName = 'ztd_' . bin2hex(random_bytes(8));
126 $rawPdo->exec(sprintf('CREATE SCHEMA "%s"', $schemaName));
127 $rawPdo->exec(sprintf('SET search_path TO "%s"', $schemaName));
128
129 $table = 'prefix_' . bin2hex(random_bytes(8));
130
131 try {
132 $rawPdo->exec("CREATE TABLE {$table} (id INTEGER PRIMARY KEY, name TEXT NOT NULL, age INTEGER NOT NULL)");
133 $rawPdo->exec("INSERT INTO {$table} (id, name, age) VALUES (1, 'Alice', 30), (2, 'Bob', 25), (3, 'Charlie', 35)");
134
135 $ztdPdo = ZtdPdo::fromPdo($rawPdo);
136
137 $ztdPdo->exec("INSERT INTO {$table} (id, name, age) VALUES (1, 'Alice', 30), (2, 'Bob', 25), (3, 'Charlie', 35)");
138
139 $ztdPdo->exec("TRUNCATE TABLE {$table}");
140
141 $stmt = $rawPdo->query("SELECT * FROM {$table}");
142 self::assertNotFalse($stmt);
143 /** @var list<Row> */
144 $rawRows = $stmt->fetchAll();
145
146 self::assertCount(3, $rawRows);
147 } finally {
148 $rawPdo->exec(sprintf('DROP SCHEMA IF EXISTS "%s" CASCADE', $schemaName));
149 }
150 }
151
152 public function testTruncateMultipleTables(): void
153 {
154 $endpoint = \Testcontainers\Testcontainers::run(PostgreSqlRelease::fromEnvironment())->getData(Endpoint::class);
155 /** @var PDO $rawPdo */
156 $rawPdo = new PDO(
157 $endpoint->dsn(),
158 $endpoint->username,
159 $endpoint->password,
160 [PDO::ATTR_ERRMODE => PDO::ERRMODE_EXCEPTION, PDO::ATTR_DEFAULT_FETCH_MODE => PDO::FETCH_ASSOC],
161 );
162
163 $schemaName = 'ztd_' . bin2hex(random_bytes(8));
164 $rawPdo->exec(sprintf('CREATE SCHEMA "%s"', $schemaName));
165 $rawPdo->exec(sprintf('SET search_path TO "%s"', $schemaName));
166
167 $alpha = 'prefix_' . bin2hex(random_bytes(8));
168 $beta = 'prefix_' . bin2hex(random_bytes(8));
169
170 try {
171 $rawPdo->exec("CREATE TABLE {$alpha} (id INTEGER PRIMARY KEY, name TEXT)");
172 $rawPdo->exec("CREATE TABLE {$beta} (id INTEGER PRIMARY KEY, name TEXT)");
173 $rawPdo->exec("INSERT INTO {$alpha} VALUES (1, 'alpha')");
174 $rawPdo->exec("INSERT INTO {$beta} VALUES (2, 'beta')");
175 $ztdPdo = ZtdPdo::fromPdo($rawPdo);
176 $ztdPdo->exec("INSERT INTO {$alpha} VALUES (1, 'alpha')");
177 $ztdPdo->exec("INSERT INTO {$beta} VALUES (2, 'beta')");
178
179 $rawPdo->exec("TRUNCATE TABLE {$alpha}, {$beta}");
180 $ztdPdo->exec("TRUNCATE TABLE {$alpha}, {$beta}");
181
182 $rawAlpha = $rawPdo->query("SELECT * FROM {$alpha}");
183 $rawBeta = $rawPdo->query("SELECT * FROM {$beta}");
184 $shadowAlpha = $ztdPdo->query("SELECT * FROM {$alpha}");
185 $shadowBeta = $ztdPdo->query("SELECT * FROM {$beta}");
186 self::assertNotFalse($rawAlpha);
187 self::assertNotFalse($rawBeta);
188 self::assertNotFalse($shadowAlpha);
189 self::assertNotFalse($shadowBeta);
190 self::assertSame($rawAlpha->fetchAll(), $shadowAlpha->fetchAll());
191 self::assertSame($rawBeta->fetchAll(), $shadowBeta->fetchAll());
192 } finally {
193 $rawPdo->exec(sprintf('DROP SCHEMA IF EXISTS "%s" CASCADE', $schemaName));
194 }
195 }
196}
197