mirror of
https://github.com/BerriAI/litellm.git
synced 2026-10-08 03:08:45 +00:00
21e9632713
`assert False` inside a `try:` raises AssertionError, which the `except Exception` right below it catches, so several tests reported green no matter what the code did. `pytest.fail` raises Failed, a BaseException, and escapes. A bare `a == b` statement is evaluated and discarded. Nine of those sat in tests, and one was comparing against a model name the router never produces. Selects B011, B015, B018, PT015, PLR0133 and PLW0127 in ruff-tests.toml alongside F821, with all 50 existing violations fixed, so no budget file or ratchet is needed. CI already runs this config over tests/.
116 lines
3.7 KiB
Python
116 lines
3.7 KiB
Python
"""
|
|
E2E Test for Container Files API.
|
|
|
|
Tests the container files endpoints using LiteLLM SDK methods.
|
|
"""
|
|
|
|
import os
|
|
import sys
|
|
import time
|
|
|
|
import pytest
|
|
|
|
sys.path.insert(0, os.path.abspath("../.."))
|
|
|
|
from litellm.containers import (
|
|
create_container,
|
|
delete_container,
|
|
)
|
|
from litellm.containers.endpoint_factory import (
|
|
list_container_files,
|
|
retrieve_container_file,
|
|
retrieve_container_file_content,
|
|
delete_container_file,
|
|
)
|
|
|
|
|
|
@pytest.mark.skipif(not os.getenv("OPENAI_API_KEY"), reason="OPENAI_API_KEY not set")
|
|
def test_container_files_api():
|
|
"""
|
|
Test container files API: list, retrieve, delete.
|
|
|
|
Flow:
|
|
1. Create a container
|
|
2. List files (should be empty)
|
|
3. Try retrieve file (should error - no files)
|
|
4. Try delete file (should error - no files)
|
|
5. Cleanup: delete container
|
|
"""
|
|
api_key = os.getenv("OPENAI_API_KEY")
|
|
|
|
# 1. Create container
|
|
print("\n1. Creating container...")
|
|
container = create_container(
|
|
name=f"test-files-api-{int(time.time())}",
|
|
custom_llm_provider="openai",
|
|
api_key=api_key,
|
|
expires_after={"anchor": "last_active_at", "minutes": 5},
|
|
)
|
|
print(f" Created: {container.id}")
|
|
|
|
try:
|
|
# 2. List files
|
|
print("2. Listing container files...")
|
|
files = list_container_files(
|
|
container_id=container.id,
|
|
custom_llm_provider="openai",
|
|
api_key=api_key,
|
|
)
|
|
assert files.object == "list"
|
|
assert isinstance(files.data, list)
|
|
assert len(files.data) == 0 # New container has no files
|
|
print(f" Files found: {len(files.data)} ✓")
|
|
|
|
# 3. Try retrieve non-existent file metadata (should raise error)
|
|
print("3. Testing retrieve_container_file (expect error)...")
|
|
try:
|
|
retrieve_container_file(
|
|
container_id=container.id,
|
|
file_id="cfile_nonexistent",
|
|
custom_llm_provider="openai",
|
|
api_key=api_key,
|
|
)
|
|
pytest.fail("Should have raised error for non-existent file")
|
|
except Exception as e:
|
|
assert "not found" in str(e).lower() or "invalid" in str(e).lower()
|
|
print(f" Got expected error ✓")
|
|
|
|
# 3b. Try retrieve non-existent file content (should raise error)
|
|
print("3b. Testing retrieve_container_file_content (expect error)...")
|
|
try:
|
|
retrieve_container_file_content(
|
|
container_id=container.id,
|
|
file_id="cfile_nonexistent",
|
|
custom_llm_provider="openai",
|
|
api_key=api_key,
|
|
)
|
|
pytest.fail("Should have raised error for non-existent file content")
|
|
except Exception as e:
|
|
print(f" Got expected error ✓")
|
|
|
|
# 4. Try delete non-existent file (should raise error)
|
|
print("4. Testing delete_container_file (expect error)...")
|
|
try:
|
|
delete_container_file(
|
|
container_id=container.id,
|
|
file_id="cfile_nonexistent",
|
|
custom_llm_provider="openai",
|
|
api_key=api_key,
|
|
)
|
|
pytest.fail("Should have raised error for non-existent file")
|
|
except Exception as e:
|
|
# Delete returns 400 for non-existent files
|
|
print(f" Got expected error ✓")
|
|
|
|
finally:
|
|
# 5. Cleanup
|
|
print("5. Deleting container...")
|
|
result = delete_container(
|
|
container_id=container.id,
|
|
custom_llm_provider="openai",
|
|
api_key=api_key,
|
|
)
|
|
assert result.deleted is True
|
|
print(f" Deleted ✓")
|
|
|
|
print("\nAll container files API tests passed! ✓")
|