mirror of
https://github.com/neondatabase/neon.git
synced 2025-12-24 06:39:58 +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
68 lines
2.0 KiB
Python
68 lines
2.0 KiB
Python
from __future__ import annotations
|
|
|
|
import datetime
|
|
from typing import TYPE_CHECKING
|
|
|
|
import pytest
|
|
from fixtures.benchmark_fixture import MetricReport, NeonBenchmarker
|
|
|
|
if TYPE_CHECKING:
|
|
from fixtures.neon_fixtures import NeonEnv
|
|
|
|
|
|
@pytest.mark.timeout(120)
|
|
def test_compute_ctl_api_latencies(
|
|
neon_simple_env: NeonEnv,
|
|
zenbenchmark: NeonBenchmarker,
|
|
):
|
|
"""
|
|
Test compute_ctl HTTP API performance. Do simple GET requests
|
|
to catch any pathological degradations in the HTTP server.
|
|
"""
|
|
env = neon_simple_env
|
|
|
|
endpoint = env.endpoints.create_start("main")
|
|
client = endpoint.http_client()
|
|
|
|
NUM_REQUESTS = 10000
|
|
|
|
status_response_latency_us = []
|
|
metrics_response_latency_us = []
|
|
|
|
for _i in range(NUM_REQUESTS):
|
|
start_time = datetime.datetime.now()
|
|
_ = client.status()
|
|
status_response_latency_us.append((datetime.datetime.now() - start_time).microseconds)
|
|
|
|
start_time = datetime.datetime.now()
|
|
_ = client.metrics_json()
|
|
metrics_response_latency_us.append((datetime.datetime.now() - start_time).microseconds)
|
|
|
|
status_response_latency_us = sorted(status_response_latency_us)
|
|
metrics_response_latency_us = sorted(metrics_response_latency_us)
|
|
|
|
zenbenchmark.record(
|
|
"status_response_latency_p50_us",
|
|
status_response_latency_us[len(status_response_latency_us) // 2],
|
|
"μs",
|
|
MetricReport.LOWER_IS_BETTER,
|
|
)
|
|
zenbenchmark.record(
|
|
"metrics_response_latency_p50_us",
|
|
metrics_response_latency_us[len(metrics_response_latency_us) // 2],
|
|
"μs",
|
|
MetricReport.LOWER_IS_BETTER,
|
|
)
|
|
zenbenchmark.record(
|
|
"status_response_latency_p99_us",
|
|
status_response_latency_us[len(status_response_latency_us) * 99 // 100],
|
|
"μs",
|
|
MetricReport.LOWER_IS_BETTER,
|
|
)
|
|
zenbenchmark.record(
|
|
"metrics_response_latency_p99_us",
|
|
metrics_response_latency_us[len(metrics_response_latency_us) * 99 // 100],
|
|
"μs",
|
|
MetricReport.LOWER_IS_BETTER,
|
|
)
|