aiohttp有一个pytest插件可以轻松构建web服务器测试程序,同时该插件还有一个用于测试其他框架(单元测试等)的测试框架包。
在写测试之前,我想你可能会想读一读如何写一个可测试的服务器程序感兴趣,因为它们之间的作用的相互的。
在使用之前,我们还需要安装下才行:
$ pip install pytest-aiohttp
如果你不想安装它,你可以在conftest.py中插入一行 pytest_plugins='aiohttp.pytest_plugin'
来代替这个包。
该模块是临时的。
对于已经废弃的API,基于向后不兼容政策,aiohttp允许仍可以继续使用一年半的时间。
不过这对aiohttp.test_tools则不适用。
同时,如有一些必要的原因,我们也会不管向后兼容期而做出更改。
aiohttp中的test_utils有一个基于aiohttp的web服务器测试模板。
其中包含两个部分: 一个是启动测试服务器,然后是向这个服务器发起HTTP请求。 TestServer使用以aiohttp.web.Application为基础的服务器。RawTestServer则使用基于aiohttp.web.WebServer的低级服务器。 发起HTTP请求到服务器你可以创建一个TestClient实例对象。 测试客户端实例使用aiohttp.ClientSession来对常规操作如ws_connect,get,post等进行支持。
#Pytest pytest-aiohttp插件允许你创建客户端并向你的应用程序发起请求来进行测试。
简易程序如下:
from aiohttp import web
async def hello(request):
return web.Response(text='Hello, world')
async def test_hello(test_client, loop):
app = web.Application()
app.router.add_get('/', hello)
client = await test_client(app)
resp = await client.get('/')
assert resp.status == 200
text = await resp.text()
assert 'Hello, world' in text
同样,它也提供访问app实例的方法,允许测试组件查看app的状态。使用fixture可以创建非常便捷的app测试客户端:
import pytest
from aiohttp import web
async def previous(request):
if request.method == 'POST':
request.app['value'] = (await request.post())['value']
return web.Response(body=b'thanks for the data')
return web.Response(
body='value: {}'.format(request.app['value']).encode('utf-8'))
@pytest.fixture
def cli(loop, test_client):
app = web.Application()
app.router.add_get('/', previous)
app.router.add_post('/', previous)
return loop.run_until_complete(test_client(app))
async def test_set_value(cli):
resp = await cli.post('/', data={'value': 'foo'})
assert resp.status == 200
assert await resp.text() == 'thanks for the data'
assert cli.server.app['value'] == 'foo'
async def test_get_value(cli):
cli.server.app['value'] = 'bar'
resp = await cli.get('/')
assert resp.status == 200
assert await resp.text() == 'value: bar'
Pytest工具箱里有以下fixture: aiohttp.test_utils.test_server(app, **kwargs) 一个创建TestServer的fixture。
async def test_f(test_server):
app = web.Application()
# 这里填写路由表
server = await test_server(app)
服务器会在测试功能结束后销毁。
app是aiohttp.web.Application组件,用于启动服务器。
kwargs是其他需要传递的参数。
aiohttp.test_utils.test_client(app, **kwargs) aiohttp.test_utils.test_client(server, **kwargs) aiohttp.test_utils.test_client(raw_server, **kwargs) 一个用户创建访问测试服务的TestClient fixture。
async def test_f(test_client):
app = web.Application()
# 这里填写路由表。
client = await test_client(app)
resp = await client.get('/')
客户端和响应在测试功能完成后会自动清除。
这个fixture可以接收aiohttp.webApplication, aiohttp.test_utils.TestServer或aiohttp.test_utils.RawTestServer实例对象。
kwargs用于接收传递给aiohttp.test_utils.TestClient的参数。
aiohttp.test_utils.raw_test_server(handler, **kwargs) 一个从给定web处理器实例创建RawTestServer的fixture。
处理器应是一个可以接受请求并且返回响应的协同程序:
async def test_f(raw_test_server, test_client):
async def handler(request):
return web.Response(text="OK")
raw_server = await raw_test_server(handler)
client = await test_client(raw_server)
resp = await client.get('/')
使用标准库里的单元测试/基础单元测试的功能来测试应用程序,提供AioHTTPTestCase类:
from aiohttp.test_utils import AioHTTPTestCase, unittest_run_loop
from aiohttp import web
class MyAppTestCase(AioHTTPTestCase):
async def get_application(self):
"""
Override the get_app method to return your application.
"""
return web.Application()
# the unittest_run_loop decorator can be used in tandem with
# the AioHTTPTestCase to simplify running
# tests that are asynchronous
@unittest_run_loop
async def test_example(self):
request = await self.client.request("GET", "/")
assert request.status == 200
text = await request.text()
assert "Hello, world" in text
# a vanilla example
def test_example(self):
async def test_get_route():
url = root + "/"
resp = await self.client.request("GET", url, loop=loop)
assert resp.status == 200
text = await resp.text()
assert "Hello, world" in text
self.loop.run_until_complete(test_get_route())
class aiohttp.test_utils.AioHTTPTestCase
一个允许使用aiohttp对web应用程序进行单元测试的基础类。
该类派生于unittest.TestCase.
提供下列功能:
client
aiohttp测试客户端(TestClient实例)。
server
aiohttp测试服务器(TestServer实例)。 新增于2.3.0版本。
loop
应用程序和服务器运行的事件循环。
app
应用程序(aiohttp.web.Application实例),由get_app()返回。
coroutine get_client()
该方法可以被覆盖。返回测试中的TestClient对象。
返回TestClient实例对象。 新增于2.3.0版本。
coroutine get_server()
该方法可以备覆盖。返回测试中的TestServer对象。
返回TestServer实例对象。 新增于2.3.0版本。
coroutine get_application()
该方法可以被覆盖。返回用于测试的aiohttp.web.Application对象。
返回aiohttp.web.Application实例对象。
coroutine setUpAsync()
默认该方法什么也不做,不过可以被覆盖用于在TestCase的setUp阶段执行异步代码。 新增于2.3.0版本。
coroutine tearDownAsync()
默认该方法什么也不做,不过可以被覆盖用于在TestCase的tearDown阶段执行异步代码。 新增于2.3.0版本。
setUp()
标准测试初始化方法。
tearDown()
标准测试析构方法。
TestClient的方法都是异步方法,你必须使用异步方法来执行它的函数。
使用unittest_run_loop()装饰器可以包装任何一个基础类中的测试方法。
class TestA(AioHTTPTestCase):
@unittest_run_loop
async def test_f(self):
resp = await self.client.get('/')
unittest_run_loop
专门用在AioHTTPTestCase的异步方法上的装饰器。
使用AioHTTPTestCase中的AioHTTPTestCase.loop来执行异步函数。
aiohttp提供创建虚假aiohttp.web.Request对象的测试工具: aiohttp.test_utils.make_mocked_request()
,在一些简单的单元测试中特别好用,比如处理器测试,或者很难在真实服务器上重现的错误之类的。
from aiohttp import web
from aiohttp.test_utils import make_mocked_request
def handler(request):
assert request.headers.get('token') == 'x'
return web.Response(body=b'data')
def test_handler():
req = make_mocked_request('GET', '/', headers={'token': 'x'})
resp = handler(req)
assert resp.body == b'data'
我们不建议在任何地方都用make_mocked_request()来测试,最好使用真实的操作。
make_mocked_request()的存在只是为了测试那些很难或根本不能通过简便方法测试的复杂案例(比如模仿网络错误)。
aiohttp.test_utils.make_mocked_request(method, path, headers=None, *, version=HttpVersion(1, 1), closing=False, app=None, match_info=sentinel, reader=sentinel, writer=sentinel, transport=sentinel, payload=sentinel, sslcontext=None, loop=...)
创建一个用于测试的仿真web.Request。
对于那些在特殊环境难以触发的错误在单元测试中非常有用。
参数:
创建高等级测试:
from aiohttp.test_utils import TestClient, loop_context
from aiohttp import request
# loop_context is provided as a utility. You can use any
# asyncio.BaseEventLoop class in it's place.
with loop_context() as loop:
app = _create_example_app()
with TestClient(app, loop=loop) as client:
async def test_get_route():
nonlocal client
resp = await client.get("/")
assert resp.status == 200
text = await resp.text()
assert "Hello, world" in text
loop.run_until_complete(test_get_route())
如果需要更细粒度的创建/拆除,可以直接用TestClient对象:
from aiohttp.test_utils import TestClient
with loop_context() as loop:
app = _create_example_app()
client = TestClient(app, loop=loop)
loop.run_until_complete(client.start_server())
root = "http://127.0.0.1:{}".format(port)
async def test_get_route():
resp = await client.get("/")
assert resp.status == 200
text = await resp.text()
assert "Hello, world" in text
loop.run_until_complete(test_get_route())
loop.run_until_complete(client.close())
你可以在api参考中找到所有工具包清单。
一些如motor, aioes等依赖asyncio循环来执行代码的库,当它们运行正常程序时,都会选一个主事件循环给asyncio.get_event_loop。问题在于,当处在测试环境中时,我们没有主事件循环,因为每个测试都有一个独立的循环。 这样当其他库尝试找这个主事件循环时就会发生出错。不过幸运的是,这问题很好解决,我们可以显式的传入循环。我们来看aioes客户端中的代码:
def __init__(self, endpoints, *, loop=None, **kwargs)
如你所见,有一个可选的loop参数。当然,我们并不打算直接测试aioes客户端只是我们的服务建立在它之上。所以如果我们想让我们的AioESService容易测试,我们可以这样写:
import asyncio
from aioes import Elasticsearch
class AioESService:
def __init__(self, loop=None):
self.es = Elasticsearch(["127.0.0.1:9200"], loop=loop)
async def get_info(self):
cluster_info = await self.es.info()
print(cluster_info)
if __name__ == "__main__":
client = AioESService()
loop = asyncio.get_event_loop()
loop.run_until_complete(client.get_info())
注意它接受的loop参数。正常情况下没有什么影响因为我们不用显示地传递loop就能让服务有一个主事件循环。问题出在我们测试时:
import pytest
from main import AioESService
class TestAioESService:
async def test_get_info(self):
cluster_info = await AioESService().get_info()
assert isinstance(cluster_info, dict)
如果尝试运行测试,一般会失败并给出类似下面的信息:
...
RuntimeError: There is no current event loop in thread 'MainThread'.
因为aioes在主线程中找不到当前的事件循环,所以就报错咯。显式地传递事件循环可以解决这个问题。 如果你的代码依靠隐式循环工作,你可以需要点小技巧。请看FAQ。
在随机TCP端口上运行给定的aiohttp.web.Application。
创建完成后服务器并没开始,请用start_server()确保服务器开启和使用close()来确保关闭。
测试服务器通常与aiohttp.test_utils.TestClient连用,后者可以提供便利的客户端方法来访问服务器。
class aiohttp.test_utils.BaseTestServer(*, scheme='http', host='127.0.0.1')
测试服务器的基础类。
参数:
class aiohttp.test_utils.RawTestServer(handler, *, scheme="http", host="127.0.0.1")
低级测试服务器(派生于BaseTestServer)
参数:
class aiohttp.test_utils.TestClient(app_or_server, *, loop=None, scheme='http', host='127.0.0.1')
一个用于制造请求来测试服务器的测试客户端。
参数:
aiohttp.test_utils.make_mocked_coro(return_value)
创建一个协程mock。
其表现形式像一个协程一般,作用是返回要返回的值(return_value)。而同时又是一个mock对象,你可以使用一般的Mock来测试它:
mocked = make_mocked_coro(1)
assert 1 == await mocked(1, 2)
mocked.assert_called_with(1, 2)
参数: return_value - 当mock对象被调用时返回的值。
像协程一样返回return_value的值。
aiohttp.test_utils.unused_port()
返回一个可以用在IPv4 TCP协议上的还没有被使用的端口。 返回一个可以使用端口值(类型为整数int)。
aiohttp.test_utils.loop_context(loop_factory=
一个上下文管理器,可以创建一个用于测试目的事件循环。
用于进行测试循环的创建和清理工作。
aiohttp.test_utils.setup_test_loop(loop_factory=
创建并返回asyncio.AbstractEventLoop实例对象。
如果要调用它,需要在结束循环时调用下teardown_test_loop.
aiohttp.test_utils.teardown_test_loop(loop)
销毁并清除setup_test_loop所创建的event_loop。
参数: loop(asyncio.AbstractEventLoop) - 需要拆除的循环。