Fix async test (opea-project#60)

* fix async test Signed-off-by: Spycsh <[email protected]> * fix yaml orchestrator --------- Signed-off-by: Spycsh <[email protected]>
poussa · May 16, 2024 · ba20162 · ba20162
1 parent 9ddc8ec
commit ba20162
Show file tree

Hide file tree

Showing 4 changed files with 4 additions and 4 deletions.
diff --git a/comps/cores/mega/orchestrator_with_yaml.py b/comps/cores/mega/orchestrator_with_yaml.py
@@ -56,7 +56,7 @@ def process_outputs(self, prev_nodes: List) -> Dict:
             all_outputs.update(self.result_dict[prev_node])
         return all_outputs
 
-    def schedule(self, initial_inputs: Dict):
+    async def schedule(self, initial_inputs: Dict):
         for node in self.topological_sort():
             if node in self.ind_nodes():
                 inputs = initial_inputs

diff --git a/tests/cores/mega/test_service_orchestrator.py b/tests/cores/mega/test_service_orchestrator.py
@@ -36,7 +36,7 @@ async def s2_add(request: TextDoc) -> TextDoc:
     return {"text": text}
 
 
-class TestServiceOrchestrator(unittest.TestCase):
+class TestServiceOrchestrator(unittest.IsolatedAsyncioTestCase):
     def setUp(self):
         self.s1 = opea_microservices["s1"]
         self.s2 = opea_microservices["s2"]

diff --git a/tests/cores/mega/test_service_orchestrator_with_gateway.py b/tests/cores/mega/test_service_orchestrator_with_gateway.py
@@ -36,7 +36,7 @@ async def s2_add(request: TextDoc) -> TextDoc:
     return {"text": text}
 
 
-class TestServiceOrchestrator(unittest.TestCase):
+class TestServiceOrchestrator(unittest.IsolatedAsyncioTestCase):
     def setUp(self):
         self.s1 = opea_microservices["s1"]
         self.s2 = opea_microservices["s2"]

diff --git a/tests/cores/mega/test_service_orchestrator_with_yaml.py b/tests/cores/mega/test_service_orchestrator_with_yaml.py
@@ -36,7 +36,7 @@ async def s2_add(request: TextDoc) -> TextDoc:
     return {"text": text}
 
 
-class TestYAMLOrchestrator(unittest.TestCase):
+class TestYAMLOrchestrator(unittest.IsolatedAsyncioTestCase):
     def setUp(self) -> None:
         self.s1 = opea_microservices["s1"]
         self.s2 = opea_microservices["s2"]