move httpserver check out of fixture

Signed-off-by: Rahul Modpur <rmodpur2@gmail.com>
This commit is contained in:
Rahul Modpur
2023-10-07 16:49:45 +05:30
committed by Joonas Koivunen
parent 5cf2b8de01
commit b461841e53
2 changed files with 13 additions and 8 deletions

View File

@@ -1014,7 +1014,7 @@ def neon_env_and_metrics_server(
httpserver: HTTPServer,
neon_env_builder: NeonEnvBuilder,
httpserver_listen_address,
) -> Iterator[Tuple[NeonEnv, SimpleQueue[Any]]]:
) -> Tuple[NeonEnv, HTTPServer, SimpleQueue[Any]]:
"""
Fixture to create a Neon environment and metrics server.
"""
@@ -1068,9 +1068,7 @@ def neon_env_and_metrics_server(
".*synthetic_size_worker: failed to calculate synthetic size for tenant .*: failed to calculate some logical_sizes"
)
yield (env, uploads)
httpserver.check()
return (env, httpserver, uploads)
@dataclass

View File

@@ -11,10 +11,13 @@ from fixtures.neon_fixtures import (
wait_for_last_flush_lsn,
)
from fixtures.types import TenantId, TimelineId
from pytest_httpserver import HTTPServer
def test_metric_collection(neon_env_and_metrics_server: Tuple[NeonEnv, SimpleQueue[Any]]):
(env, uploads) = neon_env_and_metrics_server
def test_metric_collection(
neon_env_and_metrics_server: Tuple[NeonEnv, HTTPServer, SimpleQueue[Any]]
):
(env, httpserver, uploads) = neon_env_and_metrics_server
tenant_id = env.initial_tenant
timeline_id = env.initial_timeline
@@ -107,11 +110,13 @@ def test_metric_collection(neon_env_and_metrics_server: Tuple[NeonEnv, SimpleQue
(events, is_last) = events
v.ingest(events, is_last)
httpserver.check()
def test_metric_collection_cleans_up_tempfile(
neon_env_and_metrics_server: Tuple[NeonEnv, SimpleQueue[Any]]
neon_env_and_metrics_server: Tuple[NeonEnv, HTTPServer, SimpleQueue[Any]]
):
(env, uploads) = neon_env_and_metrics_server
(env, httpserver, uploads) = neon_env_and_metrics_server
pageserver_http = env.pageserver.http_client()
tenant_id = env.initial_tenant
@@ -183,6 +188,8 @@ def test_metric_collection_cleans_up_tempfile(
), "only initial tempfile should had been removed"
assert initially.other.issuperset(later.other), "no other files should had been removed"
httpserver.check()
@dataclass
class PrefixPartitionedFiles: