stabilize test_direct_tcpip

warp-tech · Oct 17, 2022 · 38eff07 · 38eff07
1 parent 7066dd5
commit 38eff07
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/tests/test_ssh_proto.py b/tests/test_ssh_proto.py
@@ -142,8 +142,10 @@ def test_direct_tcpip(
             if response.status_code == 200:
                 break
 
-        response = requests.get(f'http://localhost:{local_port}', timeout=timeout)
-        print(response.text)
+        s = requests.Session()
+        retries = requests.adapters.Retry(total=5, backoff_factor=1)
+        s.mount('http://', requests.adapters.HTTPAdapter(max_retries=retries))
+        response = s.get(f'http://localhost:{local_port}', timeout=timeout)
         assert response.status_code == 200
         ssh_client.kill()