tiennm99/litellm
1
0
Fork 0
mirror of https://github.com/tiennm99/litellm.git synced 2026-07-12 21:04:10 +00:00
Code Issues Packages Projects Releases Wiki Activity

fix test

Browse Source
This commit is contained in:
Ishaan Jaff
2025-07-03 11:18:41 -07:00
parent ba5af3e44b
commit 96d5f891df
1 changed files with 7 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files
tests/test_litellm/proxy/client/cli/test_keys_commands.py
+7 -7
View File
@@ -40,7 +40,7 @@ def mock_keys_client():
yield MockClient
def test_keys_list_json_format(mock_keys_client, cli_runner):
def test_async_keys_list_json_format(mock_keys_client, cli_runner):
mock_keys_client.return_value.list.return_value = {
"keys": [
{
@@ -60,7 +60,7 @@ def test_keys_list_json_format(mock_keys_client, cli_runner):
mock_keys_client.return_value.list.assert_called_once()
def test_keys_list_table_format(mock_keys_client, cli_runner):
def test_async_keys_list_table_format(mock_keys_client, cli_runner):
mock_keys_client.return_value.list.return_value = {
"keys": [
{
@@ -83,7 +83,7 @@ def test_keys_list_table_format(mock_keys_client, cli_runner):
mock_keys_client.return_value.list.assert_called_once()
def test_keys_generate_success(mock_keys_client, cli_runner):
def test_async_keys_generate_success(mock_keys_client, cli_runner):
mock_keys_client.return_value.generate.return_value = {
"key": "new-key",
"spend": 100.0,
@@ -96,7 +96,7 @@ def test_keys_generate_success(mock_keys_client, cli_runner):
mock_keys_client.return_value.generate.assert_called_once()
def test_keys_delete_success(mock_keys_client, cli_runner):
def test_async_keys_delete_success(mock_keys_client, cli_runner):
mock_keys_client.return_value.delete.return_value = {
"status": "success",
"deleted_keys": ["abc123"],
@@ -108,21 +108,21 @@ def test_keys_delete_success(mock_keys_client, cli_runner):
mock_keys_client.return_value.delete.assert_called_once()
def test_keys_list_error_handling(mock_keys_client, cli_runner):
def test_async_keys_list_error_handling(mock_keys_client, cli_runner):
mock_keys_client.return_value.list.side_effect = Exception("API Error")
result = cli_runner.invoke(cli, ["keys", "list"])
assert result.exit_code != 0
assert "API Error" in str(result.exception)
def test_keys_generate_error_handling(mock_keys_client, cli_runner):
def test_async_keys_generate_error_handling(mock_keys_client, cli_runner):
mock_keys_client.return_value.generate.side_effect = Exception("API Error")
result = cli_runner.invoke(cli, ["keys", "generate", "--models", "gpt-4"])
assert result.exit_code != 0
assert "API Error" in str(result.exception)
def test_keys_delete_error_handling(mock_keys_client, cli_runner):
def test_async_keys_delete_error_handling(mock_keys_client, cli_runner):
import requests
# Mock a connection error that would normally happen in CI