crewAI/test_tracing_auth.py

#!/usr/bin/env python3
"""
Test script for issue #3559 - TraceBatchManager authentication handling
"""

import os
import sys

sys.path.insert(0, os.path.join(os.path.dirname(__file__), 'src'))

def test_tracing_auth_issue():
    """Test that tracing authentication issue is fixed"""

    try:
        from unittest.mock import patch

        from crewai.cli.authentication.token import AuthError
        from crewai.events.listeners.tracing.trace_batch_manager import (
            TraceBatchManager,
        )

        print("Test 1: TraceBatchManager creation without authentication")

        with patch(
            "crewai.events.listeners.tracing.trace_batch_manager.get_auth_token",
            side_effect=AuthError("No token found, make sure you are logged in")
        ):
            batch_manager = TraceBatchManager()
            print("✓ TraceBatchManager created successfully with empty API key")

            batch = batch_manager.initialize_batch({"user_id": "test"}, {"crew_name": "test"})
            if batch is not None:
                print(f"✓ Batch initialized successfully: {batch.batch_id}")
            else:
                print("✗ Batch initialization returned None")
                return False

    except Exception as e:
        print(f"✗ TraceBatchManager test failed: {e}")
        import traceback
        traceback.print_exc()
        return False

    try:
        from crewai import LLM, Agent, Crew, Task

        print("\nTest 2: Crew creation without authentication")

        with patch(
            "crewai.events.listeners.tracing.trace_batch_manager.get_auth_token",
            side_effect=AuthError("No token found, make sure you are logged in")
        ):
            agent = Agent(
                role="Test Agent",
                goal="Complete a simple task",
                backstory="A test agent for reproducing the bug",
                llm=LLM(model="gpt-4o-mini", api_key="fake-key")
            )

            task = Task(
                description="Say hello world",
                expected_output="A greeting message",
                agent=agent
            )

            crew = Crew(
                agents=[agent],
                tasks=[task],
                verbose=False
            )

            print(f"✓ Crew created successfully without authentication errors: {len(crew.agents)} agents, {len(crew.tasks)} tasks")

    except Exception as e:
        print(f"✗ Crew creation test failed: {e}")
        import traceback
        traceback.print_exc()
        return False

    return True

if __name__ == "__main__":
    print("Testing TraceBatchManager authentication handling...")
    success = test_tracing_auth_issue()
    if not success:
        print("\nFAILED: Issue #3559 still exists")
        exit(1)
    else:
        print("\nPASSED: Issue #3559 appears to be fixed")