@@ -33,13 +33,20 @@ def test_tunnel_no_ingress(self, tmp_path, component_tests_config):
3333 LOGGER .debug (config )
3434 with start_cloudflared (tmp_path , config , cfd_pre_args = ["tunnel" , "--ha-connections" , "1" ], cfd_args = ["run" ], new_process = True ):
3535 wait_tunnel_ready (require_min_connections = 1 )
36- resp = send_request (config .get_url ()+ "/" )
37- assert resp .status_code == 503 , "Expected cloudflared to return 503 for all requests with no ingress defined"
38- resp = send_request (config .get_url ()+ "/test" )
39- assert resp .status_code == 503 , "Expected cloudflared to return 503 for all requests with no ingress defined"
36+ expected_status_code = 503
37+ resp = send_request (config .get_url ()+ "/" , expected_status_code )
38+ assert resp .status_code == expected_status_code , "Expected cloudflared to return 503 for all requests with no ingress defined"
39+ resp = send_request (config .get_url ()+ "/test" , expected_status_code )
40+ assert resp .status_code == expected_status_code , "Expected cloudflared to return 503 for all requests with no ingress defined"
4041
42+ def retry_if_result_none (result ):
43+ '''
44+ Returns True if the result is None, indicating that the function should be retried.
45+ '''
46+ return result is None
4147
42- @retry (stop_max_attempt_number = MAX_RETRIES , wait_fixed = BACKOFF_SECS * 1000 )
43- def send_request (url , headers = {} ):
48+ @retry (retry_on_result = retry_if_result_none , stop_max_attempt_number = MAX_RETRIES , wait_fixed = BACKOFF_SECS * 1000 )
49+ def send_request (url , expected_status_code = 200 ):
4450 with requests .Session () as s :
45- return s .get (url , timeout = BACKOFF_SECS , headers = headers )
51+ resp = s .get (url , timeout = BACKOFF_SECS )
52+ return resp if resp .status_code == expected_status_code else None
0 commit comments