rust/neon
1
0
Fork 0
mirror of https://github.com/neondatabase/neon.git synced 2026-01-05 12:32:54 +00:00
Code Issues Packages Projects Releases Wiki Activity

Extend test_unlogged to include a sequence

Browse Source 
Unlogged sequences were added in v15, so let's just test to make sure
they work on Neon.
This commit is contained in:
Tristan Partin
2024-05-17 13:42:51 -05:00
committed by Tristan Partin
parent e3415706b7
commit 1988ad8db7
1 changed files with 13 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

16
test_runner/regress/test_unlogged.py
View File

@@ -1,4 +1,5 @@
from fixtures.neon_fixtures import NeonEnv, fork_at_current_lsn
from fixtures.pg_version import PgVersion
#
@@ -17,7 +18,8 @@ def test_unlogged(neon_simple_env: NeonEnv):
cur.execute("CREATE UNLOGGED TABLE iut (id int);")
# create index to test unlogged index relation as well
cur.execute("CREATE UNIQUE INDEX iut_idx ON iut (id);")
cur.execute("INSERT INTO iut values (42);")
cur.execute("ALTER TABLE iut ADD COLUMN seq int GENERATED ALWAYS AS IDENTITY;")
cur.execute("INSERT INTO iut (id) values (42);")
# create another compute to fetch inital empty contents from pageserver
fork_at_current_lsn(env, endpoint, "test_unlogged_basebackup", "test_unlogged")
@@ -26,7 +28,15 @@ def test_unlogged(neon_simple_env: NeonEnv):
conn2 = endpoint2.connect()
cur2 = conn2.cursor()
# after restart table should be empty but valid
cur2.execute("PREPARE iut_plan (int) AS INSERT INTO iut VALUES ($1)")
cur2.execute("PREPARE iut_plan (int) AS INSERT INTO iut (id) VALUES ($1)")
cur2.execute("EXECUTE iut_plan (43);")
cur2.execute("SELECT * FROM iut")
assert cur2.fetchall() == [(43,)]
results = cur2.fetchall()
# Unlogged sequences were introduced in v15. On <= v14, the sequence created
# for the GENERATED ALWAYS AS IDENTITY column is logged, and hence it keeps
# the old value (2) on restart. While on v15 and above, it's unlogged, so it
# gets reset to 1.
if env.pg_version <= PgVersion.V14:
assert results == [(43, 2)]
else:
assert results == [(43, 1)]