Your IP : 216.73.216.213
���]J �
@ s� d d l Z d d l Z d d l m Z y d d l Z Wn e k
rR d Z Yn Xd d l Z d d l m Z d d l m Z d d l m Z e j e d k d � Gd d � d e j
� � Z e d k r� e j
� d S)
� N)�mock)�log)�sslproto)�
test_utilsz
No ssl modulec s� e Z d Z � f d d � Z d d d � Z d d d � Z d d � Z d
d � Z d d
� Z d d � Z d d � Z
d d � Z � S)�SslProtoHandshakeTestsc s0 t � j � t j � | _ | j | j � d S)N)�super�setUp�asyncioZnew_event_loop�loopZset_event_loop)�self)� __class__� �2/opt/alt/python35/lib64/python3.5/test_sslproto.pyr s
z SslProtoHandshakeTests.setUpNc C sc t j � } t j � } t j | j | | | � } | j | j j � | � | j
| j j � | S)N) r Zdummy_ssl_contextr ZProtocolr Z SSLProtocolr
�assertIs�_app_transportZ get_protocolZ
addCleanup�close)r �waiterZ
sslcontextZ app_proto�protor
r
r � ssl_protocol s z#SslProtoHandshakeTests.ssl_protocolc C s t j � } t j � } d | j _ | r9 | | j _ n d d � } | | j _ t j d d | �� | j | � Wd QRX| 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 ZMock�shutdownr � do_handshakeZ side_effectZpatch�connection_made)r � ssl_protor � transportZsslpiper r
r
r r ! s z&SslProtoHandshakeTests.connection_madec s� t j d | j � } | j | � } t j d | j � � � f d d � } | j � | j | | � t j � � | j j � � Wd QRXd S)Nr
c s$ t � } | | � � j d � g S)N)� Exception�
set_result)r �exc)�
handshake_futr
r r 6 s
zBSslProtoHandshakeTests.test_cancel_handshake.<locals>.do_handshake) r �Futurer
r �cancelr r Zdisable_loggerZrun_until_complete)r r r r r
)r! r �test_cancel_handshake/ s
z,SslProtoHandshakeTests.test_cancel_handshakec C se t j d | j � } | j | � } | j | � | j � t j | j � | j | j � t
� d S)Nr
) r r" r
r r Z eof_receivedr � run_briefly�assertIsInstance� exception�ConnectionResetError)r r r r
r
r �test_eof_received_waiterB s
z/SslProtoHandshakeTests.test_eof_received_waiterc C so t j d | j � } | j | � } t j j � } t j j t j � z | j
d � Wd t j j | � Xd S)Nr
) r r" r
r r �logger�getEffectiveLevel�setLevel�logging�FATALZ _fatal_error)r r r Z log_levelr
r
r � test_fatal_error_no_name_errorJ s z5SslProtoHandshakeTests.test_fatal_error_no_name_errorc C sh t j d | j � } | j | � } | j | � | j t � t j | j � | j | j
� t � d S)Nr
) r r" r
r r �connection_lost�ConnectionAbortedErrorr r% r& r' )r r r r
r
r �test_connection_lostY s
z+SslProtoHandshakeTests.test_connection_lostc C ss t j d | j � } | j | � } d d � } | j | � } t j | j � | j j � | j | j
j � d S)Nr
c S s g S)Nr
)r r
r
r r h s zHSslProtoHandshakeTests.test_close_during_handshake.<locals>.do_handshake) r r" r
r r r r% r r Z
assertTrue�abortZcalled)r r r r r r
r
r �test_close_during_handshakec s
z2SslProtoHandshakeTests.test_close_during_handshakec C s� t j d | j � } | j | � } | j | j d � � t � } | j | j d | � | � | j | � | j | j d � � | j
d � | j | j d � � d S)Nr
�socket) r r" r
r Z assertIsNoneZ_get_extra_info�objectr r ZassertIsNotNoner0 )r r r �defaultr
r
r �(test_get_extra_info_on_closed_connectionq s
z?SslProtoHandshakeTests.test_get_extra_info_on_closed_connection) �__name__�
__module__� __qualname__r r r r$ r) r/ r2 r4 r8 r
r
)r r r s
r �__main__)r- Zunittestr Zssl� ImportErrorr r r r ZskipIfZTestCaser r9 �mainr
r
r
r �<module> s
"k