Function test_session_management_integration

tests/unit/test_session.py:409–456

Test the integration between new and legacy session management

()

def test_session_management_integration():
    """Test the integration between new and legacy session management"""
    import agentops

    # Reset client for test
    agentops._client = agentops.Client()

    # Test that we can use both new and legacy APIs together
    with (
        patch("agentops.tracer") as mock_tracer,
        patch("agentops.client.client.tracer", mock_tracer),
        patch("agentops.sdk.decorators.factory.tracer", mock_tracer),
        patch("agentops.legacy.tracer", mock_tracer),
    ):
        mock_tracer.initialized = True

        # Mock API client
        with patch("agentops.client.api.ApiClient") as mock_api:
            mock_v3 = MagicMock()
            mock_v3.fetch_auth_token.return_value = {"token": "mock-jwt-token", "project_id": "mock-project-id"}
            mock_api.return_value.v3 = mock_v3

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

            # Reset call counts after initialization
            mock_tracer.reset_mock()

            # Create mock trace context
            mock_trace_context = MagicMock()
            mock_tracer.start_trace.return_value = mock_trace_context

            # Test new API
            trace_context = agentops.start_trace(trace_name="new_api_trace")
            assert trace_context == mock_trace_context

            # Test legacy API
            session = agentops.start_session(tags=["legacy"])
            # Check that the session's trace_context has the expected properties
            assert session.trace_context is not None

            # Test ending both
            agentops.end_trace(trace_context)
            agentops.end_session(session)

            # Verify calls were made
            assert mock_tracer.start_trace.call_count >= 2
            assert mock_tracer.end_trace.call_count >= 2


# CrewAI Backwards Compatibility Tests