From 6ac7188c820b580b84b3110ba4ab583e8a40a5a5 Mon Sep 17 00:00:00 2001 From: Anastasia Lubennikova Date: Wed, 16 Oct 2024 15:48:56 +0100 Subject: [PATCH] fix test timeout logic --- .../regress/test_installed_extensions.py | 26 +++++++++++++------ 1 file changed, 18 insertions(+), 8 deletions(-) diff --git a/test_runner/regress/test_installed_extensions.py b/test_runner/regress/test_installed_extensions.py index 558348c365..7101a28fe4 100644 --- a/test_runner/regress/test_installed_extensions.py +++ b/test_runner/regress/test_installed_extensions.py @@ -101,12 +101,22 @@ def test_installed_extensions(neon_simple_env: NeonEnv): endpoint.stop() endpoint.start() - time.sleep(1) + timeout = 5 + while timeout > 0: + try: + res = client.metrics() + timeout = -1 + except Exception as e: + info("failed to get metrics, assume they are not collected yet: %s", e) + time.sleep(1) + timeout -= 1 + continue - res = client.metrics() - info("After restart metrics: %s", res) - m = parse_metrics(res) - neon_m = m.query_all("installed_extensions", {"extension_name": "neon", "versions": "1.2,1.3"}) - assert len(neon_m) == 1 - for sample in neon_m: - assert sample.value == 2 + info("After restart metrics: %s", res) + m = parse_metrics(res) + neon_m = m.query_all( + "installed_extensions", {"extension_name": "neon", "versions": "1.2,1.3"} + ) + assert len(neon_m) == 1 + for sample in neon_m: + assert sample.value == 2