python - 如何在一个文件中模拟日期时间？

Question

我有两个文件，其中使用了日期时间。有没有一种方法可以仅在一个文件中模拟它，而不是在另一个文件中模拟它？以下是我看到奇怪行为的示例。

文件:test_file_one.py

import datetime as dt
def print_datetime():
    print 'First: {}'.format(dt.datetime.utcnow())

文件:test_file_two.py

import datetime as dt
def print_datetime():
    print 'Second: {}'.format(dt.datetime.utcnow())

文件:main.py

import test_file_one as first
import test_file_two as second
import mock
from datetime import datetime, timedelta

@mock.patch('test_file_one.dt.datetime')
def main(mock_datetime):
    mock_datetime.utcnow.return_value = datetime.utcnow() + timedelta(days=1)
    first.print_datetime()
    second.print_datetime()

if __name__ == '__main__':
    main()

输出

First: 2018-06-12 08:12:43.838243
Second: 2018-06-12 08:12:43.838243

如您所见，两者都返回被 mock 的相同日期时间。

1. 为什么两人都被 mock ？
2. 如何将模拟限制为仅一个文件？

Answer 1

您可以将任意数量的结果添加到 side_effect

mock_datetime.utcnow.side_effect = [datetime.utcnow() + timedelta(days=1), datetime.utcnow() + timedelta(days=2)]