Files
neon/test_runner/regress/test_pgstat.py
Konstantin Knizhnik dfed7029e1 Make clippy happy
2024-12-04 09:40:47 +02:00

67 lines
1.8 KiB
Python

import pytest
from fixtures.neon_fixtures import AuxFileStore, NeonEnv
from fixtures.pg_version import PgVersion
#
# Test that pgstat statistic is preserved across sessions
#
@pytest.mark.parametrize(
"pageserver_aux_file_policy", [AuxFileStore.V1, AuxFileStore.V2, AuxFileStore.CrossValidation]
)
def test_pgstat(neon_simple_env: NeonEnv):
env = neon_simple_env
if env.pg_version == PgVersion.V14:
pytest.skip("PG14 doesn't support pgstat statistic persistence")
n = 10000
env.neon_cli.create_branch("test_pgstat", "empty")
endpoint = env.endpoints.create_start("test_pgstat")
con = endpoint.connect()
cur = con.cursor()
cur.execute("create table t(x integer)")
cur.execute(f"insert into t values (generate_series(1,{n}))")
cur.execute("vacuum analyze t")
cur.execute("select sum(x) from t")
cur.execute("update t set x=x+1")
cur.execute("select pg_stat_force_next_flush()")
cur.execute(
"select seq_scan,seq_tup_read,n_tup_ins,n_tup_upd,n_live_tup,n_dead_tup, vacuum_count,analyze_count from pg_stat_user_tables"
)
rec = cur.fetchall()[0]
assert (
rec[0] == 2
and rec[1] == n * 2
and rec[2] == n
and rec[3] == n
and rec[4] == n * 2
and rec[5] == n
and rec[6] == 1
and rec[7] == 1
)
endpoint.stop()
endpoint.start()
con = endpoint.connect()
cur = con.cursor()
cur.execute(
"select seq_scan,seq_tup_read,n_tup_ins,n_tup_upd,n_live_tup,n_dead_tup, vacuum_count,analyze_count from pg_stat_user_tables"
)
rec = cur.fetchall()[0]
assert (
rec[0] == 2
and rec[1] == n * 2
and rec[2] == n
and rec[3] == n
and rec[4] == n * 2
and rec[5] == n
and rec[6] == 1
and rec[7] == 1
)