File: //opt/cppython/lib/python3.8/test/__pycache__/test_asynchat.cpython-38.opt-2.pyc
U
-D�g#% � @ s� d dl mZ d dlZd dlZd dlZd dlZd dlZd dlZd dlZd dl Z d dl
Z ej Z dZ dZ
G dd� dej�ZG dd� dej�Zd d
� ZG d d � d e j�ZG d
d� de�ZG dd� de j�ZG dd� de j�ZG dd� de j�Zedkr�e �� dS )� )�supportNs QUIT
g @c @ s e Zd ZdZdd� Zdd� ZdS )� echo_server� c C s< t j�| � || _t�tjtj�| _t� | j�| _
d | _ d S �N) � threading�Thread�__init__�event�socket�AF_INET� SOCK_STREAM�sockr Z bind_port�port�start_resend_event)�selfr � r �1/opt/cppython/lib/python3.8/test/test_asynchat.pyr s
zecho_server.__init__c C s� | j �� | j�� | j �� \}}d| _t| jkrP|�d�}|sBqP| j| | _q(| j�td�| _| j rp| j �
� z<| jr�|� | jd | j � �}t
�d� | j|d � | _qrW n Y nX |�� | 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 @ s$ e Zd Zdd� Zdd� Zdd� ZdS )� echo_clientc C s^ t j�| � g | _| �tjtj� | �t |f� | �
|� d| _ dd� }t j
dkrZdd� }d S )Nr c S s d S r r �r r r r �handle_connectL s z,echo_client.__init__.<locals>.handle_connect�darwinc S s d S r r r* r r r � handle_exptR 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 D s
zecho_client.__init__c C s | j |7 _ d S r )r )r r# r r r �collect_incoming_dataU s z!echo_client.collect_incoming_datac C s | j �| j� d| _d S )Nr )r0 �appendr r* r r r �found_terminatorX s z echo_client.found_terminatorN)r& r' r( r r7 r9 r r r r r) B s r) c C s: t �� } t| �}|�� | �� | �� t�d� || fS )N�{�G�z�?)r �Eventr �startr �clearr r )r �sr r r �start_echo_server\ s
r? 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 �� | _d S r )r Zthreading_setup�_threadsr* r r r �setUpi s zTestAsynchat.setUpc C s t j| j� d S r )r Zthreading_cleanuprA r* r r r �tearDownl s zTestAsynchat.tearDownc C s� t �� }t|�}||_|�� |�� |�� t�d� t ||j
�}|� d� |� d| � |� d| � |� t � t
j| jddd� tj|td� | �|jddg� d S ) Nr: s hello s world� I'm not dead yet!�, �Zuse_poll�count�timeout�rH � hello world)r r; r r r< r r= r r r) r �pushr �asyncore�loop�usepollr � join_thread�TIMEOUT� assertEqualr0 )r ZtermZ server_chunkr r> �cr r r �line_terminator_checko s
z"TestAsynchat.line_terminator_checkc C s dD ]}| � d|� qd S )N�r � � �
�rS �r �lr r r �test_line_terminator1� s z"TestAsynchat.test_line_terminator1c C s dD ]}| � d|� qd S )NrT s
rX rY r r r �test_line_terminator2� s z"TestAsynchat.test_line_terminator2c C s dD ]}| � d|� qd S )NrT s qqqrX rY r r r �test_line_terminator3� s z"TestAsynchat.test_line_terminator3c C sj t � \}}t||j�}d}|�|� |�t� tj| jddd� tj |t
d� | � |j |d |� g� d S �Ns hello world, I'm not dead yet!
rE r: rF rI �
r? r) r rK r rL rM rN r rO rP rQ r0 )r Ztermlenr> r rR r# r r r �numeric_terminator_check� s
z%TestAsynchat.numeric_terminator_checkc C s | � d� d S )Nr �r` r* r r r �test_numeric_terminator1� s z%TestAsynchat.test_numeric_terminator1c C s | � d� d S )N� ra r* r r r �test_numeric_terminator2� s z%TestAsynchat.test_numeric_terminator2c C sn t � \}}td |j�}d}|�|� |�t� tj| jddd� tj |t
d� | � |j g � | � |j
|� d S r^ )r? r) r rK r rL rM rN r rO rP rQ r0 r �r r> r rR r# r r r �test_none_terminator� s
z!TestAsynchat.test_none_terminatorc C sl t � \}}td|j�}d}tj|t dd�}|�|� tj| j ddd� t
j |t d� | �
|jd d
g� d S ) NrW � hello world
I'm not dead yet!
� )Z buffer_sizerE r: rF rI rJ rD )r? r) r r. Zsimple_producerr �push_with_producerrL rM rN r rO rP rQ r0 )r r> r rR r# �pr r r �test_simple_producer� s
z!TestAsynchat.test_simple_producerc C s^ t � \}}td|j�}d}|�|t � tj| jddd� tj |t
d� | � |j ddg� d S ) NrW rg rE r: rF rI rJ rD )
r? r) r ri r rL rM rN r rO rP rQ r0 re r r r �test_string_producer� s
z!TestAsynchat.test_string_producerc C sb t � \}}td|j�}|�d� |�t� tj| jddd� tj |t
d� | � |j ddd g� d S )
NrW s hello world
I'm not dead yet!
rE r: rF rI rJ r rD r_ �r r> r rR r r r �test_empty_line� s
�z TestAsynchat.test_empty_linec C s� t � \}}t�� |_td|j�}|�d� |�t� |�� t j
| j ddd� |j� � t
j|td� | �|jg � | �t|j�d� d S )NrW rg rE r: rF rI r )r? r r; r r) r rK r Zclose_when_donerL rM rN r r rO rP rQ r0 Z
assertGreater�lenr rm r r r �test_close_when_done� s
z!TestAsynchat.test_close_when_donec C s� t � \}}td|j�}d}|�|� |�t|�� |�t|�� | �t|jd� | �t|jd� |�t� t j
| j ddd� t j
|td� | �|jd d d g� d S )
NrW s bytes
�
�unicoderE r: rF rI s bytes)r? r) r rK � bytearray�
memoryview� assertRaises� TypeErrorr rL rM rN r rO rP rQ r0 re r r r � test_push� s
zTestAsynchat.test_pushN)r& r' r( rN rB rC rS r[ r\ r] r` rb rd rf rk rl rn rp rw r r r r r@ f s
r@ c @ s e Zd ZdZdS )�TestAsynchat_WithPollTN)r& r' r( rN r r r r rx � s rx c @ s e Zd Zdd� ZdS )�TestAsynchatMockedc C sj t j�� }ttj�|j_t� � }|�
|� | � |j � t jj
�|d��}|�� W 5 Q R X | �|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
�
dispatcher�errorr r r �test_blockingioerror s
z'TestAsynchatMocked.test_blockingioerrorN)r& r' r( r� r r r r ry s ry c @ s e Zd Zdd� ZdS )�TestHelperFunctionsc C s, | � t�dd�d� | � t�dd�d� d S )Nzqwerty
z
r Z
qwertydkjfr )rQ r. Zfind_prefix_at_endr* r r r �test_find_prefix_at_end s z+TestHelperFunctions.test_find_prefix_at_endN)r& r' r( r� r r r r r� s r� c @ s e Zd Zdd� ZdS )�TestNotConnectedc C s t �� }| �t|jd� d S )N���)r. r/ ru �
ValueErrorr3 )r r" r r r �!test_disallow_negative_terminator s z2TestNotConnected.test_disallow_negative_terminatorN)r& r' r( r� r r r r r� s r� �__main__)�testr r. rL r| r
r4 r r rz Z
unittest.mockr2 r rP r r r/ r) r? ZTestCaser@ rx ry r� r� r& �mainr r r r �<module> s. .