push

Author: hassiebp

tests/test_core_sdk.py

@@ -2062,207 +2062,3 @@ def test_that_generation_like_properties_are_actually_created():
         assert (
             obs.completion_start_time is not None
         ), f"{obs_type} should persist completion_start_time property"
-
-
-def test_context_manager_user_propagation():
-    """Test that user context manager propagates user_id to child spans."""
-    langfuse = Langfuse()
-
-    user_id = "test_user_123"
-
-    with langfuse.start_as_current_span(name="parent-span") as parent_span:
-        with langfuse.propagate_attributes({"user_id": user_id}):
-            trace_id = parent_span.trace_id
-
-            # Create child spans that should inherit user_id
-            child_span = langfuse.start_span(name="child-span")
-            child_span.end()
-
-            # Create generation that should inherit user_id
-            generation = parent_span.start_generation(name="child-generation")
-            generation.end()
-
-    langfuse.flush()
-    sleep(2)
-
-    # Verify trace has user_id (child spans inherit via context propagation)
-    trace = get_api().trace.get(trace_id)
-    assert trace.user_id == user_id
-
-    # Verify child observations were created and have user_id
-    child_observations = [
-        obs
-        for obs in trace.observations
-        if obs.name in ["child-span", "child-generation"]
-        # 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
-
-
-def test_context_manager_session_propagation():
-    """Test that session context manager propagates session_id to child spans."""
-    langfuse = Langfuse()
-
-    session_id = "test_session_456"
-
-    with langfuse.start_as_current_span(name="parent-span") as parent_span:
-        with langfuse.propagate_attributes({"session_id": session_id}):
-            trace_id = parent_span.trace_id
-
-            # Create child spans that should inherit session_id
-            child_span = langfuse.start_span(name="child-span")
-            child_span.end()
-
-            # Create nested context to test multiple levels
-            with langfuse.start_as_current_span(name="nested-span"):
-                grandchild_span = langfuse.start_span(name="grandchild-span")
-                grandchild_span.end()
-
-    langfuse.flush()
-    sleep(2)
-
-    # Verify trace has session_id
-    trace = get_api().trace.get(trace_id)
-    assert trace.session_id == session_id
-
-    # Verify nested spans were created
-    nested_observations = [
-        obs
-        for obs in trace.observations
-        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
-
-
-def test_context_manager_metadata_propagation():
-    """Test that metadata context manager propagates metadata to child spans."""
-    langfuse = Langfuse()
-
-    with langfuse.start_as_current_span(name="parent-span") as parent_span:
-        with langfuse.propagate_attributes(
-            {
-                "experiment": "A/B",
-                "version": "1.2.3",
-                "feature_flag": "enabled",
-            }
-        ):
-            trace_id = parent_span.trace_id
-
-            # Create child spans that should inherit metadata
-            child_span = langfuse.start_span(name="child-span")
-            child_span.end()
-
-            # Create generation that should inherit metadata
-            generation = parent_span.start_generation(name="child-generation")
-            generation.end()
-
-    langfuse.flush()
-    sleep(2)
-
-    # Verify trace has metadata
-    trace = get_api().trace.get(trace_id)
-    assert trace.metadata["experiment"] == "A/B"
-    assert trace.metadata["version"] == "1.2.3"
-    assert trace.metadata["feature_flag"] == "enabled"
-
-    # Verify all observations have the metadata distributed as individual keys
-    for obs in trace.observations:
-        if obs.name in ["child-span", "child-generation", "parent-span"]:
-            # Check that metadata was set on the observation
-            assert hasattr(obs, "metadata"), f"Observation {obs.name} missing metadata"
-            assert (
-                obs.metadata["experiment"] == "A/B"
-            ), f"Observation {obs.name} missing experiment metadata"
-            assert (
-                obs.metadata["version"] == "1.2.3"
-            ), f"Observation {obs.name} missing version metadata"
-            assert (
-                obs.metadata["feature_flag"] == "enabled"
-            ), f"Observation {obs.name} missing feature_flag metadata"
-
-
-def test_context_manager_nested_contexts():
-    """Test nested context managers with overrides and merging."""
-    langfuse = Langfuse()
-
-    with langfuse.start_as_current_span(name="outer-span") as outer_span:
-        with langfuse.propagate_attributes(
-            {"user_id": "user_1", "session_id": "session_1"}
-        ):
-            with langfuse.propagate_attributes({"env": "prod", "region": "us-east"}):
-                outer_trace_id = outer_span.trace_id
-
-                # Create span in outer context
-                outer_child = langfuse.start_span(name="outer-child")
-                outer_child.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 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 = [
-        obs for obs in trace.observations if "child" in obs.name or "nested" in obs.name
-    ]
-    assert len(child_observations) >= 2
-
-    # Verify specific child spans exist and have correct metadata
-    outer_child_obs = [obs for obs in trace.observations if obs.name == "outer-child"]
-    nested_span_obs = [obs for obs in trace.observations if obs.name == "nested-span"]
-
-    assert len(outer_child_obs) == 1, "outer-child span should exist"
-    assert len(nested_span_obs) == 1, "nested-span should exist"
-
-
-def test_context_manager_baggage_propagation():
-    """Test context managers with as_baggage=True for cross-service propagation."""
-    langfuse = Langfuse()
-
-    # Test with baggage enabled (careful with sensitive data)
-    with langfuse.start_as_current_span(name="service-span") as span:
-        with langfuse.propagate_attributes(
-            {"session_id": "public_session_789"}, as_baggage=True
-        ):
-            with langfuse.propagate_attributes(
-                {"service": "api", "version": "v1.0"}, as_baggage=True
-            ):
-                trace_id = span.trace_id
-
-                # Create child spans that inherit baggage context
-                child_span = langfuse.start_span(name="external-call-span")
-                child_span.end()
-
-    langfuse.flush()
-    sleep(2)
-
-    # Verify trace properties were set
-    trace = get_api().trace.get(trace_id)
-    assert trace.session_id == "public_session_789"
-    assert trace.metadata["service"] == "api"
-    assert trace.metadata["version"] == "v1.0"
-
-    # Verify all observations have the metadata and session_id
-    for obs in trace.observations:
-        if obs.name in ["external-call-span", "service-span"]:
-            # Check that metadata was set on the observation
-            assert hasattr(obs, "metadata"), f"Observation {obs.name} missing metadata"
-            assert (
-                obs.metadata["service"] == "api"
-            ), f"Observation {obs.name} missing service metadata"
-            assert (
-                obs.metadata["version"] == "v1.0"
-            ), f"Observation {obs.name} missing version metadata"