ユニットテスト時に datetime.now() を置き換える

testutil.py

# -*- coding: utf-8 -*-
from datetime import datetime


@contextlib.contextmanager
def mock_datetime_now(module_path, return_value):
    """
    適当なモジュールの datetime.now() を mock し、適当な日付を返すようにする。

    >>> from datetime import datetime
    >>> with mock_datetime_now('some_app.some_module.datetime', datetime(2015, 1, 1)):
    ...     datetime.now()
    datetime(2015, 1, 1, 0, 0, 0)

    のように利用する。
    """
    class DummyDatetime(datetime):
        @classmethod
        def now(cls):
            return return_value

    # dateime を持つモジュールを取得する
    module_names = module_path.rstrip('.datetime').split('.')
    module = __import__('.'.join(module_names))
    for name in module_names:
        module = getattr(module, name)

    # 置き換えて yield
    module.datetime = DummyDatetime
    yield

    # 置き換えたのを戻す
    module.datetime = datetime

https://gist.github.com/podhmo/066af0861f435c5788a0
コメントしました

最終的にこうした。 : see このコメント

class DummyDatetime(object):
    """
    適当なモジュールの datetime.now() を mock し、適当な日付を返すようにする。

    >>> import mock
    >>> from datetime import datetime
    >>> with mock.patch('app.some_app.some_module.datetime',
    ...                 new=DummyDatetime(datetime(2015, 1, 1))):
    ...     datetime.now()
    datetime(2015, 1, 1, 0, 0, 0)

    のように利用する。
    """
    # XXX: メソッドがないと言われたら、随時テストに合うように追加願います。
    def __init__(self, dummy_current):
        self.now = lambda: dummy_current