Your IP : 216.73.216.213
U
i�fa/ � @ s d dl Z d dlZd dlZd dlmZ d dlmZmZ d dlmZ d dl
m Z G dd� dej �Z
G dd � d �Zejd
kr�G d d � d ee j�ZG d
d� dee j�Znxd dlZeed�r�G dd� dee j�Zeed�r�G dd� dee j�Zeed��rG dd� dee j�ZG dd � d ee j�ZdS )� N)�proactor_events)�cycle�islice)�utils)�supportc @ s> e Zd ZdZdZd dd�Zdd� Zdd� Zdd � Zd
d � Z dS )
�MyProtoNc C s2 d | _ d| _d| _|d k r.|�� | _|�� | _d S )N�INITIALr )� transport�state�nbytesZ
create_future� connected�done)�self�loop� r �I/opt/alt/python38/lib64/python3.8/test/test_asyncio/test_sock_lowlevel.py�__init__ s
zMyProto.__init__c C s@ || _ | jdkst| j��d| _| jr2| j�d � |�d� d S )Nr � CONNECTEDs% GET / HTTP/1.0
Host: example.com
)r r
�AssertionErrorr �
set_result�write)r r r r r �connection_made s zMyProto.connection_madec C s* | j dkst| j ��| jt|�7 _d S )Nr )r
r r �len)r �datar r r �
data_received s zMyProto.data_receivedc C s | j dkst| j ��d| _ d S )Nr �EOF)r
r �r r r r � eof_received" s zMyProto.eof_receivedc C s0 | j dkst| j ��d| _ | jr,| j�d � d S )N)r r ZCLOSED)r
r r
r )r �excr r r �connection_lost& s zMyProto.connection_lost)N)
�__name__�
__module__� __qualname__r r
r r r r r r r r r r
s
r c s� e Zd Zdd� Z� fdd�Z� fdd�Zdd� Zd d
� Zd d � Zd
d� Z dd� Z
dd� Z dd� Z e
jdd� �Zdd� Zdd� Zdd � Z� ZS ) �BaseSockTestsMixinc C s t �d S �N)�NotImplementedErrorr r r r �create_event_loop/ s z$BaseSockTestsMixin.create_event_loopc s$ | � � | _| �| j� t� �� d S r$ )r& r Zset_event_loop�super�setUpr �� __class__r r r( 2 s
zBaseSockTestsMixin.setUpc s4 | j �� st�| j � | �� t�� t� �� d S r$ ) r Z is_closed�
test_utilsZ run_brieflyZ
doCleanupsr Z
gc_collectr' �tearDownr r) r r r, 7 s
zBaseSockTestsMixin.tearDownc C s� t | jtj��s| j�d� |�d� | �t�� | j�| j� ||j
�� W 5 Q R X | �t�� | j�| j� |d�� W 5 Q R X | �t�� | j�| j� |d�� W 5 Q R X | �t�� | j�| j�
|t� �� W 5 Q R X | �t�� | j�| j�|�� W 5 Q R X |�d� | j�| j� ||j
�� | j�| j� |d�� | j�| j� |d��}| j�| j� |d�� |�� | �|�d�� d S )NT� GET / HTTP/1.0
� F� HTTP/1.0 200 OK)�
isinstancer r ZBaseProactorEventLoopZ set_debug� setblocking� assertRaises�
ValueError�run_until_complete� sock_connect�address� sock_sendall� sock_recv�sock_recv_into� bytearray� sock_accept�close�
assertTrue�
startswith)r �httpd�sockr r r r �_basetest_sock_client_ops@ sL
� � � �
�
� � � �z,BaseSockTestsMixin._basetest_sock_client_opsc C s� |� d� | j�| j�||j�� | j�| j�|d�� td�}t|��B}| j�| j�||d d� ��}| j�| j�|||d � �� W 5 Q R X |� � | �
|� d�� d S )NFr- r. r/ ) r1 r r4 r5 r6 r7 r: �
memoryviewr9 r< r= r> )r r? r@ r �bufr r r r �_basetest_sock_recv_intod s"
� �
��z+BaseSockTestsMixin._basetest_sock_recv_intoc C s@ t �� �.}t�� }| �||� t�� }| �||� W 5 Q R X d S r$ )r+ �run_test_server�socketrA rD �r r? r@ r r r �test_sock_client_opsu s
z'BaseSockTestsMixin.test_sock_client_opsc � s� t � � }|�d� d}d|d }| j�||�I d H | j�|dd| d �I d H t�| j�||��}| j�||�I d H }| �|� d�� |�
d �d
kr�|| j�||�I d H 7 }q�|d |� d �d � }|t |�d � }|}t
d�} tt| t |���}
| �||
� |t |�8 }| j�||�I d H }|�s0�q^tt| t |���}
| �||
� |t |�8 }�q| �|d � |I d H |�� d S �
NFi@B s
0123456789�
s POST /loop HTTP/1.0
s Content-Length: %d
s
r/ s
���� r )rF r1 r r5 r7 �asyncio� create_taskr8 r= r> �find�indexr r �bytesr � assertEqualr< ) r r6 r@ � DATA_SIZE�chunk�taskr �headers�size�checker�expectedr r r �_basetest_huge_content| sB
���
z)BaseSockTestsMixin._basetest_huge_contentc C s, t �� �}| j�| �|j�� W 5 Q R X d S r$ )r+ rE r r4 rZ r6 �r r? r r r �test_huge_content� s
�z$BaseSockTestsMixin.test_huge_contentc � s� t � � }|�d� d}d|d }| j�||�I d H | j�|dd| d �I d H t�| j�||��}t|�}t|�}| j� ||�I d H }t
|d |� �} | � | � d�� | �
d �d
kr�| j� ||�I d H }t
|d |� �} q�| d | �d �d � }
| t|
�d � } |} td�} t
t| t| ���}
| �| |
� | t| �8 } | j� ||�I d H }|d |� } | �sh�q�t
t| t| ���}
| �| |
� | t| �8 } �q>| �| d � |I d H |�� d S rI )rF r1 r r5 r7 rM rN r: rB r9 rQ r= r> rO rP r r r rR r< )r r6 r@ rS rT rU ZarrayrC r r rV rW rX rY r r r � _basetest_huge_content_recvinto� sL
���
z2BaseSockTestsMixin._basetest_huge_content_recvintoc C s, t �� �}| j�| �|j�� W 5 Q R X d S r$ )r+ rE r r4 r] r6 r[ r r r �test_huge_content_recvinto� s
�z-BaseSockTestsMixin.test_huge_content_recvintoc C sH t �� �6}t�tj�}| �||� t�tj�}| �||� W 5 Q R X d S r$ )r+ Zrun_test_unix_serverrF ZAF_UNIXrA rD rG r r r �test_unix_sock_client_ops� s
z,BaseSockTestsMixin.test_unix_sock_client_opsc C sx d }z t�� }|�d� |�� }W 5 |� � X t�� }|�d� | �t�� | j�| j� ||�� W 5 Q R X |� � d S )N�z 127.0.0.1r F)
r< rF �bind� getsocknamer1 r2 �ConnectionRefusedErrorr r4 r5 )r r6 �sr@ r r r �test_sock_client_fail� s
�z(BaseSockTestsMixin.test_sock_client_failc C s� t � � }|�d� |�d� |�d� t � � }|�|�� � | j�|�}| j�|�\}}| � |�
� d� | � ||�� � | � |� � |�� � |� � |� � |� � d S )NFr` � r )
rF r1 ra ZlistenZconnectrb r r; r4 rR Z
gettimeoutZ getpeernamer< )r ZlistenerZclient�fZconnZaddrr r r �test_sock_accept� s
z#BaseSockTestsMixin.test_sock_acceptc
s t �� ��}d }� j�� jj|jdtji��}|D ]\\}}}}}z4tj|||d�}|�d� � j�� j� ||�� W n t
k
r� Y q0X q�q0ds�t d��� jj � fdd�|d�} � j�| �\}
} � �
|
tj� � �
| tj� � j�| j� � �| jd� |
�� W 5 Q R X d S ) N�type)�familyri �protoFzCan not create socket.c s t � jd�S )N)r )r r r r r r �<lambda>" � z@BaseSockTestsMixin.test_create_connection_sock.<locals>.<lambda>)r@ r )r+ rE r r4 Z getaddrinfor6 rF Z SOCK_STREAMr1 r5 �
BaseExceptionr Zcreate_connectionZassertIsInstancerM Z Transport�Protocolr
Z
assertGreaterr r< ) r r? r@ Zinfosrj ri rk Zcnamer6 rg ZtrZprr r r �test_create_connection_sock s<
���
�
�z.BaseSockTestsMixin.test_create_connection_sock)r r! r" r& r( r, rA rD rH rZ r\ r] r^ r Z skip_unless_bind_unix_socketr_ re rh rp �
__classcell__r r r) r r# - s $,2
r# Zwin32c @ s e Zd Zdd� ZdS )�SelectEventLoopTestsc C s t �� S r$ )rM �SelectorEventLoopr r r r r& 0 s �&SelectEventLoopTests.create_event_loopN�r r! r" r& r r r r rr - s rr c @ s e Zd Zdd� ZdS )�ProactorEventLoopTestsc C s t �� S r$ )rM ZProactorEventLoopr r r r r& 6 s z(ProactorEventLoopTests.create_event_loopNru r r r r rv 3 s rv �KqueueSelectorc @ s e Zd Zdd� ZdS )�KqueueEventLoopTestsc C s t �t�� �S r$ )rM rs � selectorsrw r r r r r& @ s �z&KqueueEventLoopTests.create_event_loopNru r r r r rx = s rx �
EpollSelectorc @ s e Zd Zdd� ZdS )�EPollEventLoopTestsc C s t �t�� �S r$ )rM rs ry rz r r r r r& H s z%EPollEventLoopTests.create_event_loopNru r r r r r{ E s r{ � PollSelectorc @ s e Zd Zdd� ZdS )�PollEventLoopTestsc C s t �t�� �S r$ )rM rs ry r| r r r r r& O s z$PollEventLoopTests.create_event_loopNru r r r r r} L s r} c @ s e Zd Zdd� ZdS )rr c C s t �t�� �S r$ )rM rs ry ZSelectSelectorr r r r r& V s rt Nru r r r r rr S s )rF rM �sysr � itertoolsr r Ztest.test_asyncior r+ Ztestr ro r r# �platformZTestCaserr rv ry �hasattrrx r{ r} r r r r �<module> sB #
� �
�
� � �