mirror of
https://github.com/neondatabase/neon.git
synced 2026-01-06 13:02:55 +00:00
5bd8e2363a
This will help to keep us from using deprecated Python features going forward. Signed-off-by: Tristan Partin <tristan@neon.tech>
58 lines
2.3 KiB
Python
58 lines
2.3 KiB
Python
from __future__ import annotations
|
|
|
|
from fixtures.neon_fixtures import NeonEnv, fork_at_current_lsn
|
|
from fixtures.pg_version import PgVersion
|
|
|
|
|
|
#
|
|
# Test UNLOGGED tables/relations. Postgres copies init fork contents to main
|
|
# fork to reset them during recovery. In Neon, pageserver directly sends init
|
|
# fork contents as main fork during basebackup.
|
|
#
|
|
def test_unlogged(neon_simple_env: NeonEnv):
|
|
env = neon_simple_env
|
|
endpoint = env.endpoints.create_start("main")
|
|
|
|
conn = endpoint.connect()
|
|
cur = conn.cursor()
|
|
|
|
cur.execute("CREATE UNLOGGED TABLE iut (id int);")
|
|
# create index to test unlogged index relations as well
|
|
cur.execute("CREATE UNIQUE INDEX iut_idx ON iut (id);")
|
|
cur.execute("CREATE INDEX ON iut USING gist (int4range(id, id, '[]'));")
|
|
cur.execute("CREATE INDEX ON iut USING spgist (int4range(id, id, '[]'));")
|
|
cur.execute("CREATE INDEX ON iut USING gin ((id::text::jsonb));")
|
|
cur.execute("CREATE INDEX ON iut USING brin (id);")
|
|
cur.execute("CREATE INDEX ON iut USING hash (id);")
|
|
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", "main")
|
|
endpoint2 = env.endpoints.create_start("test_unlogged_basebackup")
|
|
|
|
conn2 = endpoint2.connect()
|
|
cur2 = conn2.cursor()
|
|
# after restart table should be empty but valid
|
|
cur2.execute("PREPARE iut_plan (int) AS INSERT INTO iut (id) VALUES ($1)")
|
|
cur2.execute("EXECUTE iut_plan (43);")
|
|
cur2.execute("SELECT * FROM iut")
|
|
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)]
|
|
|
|
# Flush all data and compact it, so we detect any errors related to
|
|
# unlogged indexes materialization.
|
|
ps_http = env.pageserver.http_client()
|
|
ps_http.timeline_compact(
|
|
tenant_id=env.initial_tenant,
|
|
timeline_id=env.initial_timeline,
|
|
force_image_layer_creation=True,
|
|
)
|