mirror of
https://github.com/neondatabase/neon.git
synced 2025-12-22 21:59:59 +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
32 lines
965 B
Python
32 lines
965 B
Python
from __future__ import annotations
|
|
|
|
from typing import TYPE_CHECKING
|
|
|
|
from fixtures.utils import query_scalar
|
|
|
|
if TYPE_CHECKING:
|
|
from fixtures.neon_fixtures import NeonEnv
|
|
|
|
|
|
#
|
|
# Test CREATE USER to check shared catalog restore
|
|
#
|
|
def test_createuser(neon_simple_env: NeonEnv):
|
|
env = neon_simple_env
|
|
endpoint = env.endpoints.create_start("main")
|
|
|
|
with endpoint.cursor() as cur:
|
|
# Cause a 'relmapper' change in the original branch
|
|
cur.execute("CREATE USER testuser with password %s", ("testpwd",))
|
|
|
|
cur.execute("CHECKPOINT")
|
|
|
|
lsn = query_scalar(cur, "SELECT pg_current_wal_insert_lsn()")
|
|
|
|
# Create a branch
|
|
env.create_branch("test_createuser2", ancestor_branch_name="main", ancestor_start_lsn=lsn)
|
|
endpoint2 = env.endpoints.create_start("test_createuser2")
|
|
|
|
# Test that you can connect to new branch as a new user
|
|
assert endpoint2.safe_psql("select current_user", user="testuser") == [("testuser",)]
|