File: //usr/local/lib/python3.7/test/test_asyncio/__pycache__/test_sslproto.cpython-37.opt-2.pyc
B
��g�f � @ sN d dl Z d dlZd dlZd dlZd dlZd dlmZ y d dlZW n ek
rX dZY nX d dlZd dlm Z d dlm
Z
d dlm Z d dlm Z d dl
mZ d dl
mZ e�edkd �G d
d � d ej��ZG d d
� d
ej�Ze�edkd �G dd� deej��Ze�edkd �e�eed�d�G dd� deej���Zedk�rJe� � dS )� N)�mock)�log)� protocols)�sslproto)�tasks)�utils)�
functionalz
No ssl modulec s� e Zd Z� fdd�Zddd�dd�Zdd�dd �Zd
d � Zd d
� Zdd� Zdd� Z dd� Z
dd� Z dd� Z dd� Z
dd� Zd d � Z� ZS ) �SslProtoHandshakeTestsc s$ t � �� t�� | _| �| j� d S )N)�super�setUp�asyncioZnew_event_loop�loopZset_event_loop)�self)� __class__� �;/usr/local/lib/python3.7/test/test_asyncio/test_sslproto.pyr s
z SslProtoHandshakeTests.setUpN)�waiter�protoc C sR t �� }|d krt�� }tj| j|||dd�}| �|j� � |� | �
|jj � |S )Ng�������?)�ssl_handshake_timeout) �
test_utils�dummy_ssl_contextr �Protocolr � SSLProtocolr
�assertIs�_app_transport� get_protocolZ
addCleanup�close)r r r �
sslcontext� ssl_protor r r � ssl_protocol s z#SslProtoHandshakeTests.ssl_protocol)� do_handshakec C s^ t �� }t �� }d|j_|r&||j_ndd� }||j_t jd|d�� |�|� W d Q R X |S )N� c S s g S )Nr )�callbackr r r �mock_handshake0 s z>SslProtoHandshakeTests.connection_made.<locals>.mock_handshakezasyncio.sslproto._SSLPipe)� return_value)r �Mock�shutdownr$ r Z side_effectZpatch�connection_made)r r r � transportZsslpiper# r r r r' ) s
z&SslProtoHandshakeTests.connection_madec C sJ t �� }t�� }t�� }| �td�� tj| j|||dd� W d Q R X d S )Nza positive numberr )r ) r r r r% �assertRaisesRegex�
ValueErrorr r r
)r r � app_protor r r r �test_handshake_timeout_zero7 s z2SslProtoHandshakeTests.test_handshake_timeout_zeroc C sJ t �� }t�� }t�� }| �td�� tj| j|||dd� W d Q R X d S )Nza positive numberi����)r ) r r r r% r) r* r r r
)r r r+ r r r r � test_handshake_timeout_negative? s z6SslProtoHandshakeTests.test_handshake_timeout_negativec C sL t j| jd�}| j|d�}| �|� |�� t�| j� | �|� � t
� d S )N)r
)r ) r �Futurer
r r' � eof_receivedr � run_briefly�assertIsInstance� exception�ConnectionResetError)r r r r r r �test_eof_received_waiterG s
z/SslProtoHandshakeTests.test_eof_received_waiterc C sT t j| jd�}| j|d�}tj�� }tj�tj � z|�
d � W d tj�|� X d S )N)r
)r ) r r. r
r r ZloggerZgetEffectiveLevelZsetLevel�loggingZFATALZ _fatal_error)r r r Z log_levelr r r � test_fatal_error_no_name_errorO s
z5SslProtoHandshakeTests.test_fatal_error_no_name_errorc C sN t j| jd�}| j|d�}| �|� |�t� t�| j� | � |�
� t� d S )N)r
)r ) r r. r
r r' �connection_lost�ConnectionAbortedErrorr r0 r1 r2 )r r r r r r �test_connection_lost^ s
z+SslProtoHandshakeTests.test_connection_lostc C sL t j| jd�}| j|d�}| �|�}t�| j� |j�� | � |j
j � d S )N)r
)r ) r r. r
r r' r r0 r r �
assertTrue�abortZcalled)r r r r( r r r �test_close_during_handshakeh s
z2SslProtoHandshakeTests.test_close_during_handshakec C s| t j| jd�}| j|d�}| �|�d�� t� }| �|�d|�|� | �|� | � |�d�� |�
d � | �|�d�� d S )N)r
)r �socket) r r. r
r � assertIsNoneZ_get_extra_info�objectr r' ZassertIsNotNoner7 )r r r �defaultr r r �(test_get_extra_info_on_closed_connections s
z?SslProtoHandshakeTests.test_get_extra_info_on_closed_connectionc C sR t j| jd�}| j|d�}t �� }|j�|� | �|j�� |� | �|j |� d S )N)r
)r )
r r. r
r r r � set_protocolr r Z
_app_protocol)r r r Z
new_app_protor r r �test_set_new_app_protocol~ s z0SslProtoHandshakeTests.test_set_new_app_protocolc C s4 | � � }| �|� |j}|�� | �|�d�� d S )Ns data)r r' r r r> �
data_received)r r �transpr r r � test_data_received_after_closing� s
z7SslProtoHandshakeTests.test_data_received_after_closingc C s4 | � � }| �|� |j}|�� | �|�d�� d S )Ns data)r r' r r r> �write)r r rE r r r �test_write_after_closing� s
z/SslProtoHandshakeTests.test_write_after_closing)�__name__�
__module__� __qualname__r r r' r, r- r4 r6 r9 r< rA rC rF rH �
__classcell__r r )r r r s
r c @ st e Zd ZdZdZdd� Zdd� Zdd� Zd d
� Zd d � Z d
d� Z
dd� Z dd� Z dd� Z
dd� Zdd� Zdd� ZdS ) � BaseStartTLSi � �< c C s t �d S )N)�NotImplementedError)r r r r �new_loop� s zBaseStartTLS.new_loopc
C s G dd� dt j�}x�dD ]�}|d|�}t�|d� | �|jd� |d|�}t�|d� | �|jd� |d|�}t�|d� | �|jd� |d|�}t�|d� | �|jd� |d |�}t�|d� | �|jd� |d
|�}| �td �� t�|d� W d Q R X qW d S ) Nc @ s$ e Zd Zdd� Zdd� Zdd� ZdS )z.BaseStartTLS.test_buf_feed_data.<locals>.Protoc S s& t |�| _t| j�| _d| _|| _d S )Nr! )� bytearray�buf�
memoryview�mv�data�usemv)r �bufsizerV r r r �__init__� s
z7BaseStartTLS.test_buf_feed_data.<locals>.Proto.__init__c S s | j r | jS | jS d S )N)rV rT rR )r �sizehintr r r �
get_buffer� s z9BaseStartTLS.test_buf_feed_data.<locals>.Proto.get_bufferc S s< | j r | j| jd |� 7 _n| j| jd |� 7 _d S )N)rV rU rT rR )r �nsizer r r �buffer_updated� s z=BaseStartTLS.test_buf_feed_data.<locals>.Proto.buffer_updatedN)rI rJ rK rX rZ r\ r r r r �Proto� s r] )FT� s 12345� s 1234� �d r z empty buffer)r �BufferedProtocolr Z _feed_data_to_buffered_proto� assertEqualrU r) � RuntimeError)r r] rV r r r r �test_buf_feed_data� s(
z BaseStartTLS.test_buf_feed_datac s� d�j �t�� �t�� ����fdd�}G �fdd�dtj�� � ���fdd�}�j|�jd��&}�j� tj
||j ��jd d
�� W d Q R X t �
������� � d S ) N� 1c s� | � �j� | �t� ��}��t|�t� �� | j�dd� | �d� | �t� ��}��t|�t� �� | �tj � | �
� d S )NT)� server_side� O) �
settimeout�TIMEOUT�recv_all�lenrc � start_tls�sendallr& r= � SHUT_RDWRr )�sockrU )� HELLO_MSGr �server_contextr r �serve� s
z=BaseStartTLS.test_start_tls_client_reg_proto_1.<locals>.servec s0 e Zd Zdd� Z� fdd�Zdd� Zdd� Zd S )
zCBaseStartTLS.test_start_tls_client_reg_proto_1.<locals>.ClientProtoc S s || _ || _d| _d S )Nr )�on_data�on_eof� con_made_cnt)r rt ru r r r rX � s zLBaseStartTLS.test_start_tls_client_reg_proto_1.<locals>.ClientProto.__init__c s | j d7 _ � �| j d� d S )Nr^ )rv rc )r �tr)r r r r' � s zSBaseStartTLS.test_start_tls_client_reg_proto_1.<locals>.ClientProto.connection_madec S s | j �|� d S )N)rt �
set_result)r rU r r r rD � s zQBaseStartTLS.test_start_tls_client_reg_proto_1.<locals>.ClientProto.data_receivedc S s | j �d� d S )NT)ru rx )r r r r r/ � s zPBaseStartTLS.test_start_tls_client_reg_proto_1.<locals>.ClientProto.eof_receivedN)rI rJ rK rX r' rD r/ r )r r r � ClientProto� s ry c � s� t jd�jd�I d H �j�� � �j�� ��jj�� �fdd�f| �� I d H \}}|��� �j�||��I d H }��� I d H d� |��� �I d H |�� d S )Ng �?)r
c s
� ���S )Nr r )ry rt ru r r �<lambda> r! zPBaseStartTLS.test_start_tls_client_reg_proto_1.<locals>.client.<locals>.<lambda>rh ) r �sleepr
�
create_future�create_connectionrG rm rc r )�addrrw r �new_tr)ry rq �client_contextr )rt ru r �client� s
"
z>BaseStartTLS.test_start_tls_client_reg_proto_1.<locals>.client)�timeout�
)r
r� )� PAYLOAD_SIZEr �simple_server_sslcontext�simple_client_sslcontextr r �
tcp_serverrj r
�run_until_complete�wait_forr~ �weakref�refr> )r rs r� �srvr )ry rq r� r rr r �!test_start_tls_client_reg_proto_1� s
$
z.BaseStartTLS.test_start_tls_client_reg_proto_1c s� d�j �t�� �t�� ����fdd�}G �fdd�dtj�� � ���fdd�}�j|�jd��"}�j� tj
||j �d d�� W d Q R X t �
������� � d S )
Nrf c s^ | � �j� | j�dd� | �d� | �t� ��}��t|�t� �� | �tj � | �
� d S )NT)rg rh ) ri rj rm rn rk rl rc r& r= ro r )rp rU )rq r rr r r rs s
z>BaseStartTLS.test_create_connection_memory_leak.<locals>.servec s0 e Zd Zdd� Z� fdd�Zdd� Zdd� Zd S )
zDBaseStartTLS.test_create_connection_memory_leak.<locals>.ClientProtoc S s || _ || _d| _d S )Nr )rt ru rv )r rt ru r r r rX ) s zMBaseStartTLS.test_create_connection_memory_leak.<locals>.ClientProto.__init__c s& || _ | jd7 _� �| jd� d S )Nr^ )rw rv rc )r rw )r r r r' . s zTBaseStartTLS.test_create_connection_memory_leak.<locals>.ClientProto.connection_madec S s | j �|� d S )N)rt rx )r rU r r r rD 5 s zRBaseStartTLS.test_create_connection_memory_leak.<locals>.ClientProto.data_receivedc S s | j �d� d S )NT)ru rx )r r r r r/ 8 s zQBaseStartTLS.test_create_connection_memory_leak.<locals>.ClientProto.eof_receivedN)rI rJ rK rX r' rD r/ r )r r r ry ( s ry c � s� t �d�I d H �j�� � �j�� ��jj�� �fdd�f| �d�i�I d H \}}��� I d H d� |��� �I d H |�� d S )Ng �?c s
� ���S )Nr r )ry rt ru r r rz B r! zQBaseStartTLS.test_create_connection_memory_leak.<locals>.client.<locals>.<lambda>�sslrh )r r{ r
r| r} rc rG r )r~ rw r )ry rq r� r )rt ru r r� ; s
z?BaseStartTLS.test_create_connection_memory_leak.<locals>.client)r� r� )r� r r� r� r r r� rj r
r� r� r~ r� r� r> )r rs r� r� r )ry rq r� r rr r �"test_create_connection_memory_leak s
z/BaseStartTLS.test_create_connection_memory_leakc s� d�j �t�� �t�� �d����fdd�}G �fdd�dtj�� G �fdd�dtj��� �����fd d
�}�j|�jd ��(}�j �
tj ||j ��j �jd �� W d Q R X d S )
Nrf r c s� | � �j� | �t� ��}��t|�t� �� | j�dd� | �d� | �t� ��}��t|�t� �� | �d� | �t� ��}��t|�t� �� | �tj � | �
� d S )NT)rg rh � 2) ri rj rk rl rc rm rn r& r= ro r )rp rU )rq r rr r r rs [ s
z=BaseStartTLS.test_start_tls_client_buf_proto_1.<locals>.servec s0 e Zd Zdd� Z� fdd�Zdd� Zdd� Zd S )
zHBaseStartTLS.test_start_tls_client_buf_proto_1.<locals>.ClientProtoFirstc S s || _ td�| _d S )Nr^ )rt rQ rR )r rt r r r rX o s zQBaseStartTLS.test_start_tls_client_buf_proto_1.<locals>.ClientProtoFirst.__init__c s � d7 � d S )Nr^ r )r rw )�client_con_made_callsr r r' s s zXBaseStartTLS.test_start_tls_client_buf_proto_1.<locals>.ClientProtoFirst.connection_madec S s | j S )N)rR )r rY r r r rZ w s zSBaseStartTLS.test_start_tls_client_buf_proto_1.<locals>.ClientProtoFirst.get_bufferc S s | j �t| jd |� �� d S )N)rt rx �bytesrR )r r[ r r r r\ z s zWBaseStartTLS.test_start_tls_client_buf_proto_1.<locals>.ClientProtoFirst.buffer_updatedN)rI rJ rK rX r' rZ r\ r )r� r r �ClientProtoFirstn s r� c s0 e Zd Zdd� Z� fdd�Zdd� Zdd� Zd S )
zIBaseStartTLS.test_start_tls_client_buf_proto_1.<locals>.ClientProtoSecondc S s || _ || _d| _d S )Nr )rt ru rv )r rt ru r r r rX s zRBaseStartTLS.test_start_tls_client_buf_proto_1.<locals>.ClientProtoSecond.__init__c s � d7 � d S )Nr^ r )r rw )r� r r r' � s zYBaseStartTLS.test_start_tls_client_buf_proto_1.<locals>.ClientProtoSecond.connection_madec S s | j �|� d S )N)rt rx )r rU r r r rD � s zWBaseStartTLS.test_start_tls_client_buf_proto_1.<locals>.ClientProtoSecond.data_receivedc S s | j �d� d S )NT)ru rx )r r r r r/ � s zVBaseStartTLS.test_start_tls_client_buf_proto_1.<locals>.ClientProtoSecond.eof_receivedN)rI rJ rK rX r' rD r/ r )r� r r �ClientProtoSecond~ s r� c � s� t jd�jd�I d H �j�� � �j�� }�j�� }�jj�� fdd�f| �� I d H \}}|��� �j�||��I d H }��� I d H d� |��� |��||�� ��|I d H d� |��� |I d H |� � ���d� d S )Ng �?)r
c s � ��S )Nr r )r� �on_data1r r rz � r! zPBaseStartTLS.test_start_tls_client_buf_proto_1.<locals>.client.<locals>.<lambda>rh r� r^ )
r r{ r
r| r} rG rm rc rB r )r~ Zon_data2ru rw r r )r� r� rq r� r� r )r� r r� � s
z>BaseStartTLS.test_start_tls_client_buf_proto_1.<locals>.client)r� )r
r� )
r� r r� r� r rb r r� rj r
r� r� r~ )r rs r� r� r )r� r� rq r� r� r rr r �!test_start_tls_client_buf_proto_1T s
z.BaseStartTLS.test_start_tls_client_buf_proto_1c s� d�j �t�� ��j�� ����fdd�}G �fdd�dtj�� � ����fdd�}�j|�jd��&}�j� tj
||j ��jd d
�� W d Q R X d S ) Nrf c sv | � �j� | �t� ��}��t|�t� �� z8y �j��jd � | �d�}W n tk
rb Y nX W d | � � X d S )Ni )
ri rj rk rl rc r
Zcall_soon_threadsaferx r8 r )rp rU )rq r �server_waits_on_handshaker r rs � s
z=BaseStartTLS.test_start_tls_slow_client_cancel.<locals>.servec s0 e Zd Zdd� Z� fdd�Zdd� Zdd� Zd S )
zCBaseStartTLS.test_start_tls_slow_client_cancel.<locals>.ClientProtoc S s || _ || _d| _d S )Nr )rt ru rv )r rt ru r r r rX � s zLBaseStartTLS.test_start_tls_slow_client_cancel.<locals>.ClientProto.__init__c s | j d7 _ � �| j d� d S )Nr^ )rv rc )r rw )r r r r' � s zSBaseStartTLS.test_start_tls_slow_client_cancel.<locals>.ClientProto.connection_madec S s | j �|� d S )N)rt rx )r rU r r r rD � s zQBaseStartTLS.test_start_tls_slow_client_cancel.<locals>.ClientProto.data_receivedc S s | j �d� d S )NT)ru rx )r r r r r/ � s zPBaseStartTLS.test_start_tls_slow_client_cancel.<locals>.ClientProto.eof_receivedN)rI rJ rK rX r' rD r/ r )r r r ry � s ry c � s� t jd�jd�I d H �j�� � �j�� ��jj�� �fdd�f| �� I d H \}}|��� �I d H ��t j��* t j�j� ||��d�jd�I d H W d Q R X d S )Ng �?)r
c s
� ���S )Nr r )ry rt ru r r rz � r! zPBaseStartTLS.test_start_tls_slow_client_cancel.<locals>.client.<locals>.<lambda>)
r r{ r
r| r} rG � assertRaises� TimeoutErrorr� rm )r~ rw r )ry rq r� r r� )rt ru r r� � s
"
z>BaseStartTLS.test_start_tls_slow_client_cancel.<locals>.client)r� r� )r
r� ) r� r r� r
r| r r r� rj r� r� r~ )r rs r� r� r )ry rq r� r r� r �!test_start_tls_slow_client_cancel� s
z.BaseStartTLS.test_start_tls_slow_client_cancelc s� d�j �d� t�� �t�� �d �� ����fdd��G �fdd�dtj��� ���fdd��� �����fd d
�}�j�|� � d S ) Nrf s answerc sf | � �j� | �|� | �t���}��t|�t��� | ��� | ��� | �t� ���| �� d S )N) ri rj Zconnectrk rl rc rm rn r )rp r~ rU )�ANSWERrq �answerr� r r r r� � s
z4BaseStartTLS.test_start_tls_server_1.<locals>.clientc s8 e Zd Zdd� Zdd� Zdd� Z� fdd�Zd d
� Zd S ) z9BaseStartTLS.test_start_tls_server_1.<locals>.ServerProtoc S s" || _ || _|| _d| _d | _d S )Nr! )�on_con� on_con_lost� on_got_hellorU r( )r r� r� r� r r r rX s
zBBaseStartTLS.test_start_tls_server_1.<locals>.ServerProto.__init__c S s || _ | j�|� d S )N)r( r� rx )r rw r r r r' s zIBaseStartTLS.test_start_tls_server_1.<locals>.ServerProto.connection_madec S s
|| _ d S )N)r( )r rw r r r �replace_transport
s zKBaseStartTLS.test_start_tls_server_1.<locals>.ServerProto.replace_transportc s0 | j |7 _ t| j �t� �kr,| j�d � d S )N)rU rl r� rx )r rU )rq r r rD s zGBaseStartTLS.test_start_tls_server_1.<locals>.ServerProto.data_receivedc S s, d | _ |d kr | j�d � n | j�|� d S )N)r( r� rx Z
set_exception)r �excr r r r7 s zIBaseStartTLS.test_start_tls_server_1.<locals>.ServerProto.connection_lostN)rI rJ rK rX r' r� rD r7 r )rq r r � ServerProto s
r� c � s� |I d H }|� �� ��| jd� �jj|| �d�jd�I d H }| �|� |I d H |� � � |I d H ��| j�� |�� d S )Nr! T)rg r )rG rc rU r
rm rj r� r )r r� r� r� rw r )r� rq r rr r r �main s
z2BaseStartTLS.test_start_tls_server_1.<locals>.mainc � s� �j �� } �j �� }�j �� }�| ||���j ��fdd�dd�I d H }|jd �� � �j� �fdd��jd��* tj��| ||��j �jd�I d H W d Q R X |� � |�
� I d H �� ��� d S )Nc s � S )Nr r )r r r rz 6 r! zHBaseStartTLS.test_start_tls_server_1.<locals>.run_main.<locals>.<lambda>z 127.0.0.1r c s
�| � �S )Nr )rp )r~ r� r r rz 9 r! )r� )r
r� ) r
r| Z
create_serverZsocketsZ getsocknameZ
tcp_clientrj r r� r Z wait_closedrc )r� r� r� �server)r� r� r� r� r� r )r~ r r �run_main/ s
z6BaseStartTLS.test_start_tls_server_1.<locals>.run_main)r� r r� r� r r r
r� )r r� r ) r� rq r� r� r� r� r� r rr r �test_start_tls_server_1� s
z$BaseStartTLS.test_start_tls_server_1c s � fdd�}� j �|� � d S )Nc � sh � � td�� � j�d d d �I d H W d Q R X t�� } � � td�� � j�d d | �I d H W d Q R X d S )NzSSLContext, gotzis not supported)r) � TypeErrorr
rm r r� )�sslctx)r r r r� F s
z4BaseStartTLS.test_start_tls_wrong_args.<locals>.main)r
r� )r r� r )r r �test_start_tls_wrong_argsE s z&BaseStartTLS.test_start_tls_wrong_argsc
s� t �� � g ��j��fdd�� d��fdd�}� �fdd�}�j|ddd ��0}��tj�� �j�||j �� W d Q R X W d Q R X ��
�� �� �g � t �
� �� ��� � � d S )
Nc s
� � |�S )N)�append)r
�ctx)�messagesr r rz V r! z5BaseStartTLS.test_handshake_timeout.<locals>.<lambda>Fc s< z,y| � d� W n tk
r( d� Y nX W d | �� X d S )Ni T)rk r8 r )rp )�server_side_abortedr r r� Z s
z3BaseStartTLS.test_handshake_timeout.<locals>.serverc � s6 t j�jjt jf| �� ddd��d�jd�I d H d S )N� g $@)r� �server_hostnamer g �?)r
)r r� r
r} r )r~ )�
client_sslctxr r r r� c s z3BaseStartTLS.test_handshake_timeout.<locals>.clientr^ )� max_clients�backlog)r r� r
�set_exception_handlerr� r� r r� r� r~ r: rc r� r� r> )r r� r� r� r )r� r� r r� r �test_handshake_timeoutP s
&
z#BaseStartTLS.test_handshake_timeoutc
s� t �� � g ��j��fdd�� dd� }� �fdd�}�j|ddd��0}��td �� �j�||j�� W d Q R X W d Q R X �� �g � d S )
Nc s
� � |�S )N)r� )r
r� )r� r r rz � r! zHBaseStartTLS.test_create_connection_ssl_slow_handshake.<locals>.<lambda>c S s8 z(y| � d� W n tk
r$ Y nX W d | �� X d S )Ni )rk r8 r )rp r r r r� � s
zFBaseStartTLS.test_create_connection_ssl_slow_handshake.<locals>.serverc � s& t j| � d�jdd��I d H \}}d S )Nr� g �?)r� r� r
r )r �open_connectionr
)r~ �reader�writer)r� r r r r� � s zFBaseStartTLS.test_create_connection_ssl_slow_handshake.<locals>.clientr^ )r� r� z SSL handshake.*is taking longer)
r r� r
r� r� r) r8 r� r~ rc )r r� r� r� r )r� r� r r �)test_create_connection_ssl_slow_handshake� s
&z6BaseStartTLS.test_create_connection_ssl_slow_handshakec
s� �j �dd� � t�� �tjdd�� �fdd�}� �fdd�}�j|d d d
��0}��tj�� �j � ||j
�� W d Q R X W d Q R X d S ) Nc S s d S )Nr )r
r� r r r rz � r! zLBaseStartTLS.test_create_connection_ssl_failed_certificate.<locals>.<lambda>F)Zdisable_verifyc sP z@y| j � dd� W n( tjk
r* Y n tk
r< Y nX W d | �� X d S )NT)rg )rm r� �SSLError�OSErrorr )rp )r� r r r� � s
zJBaseStartTLS.test_create_connection_ssl_failed_certificate.<locals>.serverc � s& t j| � d�jdd��I d H \}}d S )Nr� g �?)r� r� r
r )r r� r
)r~ r� r� )r� r r r r� � s zJBaseStartTLS.test_create_connection_ssl_failed_certificate.<locals>.clientr^ )r� r� ) r
r� r r� r� r� r� r� ZSSLCertVerificationErrorr� r~ )r r� r� r� r )r� r r� r �-test_create_connection_ssl_failed_certificate� s
z:BaseStartTLS.test_create_connection_ssl_failed_certificatec sx �j �dd� � t�� �t�� � �fdd�}� �fdd�}�j|ddd��}�j �||j��}W d Q R X ��|d � d S )
Nc S s d S )Nr )r
r� r r r rz � r! zBBaseStartTLS.test_start_tls_client_corrupted_ssl.<locals>.<lambda>c sl | � � }zLy0| j� dd� | �d� | �d� |�d� W n tjk
rP Y nX W d |�� | �� X d S )NT)rg s A
r^ s! please corrupt the SSL connection)�duprm rn rk �sendr� r� r )rp Z orig_sock)r� r r r� � s
z@BaseStartTLS.test_start_tls_client_corrupted_ssl.<locals>.serverc � sr t j| � d�jd��I d H \}}��|�� I d H d� |�d� ��tj�� |�� I d H W d Q R X |� � dS )Nr� )r� r� r
s A
� B�OK)
r r� r
rc �readlinerG r� r� r� r )r~ r� r� )r� r r r r� � s
z@BaseStartTLS.test_start_tls_client_corrupted_ssl.<locals>.clientr^ )r� r� r� ) r
r� r r� r� r� r� r~ rc )r r� r� r� Zresr )r� r r� r �#test_start_tls_client_corrupted_ssl� s
z0BaseStartTLS.test_start_tls_client_corrupted_sslN)rI rJ rK r� rj rP re r� r� r� r� r� r� r� r� r� r� r r r r rM � s /@>[=Y 0""rM c @ s e Zd Zdd� ZdS )�SelectorStartTLSTestsc C s t �� S )N)r ZSelectorEventLoop)r r r r rP � s z SelectorStartTLSTests.new_loopN)rI rJ rK rP r r r r r� � s r� �ProactorEventLoopz Windows onlyc @ s e Zd Zdd� ZdS )�ProactorStartTLSTestsc C s t �� S )N)r r� )r r r r rP � s z ProactorStartTLSTests.new_loopN)rI rJ rK rP r r r r r� � s r� �__main__) r5 r= �sysZunittestr� r r� � ImportErrorr r r r r Ztest.test_asyncior r r Z
func_testsZskipIfZTestCaser ZFunctionalTestCaseMixinrM r� Z
skipUnless�hasattrr� rI r� r r r r �<module> s>
V