scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-09-07 08:26:10 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 36

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
Show stats Download patch file Download diff file Expand all files Collapse all files

14
tests/test_litellm/proxy/client/cli/test_keys_commands.py
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