fix issues with celery tasks

osl-incubator · May 31, 2024 · 3a86023 · 3a86023
1 parent a562b48
commit 3a86023
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 10 deletions.
diff --git a/tests/conftest.py b/tests/conftest.py
@@ -2,7 +2,6 @@
 
 from __future__ import annotations
 
-import os
 import subprocess
 import time
 
@@ -24,19 +23,19 @@ def setup() -> Generator[None, None, None]:
         # # Sleep for 5 seconds
         # time.sleep(5)
 
-        # Change directory to `tests/`
-        os.chdir("tests/")
-
         # Start the Celery worker
         celery_process = subprocess.Popen(
-            ["celery", "-A", "celery_tasks", "worker", "--loglevel=debug"]
+            [
+                "celery",
+                "-A",
+                "tests.celery_tasks",
+                "worker",
+                "--loglevel=debug",
+            ],
         )
 
         time.sleep(5)
 
-        # Change directory back to the original
-        os.chdir("..")
-
         yield
 
     finally:

diff --git a/tests/test_task_celery_serial.py b/tests/test_task_celery_serial.py
@@ -39,7 +39,7 @@ def get_chord_tasks(  # type: ignore
         seconds = kwargs.get("seconds")
         task_id = kwargs.get("task_id")
         return (
-            [task_sleep.s(seconds, task_id, task_id)],
+            [task_sleep.s(seconds, task_id)],
             None,
         )
 
@@ -76,7 +76,7 @@ def test_serial_result(self, task_result: Task) -> None:
             results[task_id] = i + i
 
         for task_id, expected in results.items():
-            result = task.result.get(task_id, timeout=10)
+            result = task.result.get(task_id, timeout=10)[0]
             assert (
                 result == expected
             ), f"Expected Result: {expected}, Actual Result: {result}"