From 3a86023d557d6b976b603b3de342939837de9678 Mon Sep 17 00:00:00 2001
From: Ivan Ogasawara <ivan.ogasawara@gmail.com>
Date: Thu, 30 May 2024 23:46:19 -0400
Subject: [PATCH] fix issues with celery tasks

---
 tests/conftest.py                | 15 +++++++--------
 tests/test_task_celery_serial.py |  4 ++--
 2 files changed, 9 insertions(+), 10 deletions(-)

diff --git a/tests/conftest.py b/tests/conftest.py
index be724e9..875beab 100644
--- 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
index f4488f0..75c0d59 100644
--- 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}"