refactor: move src and tests from lib/crewai to root

tests/telemetry/test_telemetry.py

@@ -0,0 +1,115 @@
+import os
+import threading
+from unittest.mock import patch
+
+import pytest
+
+from crewai import Agent, Crew, Task
+from crewai.telemetry import Telemetry
+
+from opentelemetry import trace
+
+
+@pytest.fixture(autouse=True)
+def cleanup_telemetry():
+    Telemetry._instance = None
+    if hasattr(Telemetry, "_lock"):
+        Telemetry._lock = threading.Lock()
+    yield
+    Telemetry._instance = None
+    if hasattr(Telemetry, "_lock"):
+        Telemetry._lock = threading.Lock()
+
+
+@pytest.mark.telemetry
+@pytest.mark.parametrize(
+    "env_var,value,expected_ready",
+    [
+        ("OTEL_SDK_DISABLED", "true", False),
+        ("OTEL_SDK_DISABLED", "TRUE", False),
+        ("CREWAI_DISABLE_TELEMETRY", "true", False),
+        ("CREWAI_DISABLE_TELEMETRY", "TRUE", False),
+        ("OTEL_SDK_DISABLED", "false", True),
+        ("CREWAI_DISABLE_TELEMETRY", "false", True),
+    ],
+)
+def test_telemetry_environment_variables(env_var, value, expected_ready):
+    """Test telemetry state with different environment variable configurations."""
+    with patch.dict(os.environ, {env_var: value}):
+        with patch("crewai.telemetry.telemetry.TracerProvider"):
+            telemetry = Telemetry()
+            assert telemetry.ready is expected_ready
+
+
+@pytest.mark.telemetry
+def test_telemetry_enabled_by_default():
+    """Test that telemetry is enabled by default."""
+    with patch.dict(os.environ, {}, clear=True):
+        with patch("crewai.telemetry.telemetry.TracerProvider"):
+            telemetry = Telemetry()
+            assert telemetry.ready is True
+
+
+@pytest.mark.telemetry
+@patch("crewai.telemetry.telemetry.logger.error")
+@patch(
+    "opentelemetry.exporter.otlp.proto.http.trace_exporter.OTLPSpanExporter.export",
+    side_effect=Exception("Test exception"),
+)
+@pytest.mark.vcr(filter_headers=["authorization"])
+def test_telemetry_fails_due_connect_timeout(export_mock, logger_mock):
+    error = Exception("Test exception")
+    export_mock.side_effect = error
+
+    tracer = trace.get_tracer(__name__)
+    with tracer.start_as_current_span("test-span"):
+        agent = Agent(
+            role="agent",
+            llm="gpt-4o-mini",
+            goal="Just say hi",
+            backstory="You are a helpful assistant that just says hi",
+        )
+        task = Task(
+            description="Just say hi",
+            expected_output="hi",
+            agent=agent,
+        )
+        crew = Crew(agents=[agent], tasks=[task], name="TestCrew")
+        crew.kickoff()
+
+    trace.get_tracer_provider().force_flush()
+
+    assert export_mock.called
+    assert logger_mock.call_count == export_mock.call_count
+    for call in logger_mock.call_args_list:
+        assert call[0][0] == error
+
+
+@pytest.mark.telemetry
+def test_telemetry_singleton_pattern():
+    """Test that Telemetry uses the singleton pattern correctly."""
+    Telemetry._instance = None
+
+    telemetry1 = Telemetry()
+    telemetry2 = Telemetry()
+
+    assert telemetry1 is telemetry2
+
+    setattr(telemetry1, "test_attribute", "test_value")
+    assert hasattr(telemetry2, "test_attribute")
+    assert getattr(telemetry2, "test_attribute") == "test_value"
+
+    import threading
+
+    instances = []
+
+    def create_instance():
+        instances.append(Telemetry())
+
+    threads = [threading.Thread(target=create_instance) for _ in range(5)]
+    for thread in threads:
+        thread.start()
+    for thread in threads:
+        thread.join()
+
+    assert all(instance is telemetry1 for instance in instances)

tests/telemetry/test_telemetry_disable.py

@@ -0,0 +1,102 @@
+import os
+from unittest.mock import patch, MagicMock
+
+import pytest
+
+from crewai.telemetry import Telemetry
+
+
+@pytest.fixture(autouse=True)
+def cleanup_telemetry():
+    """Automatically clean up Telemetry singleton between tests."""
+    Telemetry._instance = None
+    yield
+    Telemetry._instance = None
+
+
+@pytest.mark.telemetry
+@pytest.mark.parametrize(
+    "env_var,value,expected_ready",
+    [
+        ("OTEL_SDK_DISABLED", "true", False),
+        ("OTEL_SDK_DISABLED", "TRUE", False),
+        ("CREWAI_DISABLE_TELEMETRY", "true", False),
+        ("CREWAI_DISABLE_TELEMETRY", "TRUE", False),
+        ("OTEL_SDK_DISABLED", "false", True),
+        ("CREWAI_DISABLE_TELEMETRY", "false", True),
+    ],
+)
+def test_telemetry_environment_variables(env_var, value, expected_ready):
+    """Test telemetry state with different environment variable configurations."""
+    with patch.dict(os.environ, {env_var: value}):
+        with patch("crewai.telemetry.telemetry.TracerProvider"):
+            telemetry = Telemetry()
+            assert telemetry.ready is expected_ready
+
+
+@pytest.mark.telemetry
+def test_telemetry_enabled_by_default():
+    """Test that telemetry is enabled by default."""
+    with patch.dict(os.environ, {}, clear=True):
+        with patch("crewai.telemetry.telemetry.TracerProvider"):
+            telemetry = Telemetry()
+            assert telemetry.ready is True
+
+
+@pytest.mark.telemetry
+def test_telemetry_disable_after_singleton_creation():
+    """Test that telemetry operations are disabled when env var is set after singleton creation."""
+    with patch.dict(os.environ, {}, clear=True):
+        with patch("crewai.telemetry.telemetry.TracerProvider"):
+            telemetry = Telemetry()
+            assert telemetry.ready is True
+
+            mock_operation = MagicMock()
+            telemetry._safe_telemetry_operation(mock_operation)
+            mock_operation.assert_called_once()
+
+            mock_operation.reset_mock()
+
+            os.environ["CREWAI_DISABLE_TELEMETRY"] = "true"
+
+            telemetry._safe_telemetry_operation(mock_operation)
+            mock_operation.assert_not_called()
+
+
+@pytest.mark.telemetry
+def test_telemetry_disable_with_multiple_instances():
+    """Test that multiple telemetry instances respect dynamically changed env vars."""
+    with patch.dict(os.environ, {}, clear=True):
+        with patch("crewai.telemetry.telemetry.TracerProvider"):
+            telemetry1 = Telemetry()
+            assert telemetry1.ready is True
+
+            os.environ["CREWAI_DISABLE_TELEMETRY"] = "true"
+
+            telemetry2 = Telemetry()
+            assert telemetry2 is telemetry1
+            assert telemetry2.ready is True
+
+            mock_operation = MagicMock()
+            telemetry2._safe_telemetry_operation(mock_operation)
+            mock_operation.assert_not_called()
+
+
+@pytest.mark.telemetry
+def test_telemetry_otel_sdk_disabled_after_creation():
+    """Test that OTEL_SDK_DISABLED also works when set after singleton creation."""
+    with patch.dict(os.environ, {}, clear=True):
+        with patch("crewai.telemetry.telemetry.TracerProvider"):
+            telemetry = Telemetry()
+            assert telemetry.ready is True
+
+            mock_operation = MagicMock()
+            telemetry._safe_telemetry_operation(mock_operation)
+            mock_operation.assert_called_once()
+
+            mock_operation.reset_mock()
+
+            os.environ["OTEL_SDK_DISABLED"] = "true"
+
+            telemetry._safe_telemetry_operation(mock_operation)
+            mock_operation.assert_not_called()