コード例 #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
ファイル: test_reliability.py プロジェクト: pombreda/gofer
    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
ファイル: test_reliability.py プロジェクト: swipswaps/gofer
    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
ファイル: test_reliability.py プロジェクト: jortel/gofer
    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
ファイル: model.py プロジェクト: darinlively/gofer
def reliable(fn):
    return reliability.endpoint(reliability.reliable(fn))