如何使用 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(...):
    ...