mirror of
https://github.com/BerriAI/litellm.git
synced 2026-10-16 04:29:31 +00:00
243ed4393d
A test that asserts on the error inside its own except block passes when the call stops raising, because nothing runs the handler. That is the exact case the test exists to catch, so the regression lands green. Rewrites all 111 such blocks into pytest.raises, which fails when the call succeeds, and selects PT017 in ruff-tests.toml so no new one lands.
112 lines
3.5 KiB
Python
112 lines
3.5 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)...")
|
|
with pytest.raises(Exception, match="(?i)not found|invalid"):
|
|
retrieve_container_file(
|
|
container_id=container.id,
|
|
file_id="cfile_nonexistent",
|
|
custom_llm_provider="openai",
|
|
api_key=api_key,
|
|
)
|
|
|
|
# 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! ✓")
|