def test_update_all(self, _impl, report): impl = Mock() impl.update.return_value = dict(failed=[]) _impl.return_value = impl cfg = {} conduit = Mock() options = {'all': True} units = [] # test handler = PackageHandler(cfg) details = handler.update(conduit, units, options) # validation impl.update.assert_called_once_with([]) report.return_value.set_succeeded.assert_called_once_with(impl.update.return_value) self.assertEqual(report.return_value, details)
def test_update_failed(self, _impl, report): impl = Mock() impl.update.return_value = dict(failed=[Mock(), Mock()]) _impl.return_value = impl cfg = {} conduit = Mock() options = {} units = [ {'name': 'dog'}, {'name': 'cat'} ] # test handler = PackageHandler(cfg) details = handler.update(conduit, units, options) # validation report.return_value.set_failed.assert_called_once_with(impl.update.return_value) self.assertEqual(report.return_value, details)