mirror of
https://github.com/BerriAI/litellm.git
synced 2026-09-07 08:26:10 +00:00
fix(tests): add .state to MockRequest in http_parsing_utils tests
MockRequest mocks were missing .state attribute, causing
_read_request_body to hit AttributeError before reading the body
and silently return {} via the catch-all handler.
This commit is contained in:
parent
7e567a5352
commit
249309d61c
1 changed files with 15 additions and 0 deletions
|
|
@ -19,6 +19,9 @@ async def test_read_request_body_valid_json():
|
|||
"""Test the function with a valid JSON payload."""
|
||||
|
||||
class MockRequest:
|
||||
def __init__(self):
|
||||
self.state = type("State", (), {})()
|
||||
|
||||
async def body(self):
|
||||
return b'{"key": "value"}'
|
||||
|
||||
|
|
@ -32,6 +35,9 @@ async def test_read_request_body_empty_body():
|
|||
"""Test the function with an empty body."""
|
||||
|
||||
class MockRequest:
|
||||
def __init__(self):
|
||||
self.state = type("State", (), {})()
|
||||
|
||||
async def body(self):
|
||||
return b""
|
||||
|
||||
|
|
@ -45,6 +51,9 @@ async def test_read_request_body_invalid_json():
|
|||
"""Test the function with an invalid JSON payload."""
|
||||
|
||||
class MockRequest:
|
||||
def __init__(self):
|
||||
self.state = type("State", (), {})()
|
||||
|
||||
async def body(self):
|
||||
return b'{"key": value}' # Missing quotes around `value`
|
||||
|
||||
|
|
@ -59,6 +68,9 @@ async def test_read_request_body_large_payload():
|
|||
large_payload = '{"key":' + '"a"' * 10**6 + "}" # Large payload
|
||||
|
||||
class MockRequest:
|
||||
def __init__(self):
|
||||
self.state = type("State", (), {})()
|
||||
|
||||
async def body(self):
|
||||
return large_payload.encode()
|
||||
|
||||
|
|
@ -72,6 +84,9 @@ async def test_read_request_body_unexpected_error():
|
|||
"""Test the function when an unexpected error occurs."""
|
||||
|
||||
class MockRequest:
|
||||
def __init__(self):
|
||||
self.state = type("State", (), {})()
|
||||
|
||||
async def body(self):
|
||||
raise ValueError("Unexpected error")
|
||||
|
||||
|
|
|
|||
Loading…
Add table
Reference in a new issue