andre/crewAI
1
0
Fork 0
mirror of https://github.com/crewAIInc/crewAI.git synced 2026-01-25 08:08:14 +00:00
Code Issues Actions 1 Packages Projects Releases Wiki Activity

Apply automatic linting fixes to tests directory

Browse Source 
Co-Authored-By: Joe Moura <joao@crewai.com>
This commit is contained in:
Devin AI
2025-05-12 13:31:07 +00:00
parent ad1ea46bbb
commit 46621113af
62 changed files with 1738 additions and 1821 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
tests/telemetry/test_telemetry.py
View File

@@ -8,7 +8,7 @@ from crewai.telemetry import Telemetry
@pytest.mark.parametrize(
"env_var,value,expected_ready",
("env_var", "value", "expected_ready"),
[
("OTEL_SDK_DISABLED", "true", False),
("OTEL_SDK_DISABLED", "TRUE", False),
@@ -18,7 +18,7 @@ from crewai.telemetry import Telemetry
("CREWAI_DISABLE_TELEMETRY", "false", True),
],
)
def test_telemetry_environment_variables(env_var, value, expected_ready):
def test_telemetry_environment_variables(env_var, value, expected_ready) -> None:
"""Test telemetry state with different environment variable configurations."""
with patch.dict(os.environ, {env_var: value}):
with patch("crewai.telemetry.telemetry.TracerProvider"):
@@ -26,7 +26,7 @@ def test_telemetry_environment_variables(env_var, value, expected_ready):
assert telemetry.ready is expected_ready
def test_telemetry_enabled_by_default():
def test_telemetry_enabled_by_default() -> None:
"""Test that telemetry is enabled by default."""
with patch.dict(os.environ, {}, clear=True):
with patch("crewai.telemetry.telemetry.TracerProvider"):
@@ -43,7 +43,7 @@ from opentelemetry import trace
side_effect=Exception("Test exception"),
)
@pytest.mark.vcr(filter_headers=["authorization"])
def test_telemetry_fails_due_connect_timeout(export_mock, logger_mock):
def test_telemetry_fails_due_connect_timeout(export_mock, logger_mock) -> None:
error = Exception("Test exception")
export_mock.side_effect = error

6
tests/telemetry/test_telemetry_disable.py
View File

@@ -6,7 +6,7 @@ import pytest
from crewai.telemetry import Telemetry
@pytest.mark.parametrize("env_var,value,expected_ready", [
@pytest.mark.parametrize(("env_var", "value", "expected_ready"), [
("OTEL_SDK_DISABLED", "true", False),
("OTEL_SDK_DISABLED", "TRUE", False),
("CREWAI_DISABLE_TELEMETRY", "true", False),
@@ -14,7 +14,7 @@ from crewai.telemetry import Telemetry
("OTEL_SDK_DISABLED", "false", True),
("CREWAI_DISABLE_TELEMETRY", "false", True),
])
def test_telemetry_environment_variables(env_var, value, expected_ready):
def test_telemetry_environment_variables(env_var, value, expected_ready) -> None:
"""Test telemetry state with different environment variable configurations."""
with patch.dict(os.environ, {env_var: value}):
with patch("crewai.telemetry.telemetry.TracerProvider"):
@@ -22,7 +22,7 @@ def test_telemetry_environment_variables(env_var, value, expected_ready):
assert telemetry.ready is expected_ready
def test_telemetry_enabled_by_default():
def test_telemetry_enabled_by_default() -> None:
"""Test that telemetry is enabled by default."""
with patch.dict(os.environ, {}, clear=True):
with patch("crewai.telemetry.telemetry.TracerProvider"):