Тестирование конструктора с использованием макета

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)