Skip to content

Commit b453363

Browse files
authored
add delay & retry to flakey weight splits test (#7324)
1 parent 46fcf92 commit b453363

File tree

1 file changed

+16
-6
lines changed

1 file changed

+16
-6
lines changed

tests/suite/test_v_s_route_weight_changes_dynamic_reload_many_splits.py

Lines changed: 16 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -173,9 +173,19 @@ def test_vsr_weight_changes_dynamic_reload_many_splits(
173173
print("Step 3: Verify hitting the other backend.")
174174
ensure_response_from_backend(backends32_url, vsr_weight_changes_dynamic_reload_many_splits_setup.vs_host)
175175
wait_and_assert_status_code(200, backends32_url, vsr_weight_changes_dynamic_reload_many_splits_setup.vs_host)
176-
wait_before_test(1)
177-
resp = requests.get(
178-
backends32_url,
179-
headers={"host": vsr_weight_changes_dynamic_reload_many_splits_setup.vs_host},
180-
)
181-
assert "backend2" in resp.text
176+
retry_count = 0
177+
backend2_detected = False
178+
179+
while retry_count < 10 and not backend2_detected:
180+
resp = requests.get(
181+
backends32_url,
182+
headers={"host": vsr_weight_changes_dynamic_reload_many_splits_setup.vs_host},
183+
)
184+
if "backend2" in resp.text:
185+
backend2_detected = True
186+
print(f"Successfully detected backend2 after {retry_count + 1} attempts")
187+
else:
188+
retry_count += 1
189+
print(f"Attempt {retry_count}/10: Expected backend2, got: {resp.text}")
190+
wait_before_test()
191+
assert backend2_detected, f"Failed to get response from backend2 after 10 attempts"

0 commit comments

Comments
 (0)