diff --git a/component-tests/requirements.txt b/component-tests/requirements.txt index e5aa3eab..edd3e1f4 100644 --- a/component-tests/requirements.txt +++ b/component-tests/requirements.txt @@ -1,3 +1,4 @@ +flaky==3.7.0 pytest==6.2.2 pyyaml==5.4.1 requests==2.25.1 diff --git a/component-tests/test_reconnect.py b/component-tests/test_reconnect.py index 44e27e3f..547dd476 100644 --- a/component-tests/test_reconnect.py +++ b/component-tests/test_reconnect.py @@ -1,12 +1,13 @@ #!/usr/bin/env python import copy - +from flaky import flaky from retrying import retry from time import sleep from util import start_cloudflared, wait_tunnel_ready, check_tunnel_not_connected, send_requests +@flaky(max_runs=3, min_passes=1) class TestReconnect(): default_ha_conns = 4 default_reconnect_secs = 5