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()
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()
def test_send(self):
request = Mock()
method = Method('', '', {})
method.sender = Mock()
method.send(request)
method.sender.send.assert_called_with(request)
def test_send(self):
request = Mock()
method = Method("", "", {})
method.sender = Mock()
method.send(request)
method.sender.send.assert_called_with(request)
