From d8525ecbfab9e1e09505aae0659ac3d11ea8b855 Mon Sep 17 00:00:00 2001 From: "Jugal D. Bhatt" <55304795+jugaldb@users.noreply.github.com> Date: Fri, 16 May 2025 20:34:22 -0500 Subject: [PATCH] Remove extra test file (#10889) --- .../test_spend_log_cleanup.py | 56 ------------------- 1 file changed, 56 deletions(-) delete mode 100644 tests/proxy_unit_tests/test_spend_log_cleanup.py diff --git a/tests/proxy_unit_tests/test_spend_log_cleanup.py b/tests/proxy_unit_tests/test_spend_log_cleanup.py deleted file mode 100644 index c5fa0cb8dd..0000000000 --- a/tests/proxy_unit_tests/test_spend_log_cleanup.py +++ /dev/null @@ -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() \ No newline at end of file