0

我正在使用 vscode-python 的扩展和 python 的 unittest 来测试是否peewee成功连接到数据库。这是正在测试的类：

import logging
from datetime import (datetime, timezone)

# 3rd party libraries
import peewee


log_fmt = "%(asctime)s - %(funcName)s - %(name)s - %(levelname)s - %(message)s"
# enable logging
logging.basicConfig(
    format=log_fmt,
    level=logging.INFO
)

logger = logging.getLogger(__name__)


class Freezer():

    db_name = "db"
    db_username = "db_user"
    db_pswd = "db_pswd"

    def __init__(self):
        self.freezer = peewee.MySQLDatabase(
            database=Freezer.db_name,
            user=Freezer.db_username,
            password=Freezer.db_pswd,
            host="localhost",
            port=3306
        )

    def open_freezer(self):
        if self.freezer.connect():
            print('CONNECTED')

调用时open_freezer()将连接到数据库并打印出来CONNECTED。这是测试套件：

import unittest
from unittest.mock import (MagicMock, patch)

from src.freezer import (Freezer, BaseModel, PlatiniumBotUser, peewee)


class FreezerTestSuites(unittest.TestCase):

    def test_open_freezer(self):
        with patch("src.freezer.peewee.MySQLDatabase", autospec=True) as mock_db:
            self.freezer = Freezer()
            mock_db.assert_called_once()
            self.freezer.open_freezer()
            mock_db.connect.assert_called()

在最后一行，akamock_db.connect.assert_called()失败并显示以下内容AssertionError：

test_open_freezer (test_freezer.FreezerTestSuites) ... 
CONNECTED
FAIL
NoneType: None

======================================================================
FAIL: test_open_freezer (test_freezer.FreezerTestSuites)
----------------------------------------------------------------------
Traceback (most recent call last):
  File "/home/gadd/vscodeworkspace/db_testing/tests/test_freezer.py", line 23, in test_open_freezer
    mock_db.connect.assert_called()
  File "/usr/lib/python3.8/unittest/mock.py", line 882, in assert_called
    raise AssertionError(msg)
AssertionError: Expected 'connect' to have been called.

----------------------------------------------------------------------
Ran 1 test in 0.117s

FAILED (failures=1)

正如您可以清楚地看到它打印出来，CONNECTED从而表明它已调用该connect()方法并成功连接，但断言失败。

我究竟做错了什么？

提前致谢。

4

1 回答 1

0

根据jonrsharpe，我调用connect的是模拟类mock_db而不是它应该返回的self.freezer. 正确的做法是：

def test_open_freezer(self):
     with patch("peewee.MySQLDatabase", autospec=True) as mock_db:
          self.freezer = Freezer()
          mock_db.assert_called()
          self.freezer.open_freezer()
          mock_db.return_value.connect.assert_called()
于 2021-03-17T11:53:56.043 回答