Your IP : 216.73.216.85
3
��2a^( �
@ s4 d dl mZ ejd�Zd dlZd dlZd dlZd dlZd dlZd dl Z d dl
Z
d dl Z
y d dl Z W n e
k
rz dZ Y nX ejZdZdZe r�G dd� de j�ZG dd � d ej�Zd
d � Ze
je d �G d
d� de
j��ZG dd� de�ZG dd� de
j�ZG dd� de
j�ZG dd� de
j�Z e dk�r0e
j � dS )� )�support�_threadNs QUIT
g @c @ s e Zd ZdZdd� Zdd� ZdS )� echo_server� c C s< t jj| � || _tjtjtj�| _tj | 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/python36/lib64/python3.6/test_asynchat.pyr s
zecho_server.__init__c
C s� | j j� | jj� | j j� \}}d| _x*t| jkrR|jd�}|sDP | j| | _q*W | jjtd�| _| j rt| j j
� y@x:| jr�|j | jd | j � �}t
jd� | j|d � | _qxW W n Y nX |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 @ s$ e Zd Zdd� Zdd� Zdd� ZdS )� echo_clientc C s^ t jj| � g | _| jtjtj� | jt |f� | j
|� d| _ dd� }t j
dkrZdd� }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 �
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 r2 r4 r r r r r& H s r& c C s: t j� } t| �}|j� | j� | j� tjd� || fS )Ng{�G�z�?)r �Eventr �startr �clearr r )r �sr r r �start_echo_serverb s
r9 z!Threading required for this test.c @ s� e Zd ZdZdd� Zdd� Zdd� Zdd � Zd
d � Zd d
� Z dd� Z
dd� Z dd� Z dd� Z
dd� Zdd� Zdd� Zd d � Zd d � Zd 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� tjd� t ||j
�}|j d� |j d| � |j d| � |j t � t
j| jddd� |jtd� |j� r�| jd� | j|jd dg� d S )
Ng{�G�z�?s hello s worlds I'm not dead yet!i, )�use_poll�count�timeout)r@ zjoin() 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_checkv s"
z"TestAsynchat.line_terminator_checkc C s xdD ]}| j d|� qW d S )Nr � � �
)r rL rM )rK )r �lr r r �test_line_terminator1� s
z"TestAsynchat.test_line_terminator1c C s xdD ]}| j d|� qW d S )Nr rL rM s
)r rL rM )rK )r rO r r r �test_line_terminator2� s
z"TestAsynchat.test_line_terminator2c C s xdD ]}| j d|� qW d S )Nr rL rM s qqq)r rL rM )rK )r rO r r r �test_line_terminator3� s
z"TestAsynchat.test_line_terminator3c C sz t � \}}t||j�}d}|j|� |jt� tj| jddd� |jt d� |j
� r^| j d� | j |j
|d |� g� d S )Ns hello world, I'm not dead yet!
i, g{�G�z�?)r> r? r@ )r@ 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 )rS )r r r r �test_numeric_terminator1� s z%TestAsynchat.test_numeric_terminator1c C s | j d� d S )N� )rS )r r r r �test_numeric_terminator2� s z%TestAsynchat.test_numeric_terminator2c C s~ t � \}}td |j�}d}|j|� |jt� tj| jddd� |jt d� |j
� r^| j d� | j |j
g � | j |j|� d S )Ns hello world, I'm not dead yet!
i, g{�G�z�?)r> r? r@ )r@ 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 � \}}td|j�}d}tj|t dd�}|j|� tj| j ddd� |j
t d� |j � rf| j
d � | j|jd
d g� d S ) NrN s hello world
I'm not dead yet!
� )� buffer_sizei, g{�G�z�?)r> r? r@ )r@ 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 sn t � \}}td|j�}d}|j|t � tj| jddd� |jt d� |j
� rX| j d� | j |j
dd g� d S )
NrN s hello world
I'm not dead yet!
i, g{�G�z�?)r> r? r@ )r@ zjoin() timed outs hello worlds I'm not dead yet!)r9 r& r rZ 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 sr t � \}}td|j�}|jd� |jt� tj| jddd� |jt d� |j
� rZ| j d� | j |j
dd d
g� d S ) NrN s hello world
I'm not dead yet!
i, g{�G�z�?)r> r? r@ )r@ 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 � \}}tj� |_td|j�}|jd� |jt� |j� t j
| j ddd� |jj � |j
td� |j� rv| jd� | j|jg � | jt|j�d� d S ) NrN s hello world
I'm not dead yet!
i, g{�G�z�?)r> r? r@ )r@ 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 � \}}td|j�}d}|j|� |jt|�� |jt|�� | jt|jd� | jt|jd� |jt� t j
| j ddd� |j t
d� | j|jd d d g� d S )
NrN s bytes
�
Zunicodei, g{�G�z�?)r> r? r@ )r@ 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 rP rQ rR rS rT rV rW r\ r] r^ r` rf r r r r r: l s
r: c @ s e Zd ZdZdS )�TestAsynchat_WithPollTN)r# r$ r% rD r r r r rg s rg c @ s e Zd Zdd� ZdS )�TestAsynchatMockedc C sj t jj� }ttj�|j_tj � }|j
|� | j |j � t jj
j|d��}|j� W d Q R X | 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% rn r r r r rh s rh c @ s e Zd Zdd� ZdS )�TestHelperFunctionsc C s, | j tjdd�d� | j tjdd�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% rp r r r r ro % s ro c @ s e Zd Zdd� ZdS )�TestNotConnectedc C s t j� }| jt|jd� d S )Nr ���)r* r+ rd �
ValueErrorr. )r r r r r �!test_disallow_negative_terminator, s z2TestNotConnected.test_disallow_negative_terminatorN)r# r$ r% rt r r r r rq + s rq �__main__) Ztestr �
import_moduleZthreadr* rB rk r
r/ r ri Z
unittest.mockr � ImportErrorr- r rF r r r+ r& r9 Z
skipUnlessZTestCaser: rg rh ro rq r# �mainr r r r �<module> s:
.
&