fix: add debugging info

Author: justinmerrell

tests/test_serverless/test_modules/test_logger.py

@@ -124,12 +124,12 @@ def test_log_job_id(self):
                 flush=True
             )
 
+            # Test with endpoint id set
             os.environ["RUNPOD_ENDPOINT_ID"] = "test_endpoint_id"
             logger.log("test_message", "INFO", job_id)
+            os.environ.pop("RUNPOD_ENDPOINT_ID")
 
             mock_print.assert_called_once_with(
                 '{"requestId": "test_job_id", "message": "test_message", "level": "INFO"}',
                 flush=True
             )
-
-            os.environ.pop("RUNPOD_ENDPOINT_ID")

tests/test_serverless/test_worker.py

@@ -268,7 +268,7 @@ async def test_run_worker_generator_handler_exception(
         '''
         # Define the mock behaviors
         mock_get_job.return_value = {
-            "id": "generator-123", "input": {"number": 1}}
+            "id": "generator-123-exception", "input": {"number": 1}}
 
         # Test generator handler
         generator_config = {
@@ -301,11 +301,12 @@ async def test_run_worker_generator_aggregate_handler(
         '''
         # Define the mock behaviors
         mock_get_job.return_value = {
-            "id": "generator-123", "input": {"number": 1}}
+            "id": "generator-123-aggregate", "input": {"number": 1}}
 
         # Test generator handler
         generator_config = {
             "handler": generator_handler, "return_aggregate_stream": True, "refresh_worker": True}
+
         runpod.serverless.start(generator_config)
 
         assert mock_send_result.called