Introduce flaky to cover some tests

CouncilDataProject · Dec 5, 2023 · c1c8469 · c1c8469
1 parent 623871b
commit c1c8469
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 1 deletion.
diff --git a/cdp_backend/tests/utils/test_file_utils.py b/cdp_backend/tests/utils/test_file_utils.py
@@ -12,6 +12,7 @@
 import imageio
 import pytest
 import requests_mock
+from flaky import flaky
 from requests import HTTPError
 
 from cdp_backend.utils import file_utils
@@ -344,6 +345,7 @@ def test_convert_video_to_mp4(
         # (EXAMPLE_M3U8_PLAYLIST_URI, None),
     ],
 )
+@flaky(max_runs=3, min_passes=1)
 def test_remote_resource_copy(
     resources_dir: Path,
     uri: str,
@@ -359,7 +361,7 @@ def test_remote_resource_copy(
 
     os.remove(actual_uri)
 
-    time.sleep(5)
+    time.sleep(10)
 
 
 def test_invalid_uri() -> None:

diff --git a/pyproject.toml b/pyproject.toml
@@ -101,6 +101,7 @@ test = [
   "pytest>=5.4.3",
   "pytest-cov>=2.9.0",
   "pytest-raises>=0.11",
+  "flaky>=3.7.0",
   # Extras
   "networkx>=2.5",
   "pydot>=1.4",