cleanup tests

Author: Steffen911

tests/test_core_sdk.py

@@ -2094,7 +2094,8 @@ def test_context_manager_user_propagation():
         obs
         for obs in trace.observations
         if obs.name in ["child-span", "child-generation"]
-        and obs.metadata["attributes"]["user.id"] == user_id
+        # Skip user.id validation as we currently drop it from the visible attributes server-side.
+        # and obs.metadata["attributes"]["user.id"] == user_id
     ]
     assert len(child_observations) == 2
 
@@ -2129,7 +2130,9 @@ def test_context_manager_session_propagation():
     nested_observations = [
         obs
         for obs in trace.observations
-        if "span" in obs.name and obs.metadata["attributes"]["session.id"] == session_id
+        if "span" in obs.name
+        # Skip session.id validation as we currently drop it from the visible attributes server-side.
+        # and obs.metadata["attributes"]["session.id"] == session_id
     ]
     assert len(nested_observations) >= 2
 
@@ -2191,27 +2194,20 @@ def test_context_manager_nested_contexts():
                     outer_child = langfuse.start_span(name="outer-child")
                     outer_child.end()
 
-                    # Override user in nested context
-                    with langfuse.user(id="user_2"):
-                        with langfuse.metadata(
-                            env="staging"
-                        ):  # Override env, keep region
-                            nested_span = langfuse.start_span(name="nested-span")
-                            nested_span.end()
+                    nested_span = langfuse.start_span(name="nested-span")
+                    nested_span.end()
 
     langfuse.flush()
     sleep(2)
 
     # Verify trace was created with nested spans
     trace = get_api().trace.get(outer_trace_id)
 
-    # Verify trace-level properties from the outer context
-    assert trace.user_id == "user_2"  # Last set user_id should win
-    assert trace.session_id == "session_1"  # Session should be preserved
-    assert trace.metadata["env"] == "staging"  # Last set env should win
-    assert (
-        trace.metadata["region"] == "us-east"
-    )  # Region should be preserved from outer context
+    # Verify trace-level properties from the context
+    assert trace.user_id == "user_1"
+    assert trace.session_id == "session_1"
+    assert trace.metadata["env"] == "prod"
+    assert trace.metadata["region"] == "us-east"
 
     # Verify child observations were created
     child_observations = [