Question

假设我有以下课程：

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

模拟出本地python变量

0 个答案: