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

Fix test_deprecation_warning to reliably detect warnings in CI

Browse Source 
Co-Authored-By: Joe Moura <joao@crewai.com>
This commit is contained in:
Devin AI
2025-05-07 08:12:02 +00:00
parent 3348fff123
commit 38f61d937a
1 changed files with 30 additions and 12 deletions
Download Patch File Download Diff File Expand all files Collapse all files

42
tests/backward_compatibility_test.py
View File

@@ -1,8 +1,38 @@
import sys
import unittest
import warnings
class BackwardCompatibilityTest(unittest.TestCase):
def setUp(self):
if "crewai.telemtry" in sys.modules:
del sys.modules["crewai.telemtry"]
warnings.resetwarnings()
def test_deprecation_warning(self):
"""Test that importing from the misspelled module raises a deprecation warning."""
with warnings.catch_warnings(record=True) as w:
warnings.simplefilter("always", DeprecationWarning)
import importlib
import crewai.telemtry
importlib.reload(crewai.telemtry)
self.assertGreaterEqual(len(w), 1)
warning_messages = [str(warning.message) for warning in w]
warning_categories = [warning.category for warning in w]
has_deprecation_warning = False
for msg, cat in zip(warning_messages, warning_categories):
if (issubclass(cat, DeprecationWarning) and
"crewai.telemtry" in msg and
"crewai.telemetry" in msg):
has_deprecation_warning = True
break
self.assertTrue(has_deprecation_warning,
f"No matching deprecation warning found. Warnings: {warning_messages}")
def test_telemtry_typo_compatibility(self):
"""Test that the backward compatibility for the telemtry typo works."""
from crewai.telemetry import Telemetry
@@ -16,15 +46,3 @@ class BackwardCompatibilityTest(unittest.TestCase):
from crewai.telemtry import Telemetry as MisspelledTelemetry
self.assertEqual(dir(MisspelledTelemetry), dir(Telemetry))
def test_deprecation_warning(self):
"""Test that importing from the misspelled module raises a deprecation warning."""
with warnings.catch_warnings(record=True) as w:
warnings.simplefilter("always")
from crewai.telemtry import Telemetry # noqa: F401
self.assertEqual(len(w), 1)
self.assertTrue(issubclass(w[0].category, DeprecationWarning))
self.assertIn("crewai.telemtry", str(w[0].message))
self.assertIn("crewai.telemetry", str(w[0].message))