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
59 lines
1.9 KiB
Python
59 lines
1.9 KiB
Python
from __future__ import annotations
|
|
|
|
import shutil
|
|
from pathlib import Path
|
|
from tempfile import NamedTemporaryFile
|
|
from typing import TYPE_CHECKING
|
|
|
|
import pytest
|
|
from fixtures.utils import subprocess_capture
|
|
|
|
if TYPE_CHECKING:
|
|
from fixtures.neon_fixtures import RemotePostgres
|
|
|
|
|
|
@pytest.mark.remote_cluster
|
|
@pytest.mark.parametrize(
|
|
"client",
|
|
[
|
|
"csharp/npgsql",
|
|
"java/jdbc",
|
|
"rust/tokio-postgres",
|
|
"python/asyncpg",
|
|
"python/pg8000",
|
|
# PostgresClientKitExample does not support SNI or connection options, so it uses workaround D (https://neon.tech/sni)
|
|
# See example itself: test_runner/pg_clients/swift/PostgresClientKitExample/Sources/PostgresClientKitExample/main.swift
|
|
"swift/PostgresClientKitExample",
|
|
"swift/PostgresNIOExample",
|
|
"typescript/postgresql-client",
|
|
"typescript/serverless-driver",
|
|
],
|
|
)
|
|
def test_pg_clients(test_output_dir: Path, remote_pg: RemotePostgres, client: str):
|
|
conn_options = remote_pg.conn_options()
|
|
|
|
env_file = None
|
|
with NamedTemporaryFile(mode="w", delete=False) as f:
|
|
env_file = f.name
|
|
f.write(
|
|
f"""
|
|
NEON_HOST={conn_options["host"]}
|
|
NEON_DATABASE={conn_options["dbname"]}
|
|
NEON_USER={conn_options["user"]}
|
|
NEON_PASSWORD={conn_options["password"]}
|
|
"""
|
|
)
|
|
|
|
image_tag = client.lower()
|
|
docker_bin = shutil.which("docker")
|
|
if docker_bin is None:
|
|
raise RuntimeError("docker is required for running this test")
|
|
|
|
build_cmd = [docker_bin, "build", "--tag", image_tag, f"{Path(__file__).parent / client}"]
|
|
subprocess_capture(test_output_dir, build_cmd, check=True)
|
|
|
|
run_cmd = [docker_bin, "run", "--rm", "--env-file", env_file, image_tag]
|
|
_, output, _ = subprocess_capture(test_output_dir, run_cmd, check=True, capture_stdout=True)
|
|
|
|
assert str(output).strip() == "1"
|