python - 使用模拟测试构造函数

Question

我需要测试我的类的构造函数调用了一些方法

class ProductionClass:
   def __init__(self):
       self.something(1, 2, 3)

   def method(self):
       self.something(1, 2, 3)

   def something(self, a, b, c):
       pass 

这个类来自'unittest.mock - 入门'。正如那里写的那样，我可以确保“方法”称为“某事”，如下所示。

real = ProductionClass()
real.something = MagicMock()
real.method()
real.something.assert_called_once_with(1, 2, 3)

但是如何对构造函数进行相同的测试？

Answer 1

您可以使用补丁(查看文档 https://docs.python.org/dev/library/unittest.mock.html )并断言在创建对象的新实例后， something方法被调用一次并使用所需的参数调用。例如，在您的示例中，它将是这样的:

from unittest.mock import MagicMock, patch
from your_module import ProductionClass

@patch('your_module.ProductionClass.something')
def test_constructor(something_mock):
    real = ProductionClass()
    assert something_mock.call_count == 1
    assert something_mock.called_with(1,2,3)