mirror of
https://github.com/python/cpython.git
synced 2025-07-24 11:44:31 +00:00
ff018e406c
* Remove unused SSLProtocol._closing attribute * test_sslproto: skip test if ssl module is missing * Python issue #23208: Don't use the traceback of the current handle if we already know the traceback of the source. The handle may be more revelant, but having 3 tracebacks (handle, source, exception) becomes more difficult to read. The handle may be preferred later but it requires more work to make this choice.
54 lines
1.6 KiB
Python
54 lines
1.6 KiB
Python
"""Tests for asyncio/sslproto.py."""
|
|
|
|
import unittest
|
|
from unittest import mock
|
|
try:
|
|
import ssl
|
|
except ImportError:
|
|
ssl = None
|
|
|
|
import asyncio
|
|
from asyncio import sslproto
|
|
from asyncio import test_utils
|
|
|
|
|
|
class SslProtoHandshakeTests(test_utils.TestCase):
|
|
|
|
def setUp(self):
|
|
self.loop = asyncio.new_event_loop()
|
|
self.set_event_loop(self.loop)
|
|
|
|
@unittest.skipIf(ssl is None, 'No ssl module')
|
|
def test_cancel_handshake(self):
|
|
# Python issue #23197: cancelling an handshake must not raise an
|
|
# exception or log an error, even if the handshake failed
|
|
sslcontext = test_utils.dummy_ssl_context()
|
|
app_proto = asyncio.Protocol()
|
|
waiter = asyncio.Future(loop=self.loop)
|
|
ssl_proto = sslproto.SSLProtocol(self.loop, app_proto, sslcontext,
|
|
waiter)
|
|
handshake_fut = asyncio.Future(loop=self.loop)
|
|
|
|
def do_handshake(callback):
|
|
exc = Exception()
|
|
callback(exc)
|
|
handshake_fut.set_result(None)
|
|
return []
|
|
|
|
waiter.cancel()
|
|
transport = mock.Mock()
|
|
sslpipe = mock.Mock()
|
|
sslpipe.shutdown.return_value = b''
|
|
sslpipe.do_handshake.side_effect = do_handshake
|
|
with mock.patch('asyncio.sslproto._SSLPipe', return_value=sslpipe):
|
|
ssl_proto.connection_made(transport)
|
|
|
|
with test_utils.disable_logger():
|
|
self.loop.run_until_complete(handshake_fut)
|
|
|
|
# Close the transport
|
|
ssl_proto._app_transport.close()
|
|
|
|
|
|
if __name__ == '__main__':
|
|
unittest.main()
|