Format: application/x-python-code
1 o
2
3 .'h
4 � @ s d dl mZmZmZmZmZ ddlmZmZm Z ddl
5 m Z m Z m
6 Z
7 mZmZmZmZ ddlmZmZmZmZmZ ddlmZ zd dl mZ W n eyU d dl mZ Y nw ded ee fd
8 d �Z ded e d ee fd
9 d�Z dee d ee fdd�Z G dd� d�Z!dS )� )�cast�List�Type�Union�
10 ValuesView� )�
11 Connection� NEED_DATA�PAUSED)�ConnectionClosed�Data� EndOfMessage�Event�InformationalResponse�Request�Response)�CLIENT�CLOSED�DONE�
12 MUST_CLOSE�SERVER)�Sentinel)�Literal�conn�returnc C sH g } | � � }|ttfv r |S tt|�}|�|� t|�tu r# |S q�N)Z
13 next_eventr r
14 r r �append�typer )r �
15 got_events�event� r �GC:\Users\shubh\Videos\agent\nova\Lib\site-packages\h11/tests/helpers.py�get_all_events s
16 �
17 �r" �datac C s | � |� t| �S r )� receive_datar" )r r# r r r! �receive_and_get# s
18 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& Z
19 out_eventsr r r r! �normalize_data_events* s ( � r+ c @ sl e Zd Zd
20 dd�Zedee fdd��Z ddee d e
21 e e e f d
22 e
23 e e e e
24 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
25 | j �� S r )r �valuesr. r r r! �connsC s
26 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� FTr3 )�
27 isinstance�listr �sendr- r$ r" ) r/ r4 r5 r6 r# �closedZ
28 send_eventZnew_datar r r r! r: H s(
29
30
31 zConnectionPair.send)r N)r3 )�__name__�
32 __module__� __qualname__r0 �propertyr r r2 r r r r r r r* r: r r r r! r, > s
33 �����r, N)"�typingr r r r r � _connectionr r r
34 Z_eventsr r r
35 r r r r �_stater r r r r Z_utilr r � ImportErrorZtyping_extensionsr" r* r% r+ r, r r r r! �<module> s $ �
36