mirror of
https://github.com/neondatabase/neon.git
synced 2026-01-10 15:02:56 +00:00
b83d722369
Going through the list of recent flaky tests, trying to fix those related to graceful shutdown. - test_forward_compatibility: flush and wait for uploads to avoid graceful shutdown - test_layer_bloating: in the end the endpoint and vanilla are still up => immediate shutdown - test_lagging_sk: pageserver shutdown is not related to the test => immediate shutdown - test_lsn_lease_size: pageserver flushing is not needed => immediate shutdown Additionally: - remove `wait_for_upload` usage from workload fixture Cc: #8708 Fixes: #8710
72 lines
2.6 KiB
Python
72 lines
2.6 KiB
Python
import os
|
|
|
|
import pytest
|
|
from fixtures.log_helper import log
|
|
from fixtures.neon_fixtures import (
|
|
NeonEnvBuilder,
|
|
logical_replication_sync,
|
|
wait_for_last_flush_lsn,
|
|
)
|
|
from fixtures.pg_version import PgVersion
|
|
|
|
|
|
def test_layer_bloating(neon_env_builder: NeonEnvBuilder, vanilla_pg):
|
|
if neon_env_builder.pg_version != PgVersion.V16:
|
|
pytest.skip("pg_log_standby_snapshot() function is available only in PG16")
|
|
|
|
env = neon_env_builder.init_start(
|
|
initial_tenant_conf={
|
|
"gc_period": "0s",
|
|
"compaction_period": "0s",
|
|
"compaction_threshold": 99999,
|
|
"image_creation_threshold": 99999,
|
|
}
|
|
)
|
|
|
|
timeline = env.initial_timeline
|
|
endpoint = env.endpoints.create_start("main", config_lines=["log_statement=all"])
|
|
|
|
pg_conn = endpoint.connect()
|
|
cur = pg_conn.cursor()
|
|
|
|
# create table...
|
|
cur.execute("create table t(pk integer primary key)")
|
|
cur.execute("create publication pub1 for table t")
|
|
# Create slot to hold WAL
|
|
cur.execute("select pg_create_logical_replication_slot('my_slot', 'pgoutput')")
|
|
|
|
# now start subscriber
|
|
vanilla_pg.start()
|
|
vanilla_pg.safe_psql("create table t(pk integer primary key)")
|
|
|
|
connstr = endpoint.connstr().replace("'", "''")
|
|
log.info(f"ep connstr is {endpoint.connstr()}, subscriber connstr {vanilla_pg.connstr()}")
|
|
vanilla_pg.safe_psql(f"create subscription sub1 connection '{connstr}' publication pub1")
|
|
|
|
cur.execute(
|
|
"""create or replace function create_snapshots(n integer) returns void as $$
|
|
declare
|
|
i integer;
|
|
begin
|
|
for i in 1..n loop
|
|
perform pg_log_standby_snapshot();
|
|
end loop;
|
|
end; $$ language plpgsql"""
|
|
)
|
|
cur.execute("set statement_timeout=0")
|
|
cur.execute("select create_snapshots(10000)")
|
|
# Wait logical replication to sync
|
|
logical_replication_sync(vanilla_pg, endpoint)
|
|
wait_for_last_flush_lsn(env, endpoint, env.initial_tenant, timeline)
|
|
env.pageserver.http_client().timeline_checkpoint(env.initial_tenant, timeline, compact=False)
|
|
|
|
# Check layer file sizes
|
|
timeline_path = f"{env.pageserver.workdir}/tenants/{env.initial_tenant}/timelines/{timeline}/"
|
|
log.info(f"Check {timeline_path}")
|
|
for filename in os.listdir(timeline_path):
|
|
if filename.startswith("00000"):
|
|
log.info(f"layer {filename} size is {os.path.getsize(timeline_path + filename)}")
|
|
assert os.path.getsize(timeline_path + filename) < 512_000_000
|
|
|
|
env.stop(immediate=True)
|