Update test_worker.py

Author: justinmerrell

tests/test_serverless/test_worker.py

@@ -267,7 +267,9 @@ async def test_run_worker_generator_handler_exception(
         # Test generator handler
         generator_config = {
             "handler": generator_handler_exception, "refresh_worker": True}
+        os.environ["RUNPOD_ENDPOINT_ID"] = "test_endpoint_id"
         runpod.serverless.start(generator_config)
+        os.environ.pop("RUNPOD_ENDPOINT_ID")
 
         assert mock_stream_result.call_count == 1
         assert not mock_run_job.called