mirror of
https://github.com/neondatabase/neon.git
synced 2026-01-15 09:22:55 +00:00
move httpserver check out of fixture
Signed-off-by: Rahul Modpur <rmodpur2@gmail.com>
This commit is contained in:
committed by
Joonas Koivunen
parent
5cf2b8de01
commit
b461841e53
@@ -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
|
||||
|
||||
@@ -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:
|
||||
|
||||
Reference in New Issue
Block a user