def test__run(self): myclass = MyClass() myclass.A = MagicMock() myclass.B = MagicMock() myclass._run() myclass.A.assert_called_with({}) myclass.B.assert_called_with({})
def test_A(self, mocked_get, mocked_print): Response = namedtuple('Response', ['data']) response = Response(data={'has_value': True}) mocked_get.side_effect = [response, 'new response'] myclass = MyClass() data = {} myclass.A(data) mocked_get.assert_called_with('some_url', None) self.assertEqual(mocked_get.call_count, 2) mocked_print.assert_called_with('new response')