added test_tenacity_retry.py

armadaproject · Jul 7, 2023 · fc30221 · fc30221
1 parent 01c1e6a
commit fc30221
Showing 1 changed file with 8 additions and 0 deletions.
diff --git a/third_party/airflow/tests/unit/test_tenacity_retry.py b/third_party/airflow/tests/unit/test_tenacity_retry.py
@@ -0,0 +1,8 @@
+from job_service_mock import JobService
+
+
+def test_tenacity_retry():
+    target_count = 3
+    JobService.tenacity_example(target_count)
+    retry_count = JobService.tenacity_example.retry.statistics["attempt_number"]
+    assert retry_count == target_count