mirror of
https://github.com/aiogram/aiogram.git
synced 2025-12-09 09:22:03 +00:00
Implement stream_content in AiohttpSession and add tests
This commit is contained in:
parent
7ab0db7991
commit
26708154b0
2 changed files with 33 additions and 3 deletions
|
|
@ -1,4 +1,4 @@
|
|||
from typing import AsyncContextManager
|
||||
from typing import AsyncContextManager, AsyncGenerator
|
||||
|
||||
import aiohttp
|
||||
import pytest
|
||||
|
|
@ -107,6 +107,26 @@ class TestAiohttpSession:
|
|||
|
||||
assert patched_raise_for_status.called_once()
|
||||
|
||||
@pytest.mark.asyncio
|
||||
async def test_stream_content(self, aresponses: ResponsesMockServer):
|
||||
aresponses.add(
|
||||
aresponses.ANY, aresponses.ANY, "get", aresponses.Response(status=200, body=b"\f" * 10)
|
||||
)
|
||||
|
||||
session = AiohttpSession()
|
||||
stream = session.stream_content(
|
||||
"https://www.python.org/static/img/python-logo.png", timeout=5, chunk_size=1
|
||||
)
|
||||
assert isinstance(stream, AsyncGenerator)
|
||||
|
||||
size = 0
|
||||
async for chunk in stream:
|
||||
assert isinstance(chunk, bytes)
|
||||
chunk_size = len(chunk)
|
||||
assert chunk_size == 1
|
||||
size += chunk_size
|
||||
assert size == 10
|
||||
|
||||
@pytest.mark.asyncio
|
||||
async def test_context_manager(self):
|
||||
session = AiohttpSession()
|
||||
|
|
|
|||
Loading…
Add table
Add a link
Reference in a new issue