mock.patch() 作为装饰器和上下文管理器之间的区别

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')与上下文管理器一起使用时完全相同时,我开始怀疑它。您知道为什么一种方法不起作用而第二种方法却有效吗?最后,我想分享我的项目树,所以也许这会帮助我们找到答案:

项目树