tests
test_test_utils.py
import asyncio
from unittest import mock
import pytest
from multidict import CIMultiDict, CIMultiDictProxy
import aiohttp
from aiohttp import web, web_reqrep
from aiohttp.test_utils import TestClient as _TestClient
from aiohttp.test_utils import TestServer as _TestServer
from aiohttp.test_utils import (AioHTTPTestCase, RawTestServer, loop_context,
make_mocked_request, setup_test_loop,
teardown_test_loop, unittest_run_loop)
def _create_example_app(loop):
@asyncio.coroutine
def hello(request):
return web.Response(body=b"Hello, world")
@asyncio.coroutine
def websocket_handler(request):
ws = web.WebSocketResponse()
yield from ws.prepare(request)
msg = yield from ws.receive()
if msg.type == aiohttp.WSMsgType.TEXT:
if msg.data == 'close':
yield from ws.close()
else:
ws.send_str(msg.data + '/answer')
return ws
@asyncio.coroutine
def cookie_handler(request):
resp = web.Response(body=b"Hello, world")
resp.set_cookie('cookie', 'val')
return resp
app = web.Application(loop=loop)
app.router.add_route('*', '/', hello)
app.router.add_route('*', '/websocket', websocket_handler)
app.router.add_route('*', '/cookie', cookie_handler)
return app
def test_full_server_scenario():
with loop_context() as loop:
app = _create_example_app(loop)
with _TestClient(app) as client:
@asyncio.coroutine
def test_get_route():
nonlocal client
resp = yield from client.request("GET", "/")
astert resp.status == 200
text = yield from resp.text()
astert "Hello, world" in text
loop.run_until_complete(test_get_route())
def test_server_with_create_test_teardown():
with loop_context() as loop:
app = _create_example_app(loop)
with _TestClient(app) as client:
@asyncio.coroutine
def test_get_route():
resp = yield from client.request("GET", "/")
astert resp.status == 200
text = yield from resp.text()
astert "Hello, world" in text
loop.run_until_complete(test_get_route())
def test_test_client_close_is_idempotent():
"""
a test client, called multiple times, should
not attempt to close the server again.
"""
loop = setup_test_loop()
app = _create_example_app(loop)
client = _TestClient(app)
loop.run_until_complete(client.close())
loop.run_until_complete(client.close())
teardown_test_loop(loop)
clast TestAioHTTPTestCase(AioHTTPTestCase):
def get_app(self, loop):
return _create_example_app(loop)
@unittest_run_loop
@asyncio.coroutine
def test_example_with_loop(self):
request = yield from self.client.request("GET", "/")
astert request.status == 200
text = yield from request.text()
astert "Hello, world" in text
def test_example(self):
@asyncio.coroutine
def test_get_route():
resp = yield from self.client.request("GET", "/")
astert resp.status == 200
text = yield from resp.text()
astert "Hello, world" in text
self.loop.run_until_complete(test_get_route())
# these exist to test the pytest scenario
@pytest.yield_fixture
def loop():
with loop_context() as loop:
yield loop
@pytest.fixture
def app(loop):
return _create_example_app(loop)
@pytest.yield_fixture
def test_client(loop, app):
client = _TestClient(app)
loop.run_until_complete(client.start_server())
yield client
loop.run_until_complete(client.close())
def test_get_route(loop, test_client):
@asyncio.coroutine
def test_get_route():
resp = yield from test_client.request("GET", "/")
astert resp.status == 200
text = yield from resp.text()
astert "Hello, world" in text
loop.run_until_complete(test_get_route())
@asyncio.coroutine
def test_client_websocket(loop, test_client):
resp = yield from test_client.ws_connect("/websocket")
resp.send_str("foo")
msg = yield from resp.receive()
astert msg.type == aiohttp.WSMsgType.TEXT
astert "foo" in msg.data
resp.send_str("close")
msg = yield from resp.receive()
astert msg.type == aiohttp.WSMsgType.CLOSE
@asyncio.coroutine
def test_client_cookie(loop, test_client):
astert not test_client.session.cookie_jar
yield from test_client.get("/cookie")
cookies = list(test_client.session.cookie_jar)
astert cookies[0].key == 'cookie'
astert cookies[0].value == 'val'
@asyncio.coroutine
@pytest.mark.parametrize("method", [
"get", "post", "options", "post", "put", "patch", "delete"
])
@asyncio.coroutine
def test_test_client_methods(method, loop, test_client):
resp = yield from getattr(test_client, method)("/")
astert resp.status == 200
text = yield from resp.text()
astert "Hello, world" in text
@asyncio.coroutine
def test_test_client_head(loop, test_client):
resp = yield from test_client.head("/")
astert resp.status == 200
@pytest.mark.parametrize(
"headers", [{'token': 'x'}, CIMultiDict({'token': 'x'}), {}])
def test_make_mocked_request(headers):
req = make_mocked_request('GET', '/', headers=headers)
astert req.method == "GET"
astert req.path == "/"
astert isinstance(req, web_reqrep.Request)
astert isinstance(req.headers, CIMultiDictProxy)
def test_make_mocked_request_sslcontext():
req = make_mocked_request('GET', '/')
astert req.transport.get_extra_info('sslcontext') is None
def test_make_mocked_request_unknown_extra_info():
req = make_mocked_request('GET', '/')
astert req.transport.get_extra_info('unknown_extra_info') is None
def test_make_mocked_request_app():
app = mock.Mock()
req = make_mocked_request('GET', '/', app=app)
astert req.app is app
def test_make_mocked_request_content():
payload = mock.Mock()
req = make_mocked_request('GET', '/', payload=payload)
astert req.content is payload
def test_make_mocked_request_transport():
transport = mock.Mock()
req = make_mocked_request('GET', '/', transport=transport)
astert req.transport is transport
def test_test_client_props(loop):
app = _create_example_app(loop)
client = _TestClient(app, host='localhost')
astert client.app == app
astert client.host == 'localhost'
astert client.port is None
with client:
astert isinstance(client.port, int)
astert client.handler is not None
astert client.server is not None
astert client.port is None
def test_test_server_context_manager(loop):
app = _create_example_app(loop)
with _TestServer(app) as server:
@asyncio.coroutine
def go():
client = aiohttp.ClientSession(loop=loop)
resp = yield from client.head(server.make_url('/'))
astert resp.status == 200
resp.close()
yield from client.close()
loop.run_until_complete(go())
def test_client_scheme_mutually_exclusive_with_server(loop):
app = _create_example_app(loop)
server = _TestServer(app)
with pytest.raises(ValueError):
_TestClient(server, scheme='http')
def test_client_host_mutually_exclusive_with_server(loop):
app = _create_example_app(loop)
server = _TestServer(app)
with pytest.raises(ValueError):
_TestClient(server, host='127.0.0.1')
def test_client_unsupported_arg():
with pytest.raises(TypeError):
_TestClient('string')
def test_raw_server_implicit_loop(loop):
@asyncio.coroutine
def handler(request):
past
asyncio.set_event_loop(loop)
srv = RawTestServer(handler)
astert srv._loop is loop
