improve ut coverage.

opea-project · Dec 12, 2024 · 98227ea · 98227ea
1 parent d68dcc8
commit 98227ea
Show file tree

Hide file tree

Showing 2 changed files with 1 addition and 3 deletions.
diff --git a/tests/cores/mega/test_dynamic_batching.py b/tests/cores/mega/test_dynamic_batching.py
@@ -3,7 +3,6 @@
 
 import asyncio
 import multiprocessing
-import os
 import unittest
 from enum import Enum
 
@@ -77,7 +76,6 @@ def tearDown(self):
         self.process1.terminate()
 
     async def test_dynamic_batching(self):
-        os.environ["LOGFLAG"] = True
         url1 = "http://localhost:8080/v1/add1"
         url2 = "http://localhost:8080/v1/add2"
 

diff --git a/tests/cores/mega/test_hybrid_service_orchestrator.py b/tests/cores/mega/test_hybrid_service_orchestrator.py
@@ -34,7 +34,7 @@ def test_add_remote_service(self):
         self.service_builder.add(opea_microservices["s1"]).add(s2)
         self.service_builder.flow_to(self.s1, s2)
         self.assertEqual(s2.endpoint_path, "http://fakehost:8008/v1/add")
-        self.assertRaises(Exception, s2._validate_env(), "N/A")
+        self.assertRaises(Exception, s2._validate_env, "N/A")
 
 
 if __name__ == "__main__":