Update test_fastapi.py

justinmerrell · justinmerrell · commit 5f2f7c66bbc2 · 2024-02-10T13:36:08.000-05:00
diff --git a/tests/test_serverless/test_modules/test_fastapi.py b/tests/test_serverless/test_modules/test_fastapi.py
@@ -231,11 +231,6 @@ def test_stream(self):
                 "error": "Stream not supported, handler must be a generator."
             }
 
-            # Test webhook caller sent
-            asyncio.run(worker_api._sim_run(input_object_with_webhook))
-            asyncio.run(worker_api._sim_stream("test-123"))
-            assert mock_threading.Thread.called
-
             # Test with generator handler
             def generator_handler(job):
                 del job
@@ -250,6 +245,11 @@ def generator_handler(job):
                 "stream": [{"output": {"result": "success"}}]
             }
 
+            # Test webhook caller sent
+            asyncio.run(generator_worker_api._sim_run(input_object_with_webhook))
+            asyncio.run(generator_worker_api._sim_stream("test-123"))
+            assert mock_threading.Thread.called
+
         loop.close()
 
     @pytest.mark.asyncio
