fix tests

Author: rcholic

tests/test_cloud_tracing.py

@@ -250,7 +250,7 @@ def test_create_tracer_pro_tier_success(self, capsys):
                 # Mock upload response
                 mock_put.return_value = Mock(status_code=200)
 
-                tracer = create_tracer(api_key="sk_pro_test123", run_id="test-run")
+                tracer = create_tracer(api_key="sk_pro_test123", run_id="test-run", upload_trace=True)
 
                 # Verify Pro tier message
                 captured = capsys.readouterr()
@@ -294,7 +294,7 @@ def test_create_tracer_api_forbidden_fallback(self, capsys):
             mock_post.return_value = mock_response
 
             with tempfile.TemporaryDirectory():
-                tracer = create_tracer(api_key="sk_free_test123", run_id="test-run")
+                tracer = create_tracer(api_key="sk_free_test123", run_id="test-run", upload_trace=True)
 
                 # Verify warning message
                 captured = capsys.readouterr()
@@ -315,7 +315,7 @@ def test_create_tracer_api_timeout_fallback(self, capsys):
             mock_post.side_effect = requests.exceptions.Timeout("Connection timeout")
 
             with tempfile.TemporaryDirectory():
-                tracer = create_tracer(api_key="sk_test123", run_id="test-run")
+                tracer = create_tracer(api_key="sk_test123", run_id="test-run", upload_trace=True)
 
                 # Verify warning message
                 captured = capsys.readouterr()
@@ -336,7 +336,7 @@ def test_create_tracer_api_connection_error_fallback(self, capsys):
             mock_post.side_effect = requests.exceptions.ConnectionError("Connection refused")
 
             with tempfile.TemporaryDirectory():
-                tracer = create_tracer(api_key="sk_test123", run_id="test-run")
+                tracer = create_tracer(api_key="sk_test123", run_id="test-run", upload_trace=True)
 
                 # Verify warning message
                 captured = capsys.readouterr()
@@ -371,7 +371,7 @@ def test_create_tracer_uses_constant_api_url(self):
                 mock_post.return_value = mock_response
                 mock_put.return_value = Mock(status_code=200)
 
-                tracer = create_tracer(api_key="sk_test123", run_id="test-run")
+                tracer = create_tracer(api_key="sk_test123", run_id="test-run", upload_trace=True)
 
                 # Verify correct API URL was used (constant)
                 assert mock_post.called
@@ -395,7 +395,7 @@ def test_create_tracer_custom_api_url(self):
                 mock_put.return_value = Mock(status_code=200)
 
                 tracer = create_tracer(
-                    api_key="sk_test123", run_id="test-run", api_url=custom_api_url
+                    api_key="sk_test123", run_id="test-run", api_url=custom_api_url, upload_trace=True
                 )
 
                 # Verify custom API URL was used
@@ -415,7 +415,7 @@ def test_create_tracer_missing_upload_url_in_response(self, capsys):
             mock_post.return_value = mock_response
 
             with tempfile.TemporaryDirectory():
-                tracer = create_tracer(api_key="sk_test123", run_id="test-run")
+                tracer = create_tracer(api_key="sk_test123", run_id="test-run", upload_trace=True)
 
                 # Verify warning message
                 captured = capsys.readouterr()
@@ -463,7 +463,7 @@ def test_create_tracer_orphaned_trace_recovery(self, capsys):
                     mock_put.return_value = Mock(status_code=200)
 
                     # Create tracer - should trigger orphaned trace recovery
-                    tracer = create_tracer(api_key="sk_test123", run_id="new-run-456")
+                    tracer = create_tracer(api_key="sk_test123", run_id="new-run-456", upload_trace=True)
 
                     # Verify recovery messages
                     captured = capsys.readouterr()

tests/test_file_size_tracking.py

@@ -186,6 +186,7 @@ def test_create_tracer_passes_logger_to_cloud_sink(self, mock_requests):
                 api_key="sk_test_key",
                 run_id="test-logger",
                 logger=mock_logger,
+                upload_trace=True,
             )
 
         # Verify tracer was created