File: //usr/local/lib/python3.10/dist-packages/h11/tests/__pycache__/helpers.cpython-310.pyc
o
���g
� @ s d dl mZmZmZmZmZ ddlmZmZm Z ddl
m Z m Z m
Z
mZmZmZmZ ddlmZmZmZmZmZ ddlmZ zd dl mZ W n eyU d dl mZ Y nw ded ee fd
d �Z ded e d ee fd
d�Z dee d ee fdd�Z G dd� d�Z!dS )� )�cast�List�Type�Union�
ValuesView� )�
Connection� NEED_DATA�PAUSED)�ConnectionClosed�Data� EndOfMessage�Event�InformationalResponse�Request�Response)�CLIENT�CLOSED�DONE�
MUST_CLOSE�SERVER)�Sentinel)�Literal�conn�returnc C sH g } | � � }|ttfv r |S tt|�}|�|� t|�tu r# |S q�N)�
next_eventr r
r r �append�typer )r �
got_events�event� r! �</usr/local/lib/python3.10/dist-packages/h11/tests/helpers.py�get_all_events s
�
�r# �datac C s | � |� t| �S r )� receive_datar# )r r$ r! r! r" �receive_and_get# s
r& � in_eventsc C s� g }| D ]A}t |�tu rtt|j�ddd�}|r@t |d �t |� u r(tu r@n nt|d j|j |d j|d jd�|d< q|�|� q|S )NF)r$ � chunk_start� chunk_end���)r r �bytesr$ r( r) r )r' �
out_eventsr r! r! r" �normalize_data_events* s ( � r- c @ sl e Zd Zd
dd�Zedee fdd��Z ddee d e
e e e f d
e
e e e e
d f defd d �ZdS )�ConnectionPairr Nc C s( t tt �ttt�i| _t ttt i| _d S r )r r r r �other��selfr! r! r" �__init__? s zConnectionPair.__init__c C s
| j �� S r )r �valuesr0 r! r! r" �connsC s
zConnectionPair.conns�match�role� send_events�expectc C s� t |t�s|g}d}d}|D ]}| j| �|�}|d u r d}q||7 }q|r1| j| j| �|� |r>| j| j| �d� t| j| j| �}|dkrN|}t |t�sV|g}||ks\J �|S )N� FTr5 )�
isinstance�listr �sendr/ r% r# ) r1 r6 r7 r8 r$ �closed�
send_event�new_datar r! r! r" r< H s(
zConnectionPair.send)r N)r5 )�__name__�
__module__� __qualname__r2 �propertyr r r4 r r r r r r r+ r< r! r! r! r" r. > s
�����r. N)"�typingr r r r r � _connectionr r r
�_eventsr r r
r r r r �_stater r r r r �_utilr r � ImportError�typing_extensionsr# r+ r& r- r. r! r! r! r" �<module> s $ �