Your IP : 216.73.216.213
3
��2a3 �
@ s� d dl Z d dlZd dlmZ y d dlZW n ek
r@ dZY nX d dlZd dlmZ d dlmZ d dlmZ ej edkd�G dd� dej
��Z e d kr�ej
� dS )
� N)�mock)�log)�sslproto)�
test_utilsz
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� Z� ZS ) �SslProtoHandshakeTestsc s$ t � j� tj� | _| j| j� d S )N)�super�setUp�asyncioZnew_event_loop�loopZset_event_loop)�self)� __class__� �2/opt/alt/python36/lib64/python3.6/test_sslproto.pyr s
z SslProtoHandshakeTests.setUpN)�waiter�protoc C sN t j� }|d krtj� }tj| j|||�}| j|jj � |� | j
|jj � |S )N) r Zdummy_ssl_contextr �Protocolr Z SSLProtocolr
�assertIs�_app_transport� get_protocolZ
addCleanup�close)r r r Z
sslcontext� ssl_protor
r
r � ssl_protocol s z#SslProtoHandshakeTests.ssl_protocol)� do_handshakec C s^ t j� }t j� }d|j_|r&||j_ndd� }||j_t jd|d�� |j|� W d Q R X |S )N� c S s g S )Nr
)�callbackr
r
r �mock_handshake) s z>SslProtoHandshakeTests.connection_made.<locals>.mock_handshakezasyncio.sslproto._SSLPipe)� return_value)r ZMockZshutdownr r Z side_effectZpatch�connection_made)r r r � transportZsslpiper r
r
r r " s
z&SslProtoHandshakeTests.connection_madec sn t j| jd�}| j|d�}t j| jd�� � fdd�}|j� tj� � | j||d� | jj� � W d Q R X d S )N)r
)r c s t � }| |� � jd � g S )N)� ExceptionZ
set_result)r �exc)�
handshake_futr
r r 7 s
zBSslProtoHandshakeTests.test_cancel_handshake.<locals>.do_handshake)r ) r �Futurer
r Zcancelr Zdisable_loggerr Zrun_until_complete)r r r r r
)r! r �test_cancel_handshake0 s
z,SslProtoHandshakeTests.test_cancel_handshakec C sL t j| jd�}| j|d�}| j|� |j� tj| j� | j|j � t
� d S )N)r
)r ) r r" r
r r Z eof_receivedr � run_briefly�assertIsInstance� exception�ConnectionResetError)r r r r
r
r �test_eof_received_waiterC s
z/SslProtoHandshakeTests.test_eof_received_waiterc C sT t j| jd�}| j|d�}tjj� }tjjtj � z|j
d � W d tjj|� 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_errorK s
z5SslProtoHandshakeTests.test_fatal_error_no_name_errorc C sN t j| jd�}| j|d�}| j|� |jt� tj| j� | j |j
� t� d S )N)r
)r ) r r" r
r r �connection_lost�ConnectionAbortedErrorr r$ r% r& )r r r r
r
r �test_connection_lostZ s
z+SslProtoHandshakeTests.test_connection_lostc C sL t j| jd�}| j|d�}| j|�}tj| j� |jj� | j |j
j � d S )N)r
)r ) r r" r
r r r r$ r r Z
assertTrue�abortZcalled)r r r r r
r
r �test_close_during_handshaked s
z2SslProtoHandshakeTests.test_close_during_handshakec C s| t j| jd�}| j|d�}| j|jd�� t� }| j|jd|�|� | j|� | j |jd�� |j
d � | j|jd�� d S )N)r
)r Zsocket) r r" r
r � assertIsNoneZ_get_extra_info�objectr r ZassertIsNotNoner+ )r r r �defaultr
r
r �(test_get_extra_info_on_closed_connectiono s
z?SslProtoHandshakeTests.test_get_extra_info_on_closed_connectionc C sR t j| jd�}| j|d�}t j� }|jj|� | j|jj� |� | j|j |� d S )N)r
)r )
r r" r
r r r Z set_protocolr r Z
_app_protocol)r r r Z
new_app_protor
r
r �test_set_new_app_protocolz s z0SslProtoHandshakeTests.test_set_new_app_protocolc C s4 | j � }| j|� |j}|j� | j|jd�� d S )Ns data)r r r r r0 Z
data_received)r r �transpr
r
r � test_data_received_after_closing� s
z7SslProtoHandshakeTests.test_data_received_after_closingc C s4 | j � }| j|� |j}|j� | j|jd�� d S )Ns data)r r r r r0 �write)r r r5 r
r
r �test_write_after_closing� s
z/SslProtoHandshakeTests.test_write_after_closing)�__name__�
__module__� __qualname__r r r r# r( r* r- r/ r3 r4 r6 r8 �
__classcell__r
r
)r r r s
r �__main__)r) Zunittestr Zssl� ImportErrorr r r r ZskipIfZTestCaser r9 �mainr
r
r
r �<module> s