def test_callback(self, mocker, aiohttp_session_mock):
# Setup
import asyncio
expected_response = mocker.Mock(spec=aiohttp_.aiohttp.ClientResponse)
@asyncio.coroutine
def request(*args, **kwargs):
return expected_response
aiohttp_session_mock.request = request
client = aiohttp_.AiohttpClient(aiohttp_session_mock)
client._sync_callback_adapter = asyncio.coroutine
# Run
globals_ = dict(globals(), client=client)
locals_ = {"asyncio": asyncio}
exec(
"""@asyncio.coroutine
def call():
response = yield from client.send((1, 2, {}))
response = yield from client.apply_callback(lambda x: 2, response)
return response
""",
globals_,
locals_,
)
loop = asyncio.get_event_loop()
value = loop.run_until_complete(
asyncio.ensure_future(locals_["call"]()))
# Verify
assert value == 2
async def test_request_send(self, mocker, aiohttp_session_mock):
# Setup
expected_response = mocker.Mock()
async def request(*args, **kwargs):
return expected_response
aiohttp_session_mock.request = request
client = aiohttp_.AiohttpClient(aiohttp_session_mock)
# Run
response = await client.send((1, 2, {}))
# Verify
assert response == expected_response
def test_request_send(self, aiohttp_session_mock):
# Setup
import asyncio
@asyncio.coroutine
def request(*args, **kwargs):
return 0
aiohttp_session_mock.request = request
client = aiohttp_.AiohttpClient(aiohttp_session_mock)
request = aiohttp_.Request(client)
# Run
response = request.send(1, 2, {})
loop = asyncio.get_event_loop()
value = loop.run_until_complete(asyncio.ensure_future(response))
# Verify
assert value == 0
def test_request_send(self, mocker, aiohttp_session_mock):
# Setup
import asyncio
expected_response = mocker.Mock()
@asyncio.coroutine
def request(*args, **kwargs):
return expected_response
aiohttp_session_mock.request = request
client = aiohttp_.AiohttpClient(aiohttp_session_mock)
# Run
response = client.send((1, 2, {}))
loop = asyncio.get_event_loop()
value = loop.run_until_complete(asyncio.ensure_future(response))
# Verify
assert value == expected_response
def test_callback(self, aiohttp_session_mock):
# Setup
import asyncio
@asyncio.coroutine
def request(*args, **kwargs):
return 2
aiohttp_session_mock.request = request
client = aiohttp_.AiohttpClient(aiohttp_session_mock)
client._sync_callback_adapter = asyncio.coroutine
request = aiohttp_.Request(client)
# Run
request.add_callback(lambda x: 2)
response = request.send(1, 2, {})
loop = asyncio.get_event_loop()
value = loop.run_until_complete(asyncio.ensure_future(response))
# Verify
assert value == 2
async def test_callback(self, mocker, aiohttp_session_mock):
# Setup
expected_response = mocker.Mock(spec=aiohttp_.aiohttp.ClientResponse)
expected_response.text = AsyncMock()
async def request(*args, **kwargs):
return expected_response
aiohttp_session_mock.request = request
client = aiohttp_.AiohttpClient(aiohttp_session_mock)
# Run
async def call():
response = await client.send((1, 2, {}))
response = await client.apply_callback(lambda x: 2, response)
return response
value = await call()
# Verify
assert value == 2
assert expected_response.text.called
def test_create_request(self, aiohttp_session_mock):
aiohttp = aiohttp_.AiohttpClient(aiohttp_session_mock)
assert isinstance(aiohttp.create_request(), aiohttp_.Request)
