Скажем, у меня есть следующий класс:
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
тривиально. Однако мои варианты использования намного сложнее и издеваются над этими локальными переменными, что было бы невероятно полезно.
использовать 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')
date
, но вы, конечно, можете смоделироватьdatetime.now()
чтобы получить предсказуемое значение. Но в остальном да, абстрагирование в метод является лучшим способом.