模拟一个本地 python 变量答案

【问题标题】：Mock out a local python variable模拟一个本地 python 变量
【发布时间】：2018-09-14 08:35:27
【问题描述】：

假设我有以下课程：

class Person:
    def __init__(self, name):
        self.name = name

    def print_name(self):
        date = datetime.now().strftime('%Y-%m-%d')
        return 'Hello {}. Today is {}'.format(self.name, date)

我正在尝试模拟 date 变量，以便为 print_name 方法编写成功的测试。我一直在更改函数，所以 date 从函数返回，这很容易模拟。

class Person:
    def __init__(self, name):
        self.name = name

    def print_name(self):
        date = self._get_date_now()
        return 'Hello {}. Today is {}'.format(self.name, date)

    @staticmethod
    def _get_date_now():
        return datetime.now().strftime('%Y-%m-%d')

单元测试看起来像这样：

from mock import patch

class MyTests(unittest.TestCase):

    @patch.object(Person, '_get_date_now')
    def test_hello(self, date_mock):
        date_mock.return_value = '2018-10-10'

        person = Person('Jim')
        result = person.print_name()
        self.assertEqual(result, 'Hello Jim. Today is 2018-10-10')

无论如何我可以直接模拟date 变量，或者这是最好的方法。

注意这是一个简单的例子，创建_get_date_now 函数很简单。但是，我的用例要复杂得多，模拟出这些局部变量会非常有帮助。

【问题讨论】：

你不能模拟出date，但你当然可以模拟datetime.now() 以返回一个可预测的值。但除此之外是的，抽象成一个方法是最好的方法。

标签： python unit-testing mocking python-unittest python-mock

【解决方案1】：

使用freezegun

from freezegun import freeze_time

@freeze_time("2018-10-10")
def test_hello(self):

    person = Person('Jim')
    result = person.print_name()
    self.assertEqual(result, 'Hello Jim. Today is 2018-10-10')

【讨论】：