test(budget): validate warning message content in timezone fallback test

This commit is contained in:
LeeJuOh 2026-03-07 14:06:39 +09:00
parent 383f6d45d8
commit a9ef3cd18c

View file

@ -191,6 +191,7 @@ class TestStandardizedResetTime(unittest.TestCase):
with patch("litellm.litellm_core_utils.duration_parser.verbose_logger") as mock_logger:
get_next_standardized_reset_time("1d", base_time, "NonExistentTimeZone")
mock_logger.warning.assert_called_once()
assert "NonExistentTimeZone" in mock_logger.warning.call_args.args[1]
if __name__ == "__main__":