如何为一小时后函数返回时间编写单元测试

Question

我有以下函数，它在当前时间添加给定的小时数并返回新时间

func getExpiryTime(hour int) *string {

   const layout = "2006-01-02T15:04:05Z"

   expiryTime := time.Now().Local().Add(time.Hour *time.Duration(hour))

   return aws.String(expiryTime.Format(layout))

}

为此功能编写单元测试的最佳方法是什么？

Answer 1

您可以尝试模拟时间提供者并在模拟中设置 time.Now() 函数以返回您控制的预设值。在生产中，您可以将其转发给实际时间的调用。