Deflake Serve tests (#12542)

2026-07-06 05:16:30 +08:00 · 2020-12-01 13:42:21 -08:00
parent 4288b5b9ff
commit 7022278ce9
2 changed files with 11 additions and 1 deletions
@@ -364,7 +364,16 @@ def test_delete_backend(serve_instance):
    client.create_backend("delete:v1", function2)
    client.set_traffic("delete_backend", {"delete:v1": 1.0})

-    assert requests.get("http://127.0.0.1:8000/delete-backend").text == "olleh"
+    for _ in range(10):
+        try:
+            assert requests.get(
+                "http://127.0.0.1:8000/delete-backend").text == "olleh"
+            break
+        except AssertionError:
+            time.sleep(0.5)  # wait for the traffic policy to propogate
+    else:
+        assert requests.get(
+            "http://127.0.0.1:8000/delete-backend").text == "olleh"


@pytest.mark.parametrize("route", [None, "/delete-endpoint"])