Function test_multiple_concurrent_traces

tests/unit/test_session.py:322–348

Test that multiple traces can be started concurrently

(mock_tracing_core, mock_api_client, reset_client)

def test_multiple_concurrent_traces(mock_tracing_core, mock_api_client, reset_client):
    """Test that multiple traces can be started concurrently"""
    import agentops

    # Initialize AgentOps
    agentops.init(api_key="test-api-key", auto_start_session=False)

    # Create mock trace contexts for different traces
    mock_trace_context_1 = MagicMock()
    mock_trace_context_2 = MagicMock()

    # Mock start_trace to return different contexts
    mock_tracing_core.start_trace.side_effect = [
        mock_trace_context_1,
        mock_trace_context_2,
    ]

    # Start multiple traces
    trace1 = agentops.start_trace(trace_name="trace1", tags=["test1"])
    trace2 = agentops.start_trace(trace_name="trace2", tags=["test2"])

    # Verify both traces were created
    assert trace1 == mock_trace_context_1
    assert trace2 == mock_trace_context_2

    # Verify start_trace was called twice
    assert mock_tracing_core.start_trace.call_count == 2


def test_trace_context_properties(mock_trace_context):