示例#1
0
    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()
示例#2
0
    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()
示例#3
0
    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()
示例#4
0
    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()
示例#5
0
文件: model.py 项目: swipswaps/gofer
def reliable(fn):
    return reliability.endpoint(reliability.reliable(fn))
示例#6
0
def reliable(fn):
    return reliability.endpoint(reliability.reliable(fn))