python - 使pytest对数据库使用固定日期

Question

我正在使用 pytest 来测试一些数学函数。我希望数据库认为这是过去的日期。

例如，我会将数据库设置为认为现在是 2019 年 8 月 1 日。

有没有办法用 pytest 配置它？

目前我的数据库是这样设置的:

@pytest.fixture(autouse=True)
def autouse_db(db):
    pass

Answer 1

使用freezegun用于 monkeypatching datetime 函数。示例:

import datetime
import pytest
from freezegun import freeze_time


@pytest.fixture
def fast_forward_to_2038():
    # use me to test the year 2038 problem
    with freeze_time('2038-01-19 03:14:08'):
        # year 2038 problem should be reproducible now
        yield


@pytest.mark.usefixtures('fast_forward_to_2038')
def test_2038_problem():
    assert datetime.datetime.now() == datetime.datetime(2038, 1, 19, 03, 14, 08)
    ...

查看包中的更多示例 readme .