Your IP : 216.73.216.85
��Yf - �
@ s� d d l m Z e j d � Z d d l Z d d l Z d d l Z d d l Z d d l Z d d l Z d d l
Z
d d l Z d d l Z
y d d l
Z
Wn e k
r� d Z
Yn Xe j Z d Z d Z e
rGd d � d e
j � Z Gd d � d e j � Z d
d � Z e
j e
d � Gd
d � d e
j � � Z Gd d � d e � Z Gd d � d e
j � Z Gd d � d e
j � Z Gd d � d e
j � Z Gd d � d e
j � Z e d k r�e
j � d S)� )�support�_threadNs QUIT
g @c @ s. e Z d Z d Z d d � Z d d � Z d S)� echo_server� c C sV t j j | � | | _ t j t j t j � | _ t j | j � | _
d | _ d S)N) � threading�Thread�__init__�event�socket�AF_INET� SOCK_STREAM�sockr Z bind_port�port�start_resend_event)�selfr � r �7/opt/alt/python35/lib64/python3.5/test/test_asynchat.pyr s
zecho_server.__init__c
C s | j j � | j j � | j j � \ } } d | _ x9 t | j k rs | j d � } | s` P| j | | _ q; W| j j t d � | _ | j r� | j j
� yV xO | j r� | j | j d | j � � } t
j d � | j | d � | _ q� WWn Yn X| j � | j j � d S)N� r g����MbP?)r
Zlistenr �setZaccept�buffer� SERVER_QUIT�recv�replacer �wait�send�
chunk_size�time�sleep�close)r Zconn�client�data�nr r r �run) s*
zecho_server.runN)�__name__�
__module__� __qualname__r r r" r r r r r s r c @ s4 e Z d Z d d � Z d d � Z d d � Z d S)� echo_clientc C s� t j j | � g | _ | j t j t j � | j t | f � | j
| � d | _ d d � } t j
d k r d d � } d S)Nr c S s d S)Nr )r r r r �handle_connectS s z,echo_client.__init__.<locals>.handle_connect�darwinc S s d S)Nr )r r r r � handle_exptY s z)echo_client.__init__.<locals>.handle_expt)�asynchat�
async_chatr �contentsZ
create_socketr
r r Zconnect�HOST�set_terminatorr �sys�platform)r �
terminatorZ server_portr' r) r r r r K s
zecho_client.__init__c C s | j | 7_ d S)N)r )r r r r r �collect_incoming_data\ s z!echo_client.collect_incoming_datac C s | j j | j � d | _ d S)Nr )r, �appendr )r r r r �found_terminator_ s z echo_client.found_terminatorN)r# r$ r% r r2 r4 r r r r r& I s r& c C sM t j � } t | � } | j � | j � | j � t j d � | | f S)Ng{�G�z�?)r �Eventr �startr �clearr r )r �sr r r �start_echo_serverc s
r9 z!Threading required for this test.c @ s� e Z d Z d Z d d � Z d d � Z d d � Z d d � Z d
d � Z d d
� Z d d � Z
d d � Z d d � Z d d � Z
d d � Z d d � Z d d � Z d d � Z d d � Z d S)!� TestAsynchatFc C s t j � | _ d S)N)r Zthreading_setup�_threads)r r r r �setUpq s zTestAsynchat.setUpc C s t j | j � d S)N)r Zthreading_cleanupr; )r r r r �tearDownt s zTestAsynchat.tearDownc C s� t j � } t | � } | | _ | j � | j � | j � t j d � t | | j
� } | j d � | j d | � | j d | � | j t � t
j d | j d d d d � | j d t � | j � r� | j d � | j | j d
d g � d S) Ng{�G�z�?s hello s worlds I'm not dead yet!�use_poll�counti, �timeoutzjoin() timed outs hello world)r r5 r r r6 r r7 r r r& r �pushr �asyncore�loop�usepoll�join�TIMEOUT�is_alive�fail� assertEqualr, )r ZtermZ server_chunkr r8 �cr r r �line_terminator_checkw s"
z"TestAsynchat.line_terminator_checkc C s% x d D] } | j d | � q Wd S)Nr � � s
)r rL rM )rK )r �lr r r �test_line_terminator1� s
z"TestAsynchat.test_line_terminator1c C s% x d D] } | j d | � q Wd S)Nr rL rM s
)r rL rM )rK )r rN r r r �test_line_terminator2� s
z"TestAsynchat.test_line_terminator2c C s% x d D] } | j d | � q Wd S)Nr rL rM s qqq)r rL rM )rK )r rN r r r �test_line_terminator3� s
z"TestAsynchat.test_line_terminator3c C s� t � \ } } t | | j � } d } | j | � | j t � t j d | j d d d d � | j d t � | j
� r� | j d � | j | j
| d | � g � d S)Ns hello world, I'm not dead yet!
r> r? i, r@ g{�G�z�?zjoin() timed out)r9 r& r rA r rB rC rD rE rF rG rH rI r, )r Ztermlenr8 r rJ r r r r �numeric_terminator_check� s
z%TestAsynchat.numeric_terminator_checkc C s | j d � d S)Nr )rR )r r r r �test_numeric_terminator1� s z%TestAsynchat.test_numeric_terminator1c C s | j d � d S)N� )rR )r r r r �test_numeric_terminator2� s z%TestAsynchat.test_numeric_terminator2c C s� t � \ } } t d | j � } d } | j | � | j t � t j d | j d d d d � | j d t � | j
� r� | j d � | j | j
g � | j | j | � d S)Ns hello world, I'm not dead yet!
r> r? i, r@ g{�G�z�?zjoin() timed out)r9 r& r rA r rB rC rD rE rF rG rH rI r, r )r r8 r rJ r r r r �test_none_terminator� s
z!TestAsynchat.test_none_terminatorc C s� t � \ } } t d | j � } d } t j | t d d �} | j | � t j d | j d d d d � | j
d t � | j � r� | j
d
� | j | j d d g � d S)
Ns
s hello world
I'm not dead yet!
Z buffer_size� r> r? i, r@ g{�G�z�?zjoin() timed outs hello worlds I'm not dead yet!)r9 r& r r* Zsimple_producerr �push_with_producerrB rC rD rE rF rG rH rI r, )r r8 r rJ r �pr r r �test_simple_producer� s
z!TestAsynchat.test_simple_producerc C s� t � \ } } t d | j � } d } | j | t � t j d | j d d d d � | j d t � | j
� r� | j d � | j | j
d d
g � d S) Ns
s hello world
I'm not dead yet!
r> r? i, r@ g{�G�z�?zjoin() timed outs hello worlds I'm not dead yet!)r9 r& r rX r rB rC rD rE rF rG rH rI r, )r r8 r rJ r r r r �test_string_producer� s
z!TestAsynchat.test_string_producerc C s� t � \ } } t d | j � } | j d � | j t � t j d | j d d d d � | j d t � | j
� r� | j d � | j | j
d d
d g � d S) Ns
s hello world
I'm not dead yet!
r> r? i, r@ g{�G�z�?zjoin() timed outs hello worldr s I'm not dead yet!)r9 r& r rA r rB rC rD rE rF rG rH rI r, )r r8 r rJ r r r �test_empty_line� s
z TestAsynchat.test_empty_linec C s� t � \ } } t j � | _ t d | j � } | j d � | j t � | j � t j
d | j d d d d � | j j � | j
d t � | j � r� | j d � | j | j g � | j t | j � d � d S)
Ns
s hello world
I'm not dead yet!
r> r? i, r@ g{�G�z�?zjoin() timed outr )r9 r r5 r r& r rA r Zclose_when_donerB rC rD r rE rF rG rH rI r, Z
assertGreater�lenr )r r8 r rJ r r r �test_close_when_done� s
z!TestAsynchat.test_close_when_donec C s� t � \ } } t d | j � } d } | j | � | j t | � � | j t | � � | j t | j d � | j t | j d � | j t � t j
d | j d d d d � | j d t
� | j | j d
d
d
g � d S) Ns
s bytes
�
Zunicoder> r? i, r@ g{�G�z�?s bytes)r9 r& r rA � bytearray�
memoryview� assertRaises� TypeErrorr rB rC rD rE rF rI r, )r r8 r rJ r r r r � test_push s
zTestAsynchat.test_pushN)r# r$ r% rD r< r= rK rO rP rQ rR rS rU rV rZ r[ r\ r^ rd r r r r r: m s
r: c @ s e Z d Z d Z d S)�TestAsynchat_WithPollTN)r# r$ r% rD r r r r re s re c @ s e Z d Z d d � Z d S)�TestAsynchatMockedc C s� t j j � } t t j � | j _ t j � } | j
| � | j | j � t j j
j | d � � } | j � Wd QRX| j | j � d S)NZ handle_error)�unittestZmockZMock�BlockingIOError�errnoZEAGAINr Z side_effectr* r+ Z
set_socketZ
addCleanupZ del_channelZpatch�objectZ handle_readZ assertFalseZcalled)r r
Z
dispatcher�errorr r r �test_blockingioerror s
z'TestAsynchatMocked.test_blockingioerrorN)r# r$ r% rl r r r r rf s rf c @ s e Z d Z d d � Z d S)�TestHelperFunctionsc C s< | j t j d d � d � | j t j d d � d � d S)Nzqwerty
z
r Z
qwertydkjfr )rI r* Zfind_prefix_at_end)r r r r �test_find_prefix_at_end' s z+TestHelperFunctions.test_find_prefix_at_endN)r# r$ r% rn r r r r rm & s rm c @ s( e Z d Z d d � Z d d � Z d S)�TestFifoc C s8 | j t � � } t j � } Wd QRX| j t | j � d � | j d � | j d � | j t | � d � | j | j � d � | j | j
� d � | j t | � d � | j | j � d � | j | j � d � | j | j
� d
� | j t | � d � | j | j � d � | j | j
� d � d S) Nz(fifo class will be removed in Python 3.6� � arL r Fr T)r rp )r rq )r N) � assertWarns�DeprecationWarningr* �fiforI �str�warningrA r] �first�popZis_empty)r �cm�fr r r �
test_basic- s
zTestFifo.test_basicc C s� | j t � � } t j d d d g � } Wd QRX| j t | j � d � | j t | � d � | j | j � d � | j | j � d � | j | j � d � | j | j � d
� d S) N� x� rM z(fifo class will be removed in Python 3.6r r )r r| )r r} )r rM )r N) rr rs r* rt rI ru rv r] rx )r ry rz r r r �test_given_list? s zTestFifo.test_given_listN)r# r$ r% r{ r~ r r r r ro , s ro c @ s e Z d Z d d � Z d S)�TestNotConnectedc C s& t j � } | j t | j d � d S)Nr ���)r* r+ rb �
ValueErrorr. )r r r r r �!test_disallow_negative_terminatorL s z2TestNotConnected.test_disallow_negative_terminatorN)r# r$ r% r� r r r r r K s r �__main__)!Ztestr �
import_moduleZthreadr* rB ri r
r/ r rg �warningsZ
unittest.mockr � ImportErrorr- r rF r r r+ r& r9 Z
skipUnlessZTestCaser: re rf rm ro r r# �mainr r r r �<module> s<
.
�