Format: application/x-python-code
1 o
2
3 .'h@ � @ s� d dl mZmZmZmZ d dlZddlmZmZm Z m
4 Z
5 m Z m Z m
6 Z
7 ddlmZmZ ddlmZmZmZmZmZ ddlmZ ddlmZmZm Z m Z m Z m Z m Z m!Z!m"Z"m#Z# dd l$m%Z% dd
8 l&m'Z'm(Z(m)Z)m*Z*m+Z+m,Z,m-Z- d d l.m/Z/ ee fe d
9 dddgd�dfe"e!fe
10 ddgdd�dfe"e!fe
11 dg dd�dfe"e!fe ddgdd�dfe"e!fe dg dd�d fgZ0d ed d ed e1fd!d"�Z2d ed ed#ed dfd$d%�Z3d&e1d efd'd(�Z4d)ed&e1d#ed dfd*d+�Z5dYd,d-�Z6dYd.d/�Z7dYd0d1�Z8dYd2d3�Z9dYd4d5�Z:d)ed6e1d7e;d eeddf fd8d9�Z<d:ed ee
12 fd;d<�Z=dZd>ed&e1d#ed7e;d df
13 d?d@�Z>dYdAdB�Z?dYdCdD�Z@dYdEdF�ZAdYdGdH�ZBdYdIdJ�ZCdYdKdL�ZDdYdMdN�ZEdYdOdP�ZFdYdQdR�ZGdYdSdT�ZHdYdUdV�ZIdYdWdX�ZJdS )[� )�Any�Callable� Generator�ListN� )�ConnectionClosed�Data� EndOfMessage�Event�InformationalResponse�Request�Response)�Headers�normalize_and_validate)�_obsolete_line_fold�
14 ChunkedReader�ContentLengthReader� Http10Reader�READERS��
15 ReceiveBuffer)
16 �CLIENT�CLOSED�DONE�IDLE�MIGHT_SWITCH_PROTOCOL�
17 MUST_CLOSE� SEND_BODY�
18 SEND_RESPONSE�SERVER�SWITCHED_PROTOCOL)�LocalProtocolError)�
19 ChunkedWriter�ContentLengthWriter� Http10Writer�write_any_response�
20 write_headers�
21 write_request�WRITERS� )�normalize_data_events�GETz/a��Host�foo��
22 Connection�close��method�target�headerss1 GET /a HTTP/1.1
23 Host: foo
24 Connection: close
25
26 �� � OK)� status_coder5 �reasons& HTTP/1.1 200 OK
27 Connection: close
28
29 s HTTP/1.1 200 OK
30
31 �e )ZUpgradeZ websockets Upgrades, HTTP/1.1 101 Upgrade
32 Upgrade: websocket
33
34 s HTTP/1.1 101 Upgrade
35
36 �writer).N�obj�returnc C s g }| ||j � d�|�S )N� )�append�join)r; r< Zgot_list� rA �GC:\Users\shubh\Videos\agent\nova\Lib\site-packages\h11/tests/test_io.py�dowriteR s
37 rC �expectedc C s t | |�}||ks J �d S �N)rC )r; r< rD �gotrA rA rB �twX s
38 rG �datac C s t � }|| 7 }|S rE r )rH �bufrA rA rB �makebuf] s rJ �readerc s� dt dd f� fdd� }t|�}|| |�� |rJ �t� }tt|��D ]}| |�d u s,J �||||d � 7 }q"|| |�� t|�}|d7 }|| |�� t|�dksSJ �d S )NrF r= c sF | � ksJ �t | dg �D ]\}}t|�tu sJ �t|�tu s J �q d S )Nr5 )�getattr�type�bytes)rF �name�value�rD rA rB �checkd s
39 �ztr.<locals>.checkr) � trailing)r rJ r �range�lenrN )rK rH rD rR rI �irA rQ rB �trc s rW c C s, t D ]\\} }}}tt| |f ||� qd S rE )� SIMPLE_CASESrG r( �Zrole�state�event�binaryrA rA rB �test_writers_simple � �r] c C s, t D ]\\} }}}tt| |f ||� qd S rE )rX rW r rY rA rA rB �test_readers_simple� r^ r_ c C s� t ttddg�d� t ttg �d� t�t�� t ttddddgd d
40 �d � W d � n1 s0w Y t�t�� t t t
41 d dgd d �d � W d � d S 1 sRw Y d S )
42 N�r. �bar)ZbazZquuxs foo: bar
43 baz: quux
44
45 s
46 r+ �/r, r/ �1.0�r3 r4 r5 � http_versionr6 )r8 r5 re ) rG r& r r �pytest�raisesr! r' r r% r
47 rA rA rA rB �test_writers_unusual� s6
48 � �� � ��"�rh c C s� t tttf dtdddgdd�� t tttf dtddg dd�� t tttf dtd dgdd
49 d �� t tttf d td d
50 gdd
51 d �� t tttf dtd dgdd
52 d �� t tttf dtd dgdd
53 d �� t tttf dtd dgddd �� t tttf dtd dgddd �� t tttf dtd ddgddd �� t tttf dtd ddgddd �� t tttf d tddg d �d �� t� t
54 �� t tttf d d � W d � n1 s�w Y t� t
55 �� t tttf d d � W d � n1 s�w Y t� t
56 �� t tttf d!d � W d � n 1 �sw Y t� t
57 �� t tttf d!d � W d � n 1 �s5w Y t� t
58 �� t tttf d"d � W d � d S 1 �sVw Y d S )#Ns$ HEAD /foo HTTP/1.0
59 Some: header
60
61 �HEAD�/foo��Some�headerrc rd s HEAD /foo HTTP/1.0
62
63 s! HTTP/1.0 200 OK
64 Some: header
65
66 r6 r7 )r8 r5 re r9 s$ HTTP/1.0 200 OK
67 Foo: a a a a a
68
69 )�Fooz a a a a as HTTP/1.0 200 OK
70 Foo:
71
72 )rn � s HTTP/1.0 200 OK
73 Foo:
74
75 s HTTP/1.0 200
76 Foo: bar
77
78 )rn ra r> s# HTTP/1.1 200 OK
79 SomeHeader: val
80
81 )Z
82 SomeHeader�valz1.1�OKs5 HTTP/1.1 200 OK
83 SomeHeader1: val1
84 SomeHeader2: val2
85
86 )Z SomeHeader1Zval1)Z SomeHeader2Zval2s7 HTTP/1.1 200 OK
87 SomeHeader1: val1
88 SomeHeader2: val2
89
90 s� HEAD /foo HTTP/1.1
91 Host: example.com
92 Some: multi-line
93 header
94 nonsense
95 I guess
96 Connection: close
97 More-nonsense: in the
98 last header
99
100 )�r- z example.com)rl z"multi-line header nonsense I guessr/ )z
101 More-nonsensezin the last headerr2 s& HEAD /foo HTTP/1.1
102 folded: line
103
104 s# HEAD /foo HTTP/1.1
105 foo : line
106
107 s" HEAD /foo HTTP/1.1
108 foo : line
109
110 s HEAD /foo HTTP/1.1
111 : line
112
113 ) rW r r r r r r r
114 rf rg r! rA rA rA rB �test_readers_unusual� s�
115 ��
116 �
117 ��
118 ��
119
120 ��
121
122 ��
123
124 ��
125
126 ��
127 ��
128 ��
129 ��
130 � �
131 � �
132 � �
133 � � $�rs c C s& t tg d���dtd�dgksJ �d S )N)� aaas bbbs ccc� dddrt s bbb cccru )�listr � bytearrayrA rA rA rB � test__obsolete_line_fold_bytesH s
134 �rx rI �do_eofc c sJ � | |�}|d u r n |V t |�tu rnq|r#|r J �| �� V d S d S rE )rM r Zread_eof)rK rI ry r[ rA rA rB �_run_reader_iterU s � � �rz �argsc G s t t| � �}t|�S rE )rv rz r* )r{ �eventsrA rA rB � _run_readerf s r} F�thunkc C s� t d� t|�}t| � ||�|ksJ �t d� | � }t� }g }tt|��D ]}|t||d�7 }||||d � 7 }q%|t|||�7 }t|�|ksJJ �tdd� |D ��}|rh|sjt|d �}t| � |d�|kslJ �d S d S d S )NzTest 1zTest 2Fr) c s s � | ] }t |�tu V qd S rE )rM r )�.0r[ rA rA rB � <genexpr>| s � z t_body_reader.<locals>.<genexpr>rS )�printrJ r} r rT rU r* �any) r~ rH rD ry rI rK r| rV Z is_completerA rA rB �
135 t_body_readerk s$ �r� c C s4 t dd� dt� g� t dd� dtdd�t� g� d S )Nc S � t d�S )Nr �r rA rA rA rB �<lambda>� � z*test_ContentLengthReader.<locals>.<lambda>r> c S r� )N�
136 r� rA rA rA rB r� � r� s
137 0123456789�rH )r� r r rA rA rA rB �test_ContentLengthReader� s �r� c C sL t tdt� gdd� t tdtdd�gdd� t tdtdd�t� gdd� d S )Nr> T)ry s asdfr� F)r� r r r rA rA rA rB �test_Http10Reader� s
138
139 �r� c C s t tdt� g� t tdtdgd�g� t tdtdd�tdgd�g� t tdtdd�t� g� t td td
140 d�t� g� t�t�� t td td d�g� W d � n1 sTw Y t�t�� t td
141 d � W d � n1 sow Y t tdtdd�t� g� t tdtdd�t� g� d S )N� 0
142
143 s 0
144 Some: header
145
146 rk �r5 s3 5
147 01234
148 10
149 0123456789abcdef
150 0
151 Some: header
152
153 s 012340123456789abcdefr� s% 5
154 01234
155 10
156 0123456789abcdef
157 0
158
159 s� aA
160 xxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxx
161 0
162
163 s� xxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxsi 9999999999999999999999999999999999999999999999999999999999999999999999999999999999999999999999999999
164 xxxs xxxs 10
165 xxxsG 5; hello=there
166 xxxxx
167 0; random="junk"; some=more; canbe=lonnnnngg
168
169 s xxxxxs 5
170 01234
171 0
172
173 s 01234)r� r r r rf rg r! rA rA rA rB �test_ChunkedReader� sN �
174 �� �� � �� �r� c C s� t d�} t| tdd��dksJ �t| tdd��dks J �t| t� �dks&J �t d�} t�t�� t| tdd�� W d � n1 sBw Y t d�} t| tdd�� t�t�� t| tdd�� W d � n1 skw Y t d�} t| tdd�� t�t�� t| t� � W d � n1 s�w Y t d�} t| tdd��dk t| tdd��dk t�t�� t| tdgd �� W d � d S 1 s�w Y d S )
175 N� s 123r� s 45r> s 123456s 456�ZEtagZasdfr� )r# rC r r rf rg r! ��wrA rA rB �test_ContentLengthWriter� s0 � � � "�r� c C s� t � } t| tdd��dksJ �t| tdd��dksJ �t| tdd��dks'J �t| t� �dks1J �t| tdd gd
176 ��d ks?J �d S ) Nrt r� s 3
177 aaa
178 s aaaaaaaaaaaaaaaaaaaas 14
179 aaaaaaaaaaaaaaaaaaaa
180 r> r� r� )�a�br� s 0
181 Etag: asdf
182 a: b
183
184 )r" rC r r r� rA rA rB �test_ChunkedWriter� s �r� c C st t � } t| tdd��dksJ �t| t� �dksJ �t�t�� t| tdgd�� W d � d S 1 s3w Y d S )Ns 1234r� r> r� r� )r$ rC r r rf rg r! r� rA rA rB �test_Http10Writer� s "�r� c C �D t �t�� ttttf dd � W d � d S 1 sw Y d S )Ns HTTP/1.0 200 OK xxxx
185
186 )rf rg r! rW r r r rA rA rA rB �&test_reject_garbage_after_request_line s "�r� c C r� )Ns& HEAD /foo HTTP/1.1 xxxxxx
187 Host: a
188
189 �rf rg r! rW r r r rA rA rA rB �'test_reject_garbage_after_response_line �
190 �"�r� c C r� )Ns% HEAD /foo HTTP/1.1
191 Host: foo bar
192
193 r� rA rA rA rB �"test_reject_garbage_in_header_line r� r� c C sh dD ]/} t d�}|�| � |�d� t�t�� tttt f |d � W d � n1 s,w Y qd S )Ns �s HEAD /s HTTP/1.1
194 Host: foobar
195
196 )
197 rw r? �extendrf rg r! rW r r r )Zbad_char�messagerA rA rB � test_reject_non_vchar_in_path s
198
199 ���r� c C s& t tttf dtddddgd�� d S )Ns\ HEAD /foo HTTP/1.1
200 Host: foo
201 Set-Cookie: ___utmvafIumyLc=kUdUpAt; path=/; Max-Age=900
202
203 ri rj r, )z
204 Set-Cookiez-___utmvafIumyLc=kUdUpAt; path=/; Max-Age=900r2 )rW r r r r rA rA rA rB �"test_allow_some_garbage_in_cookies% s
205 ���r� c C s t ttddg�d� d S )Nr` rr s Host: example.com
206 foo: bar
207
208 )rG r& r rA rA rA rB �test_host_comes_first7 s
209
210 �r� )r= N)F)K�typingr r r r rf Z_eventsr r r r
211 r r r
212 �_headersr r �_readersr r r r r Z_receivebufferr �_stater r r r r r r r r r Z_utilr! �_writersr" r# r$ r% r& r' r( �helpersr* rX rN rC rG rJ rW r] r_ rh rs rx �boolrz r} r� r� r� r� r� r� r� r� r� r� r� r� r� rA rA rA rB �<module> s� $ 0 $ ��
213 � ��� ��#
214
215
216
217
218 !
219 ���
220 �
221
222
223
224
225 =
226
227
228
229
230
231
232
233