如何使用 py.test 对 python 的 datetime.datetime.now 进行猴子补丁?

作者: 人在囧途开心一笑
来源: 51数据库
2022-10-20

问题描述

我需要测试使用 datetime.datetime.now() 的函数.最简单的方法是什么?

I need to test functions which uses datetime.datetime.now(). What is the easiest way to do this?

推荐答案

你需要monkeypatch datetime.now 函数.在下面的示例中，我正在创建可以稍后在其他测试中重复使用的夹具:

You need to monkeypatch datetime.now function. In example below, I'm creating fixture which I can re-use later in other tests:

import datetime
import pytest

FAKE_TIME = datetime.datetime(2020, 12, 25, 17, 5, 55)

@pytest.fixture
def patch_datetime_now(monkeypatch):

    class mydatetime:
        @classmethod
        def now(cls):
            return FAKE_TIME

    monkeypatch.setattr(datetime, 'datetime', mydatetime)


def test_patch_datetime(patch_datetime_now):
    assert datetime.datetime.now() == FAKE_TIME

推荐阅读