Патч __call__ функции

[РЕДАКТИРОВАТЬ]

Возможно, самая интересная часть этого вопроса - Почему я не могу исправить somefunction.__call__?

Поскольку функция не использует код __call__, а __call__ (объект-метод-обертка) использует код функции.

Я не нахожу никакой документации об этом, но я могу доказать это (Python2.7):

>>> def f():
...     return "f"
... 
>>> def g():
...     return "g"
... 
>>> f
<function f at 0x7f1576381848>
>>> f.__call__
<method-wrapper '__call__' of function object at 0x7f1576381848>
>>> g
<function g at 0x7f15763817d0>
>>> g.__call__
<method-wrapper '__call__' of function object at 0x7f15763817d0>

Замените код f на код g:

>>> f.func_code = g.func_code
>>> f()
'g'
>>> f.__call__()
'g'

Конечно, ссылки f и f.__call__ не изменены:

>>> f
<function f at 0x7f1576381848>
>>> f.__call__
<method-wrapper '__call__' of function object at 0x7f1576381848>

Восстановите исходную реализацию и скопируйте ссылки __call__:

>>> def f():
...     return "f"
... 
>>> f()
'f'
>>> f.__call__ = g.__call__
>>> f()
'f'
>>> f.__call__()
'g'

Это не влияет на функцию f. Примечание. В Python 3 вы должны использовать __code__ вместо func_code.

Я надеюсь, что кто-то может указать мне на документацию, которая объясняет это поведение.

У вас есть способ обойти это: в utils вы можете определить

class Utcnow(object):
    def __call__(self):
        return datetime.datetime.utcnow()


utcnow = Utcnow()

И теперь ваш патч может работать как шарм.

Следуйте первоначальному ответу, который я считаю даже лучшим способом реализовать ваши тесты.

У меня есть собственное золотое правило : никогда не исправляйте защищенные методы . В этом случае все немного гладче, потому что защищенный метод был введен только для тестирования, но я не могу понять, почему.

Настоящая проблема здесь в том, что вы не можете установить патч datetime.datetime.utcnow напрямую (это расширение C, как вы написали в комментарии выше). Что вы можете сделать, это исправить datetime, обернув стандартное поведение и переопределить функцию utcnow:

>>> with mock.patch("datetime.datetime", mock.Mock(wraps=datetime.datetime, utcnow=mock.Mock(return_value=3))):
...  print(datetime.datetime.utcnow())
... 
3

Хорошо, это не совсем понятно и аккуратно, но вы можете представить свою собственную функцию, как

def mock_utcnow(return_value):
    return mock.Mock(wraps=datetime.datetime, 
                     utcnow=mock.Mock(return_value=return_value)):

И сейчас

mock.patch("datetime.datetime", mock_utcnow(***))

Делать то, что вам нужно, без какого-либо другого слоя и для каждого вида импорта.

Другим решением может быть импорт datetime в utils и исправление ***.utils.datetime; это может дать вам некоторую свободу изменять ссылочную реализацию datetime без изменения ваших тестов (в этом случае позаботьтесь также об изменении аргумента mock_utcnow() wraps).

Как прокомментировал вопрос, поскольку datetime.datetime написано на C, Mock не может заменить атрибуты класса (см. Насмешка от datetime.today Неда Батчелдера). Вместо этого вы можете использовать freezegun.

$ pip install freezegun

Вот пример:

import datetime

from freezegun import freeze_time

def my_now():
    return datetime.datetime.utcnow()


@freeze_time('2000-01-01 12:00:01')
def test_freezegun():
    assert my_now() == datetime.datetime(2000, 1, 1, 12, 00, 1)

Как вы упоминаете, альтернативой является отслеживание импорта каждого модуля datetime и их исправление. По сути, это то, что делает freezegun . Он принимает объект, имитирующий datetime, перебирает sys.modules, чтобы найти, куда datetime был импортирован, и заменяет каждый экземпляр. Я предполагаю, что это спорно , можно ли сделать это элегантно в одной функции .