Your IP : 216.73.216.213
���] - �
@ 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 �2/opt/alt/python35/lib64/python3.5/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
�listenr �set�accept�buffer� SERVER_QUIT�recv�replacer �wait�send�
chunk_size�time�sleep�close)r �conn�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 �connect�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 r6 r8 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
r= 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 r9 r r r: r r; r r r) r �pushr �asyncore�loop�usepoll�join�TIMEOUT�is_alive�fail� assertEqualr/ )r ZtermZ server_chunkr r< �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 rP rQ )rO )r �lr r r �test_line_terminator1� s
z"TestAsynchat.test_line_terminator1c C s% x d D] } | j d | � q Wd S)Nr rP rQ s
)r rP rQ )rO )r rR r r r �test_line_terminator2� s
z"TestAsynchat.test_line_terminator2c C s% x d D] } | j d | � q Wd S)Nr rP rQ s qqq)r rP rQ )rO )r rR 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!
rB rC i, rD g{�G�z�?zjoin() timed out)r= r) r rE r rF rG rH rI rJ rK rL rM r/ )r Ztermlenr< r rN r# r r r �numeric_terminator_check� s
z%TestAsynchat.numeric_terminator_checkc C s | j d � d S)Nr )rV )r r r r �test_numeric_terminator1� s z%TestAsynchat.test_numeric_terminator1c C s | j d � d S)N� )rV )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!
rB rC i, rD g{�G�z�?zjoin() timed out)r= r) r rE r rF rG rH rI rJ rK rL rM r/ r )r r< r rN 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� rB rC i, rD g{�G�z�?zjoin() timed outs hello worlds I'm not dead yet!)r= r) r r- Zsimple_producerr �push_with_producerrF rG rH rI rJ rK rL rM r/ )r r< r rN 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!
rB rC i, rD g{�G�z�?zjoin() timed outs hello worlds I'm not dead yet!)r= r) r r\ r rF rG rH rI rJ rK rL rM r/ )r r< r rN 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!
rB rC i, rD g{�G�z�?zjoin() timed outs hello worldr s I'm not dead yet!)r= r) r rE r rF rG rH rI rJ rK rL rM r/ )r r< r rN 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!
rB rC i, rD g{�G�z�?zjoin() timed outr )r= r r9 r r) r rE r Zclose_when_donerF rG rH r rI rJ rK rL rM r/ Z
assertGreater�lenr )r r< r rN 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
�
�unicoderB rC i, rD g{�G�z�?s bytes)r= r) r rE � bytearray�
memoryview� assertRaises� TypeErrorr rF rG rH rI rJ rM r/ )r r< r rN r# r r r � test_push s
zTestAsynchat.test_pushN)r& r' r( rH r@ rA rO rS rT rU rV rW rY rZ r^ r_ r` rb ri r r r r r> m s
r> c @ s e Z d Z d Z d S)�TestAsynchat_WithPollTN)r& r' r( rH r r r r rj s rj 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�errno�EAGAINr 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( rr r r r r rk s rk 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 )rM 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( rt r r r r rs & s rs 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� � arP r Fr T)r rv )r rw )r N) � assertWarns�DeprecationWarningr- �fiforM �str�warningrE ra �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� rQ z(fifo class will be removed in Python 3.6r r )r r� )r r� )r rQ )r N) rx ry r- rz rM r{ r| ra r~ )r r r� r r r �test_given_list? s zTestFifo.test_given_listN)r& r' r( r� r� r r r r ru , s ru 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. rg �
ValueErrorr2 )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_module�threadr- rF rn r
r3 r rl �warningsZ
unittest.mockr � ImportErrorr1 r rJ r r r. r) r= Z
skipUnlessZTestCaser> rj rk rs ru r� r& �mainr r r r �<module> s<
.
�