fix: add annotation to tests

Signed-off-by: Guilhem Barthés <guilhem.barthes@owkin.com>
Substra · Sep 6, 2024 · 4506487 · 4506487
1 parent 7801d7f
commit 4506487
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/backend/builder/tests/test_task_build_image.py b/backend/builder/tests/test_task_build_image.py
@@ -39,6 +39,7 @@ def test_catch_all_exceptions(celery_app, celery_worker, mocker):
         r.get()
 
 
+@pytest.mark.django_db
 @pytest.mark.parametrize("execution_number", range(10))
 def test_order_building_success(celery_app, celery_worker, mocker, execution_number):
     function_1 = orc_mock.FunctionFactory()
@@ -63,6 +64,7 @@ def test_order_building_success(celery_app, celery_worker, mocker, execution_num
     assert result_2.state == "WAITING"
 
 
+@pytest.mark.django_db
 @pytest.mark.parametrize("execution_number", range(10))
 def test_order_building_retry(celery_app, celery_worker, mocker, execution_number):
     function_retry = orc_mock.FunctionFactory()
@@ -100,6 +102,7 @@ def side_effect(*args, **kwargs):
     assert result_other.state == "WAITING"
 
 
+@pytest.mark.django_db
 def test_ssl_connection_timeout(celery_app, celery_worker, mocker):
     """
     Test that in case of a SSL connection timeout, the task is retried max_retries times,