我有以下代码
@ddt
@patch('mymodule.myclass', MagicMock)
class MyTest():
@data([val1, val2])
@unpack
def test_run(self, val1, val2):
ClassA().run(val1, val2)
如何在mymodule.myclass
内声明某些方法?
答案 0 :(得分:1)
当你在类级别修补它时,你会在测试方法中获得对mock作为参数的引用,例如:
@ddt
@patch('mymodule.myclass', MagicMock)
class MyTest():
@data([val1, val2])
@unpack
def test_run(self, val1, val2, my_class_mock):
# use my_class_mock
您可以将此模拟用于多种用途,但在这种情况下,您仍然需要在测试方法中修补您的类以进行断言。