模拟时间戳单元测试 Python

Question

我有那个代码：

from datetime import datetime, timedelta
    def func():
   
        date = datetime.today() - timedelta(days=1)
        timestamp = (str(date.timestamp()).split(".")[0]) + "000"
    
        return DataAccess().get(
            sort_key="CampaignRecipientsReport#" + timestamp,
        )

现在我需要模拟一个变量时间戳以在我的单元测试中使用它

mock_data_access.return_value.get.assert_called_once_with(
      sort_key="CampaignRecipientsReport#" + timestamp,
)

Answer 1

不，您不需要模拟函数的局部变量timestamp 。 相反，您需要重新设计您的 function。 你的问题是，你的 function 是状态相关的，因此有副作用。

即你应该能够传递时间戳。 如果您经常需要当前日期时间，则可以将其默认为该值，前提是未指定任何值。 这使得 function 更加灵活，状态依赖性和可测试性更低。

from datetime import datetime, timedelta
from typing import Optional


def func(timestamp: Optional[datetime] = None):
   
    if timestamp is None:
        timestamp = datetime.today() - timedelta(days=1)

    timestamp = (str(timestamp.timestamp()).split(".")[0]) + "000"
    
    return DataAccess().get(
        sort_key="CampaignRecipientsReport#" + timestamp
    )