fix: tests

Author: justinmerrell

runpod/serverless/__init__.py

@@ -136,8 +136,7 @@ def start(config: Dict[str, Any]):
         )
 
     elif realtime_port:
-        print(f"Starting API server for realtime on port {realtime_port}.")
-        log.info("Starting API server for realtime.")
+        log.info(f"Starting API server for realtime on port {realtime_port}.")
         api_server = rp_fastapi.WorkerAPI(config)
 
         api_server.start_uvicorn(

tests/test_serverless/test_worker.py

@@ -139,9 +139,11 @@ def test_worker_bad_local(self):
         known_args.test_output = '{"test": "test"}'
 
         with patch("argparse.ArgumentParser.parse_known_args") as mock_parse_known_args, \
+                patch("runpod.serverless.os") as mock_os, \
                 self.assertRaises(SystemExit):
 
             mock_parse_known_args.return_value = known_args, []
+            mock_os.environ.get.return_value = None
             runpod.serverless.start({"handler": self.mock_handler})
 
             # Confirm that the log level is set to WARN