fix tests

Author: rcholic

sentience/cloud_tracing.py

@@ -452,6 +452,19 @@ def _extract_stats_from_trace(self) -> TraceStats:
             TraceStats with stats fields for /v1/traces/complete
         """
         try:
+            # Check if file exists before reading
+            if not self._path.exists():
+                if self.logger:
+                    self.logger.warning(f"Trace file not found: {self._path}")
+                return TraceStats(
+                    total_steps=0,
+                    total_events=0,
+                    duration_ms=None,
+                    final_status="unknown",
+                    started_at=None,
+                    ended_at=None,
+                )
+
             # Read trace file to extract stats
             events = TraceFileManager.read_events(self._path)
             # Use TraceFileManager to extract stats (with custom status inference)
@@ -529,6 +542,12 @@ def _extract_screenshots_from_trace(self) -> dict[int, dict[str, Any]]:
         sequence = 0
 
         try:
+            # Check if file exists before reading
+            if not self._path.exists():
+                if self.logger:
+                    self.logger.warning(f"Trace file not found: {self._path}")
+                return screenshots
+
             events = TraceFileManager.read_events(self._path)
             for event in events:
                 # Check if this is a snapshot event with screenshot
@@ -557,6 +576,15 @@ def _create_cleaned_trace(self, output_path: Path) -> None:
             output_path: Path to write cleaned trace file
         """
         try:
+            # Check if file exists before reading
+            if not self._path.exists():
+                if self.logger:
+                    self.logger.warning(f"Trace file not found: {self._path}")
+                # Create empty cleaned trace file
+                output_path.parent.mkdir(parents=True, exist_ok=True)
+                output_path.touch()
+                return
+
             events = TraceFileManager.read_events(self._path)
             with open(output_path, "w", encoding="utf-8") as outfile:
                 for event in events:

tests/test_cloud_tracing.py

@@ -418,7 +418,12 @@ def test_create_tracer_pro_tier_success(self, capsys):
                 ), f"Expected CloudTraceSink, got {type(tracer.sink)}"
                 assert tracer.sink.run_id == "test-run"  # Verify run_id is passed
 
-                # Cleanup
+                # Verify the init API was called
+                assert mock_post.called
+                assert mock_post.call_count == 1
+
+                # Cleanup - emit at least one event so file exists before close
+                tracer.emit("test", {"v": 1, "seq": 1})
                 tracer.close()
 
     def test_create_tracer_free_tier_fallback(self, capsys):