代码之家 › 专栏 › 技术社区 › Johnny Metz

模拟本地python变量

python-mock python-unittest mocking unit-testing python

Johnny Metz · 技术社区 · 6 年前

假设我有以下课程:

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 方法。我一直在改变功能,所以

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

日期 直接变量,或者这是最好的方法。

注意这是一个简单的示例,创建 _get_date_now 功能是微不足道的。然而,我的用例要复杂得多,模拟这些局部变量将非常有用。

1 回复 | 直到 6 年前

Gang 6 年前

使用 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')