Function test_crewai_kwargs_force_flush

tests/unit/test_session.py:593–621

Test that when using the CrewAI < 0.105.0 pattern (end_session with kwargs), the spans are properly exported to the backend with force_flush. This is a more comprehensive test that ensures spans are actually sent to the backend when using the CrewAI integration pattern.

(mock_tracing_core, mock_api_client, mock_trace_context, reset_client)

def test_crewai_kwargs_force_flush(mock_tracing_core, mock_api_client, mock_trace_context, reset_client):
    """
    Test that when using the CrewAI < 0.105.0 pattern (end_session with kwargs),
    the spans are properly exported to the backend with force_flush.

    This is a more comprehensive test that ensures spans are actually sent
    to the backend when using the CrewAI integration pattern.
    """
    import agentops
    import time

    # Configure mocks
    mock_tracing_core.start_trace.return_value = mock_trace_context

    # Initialize AgentOps with API key
    agentops.init(api_key="test-api-key")

    # Create a session
    agentops.start_session(tags=["test", "crewai-integration"])

    # Simulate some work
    time.sleep(0.1)

    # End session with kwargs (CrewAI < 0.105.0 pattern)
    agentops.end_session(end_state="Success", end_state_reason="Test Finished", is_auto_end=True)

    # Explicitly ensure the core isn't already shut down for the test
    # Note: We're using mocks, so we check the mock's initialized state
    assert mock_tracing_core.initialized, "Mocked tracer should still be initialized"


def test_crewai_task_instrumentation(mock_tracing_core, mock_api_client, mock_trace_context, reset_client):