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
27 lines
786 B
Python
27 lines
786 B
Python
from __future__ import annotations
|
|
|
|
from typing import TYPE_CHECKING
|
|
|
|
from fixtures.log_helper import log
|
|
from fixtures.utils import wait_until
|
|
|
|
if TYPE_CHECKING:
|
|
from fixtures.common_types import TenantId, TimelineId
|
|
from fixtures.safekeeper.http import SafekeeperHttpClient
|
|
|
|
|
|
def wait_walreceivers_absent(
|
|
sk_http_cli: SafekeeperHttpClient, tenant_id: TenantId, timeline_id: TimelineId
|
|
):
|
|
"""
|
|
Wait until there is no walreceiver connections from the compute(s) on the
|
|
safekeeper.
|
|
"""
|
|
|
|
def walreceivers_absent():
|
|
status = sk_http_cli.timeline_status(tenant_id, timeline_id)
|
|
log.info(f"waiting for walreceivers to be gone, currently {status.walreceivers}")
|
|
assert len(status.walreceivers) == 0
|
|
|
|
wait_until(walreceivers_absent)
|