pytest - 测试期间假时间变化

Question

我有以下代码来在测试期间设置假时间。

我想在测试期间更改时间。也就是说，测试应该从 9:00 开始，然后像 10:00 一样继续。

from __future__ import annotations

import datetime
import logging

import pytest

LOGGER = logging.getLogger(__name__)


@pytest.fixture(params=[datetime.datetime(2020, 12, 25, 17, 5, 55)])
def patch_datetime_now(request, monkeypatch):
    class mydatetime(datetime.datetime):
        @classmethod
        def now(cls):
            return request.param

    class mydate(datetime.date):
        @classmethod
        def today(cls):
            return request.param.date()

    monkeypatch.setattr(datetime, "datetime", mydatetime)
    monkeypatch.setattr(datetime, "date", mydate)


@pytest.mark.usefixtures("patch_datetime_now")
@pytest.mark.parametrize(
    "patch_datetime_now", [(datetime.datetime(2020, 12, 9, 11, 22, 00))], indirect=True
)
def test_update_data():
    fakeTime = datetime.datetime.now()
    # Do some stuff

    # Change the fake time

    # Do some other stuff

如何更改考试期间的假时间？ “datetime”在测试的代码中使用，因此不是更改“fakeTime”变量内容，而是更改日期时间模型返回的时间。

也许我需要完全改变模拟方法，我只是分享我当前的代码。

Answer 1

按照这个答案@MrBeanBremen提供的另一个问题 我像这样更新了我的代码：

from __future__ import annotations

import datetime
import logging

import pytest

LOGGER = logging.getLogger(__name__)


@pytest.fixture(params=[datetime.datetime(2020, 12, 25, 17, 5, 55)])
def patch_datetime_now(request, monkeypatch):
    def _delta(timedelta=None, **kwargs):
        """ Moves time fwd/bwd by the delta"""
        from datetime import timedelta as td
        if not timedelta:
            timedelta = td(**kwargs)
        request.param += timedelta

    class mydatetime(datetime.datetime):
        @classmethod
        def now(cls):
            return request.param

        @classmethod
        def delta(cls,*args,**kwargs):
            _delta(*args,**kwargs)

    class mydate(datetime.date):
        @classmethod
        def today(cls):
            return request.param.date()

        @classmethod
        def delta(cls,*args,**kwargs):
            _delta(*args,**kwargs)

    monkeypatch.setattr(datetime, "datetime", mydatetime)
    monkeypatch.setattr(datetime, "date", mydate)


# Test & Example using the fixture above
@pytest.mark.usefixtures("patch_datetime_now")
@pytest.mark.parametrize(
    "patch_datetime_now", [(datetime.datetime(2020, 12, 9, 11, 22, 00))], indirect=True
)
def test_update_data():
    # The test starts with the fake time set in the parameter above
    fakeTime = datetime.datetime.now()
    # Assert (verify) that the fake time is used
    assert fakeTime == datetime.datetime(2020, 12, 9, 11, 22, 00)
    # Move the fake time 1 hour and 10 seconds in the future
    datetime.datetime.delta(hours=1,seconds=10)
    # Get the new fake time
    fakeTime = datetime.datetime.now()
    # Assert (verify) that the fake time was updated as expected
    assert fakeTime == datetime.datetime(2020, 12, 9, 12, 22, 10)