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
52 lines
1.4 KiB
Python
52 lines
1.4 KiB
Python
from __future__ import annotations
|
|
|
|
from typing import TYPE_CHECKING
|
|
|
|
from fixtures.log_helper import log
|
|
|
|
if TYPE_CHECKING:
|
|
from fixtures.neon_fixtures import NeonEnvBuilder
|
|
|
|
|
|
def test_oid_overflow(neon_env_builder: NeonEnvBuilder):
|
|
env = neon_env_builder.init_start()
|
|
|
|
endpoint = env.endpoints.create_start("main")
|
|
|
|
conn = endpoint.connect()
|
|
cur = conn.cursor()
|
|
|
|
cur.execute("CREATE EXTENSION neon_test_utils")
|
|
|
|
cur.execute("CREATE TABLE t1(x integer)")
|
|
cur.execute("INSERT INTO t1 values (1)")
|
|
cur.execute("CREATE TABLE t2(x integer)")
|
|
cur.execute("INSERT INTO t2 values (2)")
|
|
|
|
cur.execute("SELECT x from t1")
|
|
assert cur.fetchone() == (1,)
|
|
cur.execute("SELECT x from t2")
|
|
assert cur.fetchone() == (2,)
|
|
|
|
cur.execute("VACUUM FULL t1")
|
|
cur.execute("VACUUM FULL t1")
|
|
cur.execute("vacuum pg_class")
|
|
cur.execute("SELECT relfilenode FROM pg_class where relname='t1'")
|
|
oid = cur.fetchall()[0][0]
|
|
log.info(f"t1.relfilenode={oid}")
|
|
|
|
cur.execute("set statement_timeout=0")
|
|
cur.execute(f"select test_consume_oids({oid - 1})")
|
|
cur.execute("VACUUM FULL t2")
|
|
|
|
cur.execute("SELECT relfilenode FROM pg_class where relname='t2'")
|
|
oid = cur.fetchall()[0][0]
|
|
log.info(f"t2.relfilenode={oid}")
|
|
|
|
endpoint.clear_buffers(cursor=cur)
|
|
|
|
cur.execute("SELECT x from t1")
|
|
assert cur.fetchone() == (1,)
|
|
cur.execute("SELECT x from t2")
|
|
assert cur.fetchone() == (2,)
|