如何仅Mock datetime.date.today()而不影响date类的其他方法？

这个坑我之前踩过好几次！直接patch整个date类或者硬改它的today方法确实容易出问题，咱们先理清楚你遇到的问题根源，再一步步解决。

问题出在哪？ #

1. patch整个date类导致isinstance失效：你第三个测试里patch("mock_experiment.main.date")把main模块里的date替换成了MagicMock对象，这时候isinstance(dt, date)里的date已经不是真实的datetime.date类型了，自然会抛出类型错误。
2. 直接patchdate.today失败：datetime.date是Python内置的不可变类型，它的类属性（比如today）不允许被直接修改，所以直接patch这个方法会报错。

解决方案1：用unittest.mock的wraps参数保留原类功能 #

我们可以让Mock对象包装真实的date类，只替换today方法的返回值，其他方法依然调用原类的实现。修改你的测试代码如下：

def test_with_patch_on_date():
    with patch("mock_experiment.main.date", wraps=date) as patch_date:
        # 只mocktoday方法的返回值
        patch_date.today.return_value = date(2025, 1, 2)
        # 测试传入date对象的情况
        assert main.date_distance_from_today(date(2025, 1, 1)) == timedelta(1)
        # 顺便测试传入字符串的情况，确保isinstance和strptime都正常工作
        assert main.date_distance_from_today("2025-01-01") == timedelta(1)

为什么这样可行？

• wraps=date让Mock对象在遇到未被显式mock的方法时，自动转发给真实的date类执行，所以isinstance、strptime这些方法都能正常工作。
• 我们只需要显式指定patch_date.today.return_value，就能覆盖today方法的返回值，完全不影响其他功能。

解决方案2：用freezegun库（更简洁的时间冻结工具） #

如果你的测试里经常需要Mock时间，推荐直接用freezegun这个第三方库，它专门用来冻结系统时间，不需要手动处理Mock细节，代码会更简洁：

首先安装库：

pip install freezegun

然后修改测试代码：

from freezegun import freeze_time

def test_with_freezegun():
    # 直接冻结到指定日期
    with freeze_time("2025-01-02"):
        assert main.date_distance_from_today(date(2025, 1, 1)) == timedelta(1)
        assert main.date_distance_from_today("2025-01-01") == timedelta(1)

这个方法更省心，它会全局冻结Python的时间相关方法，包括date.today()、datetime.now()等，而且完全不需要担心影响其他date类的方法。

验证结果 #

修改后再运行测试，三个测试应该都能正常通过，不管哪天跑测试结果都是稳定的，不会依赖当前系统日期。

备注：内容来源于stack exchange，提问作者scotscotmcc