Python单元测试中Mock时间戳变量的实现方法问询

这里有几个实用的方案，帮你在单元测试中精准控制timestamp变量，完成断言验证：

方案1：Mock datetime.today() 控制日期生成 #

这是最直接的方式，通过控制datetime.today()的返回值，让函数生成的timestamp完全符合我们的预期：

from unittest.mock import patch, Mock
from datetime import datetime, timedelta
import your_module  # 替换成包含func函数的实际模块名

def test_func_with_mocked_datetime():
    # 定义一个固定的测试日期（模拟"昨天"的时间）
    test_yesterday = datetime(2024, 5, 20, 14, 30, 0)
    # 计算对应的预期timestamp字符串（和原函数逻辑一致：取时间戳整数部分+000）
    expected_timestamp = str(int(test_yesterday.timestamp())) + "000"
    
    # Mock datetime.today()，让它返回"今天"（即test_yesterday + 1天）
    with patch('your_module.datetime.today', return_value=test_yesterday + timedelta(days=1)):
        # Mock DataAccess类，避免实际调用数据库
        mock_data_access = Mock()
        with patch('your_module.DataAccess', return_value=mock_data_access):
            # 执行被测试函数
            your_module.func()
            
            # 验证get方法的调用参数是否符合预期
            mock_data_access.get.assert_called_once_with(
                sort_key=f"CampaignRecipientsReport#{expected_timestamp}"
            )

这个方案的优势是完全复刻原函数的timestamp生成逻辑，不会因为逻辑变动导致测试失效。

方案2：使用freezegun冻结系统时间 #

如果你不想手动处理datetime的Mock，可以用freezegun这个第三方库一键冻结系统时间，轻松控制"昨天"的日期：

首先安装依赖：

pip install freezegun

然后编写测试代码：

from freezegun import freeze_time
from unittest.mock import Mock
from datetime import datetime
import your_module

def test_func_with_frozen_time():
    # 把系统时间冻结到2024-05-21，此时"昨天"就是2024-05-20
    with freeze_time("2024-05-21"):
        # 计算预期的timestamp
        test_yesterday = datetime(2024, 5, 20)
        expected_timestamp = str(int(test_yesterday.timestamp())) + "000"
        
        # Mock DataAccess类
        mock_data_access = Mock()
        with patch('your_module.DataAccess', return_value=mock_data_access):
            your_module.func()
            
            # 验证调用参数
            mock_data_access.get.assert_called_once_with(
                sort_key=f"CampaignRecipientsReport#{expected_timestamp}"
            )

freezegun会自动接管所有时间相关的调用，用起来非常省心，适合需要频繁模拟不同时间场景的测试。

方案3：精准Mock时间戳生成逻辑 #

如果只想针对timestamp的生成做Mock，而不改动整个datetime的行为，可以直接Mockdate.timestamp()方法的返回值：

from unittest.mock import patch, Mock
import your_module

def test_func_mock_timestamp_directly():
    # 定义我们想要的最终timestamp后缀
    expected_timestamp = "1716210600000"
    # 提取原函数中需要的时间戳整数部分（去掉最后三个0）
    mock_timestamp_value = int(expected_timestamp[:-3])
    
    # 创建一个Mock的date对象，让它的timestamp()返回我们指定的数值
    mock_date = Mock()
    mock_date.timestamp.return_value = mock_timestamp_value
    
    # Mock datetime.today() - timedelta(days=1) 的结果为这个mock_date
    with patch('your_module.datetime') as mock_datetime:
        mock_today = Mock()
        # 让"今天"减去1天返回我们的mock_date
        mock_today.__sub__.return_value = mock_date
        mock_datetime.today.return_value = mock_today
        
        # Mock DataAccess类
        mock_data_access = Mock()
        with patch('your_module.DataAccess', return_value=mock_data_access):
            your_module.func()
            
            # 完成断言
            mock_data_access.get.assert_called_once_with(
                sort_key=f"CampaignRecipientsReport#{expected_timestamp}"
            )

这个方案的粒度更细，适合不需要改动整个日期，只需要控制timestamp输出的场景。

内容的提问来源于stack exchange，提问作者Giuseppe Fantoni