0

我有以下需要测试烧瓶路线的单元测试。

@unittest.mock.patch('flask_login.utils._get_user')
@unittest.mock.patch('flask_sqlalchemy.SQLAlchemy')
def test_attendances_management(self, SQLAlchemy, current_user):

    user = MagicMock()

    sqlalchemy = SQLAlchemy()
    sqlalchemy.engine.return_value = MagicMock()

    sqlalchemy.engine.raw_connection.return_value = MagicMock()

    sqlalchemy.engine.raw_connection.cursor.return_value = MagicMock()

    sqlalchemy.engine.raw_connection.cursor.callproc("get_project_members", [3, 9, 2020]).return_value = ((1, 3, 4),)

    current_user.return_value = user
    current_user.return_value.role.name = 'Administrator'
    current_user.return_value.id = 1

    response = self.client.get('attendances/management/3/2020/11')
    self.assertEqual(response.status_code, 200)

一切正常（模拟数据库引擎、raw_connection 和游标），但我有一个问题，即 get_project_members 不存在。我需要模拟哪个部分flask-sqlalchemy或MySQLdb库以伪造某些过程实际存在，因为我使用它flask-testing来创建例如某些表`ExampleTable. 表.create（db.session.bind），我不知道如何创建存储过程或如何使用mock来模拟该过程。

异常日志：

    File "C\:my_app\app\attendances\views.py", line 169, in management
    cursor.callproc("get_project_members", [project_id, month, year])
  File "c:\my_app\flaskvenv\lib\site-packages\MySQLdb\cursors.py", line 365, in callproc
    self._query(q)
  File "c:\my_app\flaskvenv\lib\site-packages\MySQLdb\cursors.py", line 411, in _query
    rowcount = self._do_query(q)
  File "c:\my_app\flaskvenv\lib\site-packages\MySQLdb\cursors.py", line 374, in _do_query
    db.query(q)
  File "c:\my_app\flaskvenv\lib\site-packages\MySQLdb\connections.py", line 277, in query
    _mysql.connection.query(self, query)
_mysql_exceptions.OperationalError: (1305, 'PROCEDURE my_app_test_db.get_project_members does not exist')
4

1 回答 1

0

我刚刚通过cursor.execute使用原始查询创建 MySQL 过程的方法解决了这个问题。另外，我已经有 db 作为 SQLAlchemy 的实例，所以我不需要模拟已经定义的连接。

    @unittest.mock.patch('flask_login.utils._get_user')
    def test_attendances_management(self, current_user):

        user = MagicMock()
        q = "CREATE PROCEDURE `get_project_members` (IN `project_id` INT, IN `month` INT, IN `year` INT) BEGIN END"
        connection = db.engine.raw_connection()
        cursor = connection.cursor()
        cursor.execute(q)

        current_user.return_value = user
        current_user.return_value.role.name = 'Administrator'
        current_user.return_value.id = 1

        response = self.client.get('attendances/management/3/2020/11')
        self.assertEqual(response.status_code, 200)

类似的答案： 使用 SQLAlchemy 创建存储过程

于 2020-10-19T15:02:59.317 回答