jeans/core
jeans/core
1
0
Fork 0
mirror of https://github.com/home-assistant/core.git synced 2025-04-25 01:38:02 +00:00
Code Issues Packages Projects Releases Wiki Activity

Add body_exists to MockRequest in aiohttp util (#100932)

Browse Source 
* Add body_exists to MockRequest in aiohttp util

* Add body_exists to MockRequest in aiohttp util

* Add body_exists to MockRequest in aiohttp util
This commit is contained in:
Joost Lekkerkerker 2023-09-26 17:52:29 +02:00 committed by GitHub
parent 785b46af22
commit 59207be5f8
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
2 changed files with 13 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
homeassistant/util/aiohttp.py
View File

@ -66,6 +66,11 @@ class MockRequest:
"""Return the body as text.""" """Return the body as text."""
return MockStreamReader(self._content) return MockStreamReader(self._content)
@property
def body_exists(self) -> bool:
"""Return True if request has HTTP BODY, False otherwise."""
return bool(self._text)
async def json(self, loads: JSONDecoder = json_loads) -> Any: async def json(self, loads: JSONDecoder = json_loads) -> Any:
"""Return the body as JSON.""" """Return the body as JSON."""
return loads(self._text) return loads(self._text)

9
tests/util/test_aiohttp.py
View File

@ -12,12 +12,19 @@ async def test_request_json() -> None:
async def test_request_text() -> None: async def test_request_text() -> None:
"""Test a JSON request.""" """Test bytes in request."""
request = aiohttp.MockRequest(b"hello", status=201, mock_source="test") request = aiohttp.MockRequest(b"hello", status=201, mock_source="test")
assert request.body_exists
assert request.status == 201 assert request.status == 201
assert await request.text() == "hello" assert await request.text() == "hello"
async def test_request_body_exists() -> None:
"""Test body exists."""
request = aiohttp.MockRequest(b"", mock_source="test")
assert not request.body_exists
async def test_request_post_query() -> None: async def test_request_post_query() -> None:
"""Test a JSON request.""" """Test a JSON request."""
request = aiohttp.MockRequest( request = aiohttp.MockRequest(