Update test_worker.py

justinmerrell · justinmerrell · commit 5a13932b4c65 · 2024-02-12T17:07:47.000-05:00
diff --git a/tests/test_serverless/test_worker.py b/tests/test_serverless/test_worker.py
@@ -274,7 +274,9 @@ async def test_run_worker_generator_handler_exception(
         except KeyError:
             pass
 
-        runpod.serverless.start(generator_config)
+        with patch("os.environ.get") as mock_get:
+            mock_get.return_value = None
+            runpod.serverless.start(generator_config)
 
         assert mock_stream_result.call_count == 1
         assert not mock_run_job.called
