Using CONTAINER_TIMEOUT for container restart

celery · Oct 25, 2023 · bbda394 · bbda394
1 parent 109b9b7
commit bbda394
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/src/pytest_celery/api/base.py b/src/pytest_celery/api/base.py
@@ -10,6 +10,7 @@
 from pytest_docker_tools.wrappers.container import wait_for_callable
 
 from pytest_celery.api.container import CeleryTestContainer
+from pytest_celery.defaults import CONTAINER_TIMEOUT
 from pytest_celery.defaults import RESULT_TIMEOUT
 
 
@@ -61,7 +62,7 @@ def kill(self, reload_container: bool = True) -> None:
             self.container.reload()
 
     def restart(self, reload_container: bool = True) -> None:
-        self.container.restart()
+        self.container.restart(timeout=CONTAINER_TIMEOUT)
         if reload_container:
             self.container.reload()
         if self.app: