mirror of
https://github.com/neondatabase/neon.git
synced 2026-01-04 03:52:56 +00:00
30a7dd630c
## Problem `TYPE_CHECKING` is used inconsistently across Python tests. ## Summary of changes - Update `ruff`: 0.7.0 -> 0.11.2 - Enable TC (flake8-type-checking): https://docs.astral.sh/ruff/rules/#flake8-type-checking-tc - (auto)fix all new issues
50 lines
1.7 KiB
Python
50 lines
1.7 KiB
Python
from __future__ import annotations
|
|
|
|
import time
|
|
from typing import TYPE_CHECKING
|
|
|
|
if TYPE_CHECKING:
|
|
from fixtures.neon_fixtures import NeonEnvBuilder
|
|
|
|
|
|
#
|
|
# Test truncation of FSM and VM forks of a relation
|
|
#
|
|
def test_truncate(neon_env_builder: NeonEnvBuilder, zenbenchmark):
|
|
env = neon_env_builder.init_start()
|
|
n_records = 10000
|
|
n_iter = 10
|
|
|
|
# Problems with FSM/VM forks truncation are most frequently detected during page reconstruction triggered
|
|
# by image layer generation. So adjust default parameters to make it happen more frequently.
|
|
tenant, _ = env.create_tenant(
|
|
conf={
|
|
# disable automatic GC
|
|
"gc_period": "0s",
|
|
# Compact and create images aggressively
|
|
"checkpoint_distance": "1000000",
|
|
"compaction_period": "1 s",
|
|
"compaction_threshold": "3",
|
|
"image_creation_threshold": "1",
|
|
"compaction_target_size": "1000000",
|
|
}
|
|
)
|
|
|
|
endpoint = env.endpoints.create_start("main", tenant_id=tenant)
|
|
cur = endpoint.connect().cursor()
|
|
cur.execute("create table t1(x integer)")
|
|
cur.execute(f"insert into t1 values (generate_series(1,{n_records}))")
|
|
cur.execute("vacuum t1")
|
|
for _ in range(n_iter):
|
|
cur.execute(f"delete from t1 where x>{n_records // 2}")
|
|
cur.execute("vacuum t1")
|
|
time.sleep(1) # let pageserver a chance to create image layers
|
|
cur.execute(f"insert into t1 values (generate_series({n_records // 2 + 1}, {n_records}))")
|
|
cur.execute("vacuum t1")
|
|
time.sleep(1) # let pageserver a chance to create image layers
|
|
|
|
cur.execute("select count(*) from t1")
|
|
res = cur.fetchone()
|
|
assert res is not None
|
|
assert res[0] == n_records
|