.

Author: sentrivana

tests/integrations/ray/test_ray.py

@@ -1,6 +1,8 @@
 import json
 import os
 import pytest
+import shutil
+import uuid
 
 import ray
 
@@ -26,9 +28,6 @@ def capture_envelope(self, envelope: Envelope) -> None:
 
 
 class RayLoggingTransport(TestTransport):
-    def __init__(self):
-        super().__init__()
-
     def capture_envelope(self, envelope: Envelope) -> None:
         print(envelope.serialize().decode("utf-8", "replace"))
 
@@ -45,8 +44,18 @@ def setup_sentry(transport=None):
     )
 
 
-def read_error_from_log(job_id):
-    log_dir = "/tmp/ray/session_latest/logs/"
+def read_error_from_log(job_id, ray_temp_dir):
+    # Find the actual session directory that Ray created
+    session_dirs = [d for d in os.listdir(ray_temp_dir) if d.startswith("session_")]
+    if not session_dirs:
+        raise FileNotFoundError(f"No session directory found in {ray_temp_dir}")
+
+    session_dir = os.path.join(ray_temp_dir, session_dirs[0])
+    log_dir = os.path.join(session_dir, "logs")
+
+    if not os.path.exists(log_dir):
+        raise FileNotFoundError(f"No logs directory found at {log_dir}")
+
     log_file = [
         f
         for f in os.listdir(log_dir)
@@ -132,33 +141,44 @@ def example_task():
 def test_errors_in_ray_tasks():
     setup_sentry_with_logging_transport()
 
-    ray.init(
-        runtime_env={
-            "worker_process_setup_hook": setup_sentry_with_logging_transport,
-            "working_dir": "./",
-        }
-    )
-
-    # Setup ray task
-    @ray.remote
-    def example_task():
-        1 / 0
-
-    with sentry_sdk.start_transaction(op="task", name="ray test transaction"):
-        with pytest.raises(ZeroDivisionError):
-            future = example_task.remote()
-            ray.get(future)
-
-    job_id = future.job_id().hex()
-    error = read_error_from_log(job_id)
-
-    assert error["level"] == "error"
-    assert (
-        error["transaction"]
-        == "tests.integrations.ray.test_ray.test_errors_in_ray_tasks.<locals>.example_task"
-    )
-    assert error["exception"]["values"][0]["mechanism"]["type"] == "ray"
-    assert not error["exception"]["values"][0]["mechanism"]["handled"]
+    # Create a short temp directory to avoid Unix socket path length limits
+    ray_temp_dir = os.path.join("/tmp", f"ray_test_{uuid.uuid4().hex[:8]}")
+    os.makedirs(ray_temp_dir, exist_ok=True)
+
+    try:
+        ray.init(
+            runtime_env={
+                "worker_process_setup_hook": setup_sentry_with_logging_transport,
+                "working_dir": "./",
+            },
+            _temp_dir=ray_temp_dir,
+        )
+
+        # Setup ray task
+        @ray.remote
+        def example_task():
+            1 / 0
+
+        with sentry_sdk.start_transaction(op="task", name="ray test transaction"):
+            with pytest.raises(ZeroDivisionError):
+                future = example_task.remote()
+                ray.get(future)
+
+        job_id = future.job_id().hex()
+        error = read_error_from_log(job_id, ray_temp_dir)
+
+        assert error["level"] == "error"
+        assert (
+            error["transaction"]
+            == "tests.integrations.ray.test_ray.test_errors_in_ray_tasks.<locals>.example_task"
+        )
+        assert error["exception"]["values"][0]["mechanism"]["type"] == "ray"
+        assert not error["exception"]["values"][0]["mechanism"]["handled"]
+
+    finally:
+        # Clean up the temporary directory
+        if os.path.exists(ray_temp_dir):
+            shutil.rmtree(ray_temp_dir, ignore_errors=True)
 
 
 def test_tracing_in_ray_actors():
@@ -200,33 +220,44 @@ def increment(self):
 def test_errors_in_ray_actors():
     setup_sentry_with_logging_transport()
 
-    ray.init(
-        runtime_env={
-            "worker_process_setup_hook": setup_sentry_with_logging_transport,
-            "working_dir": "./",
-        }
-    )
-
-    # Setup ray actor
-    @ray.remote
-    class Counter:
-        def __init__(self):
-            self.n = 0
-
-        def increment(self):
-            with sentry_sdk.start_span(op="task", name="example actor execution"):
-                1 / 0
-
-            return sentry_sdk.get_client().transport.envelopes
-
-    with sentry_sdk.start_transaction(op="task", name="ray test transaction"):
-        with pytest.raises(ZeroDivisionError):
-            counter = Counter.remote()
-            future = counter.increment.remote()
-            ray.get(future)
-
-    job_id = future.job_id().hex()
-    error = read_error_from_log(job_id)
-
-    # We do not capture errors in ray actors yet
-    assert error is None
+    # Create a short temp directory to avoid Unix socket path length limits
+    ray_temp_dir = os.path.join("/tmp", f"ray_test_{uuid.uuid4().hex[:8]}")
+    os.makedirs(ray_temp_dir, exist_ok=True)
+
+    try:
+        ray.init(
+            runtime_env={
+                "worker_process_setup_hook": setup_sentry_with_logging_transport,
+                "working_dir": "./",
+            },
+            _temp_dir=ray_temp_dir,
+        )
+
+        # Setup ray actor
+        @ray.remote
+        class Counter:
+            def __init__(self):
+                self.n = 0
+
+            def increment(self):
+                with sentry_sdk.start_span(op="task", name="example actor execution"):
+                    1 / 0
+
+                return sentry_sdk.get_client().transport.envelopes
+
+        with sentry_sdk.start_transaction(op="task", name="ray test transaction"):
+            with pytest.raises(ZeroDivisionError):
+                counter = Counter.remote()
+                future = counter.increment.remote()
+                ray.get(future)
+
+        job_id = future.job_id().hex()
+        error = read_error_from_log(job_id, ray_temp_dir)
+
+        # We do not capture errors in ray actors yet
+        assert error is None
+
+    finally:
+        # Clean up the temporary directory
+        if os.path.exists(ray_temp_dir):
+            shutil.rmtree(ray_temp_dir, ignore_errors=True)