mocking ignores base_url
hydrargyrum opened this issue · 2 comments
hydrargyrum commented
aiohttp.ClientSession
takes a base_url
(https://docs.aiohttp.org/en/stable/client_reference.html?highlight=base_url#aiohttp.ClientSession) that is then used to build the URL with .request()
and other methods.
Sample usage:
In [3]: session = aiohttp.ClientSession("http://httpbin.org/")
In [4]: await session.get("/get")
Out[4]:
<ClientResponse(http://httpbin.org/get) [200 OK]>
[…]
Unfortunately, aioresponses
ignores that, so it's not possible to mock the full URL, because aioresponses would only pass /get
to its matchers.
pcrespov commented
A small test that illustrates it (in case is useful)
import pytest
from aiohttp import ClientSession
from aioresponses import aioresponses
@pytest.fixture
def mock_aioresponse():
with aioresponses() as m:
m.get("http://example.com/foo", status=200)
yield m
async def test_with_full_url(mock_aioresponse: aioresponses):
async with ClientSession() as session:
async with session.get("http://example.com/foo") as response:
assert response.ok
async def test_with_relative_url(mock_aioresponse: aioresponses):
async with ClientSession(base_url="http://example.com") as session:
async with session.get("/foo") as response:
assert response.ok
test_aioreponses.py::test_with_full_url PASSED
test_aioreponses.py::test_with_relative_url FAILED
rsaleev commented
A dirty hack solves the problem.
# construct URL with ClientSession._base_url
if orig_self._base_url:
url_origin = f"{orig_self._base_url}/{url}"
url = f"{orig_self._base_url}/{url}"
url = normalize_url(merge_params(url, kwargs.get('params')))
url_str = str(url)
for prefix in self._passthrough:
if url_str.startswith(prefix):
return (await self.patcher.temp_original(
orig_self, method, url_origin, *args, **kwargs
))
# add ClientSession headers
if orig_self.headers:
kwargs['headers'] = orig_self.headers
key = (method, url)
self.requests.setdefault(key, [])
request_call = self._build_request_call(method, *args, ,**kwargs)
self.requests[key].append(request_call)
response = await self.match(method, url, **kwargs)