python/aiogram
1
0
Fork 0
mirror of https://github.com/aiogram/aiogram.git synced 2025-12-12 10:11:52 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

#1191 Added possibility to pass custom headers to URLInputFile object (#1197)

Browse source
This commit is contained in:
Alex Root Junior 2023-06-25 01:39:26 +03:00 • committed by GitHub
parent 484a61bdc1
commit d29b18da8c
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
10 changed files with 53 additions and 16 deletions
Download patch file Download diff file Expand all files Collapse all files

10
tests/test_api/test_client/test_session/test_base_session.py
View file

@ -1,6 +1,6 @@
import datetime
import json
from typing import Any, AsyncContextManager, AsyncGenerator, Optional
from typing import Any, AsyncContextManager, AsyncGenerator, Dict, Optional
from unittest.mock import AsyncMock, patch
import pytest
@ -44,7 +44,12 @@ class CustomSession(BaseSession):
assert isinstance(method, TelegramMethod)
async def stream_content(
self, url: str, timeout: int, chunk_size: int, raise_for_status: bool
self,
url: str,
headers: Optional[Dict[str, Any]] = None,
timeout: int = 30,
chunk_size: int = 65536,
raise_for_status: bool = True,
) -> AsyncGenerator[bytes, None]: # pragma: no cover
assert isinstance(url, str)
assert isinstance(timeout, int)
@ -215,6 +220,7 @@ class TestBaseSession:
session = CustomSession()
stream = session.stream_content(
"https://www.python.org/static/img/python-logo.png",
headers={},
timeout=5,
chunk_size=65536,
raise_for_status=True,