81 lines
2.1 KiB
Python
81 lines
2.1 KiB
Python
import pytest
|
|
from fastapi import Request
|
|
from fastapi.testclient import TestClient
|
|
from starlette.datastructures import Headers
|
|
from starlette.requests import HTTPConnection
|
|
import os
|
|
import sys
|
|
|
|
sys.path.insert(
|
|
0, os.path.abspath("../..")
|
|
) # Adds the parent directory to the system-path
|
|
|
|
from litellm.proxy.common_utils.http_parsing_utils import _read_request_body
|
|
from litellm.proxy._types import ProxyException
|
|
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_read_request_body_valid_json():
|
|
"""Test the function with a valid JSON payload."""
|
|
|
|
class MockRequest:
|
|
async def body(self):
|
|
return b'{"key": "value"}'
|
|
|
|
request = MockRequest()
|
|
result = await _read_request_body(request)
|
|
assert result == {"key": "value"}
|
|
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_read_request_body_empty_body():
|
|
"""Test the function with an empty body."""
|
|
|
|
class MockRequest:
|
|
async def body(self):
|
|
return b""
|
|
|
|
request = MockRequest()
|
|
result = await _read_request_body(request)
|
|
assert result == {}
|
|
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_read_request_body_invalid_json():
|
|
"""Test the function with an invalid JSON payload."""
|
|
|
|
class MockRequest:
|
|
async def body(self):
|
|
return b'{"key": value}' # Missing quotes around `value`
|
|
|
|
request = MockRequest()
|
|
with pytest.raises(ProxyException):
|
|
await _read_request_body(request)
|
|
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_read_request_body_large_payload():
|
|
"""Test the function with a very large payload."""
|
|
large_payload = '{"key":' + '"a"' * 10**6 + "}" # Large payload
|
|
|
|
class MockRequest:
|
|
async def body(self):
|
|
return large_payload.encode()
|
|
|
|
request = MockRequest()
|
|
with pytest.raises(ProxyException):
|
|
await _read_request_body(request)
|
|
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_read_request_body_unexpected_error():
|
|
"""Test the function when an unexpected error occurs."""
|
|
|
|
class MockRequest:
|
|
async def body(self):
|
|
raise ValueError("Unexpected error")
|
|
|
|
request = MockRequest()
|
|
result = await _read_request_body(request)
|
|
assert result == {} # Ensure fallback behavior
|