def test_endpoint(self, messenger): fn = Mock() url = Mock() args = (messenger.return_value, ) # test wrapped = endpoint(fn) wrapped(url) # validation messenger.assert_called_once_with(url) messenger.return_value.open.assert_called_once_with() fn.assert_called_once_with(*args) messenger.return_value.close.assert_called_once_with()
def test_endpoint(self, messenger): fn = Mock() url = Mock() args = (messenger.return_value,) # test wrapped = endpoint(fn) wrapped(url) # validation messenger.assert_called_once_with(url) messenger.return_value.open.assert_called_once_with() fn.assert_called_once_with(*args) messenger.return_value.close.assert_called_once_with()
def test_endpoint(self, messenger): fn = Mock() url = Mock() args = (messenger.return_value, ) def _enter(): messenger.return_value.open() return messenger.return_value def _exit(*unused): messenger.return_value.close() messenger.return_value.__enter__ = Mock(side_effect=_enter) messenger.return_value.__exit__ = Mock(side_effect=_exit) # test wrapped = endpoint(fn) wrapped(url) # validation messenger.assert_called_once_with(url) messenger.return_value.open.assert_called_once_with() fn.assert_called_once_with(*args) messenger.return_value.close.assert_called_once_with()
def test_endpoint(self, messenger): fn = Mock() url = Mock() args = (messenger.return_value,) def _enter(): messenger.return_value.open() return messenger.return_value def _exit(*unused): messenger.return_value.close() messenger.return_value.__enter__ = Mock(side_effect=_enter) messenger.return_value.__exit__ = Mock(side_effect=_exit) # test wrapped = endpoint(fn) wrapped(url) # validation messenger.assert_called_once_with(url) messenger.return_value.open.assert_called_once_with() fn.assert_called_once_with(*args) messenger.return_value.close.assert_called_once_with()
def reliable(fn): return reliability.endpoint(reliability.reliable(fn))