python - 模拟 timedelta.total_seconds()

Question

有没有办法从下面的代码中模拟total_seconds()？

start = datetime.now()
...
end = datetime.now()
diff = (end - start).total_seconds()

我已经尝试过，但出现此错误

TypeError: unorderable types: MagicMock() > int()

Answer 1

这是一个可靠的例子，但它可以给出一些提示:

from datetime import datetime
import unittest


def need_test():
    start = datetime.datetime.now()
    end = datetime.datetime.now()
    diff = (end - start).total_seconds()
    return diff


class SimpleTestCase(unittest.TestCase):

    def setUp(self):
        datetime_mock = mock.patch(__name__ + ".datetime")
        self.datetime_mock = datetime_mock.start()

    def test_need_test(self):
        self.datetime_mock.datetime.now().__sub__().total_seconds.return_value = 123
        self.assertEqual(need_test(), 123)

首先，应在 mock.patch.object(module.where.is.your.function, "datetime"上更改 mock.patch(__name__ + ".datetime") )。之后，self.datetime_mock.datetime.now()(也许)可以替换为self.datetime_mock.now()

此示例不是进行此类测试的正确方法，但它可以解决您的问题。