Update test_fastapi.py

justinmerrell · justinmerrell · commit 8bf017062ba1 · 2024-02-10T13:09:33.000-05:00
diff --git a/tests/test_serverless/test_modules/test_fastapi.py b/tests/test_serverless/test_modules/test_fastapi.py
@@ -242,6 +242,12 @@ def test_stream(self):
                 "error": "Stream not supported, handler must be a generator."
             }
 
+            # Test webhook caller sent
+            with patch(f"{module_location}._send_webhook_async", return_value=True) as mock_send_webhook:  # pylint: disable=line-too-long
+                asyncio.run(worker_api._sim_run(input_object_with_webhook))
+                asyncio.run(worker_api._sim_stream("test-123"))
+                assert mock_send_webhook.called
+
             # Test with generator handler
             def generator_handler(job):
                 del job
@@ -256,12 +262,6 @@ def generator_handler(job):
                 "stream": [{"output": {"result": "success"}}]
             }
 
-            # Test webhook caller sent
-            with patch(f"{module_location}._send_webhook_async", return_value=True) as mock_send_webhook:  # pylint: disable=line-too-long
-                asyncio.run(worker_api._sim_run(input_object_with_webhook))
-                asyncio.run(worker_api._sim_stream("test-123"))
-                assert mock_send_webhook.called
-
         loop.close()
 
     @pytest.mark.asyncio
@@ -305,6 +305,12 @@ def test_status(self):
                 "output": {"result": "success"}
             }
 
+            # Test webhook caller sent
+            with patch(f"{module_location}._send_webhook_async", return_value=True) as mock_send_webhook:  # pylint: disable=line-too-long
+                asyncio.run(worker_api._sim_run(input_object_with_webhook))
+                asyncio.run(worker_api._sim_status("test-123"))
+                assert mock_send_webhook.called
+
             # Test with generator handler
             def generator_handler(job):
                 del job
@@ -326,10 +332,4 @@ def generator_handler(job):
                 error_worker_api._sim_status("test-123"))
             assert "error" in error_status_return
 
-            # Test webhook caller sent
-            with patch(f"{module_location}._send_webhook_async", return_value=True) as mock_send_webhook:  # pylint: disable=line-too-long
-                asyncio.run(worker_api._sim_run(input_object_with_webhook))
-                asyncio.run(worker_api._sim_status("test-123"))
-                assert mock_send_webhook.called
-
         loop.close()
