retrying project creation

zauberzeug · Aug 21, 2023 · 2b7d792 · 2b7d792
1 parent a4de8a2
commit 2b7d792
Showing 1 changed file with 7 additions and 2 deletions.
diff --git a/mock_trainer/mock_trainer_tests/detections/test_detections.py b/mock_trainer/mock_trainer_tests/detections/test_detections.py
@@ -17,8 +17,13 @@ def create_project():
     test_helper.LiveServerSession().delete(f"/zauberzeug/projects/pytest?keep_images=true")
     project_configuration = {'project_name': 'pytest', 'inbox': 1, 'annotate': 2, 'review': 3, 'complete': 4, 'image_style': 'plain',
                              'box_categories': 1, 'point_categories': 1, 'segmentation_categories': 1, 'thumbs': False, 'trainings': 1}
-    assert test_helper.LiveServerSession().post(f"/zauberzeug/projects/generator",
-                                                json=project_configuration).status_code == 200
+    for _ in range(10):
+        test_helper.LiveServerSession().get("/status")
+        response = test_helper.LiveServerSession().post(f"/zauberzeug/projects/generator", json=project_configuration)
+        if response.status_code == 200:
+            break
+    else:
+        raise Exception('Could not create project')
     yield
     test_helper.LiveServerSession().delete(f"/zauberzeug/projects/pytest?keep_images=true")