嘲笑时间。睡眠会导致测试失败

Question

我写了这项测试，但是为了不延迟测试，我嘲笑时间。睡觉和测试将遇到失败。

from unittest.mock import patch
from django.core.management import call_command
from django.db.utils import OperationalError
from django.test import TestCase

class CommandsTest(TestCase):
    @patch('time.sleep', return_value=None)
    def test_wait_for_db(self):
        """Test waiting for db"""
        with patch('django.utils.connection.BaseConnectionHandler.__getitem__') as gi:
            gi.side_effect = [OperationalError] * 5 + [True]
            call_command('wait_for_db')
            self.assertEqual(gi.call_count, 6)

通过对第二行（@Patch）评论，该程序将正确运行。 这是错误：

ERROR: test_wait_for_db (core.tests.test_commands.CommandsTest)
Test waiting for db
----------------------------------------------------------------------
Traceback (most recent call last):
  File "/usr/local/lib/python3.10/unittest/mock.py", line 1369, in patched
    return func(*newargs, **newkeywargs)
TypeError: CommandsTest.test_wait_for_db() takes 1 positional argument but 2 were given

Answer 1

您需要将参数添加到test_wait_for_db中。
模拟函数作为参数传递给该功能，

class CommandsTest(TestCase):
    @patch('time.sleep', return_value=None)
    def test_wait_for_db(self, mocked_sleep):

由于您使用装饰器，因此在测试中将 然后可以测试确定是否确实被调用。更多信息。