Your IP : 216.73.216.213
�
g f�+ �
@ 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
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 n 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 � n 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/python34/lib64/python3.4/test/test_asynchat.pyr s
zecho_server.__init__c
C s' | j j d � | j j � | j j � \ } } d | _ x< t | j k ry | j d � } | sf Pn | j | | _ q> W| j j t d � | _ | j r� | j j
� n yV xO | j r� | j | j d | j � � } t
j d � | j | d � | _ q� WWn Yn X| j � | j j � d S)Nr � 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 � } n d S)Nr c S s d S)Nr )r r r r �handle_connectR s z,echo_client.__init__.<locals>.handle_connect�darwinc S s d S)Nr )r r r r � handle_exptX s z)echo_client.__init__.<locals>.handle_expt)�asynchat�
async_chatr �contentsZ
create_socketr
r r Zconnect�HOST�set_terminatorr �sys�platform)r Z
terminatorZ server_portr' r) r r r r J 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 r1 r3 r r r r r& H 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_serverb s
r8 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 �setUpp s zTestAsynchat.setUpc C s t j | j � d S)N)r Zthreading_cleanupr: )r r r r �tearDowns 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 � n | 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 r4 r r r5 r r6 r r r& r �pushr �asyncore�loop�usepoll�join�TIMEOUT�is_alive�fail� assertEqualr, )r ZtermZ server_chunkr r7 �cr r r �line_terminator_checkv s"
z"TestAsynchat.line_terminator_checkc C s% x d D] } | j d | � q Wd S)Nr � � s
)r rK rL )rJ )r �lr r r �test_line_terminator1� s
z"TestAsynchat.test_line_terminator1c C s% x d D] } | j d | � q Wd S)Nr rK rL s
)r rK rL )rJ )r rM r r r �test_line_terminator2� s
z"TestAsynchat.test_line_terminator2c C s% x d D] } | j d | � q Wd S)Nr rK rL s qqq)r rK rL )rJ )r rM 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 � n | j | j
| d | � g � d S)Ns hello world, I'm not dead yet!
r= r> i, r? g{�G�z�?zjoin() timed out)r8 r& r r@ r rA rB rC rD rE rF rG rH r, )r Ztermlenr7 r rI r r r r �numeric_terminator_check� s
z%TestAsynchat.numeric_terminator_checkc C s | j d � d S)Nr )rQ )r r r r �test_numeric_terminator1� s z%TestAsynchat.test_numeric_terminator1c C s | j d � d S)N� )rQ )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 � n | 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)r8 r& r r@ r rA rB rC rD rE rF rG rH r, r )r r7 r rI 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
� n | 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!)r8 r& r r* Zsimple_producerr �push_with_producerrA rB rC rD rE rF rG rH r, )r r7 r rI 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 � n | 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!)r8 r& r rW r rA rB rC rD rE rF rG rH r, )r r7 r rI 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 � n | 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!)r8 r& r r@ r rA rB rC rD rE rF rG rH r, )r r7 r rI 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 � n | 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 )r8 r r4 r r& r r@ r Zclose_when_donerA rB rC r rD rE rF rG rH r, Z
assertGreater�lenr )r r7 r rI 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)r8 r& r r@ � bytearray�
memoryview� assertRaises� TypeErrorr rA rB rC rD rE rH r, )r r7 r rI r r r r � test_push s
zTestAsynchat.test_pushN)r# r$ r% rC r; r< rJ rN rO rP rQ rR rT rU rY rZ r[ r] rc r r r r r9 l s
r9 c @ s e Z d Z d Z d S)�TestAsynchat_WithPollTN)r# r$ r% rC r r r r rd s rd 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 QX| 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% rk r r r r re s re 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 )rH 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% rm r r r r rl % s rl c @ s( e Z d Z d d � Z d d � Z d S)�TestFifoc C s t j � } | 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) N� � arK r Fr T)r ro )r rp )r N)r* �fifor@ rH r\ �first�popZis_empty)r �fr r r �
test_basic, s
zTestFifo.test_basicc C s� t j d d d g � } | j t | � d � | j | j � d � | j | j � d � | j | j � d � | j | j � d � d S)
N� x� rL r r )r rv )r rw )r rL )r N)r* rq rH r\ rs )r rt r r r �test_given_list; s zTestFifo.test_given_listN)r# r$ r% ru rx r r r r rn + s rn 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+ ra �
ValueErrorr. )r r r r r �!test_disallow_negative_terminatorE s z2TestNotConnected.test_disallow_negative_terminatorN)r# r$ r% r| r r r r ry D s ry �__main__) Ztestr �
import_moduleZthreadr* rA rh r
r/ r rf Z
unittest.mockr � ImportErrorr- r rE r r r+ r& r8 Z
skipUnlessZTestCaser9 rd re rl rn ry r# �mainr r r r �<module> s:
.
�