模拟一个本地 python 变量

Question

假设我有以下 class:

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 函数很简单。但是，我的用例要复杂得多，模拟这些局部变量会非常有帮助。

Answer 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')

模拟一个本地 python 变量

Mock out a local python variable

python

unit-testing

mocking

python-mock

python-unittest