File: //usr/local/lib/python3.10/test/__pycache__/test_asynchat.cpython-310.pyc
o
�i�% � @ sN d dl mZ d dlmZ d dlmZ d dlZd dlZd dlZd dlZd dl Z d dl
Z
d dl Z
d dl Z e �
� � e �de� d dlZd dlZW d � n1 sOw Y ejZdZG dd� dej�ZG d d
� d
ej�Zd d � ZG d
d� de
j�ZG dd� de�ZG dd� de
j�Z G dd� de
j�Z G dd� de
j�Z e dkr�e
� � dS dS )� )�support)�
socket_helper)�threading_helperN�ignores QUIT
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 �//usr/local/lib/python3.10/test/test_asynchat.pyr s
zecho_server.__init__c C s� | j �� | j�� | j �� \}}d| _t| jvr,|�d�}|s!n | j| | _t| jvs| j�td�| _| j r<| j �
� z | jr[|� | jd | j � �}t
�d� | j|d � | _| js@W n Y |�� | 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' s0
�
��zecho_server.runN)�__name__�
__module__� __qualname__r r r( r r r r r s r c @ s> e Zd Zdd� Zdd� Zejdkrdd� Zdd � Zd
d � Z d S )
� echo_clientc C sD t j�| � g | _| �tjtj� | �t |f� | �
|� d| _ d S �Nr ) �asynchat�
async_chatr �contentsZ
create_socketr
r r �connect�HOST�set_terminatorr )r �
terminatorZ server_portr r r r I s
zecho_client.__init__c C � d S r r �r r r r �handle_connectQ � zecho_client.handle_connect�darwinc C r5 r r r6 r r r � handle_exptW r8 zecho_client.handle_exptc C s | j |7 _ d S r )r )r r& r r r �collect_incoming_dataZ s z!echo_client.collect_incoming_datac C s | j �| j� d| _d S r- )r0 �appendr r6 r r r �found_terminator] s
z echo_client.found_terminatorN)
r) r* r+ r r7 �sys�platformr: r; r= r r r r r, G s
r, c C s: t �� } t| �}|�� | �� | �� t�d� || fS )N�{�G�z�?)r �Eventr �startr �clearr! r" )r �sr r r �start_echo_servera s
rE 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�_threadsr6 r r r �setUpn � zTestAsynchat.setUpc C s t j| j� d S r )r Zthreading_cleanuprG r6 r r r �tearDownq s zTestAsynchat.tearDownc C s� t �� }t|�}||_|�� |�� |�� t�d� t ||j
�}|� d� |� d| � |� d| � |� t � t
j| jddd� t�|� | �|jddg� d S )Nr@ s hello s world� I'm not dead yet!�, �Zuse_poll�count�timeout� hello world)r rA r r rB r rC r! r" r, r �pushr �asyncore�loop�usepollr � join_thread� assertEqualr0 )r ZtermZ server_chunkr rD �cr r r �line_terminator_checkt s
z"TestAsynchat.line_terminator_checkc C � dD ]}| � d|� qd S )N�r � � �
�rX �r �lr r r �test_line_terminator1� � �z"TestAsynchat.test_line_terminator1c C rY )NrZ s
r^ r_ r r r �test_line_terminator2� rb z"TestAsynchat.test_line_terminator2c C rY )NrZ s qqqr^ r_ r r r �test_line_terminator3� rb z"TestAsynchat.test_line_terminator3c C sf t � \}}t||j�}d}|�|� |�t� tj| jddd� t� |� | �
|j |d |� g� d S �Ns hello world, I'm not dead yet!
rL r@ rM � rE r, r rQ r rR rS rT r rU rV r0 )r ZtermlenrD r rW r& r r r �numeric_terminator_check� s
z%TestAsynchat.numeric_terminator_checkc C � | � d� d S )Nr �rg r6 r r r �test_numeric_terminator1� s z%TestAsynchat.test_numeric_terminator1c C rh )N� ri r6 r r r �test_numeric_terminator2� rI z%TestAsynchat.test_numeric_terminator2c C sj t � \}}td |j�}d}|�|� |�t� tj| jddd� t� |� | �
|j g � | �
|j |� d S re )
rE r, r rQ r rR rS rT r rU rV r0 r �r rD r rW r& r r r �test_none_terminator� s
z!TestAsynchat.test_none_terminatorc C sh t � \}}td|j�}d}tj|t dd�}|�|� tj| j ddd� t
� |� | � |j
dd g� d S )
Nr] � hello world
I'm not dead yet!
� )Z buffer_sizerL r@ rM rP rK )rE r, r r. Zsimple_producerr �push_with_producerrR rS rT r rU rV r0 )r rD r rW r& �pr r r �test_simple_producer� s
z!TestAsynchat.test_simple_producerc C sZ t � \}}td|j�}d}|�|t � tj| jddd� t� |� | �
|j ddg� d S )Nr] ro rL r@ rM rP rK ) rE r, r rq r rR rS rT r rU rV r0 rm r r r �test_string_producer� s
z!TestAsynchat.test_string_producerc C s\ t � \}}td|j�}|�d� |�t� tj| jddd� t� |� | �
|j g d�� d S )Nr] s hello world
I'm not dead yet!
rL r@ rM )rP r rK rf �r rD r rW 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
�|� | �|jg � | �t|j�d� d S )Nr] ro rL r@ rM r )rE r rA r r, r rQ r Zclose_when_donerR rS rT r r rU rV r0 Z
assertGreater�lenr ru 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 �
|� | �|jg d�� d S ) Nr] s bytes
�
�unicoderL r@ rM )� bytesr{ r{ )rE r, r rQ � bytearray�
memoryview� assertRaises� TypeErrorr rR rS rT r rU rV r0 rm r r r � test_push� s
zTestAsynchat.test_pushN)r) r* r+ rT rH rJ rX ra rc rd rg rj rl rn rs rt rv rx r� r r r r rF k s"
rF c @ s e Zd ZdZdS )�TestAsynchat_WithPollTN)r) r* r+ rT r r r r r� s r� c @ � e Zd Zdd� ZdS )�TestAsynchatMockedc C s~ t j�� }ttj�|j_t� � }|�
|� | � |j � t jj
�|d�� }|�� W d � n1 s2w Y | �|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 r� � r� c @ r� )�TestHelperFunctionsc C s, | � t�dd�d� | � t�dd�d� d S )Nzqwerty
z
r Z
qwertydkjfr )rV r. Zfind_prefix_at_endr6 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� r� r� c @ r� )�TestNotConnectedc C s t �� }| �t|jd� d S )N���)r. r/ r~ �
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� r� r� �__main__)!�testr Z test.supportr r r� r
r> r r! r� Z
unittest.mock�warnings�catch_warnings� simplefilter�DeprecationWarningr. rR r2 r r
r r/ r, rE ZTestCaserF r� r� r� r� r) �mainr r r r �<module> s<
�.
�