如何使用 unittests mock.patch 模拟两个函数

Question

我需要在我的 API post 方法中模拟两个函数。 只模拟一个函数效果很好，但是当我需要模拟两个函数时，我不知道该怎么做？

@unittest.mock.patch('path.to.first.function.thefirstfunction')
def test_connector_post(mock_thefirstfunction):
... 

有任何想法吗？

Answer 1

堆叠装饰器！

@unittest.mock.patch("functionA")
@unittest.mock.patch("functionB")
def test_foo(...):
    ...