fixes #51: change from loop to asyncio.run

ai-cfia · Mar 1, 2024 · cbfe42c · cbfe42c
1 parent d94fb95
commit cbfe42c
Showing 1 changed file with 7 additions and 8 deletions.
diff --git a/tests/test_inference_request.py b/tests/test_inference_request.py
@@ -12,10 +12,9 @@ def setUp(self) -> None:
         """
         Set up the test environment before running each test case.
         """
-         # Start the test pipeline
-        self.loop = asyncio.get_event_loop()    
+         # Start the test pipeline  
         self.test = app.test_client()
-        response = self.loop.run_until_complete(
+        response = asyncio.run(
             self.test.get("/test")
         )
         self.pipeline = json.loads(asyncio.run(response.get_data()))[0]
@@ -73,7 +72,7 @@ def test_inference_request_successful(self, MockFromConnectionString):
         }
 
         # Test the answers from inference_request
-        response = self.loop.run_until_complete(
+        response = asyncio.run(
             self.test.post(
                 '/inf',
                 headers={
@@ -121,7 +120,7 @@ def test_inference_request_unsuccessfull(self, MockFromConnectionString):
         expected = 500
 
         # Test the answers from inference_request
-        response = self.loop.run_until_complete(
+        response = asyncio.run(
             self.test.post(
                 '/inf',
                 headers={
@@ -158,7 +157,7 @@ def test_inference_request_missing_argument(self):
         for k, v in data.items():
             if k != "model_name":
                 data[k] = ""        
-                response = self.loop.run_until_complete(
+                response = asyncio.run(
                     self.test.post(
                         '/inf',
                         headers={
@@ -188,7 +187,7 @@ def test_inference_request_wrong_pipeline_name(self):
         expected = ("Model wrong_pipeline_name not found")
 
         # Test the answers from inference_request
-        response = self.loop.run_until_complete(
+        response = asyncio.run(
             self.test.post(
                 '/inf',
                 headers={
@@ -217,7 +216,7 @@ def test_inference_request_wrong_header(self):
         expected = ("Invalid image header")
 
         # Test the answers from inference_request
-        response = self.loop.run_until_complete(
+        response = asyncio.run(
             self.test.post(
                 '/inf',
                 headers={