improve ut coverage.

opea-project · Dec 12, 2024 · 04b8fd6 · 04b8fd6
1 parent 60fa89a
commit 04b8fd6
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 2 deletions.
diff --git a/tests/cores/mega/test_dynamic_batching.py b/tests/cores/mega/test_dynamic_batching.py
@@ -1,6 +1,7 @@
 # Copyright (C) 2024 Intel Corporation
 # SPDX-License-Identifier: Apache-2.0
 
+import os
 import asyncio
 import multiprocessing
 import unittest
@@ -76,6 +77,7 @@ 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"
 
@@ -89,6 +91,5 @@ async def test_dynamic_batching(self):
         self.assertEqual(response1["result"], "processed: Hello, ")
         self.assertEqual(response2["result"], "processed: OPEA Project!")
 
-
 if __name__ == "__main__":
     unittest.main()
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')
 
 if __name__ == "__main__":
     unittest.main()