Asm*_*oox 6 python unit-testing mocking aiohttp
我有一个模拟问题,当使用装饰器时我无法解决几个小时@mock.patch,但当尝试使用上下文管理器时立即解决。我想做的是问它在幕后工作方式是否还有更多差异?说得更多,我的意思不仅仅是设置模拟上下文。
我的示例是关于模拟 aiohttp 服务器的数据库连接。测试如下:
后端/测试/test_server.py
# source: https://docs.aiohttp.org/en/stable/testing.html#unittest
from aiohttp.test_utils import AioHTTPTestCase, unittest_run_loop
from alchemy_mock.mocking import AlchemyMagicMock
from backend.app import create_app
from datetime import datetime
from unittest import mock
import aiohttp
import asyncio
import json
class MyAppTestCase(AioHTTPTestCase):
"""
Class for testing aiohttp server
Methods
-------
...
test_get_version()
Test response for GET /version
"""
@mock.patch('backend.app.SessionProvider')
async def get_application(self, mock_sess_prov):
mock_sess_prov.session = AlchemyMagicMock()
mock_sess_prov.start_session = lambda: None
return create_app()
...
@unittest_run_loop
async def test_get_version(self):
"""Test response for GET /version
------------
Verification
Answer comparison
"""
resp: aiohttp.ClientResponse =\
await self.client.request("GET", "/version")
assert resp.status == 200
text = await resp.text()
assert "Project version is 0.1" in text
Run Code Online (Sandbox Code Playgroud)
所以问题出在get_application()方法上。当我如上所述模拟它时,我收到错误,因为目标类SessionProvider未被模拟并尝试创建 SqlAlchemy 会话对象。当我重建提到的功能时,如下所示:
async def get_application(self):
with mock.patch('backend.app.SessionProvider') as mock_sess_prov:
mock_sess_prov.session = AlchemyMagicMock()
mock_sess_prov.start_session = lambda: None
return create_app()
Run Code Online (Sandbox Code Playgroud)
一切正常。为了更多地了解我的应用程序如何工作,我粘贴了我的应用程序代码和会话管理器:
后端/后端/app.py:
from aiohttp import web
from backend.schema import ReservationSchema
import marshmallow.exceptions
from backend.models import Reservation, ValidationError
from backend.session_provider import SessionProvider
from sqlalchemy.exc import IntegrityError
import json
import logging
async def version(request) -> web.Response:
"""GET /version endpoint
Returns API version
Returns
-------
web.Response
Http response object
"""
text: str = "Project version is 0.1"
return web.Response(text=text)
async def new_reservation(request) -> web.Response:
"""POST /new-reservation endpoint
Create new reservation
Returns
-------
web.Response
Http response object
"""
# TODO: How to check that object?
body = await request.json()
# json_obj = json.loads(body)
if type(body) is str:
body = json.loads(body)
try:
reservation: Reservation = ReservationSchema().load(body)
SessionProvider.session.add(reservation)
SessionProvider.session.commit()
return web.Response(text='OK')
except IntegrityError:
return web.Response(text='ERR: No restaurant has given id')
except marshmallow.exceptions.ValidationError as e:
return web.Response(text='ERR: Failed validation:\n' + str(e))
except ValidationError as e:
return web.Response(text='ERR: Failed validation:\n' + str(e))
def create_app() -> web.Application:
"""Creates application object with routing
Returns
-------
web.Application
Application object
"""
SessionProvider.start_session()
app = web.Application()
app.add_routes([web.get('/version', version),
web.post('/new-reservation', new_reservation)])
return app
if __name__ == '__main__':
logging.basicConfig(level=logging.DEBUG)
app = create_app()
web.run_app(app, port=8000)
Run Code Online (Sandbox Code Playgroud)
后端/后端/session_provider.py:
from backend.config import DATABASE_URI
from sqlalchemy import create_engine
from sqlalchemy.orm import sessionmaker
class SessionProvider:
engine = None
sess_maker = None
session = None
@classmethod
def start_session(cls):
# DATABASE_URI = f"postgres+psycopg2://postgres:example@localhost:5433/restro"
cls._engine = create_engine(DATABASE_URI)
cls.sess_maker = sessionmaker(bind=cls._engine)
# create a Session
cls.session = cls.sess_maker()
Run Code Online (Sandbox Code Playgroud)
所以我的第一个猜测是它可能与路径有关,但是当我注意到mock.patch('backend.app.SessionProvider')与上下文管理器一起使用时完全相同时,我开始怀疑它。您知道为什么一种方法不起作用而第二种方法却有效吗?最后,我想分享我的项目树,所以也许这会帮助我们找到答案: