Exemple #1
0
    def test_close(self):
        connection = Mock()
        sender = Mock()
        receiver = Mock()
        method = Method('', '', {})
        method.connection = connection
        method.sender = sender
        method.receiver = receiver

        # test
        method.close()

        # validation
        receiver.close.assert_called_once_with()
        sender.close.assert_called_once_with()
Exemple #2
0
    def test_close(self):
        connection = Mock()
        sender = Mock()
        receiver = Mock()
        method = Method("", "", {})
        method.connection = connection
        method.sender = sender
        method.receiver = receiver

        sender.close.side_effect = ValueError
        receiver.close.side_effect = ValueError

        # test
        method.close()

        # validation
        receiver.close.assert_called_once_with()
        sender.close.assert_called_once_with()