ai-workspace-services/litellm
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Remove extra test file (#10889)

Browse Source
This commit is contained in:
Jugal D. Bhatt 2025-05-16 20:34:22 -05:00 committed by GitHub
parent 7702663666
commit d8525ecbfa
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 0 additions and 56 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

56
tests/proxy_unit_tests/test_spend_log_cleanup.py
View File

@ -1,56 +0,0 @@
"""
Test cases for spend log cleanup functionality
"""
import pytest
from litellm.proxy.db.db_transaction_queue.spend_log_cleanup import _should_delete_spend_logs
from litellm.proxy.proxy_server import general_settings
@pytest.mark.asyncio
async def test_should_delete_spend_logs():
"""
Test the _should_delete_spend_logs function with various scenarios
"""
# Test case 1: No maximum_spend_logs_retention_period set
general_settings.clear()
assert _should_delete_spend_logs() is False
# Test case 2: Valid integer maximum_spend_logs_retention_period (in seconds)
general_settings["maximum_spend_logs_retention_period"] = 3600
assert _should_delete_spend_logs() is False
# Test case 3: Valid duration string - days
general_settings["maximum_spend_logs_retention_period"] = "30d"
assert _should_delete_spend_logs() is True
# Test case 4: Valid duration string - hours
general_settings["maximum_spend_logs_retention_period"] = "24h"
assert _should_delete_spend_logs() is True
# Test case 5: Valid duration string - minutes
general_settings["maximum_spend_logs_retention_period"] = "60m"
assert _should_delete_spend_logs() is True
# Test case 6: Valid duration string - seconds
general_settings["maximum_spend_logs_retention_period"] = "3600s"
assert _should_delete_spend_logs() is True
# Test case 7: Valid duration string - weeks
general_settings["maximum_spend_logs_retention_period"] = "1w"
assert _should_delete_spend_logs() is True
# Test case 8: Valid duration string - months
general_settings["maximum_spend_logs_retention_period"] = "1mo"
assert _should_delete_spend_logs() is True
# Test case 9: Invalid duration string
general_settings["maximum_spend_logs_retention_period"] = "invalid"
assert _should_delete_spend_logs() is False
# Test case 10: None value
general_settings["maximum_spend_logs_retention_period"] = None
assert _should_delete_spend_logs() is False
# Clean up
general_settings.clear()