Попытка смоделировать datetime.date.today (), но не работает

Question

Может кто-нибудь сказать мне, почему это не работает?

>>> import mock
>>> @mock.patch('datetime.date.today')
... def today(cls):
...  return date(2010, 1, 1)
...
>>> from datetime import date
>>> date.today()
datetime.date(2010, 12, 19)

Возможно, кто-то может предложить лучший способ?

Answer 1

Другой вариант - использовать https://github.com/spulec/freezegun/

Установить его:

pip install freezegun

И использовать его:

from freezegun import freeze_time

@freeze_time("2012-01-01")
def test_something():

    from datetime import datetime
    print(datetime.now()) #  2012-01-01 00:00:00

    from datetime import date
    print(date.today()) #  2012-01-01

Это также влияет на другое время и датувызовы вызовов методов из других модулей:

other_module.py:

from datetime import datetime

def other_method():
    print(datetime.now())    

main.py:

from freezegun import freeze_time

@freeze_time("2012-01-01")
def test_something():

    import other_module
    other_module.other_method()

И наконец:

$ python main.py
# 2012-01-01

Answer 2

Есть несколько проблем.

Во-первых, способ использования mock.patch не совсем верен. При использовании в качестве декоратора он заменяет данную функцию / класс (в данном случае datetime.date.today) на Mock объект только в пределах декорированной функции . Таким образом, только в вашем today() будет datetime.date.today другая функция, которая, кажется, не соответствует вашим ожиданиям.

То, что вы действительно хотите, похоже, выглядит примерно так:

@mock.patch('datetime.date.today')
def test():
    datetime.date.today.return_value = date(2010, 1, 1)
    print datetime.date.today()

К сожалению, это не сработает:

>>> test()
Traceback (most recent call last):
  File "<stdin>", line 1, in <module>
  File "build/bdist.macosx-10.6-universal/egg/mock.py", line 557, in patched
  File "build/bdist.macosx-10.6-universal/egg/mock.py", line 620, in __enter__
TypeError: can't set attributes of built-in/extension type 'datetime.date'

Это невозможно, поскольку встроенные типы Python являются неизменяемыми - см. этот ответ для получения более подробной информации.

В этом случае я бы создал подкласс datetime.date и создал бы правильную функцию:

import datetime
class NewDate(datetime.date):
    @classmethod
    def today(cls):
        return cls(2010, 1, 1)
datetime.date = NewDate

А теперь вы можете сделать:

>>> datetime.date.today()
NewDate(2010, 1, 1)

Answer 3

Что бы это ни стоило, в документах Mock конкретно говорится о datetime.date.today, и это можно сделать без необходимости создания фиктивного класса:

https://docs.python.org/3/library/unittest.mock-examples.html#partial-mocking

>>> from datetime import date
>>> with patch('mymodule.date') as mock_date:
...     mock_date.today.return_value = date(2010, 10, 8)
...     mock_date.side_effect = lambda *args, **kw: date(*args, **kw)
...
...     assert mymodule.date.today() == date(2010, 10, 8)
...     assert mymodule.date(2009, 6, 8) == date(2009, 6, 8)
...

Answer 4

Полагаю, я немного опоздал, но я думаю, что главная проблема здесь в том, что вы исправляете datetime.date.today напрямую, и, согласно документации, это неправильно.

Вы должны исправить ссылку, импортированную в файл, где находится тестируемая функция, например.

Допустим, у вас есть файл functions.py, в котором есть следующее:

import datetime

def get_today():
    return datetime.date.today()

тогда в вашем тесте у вас должно быть что-то вроде этого

import datetime
import unittest

from functions import get_today
from mock import patch, Mock

class GetTodayTest(unittest.TestCase):

    @patch('functions.datetime')
    def test_get_today(self, datetime_mock):
        datetime_mock.date.today = Mock(return_value=datetime.strptime('Jun 1 2005', '%b %d %Y'))
        value = get_today()
        # then assert your thing...

Надеюсь, это немного поможет.

Answer 5

Для добавления к решению Даниэля Г. :

from datetime import date

class FakeDate(date):
    "A manipulable date replacement"
    def __new__(cls, *args, **kwargs):
        return date.__new__(date, *args, **kwargs)

Это создает класс, который при создании экземпляра будет возвращать обычный объект datetime.date, но который также может бытьизменено.

@mock.patch('datetime.date', FakeDate)
def test():
    from datetime import date
    FakeDate.today = classmethod(lambda cls: date(2010, 1, 1))
    return date.today()

test() # datetime.date(2010, 1, 1)

Answer 6

Я столкнулся с такой же ситуацией пару дней назад, и мое решение состояло в том, чтобы определить функцию в модуле для тестирования и просто высмеивать это:

def get_date_now():
    return datetime.datetime.now()

Сегодня я узнал о FreezeGun , и, кажется, он прекрасно освещает этот случай

from freezegun import freeze_time
import datetime
import unittest


@freeze_time("2012-01-14")
def test():
    assert datetime.datetime.now() == datetime.datetime(2012, 1, 14)

Answer 7

Вы можете использовать следующий подход, основанный на решении Daniel G.Преимущество этого метода состоит в том, что он не нарушает проверку типов с помощью isinstance(d, datetime.date).

import mock

def fixed_today(today):
    from datetime import date

    class FakeDateType(type):
        def __instancecheck__(self, instance):
            return isinstance(instance, date)

    class FakeDate(date):
        __metaclass__ = FakeDateType

        def __new__(cls, *args, **kwargs):
            return date.__new__(date, *args, **kwargs)

        @staticmethod
        def today():
            return today

    return mock.patch("datetime.date", FakeDate)

. По сути, мы заменяем класс datetime.date на основе C собственным подклассом python, который создает оригинальные экземпляры datetime.date и отвечаетisinstance() запрашивает в точности как нативный datetime.date.

Используйте его в качестве менеджера контекста в своих тестах:

with fixed_today(datetime.date(2013, 11, 22)):
    # run the code under test
    # note, that these type checks will not break when patch is active:
    assert isinstance(datetime.date.today(), datetime.date)

Аналогичный подход можно использовать для макетирования datetime.datetime.now() функции.

Answer 8

Самый простой способ для меня это сделать:

from unittest import patch, Mock

def test():
    datetime_mock = Mock(wraps=datetime)
    datetime_mock.now = Mock(return_value=datetime(1999, 1, 1)
    patch('target_module.datetime', new=datetime_mock).start()

ВНИМАНИЕ для этого решения: все функции с datetime module с target_module перестанут работать.

Answer 9

Вообще говоря, вы бы импортировали в модуль datetime или, возможно, datetime.date. Более эффективный способ насмешки над этим методом - установить его на модуль, который его импортирует. Пример:

a.py

from datetime import date

def my_method():
    return date.today()

Тогда для вашего теста сам фиктивный объект будет передан в качестве аргумента методу теста. Вы должны настроить макет с желаемым значением результата, а затем вызвать тестируемый метод. Тогда вы бы утверждали, что ваш метод сделал то, что вы хотите.

>>> import mock
>>> import a
>>> @mock.patch('a.date')
... def test_my_method(date_mock):
...     date_mock.today.return_value = mock.sentinel.today
...     result = a.my_method()
...     print result
...     date_mock.today.assert_called_once_with()
...     assert mock.sentinel.today == result
...
>>> test_my_method()
sentinel.today

Слово предупреждения. Безусловно, можно пойти за борт с издевательством. Когда вы это делаете, это делает ваши тесты более длинными, трудными для понимания и невозможными для обслуживания. Прежде чем издеваться над таким простым методом, как datetime.date.today, спросите себя, действительно ли вам нужен для насмешки. Если ваш тест является коротким и точным и работает нормально, без насмешки над функцией, вы можете просто смотреть на внутреннюю деталь кода, который вы тестируете, а не на объект, который вам нужно смоделировать.

Answer 10

Несколько решений обсуждаются в http://blog.xelnor.net/python-mocking-datetime/. В итоге:

Ложный объект - Простой и эффективный, но нарушающий проверку isinstance ():

target = datetime.datetime(2009, 1, 1)
with mock.patch.object(datetime, 'datetime', mock.Mock(wraps=datetime.datetime)) as patched:
    patched.now.return_value = target
    print(datetime.datetime.now())

Макет класса

import datetime
import mock

real_datetime_class = datetime.datetime

def mock_datetime_now(target, dt):
    class DatetimeSubclassMeta(type):
        @classmethod
        def __instancecheck__(mcs, obj):
            return isinstance(obj, real_datetime_class)

    class BaseMockedDatetime(real_datetime_class):
        @classmethod
        def now(cls, tz=None):
            return target.replace(tzinfo=tz)

        @classmethod
        def utcnow(cls):
            return target

    # Python2 & Python3 compatible metaclass
    MockedDatetime = DatetimeSubclassMeta('datetime', (BaseMockedDatetime,), {})

    return mock.patch.object(dt, 'datetime', MockedDatetime)

Использовать как:

with mock_datetime_now(target, datetime):
   ....