def test_close(self): url = Mock() messenger = Endpoint(url) messenger.connection = Mock() messenger.channel = Mock() messenger.close() messenger.channel.close.assert_called_once_with()
def test_close_failed(self): url = Mock() messenger = Endpoint(url) messenger.connection = Mock() messenger.channel = Mock() messenger.channel.close.side_effect = ValueError messenger.close() messenger.channel.close.assert_called_once_with()