litellm - 💡(How to fix) Fix [Bug]: CI test fails due to `test_check_responses_cost.py` mismatch with the latest commit [1 participants]

TL;DR

Update the test file to use AsyncMock instead of MagicMock for mock_prisma_client.db.litellm_managedobjecttable.update_many to fix the CI test failure.

Guidance

• Identify the specific test file that needs to be updated, likely related to Proxy Legacy Tests / auth-and-jwt (pull_request).
• Update the mock_prisma_client.db.litellm_managedobjecttable.update_many configuration to use AsyncMock to match the asynchronous nature of the _cleanup_stale_managed_objects function.
• Verify that the test failure is resolved by checking the CI test results after updating the test file.
• Review the test file to ensure that it correctly handles the asynchronous cleanup process and updates the assertions accordingly.

Example

from unittest.mock import AsyncMock

# Update the mock configuration to use AsyncMock
mock_prisma_client.db.litellm_managedobjecttable.update_many = AsyncMock()

Notes

The fix assumes that the issue is solely due to the mismatch between the synchronous MagicMock and the asynchronous execute_raw function. If other factors contribute to the test failure, additional updates may be necessary.

Recommendation

Apply workaround: Update the test file to use AsyncMock as described, to resolve the CI test failure.