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

1

Скажем, у меня есть следующий класс:

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)

Я пытаюсь print_name переменную 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
    Вы не можете смоделировать date , но вы, конечно, можете смоделировать datetime.now() чтобы получить предсказуемое значение. Но в остальном да, абстрагирование в метод является лучшим способом.
Теги:
unit-testing
mocking
python-unittest
python-mock

1 ответ

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

Ещё вопросы

Сообщество Overcoder
Наверх
Меню