Макет локальной переменной Python - PullRequest
Новая
       55

Макет локальной переменной Python

0 голосов
/ 14 сентября 2018

Допустим, у меня есть следующий класс: 

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 тривиально. Тем не менее, мои варианты использования гораздо сложнее, и макетирование этих локальных переменных было бы невероятно полезным.

1 Ответ

0 голосов
/ 15 сентября 2018

использование 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')
...