如何强制MagicMock复制函数签名？

Question

我想模拟一些不重要的测试对象函数(其他函数)- time.sleep()等等。

我可以用简单的模拟来代替它们，这是可行的。但是我希望它报告错误，如果它们是用不正确的参数调用的，没有命名的参数等等。

如果你的参数列表与给定的函数不同，有什么方法可以说模拟‘引发异常’吗？

我想要模拟的函数示例：

def mockme(arg1, arg2):
    pass

我想要的：

>>> m = mock.CallableMock(signature=mockme)
>>> m(1,2)
<MagicMock name='mockme()' id='140435553563920'>
>>> m(1,2,3)
TypeError: <MagicMock name='mockme()' id='140435553563920'> takes exactly 2 arguments (3 given)

知道怎么做这个吗？

Answer 1

看看unittest.mock.create_autospec，它应该能做你想做的事情

def some (a,b):
    pass

some_mock=mock.create_autospec(some)
some_mock(1)