U
|
M ±d@ ã
|
@ sì d dl mZmZmZmZ d dlZddlmZmZm Z m
|
Z
|
m Z m Z m Z ddlmZmZ ddlmZmZmZmZmZ ddlmZ ddlmZmZm Z m Z m Z m Z m Z m!Z!m"Z"m#Z# dd l$m%Z% dd
|
l&m'Z'm(Z(m)Z)m*Z*m+Z+m,Z,m-Z- d d l.m/Z/ ee fe d dddgddfe"e!fe ddgdddfe"e!fe dg dddfe"e!fe ddgdddfe"e!fe dg ddd fgZ0ed ee1d d d Z2eeedd!d"d#Z3e1ed$d%d&Z4ee1edd'd(d)Z5dd*d+d,Z6dd*d-d.Z7dd*d/d0Z8dd*d1d2Z9dd*d3d4Z:ee1e;eeddf d5d6d7Z<eee
|
d8d9d:Z=dWee1ee;dd<d=d>Z>dd*d?d@Z?dd*dAdBZ@dd*dCdDZAdd*dEdFZBdd*dGdHZCdd*dIdJZDdd*dKdLZEdd*dMdNZFdd*dOdPZGdd*dQdRZHdd*dSdTZIdd*dUdVZJdS )Xé )ÚAnyÚCallableÚ GeneratorÚListNé )ÚConnectionClosedÚDataÚ EndOfMessageÚEventÚInformationalResponseÚRequestÚResponse)ÚHeadersÚnormalize_and_validate)Ú_obsolete_line_foldÚ ChunkedReaderÚContentLengthReaderÚ Http10ReaderÚREADERS©Ú ReceiveBuffer)
|
ÚCLIENTÚCLOSEDÚDONEÚIDLEÚMIGHT_SWITCH_PROTOCOLÚ
|
MUST_CLOSEÚ SEND_BODYÚ SEND_RESPONSEÚSERVERÚSWITCHED_PROTOCOL)ÚLocalProtocolError)Ú ChunkedWriterÚContentLengthWriterÚ Http10WriterÚwrite_any_responseÚ write_headersÚ write_requestÚWRITERSé )Únormalize_data_eventsÚGETz/a©ÚHostÚfoo©Ú
|
ConnectionÚclose©ÚmethodÚtargetÚheaderss1 GET /a HTTP/1.1
|
Host: foo
|
Connection: close
|
|
éÈ ó OK)Ú status_coder5 Úreasons& HTTP/1.1 200 OK
|
Connection: close
|
|
s HTTP/1.1 200 OK
|
|
ée )ZUpgradeZ websockets Upgrades, HTTP/1.1 101 Upgrade
|
Upgrade: websocket
|
|
s HTTP/1.1 101 Upgrade
|
|
).N)ÚwriterÚobjÚreturnc C s g }| ||j d |¡S )Nó )ÚappendÚjoin)r; r< Zgot_list© rA úHd:\z\workplace\vscode\pyvenv\venv\Lib\site-packages\h11/tests/test_io.pyÚdowriteR s rC )r; r< Úexpectedr= c C s t | |}||kstd S ©N)rC ÚAssertionError)r; r< rD ÚgotrA rA rB ÚtwX s
|
rH )Údatar= c C s t }|| 7 }|S rE r )rI ÚbufrA rA rB Úmakebuf] s rK )ÚreaderrI rD r= c s¨ t d d fdd }t|}|| | |r0tt }tt|D ](}| |d ksVt||||d
7 }qB|| | t|}|d7 }|| | t|dks¤td S )N)rG r= c sF | ks t t| dg D ](\}}t|tks0t t|tkst qd S )Nr5 )rF ÚgetattrÚtypeÚbytes)rG ÚnameÚvalue©rD rA rB Úcheckd s ztr.<locals>.checkr) ó trailing)r rK rF r ÚrangeÚlenrO )rL rI rD rS rJ ÚirA rR rB Útrc s rX )r= c C s, t D ]"\\} }}}tt| |f || qd S rE )Ú SIMPLE_CASESrH r( ©ÚroleÚstateÚeventÚbinaryrA rA rB Útest_writers_simple s r_ c C s, t D ]"\\} }}}tt| |f || qd S rE )rY rX r rZ rA rA rB Útest_readers_simple s r` c C s t ttddgd t ttg d t t¡" t ttddddgd d
|
d W 5 Q R X t t¡ t t t
|
d dgd d d W 5 Q R X d S ) N©r. Úbar)ZbazZquuxs foo: bar
|
baz: quux
|
|
s
|
r+ ú/r, r/ ú1.0©r3 r4 r5 Ú http_versionr6 )r8 r5 rf ) rH r& r r ÚpytestÚraisesr! r' r r% r rA rA rA rB Útest_writers_unusual s6
|
ý üø
|
ÿûri c
|
C sR t tttf dtdddgdd t tttf dtddg dd t tttf dtd dgdd
|
d t tttf d td d gdd
|
d t tttf dtd dgdd
|
d t tttf dtd dgdd
|
d t tttf dtd dgddd t tttf dtd dgddd t tttf dtd ddgddd t tttf dtd ddgddd t tttf d tddd d d d gd! t t
|
¡ t tttf d"d W 5 Q R X t t
|
¡ t tttf d#d W 5 Q R X t t
|
¡ t tttf d$d W 5 Q R X t t
|
¡ t tttf d$d W 5 Q R X t t
|
¡ t tttf d%d W 5 Q R X d S )&Ns$ HEAD /foo HTTP/1.0
|
Some: header
|
|
ÚHEADú/foo©ÚSomeÚheaderrd re s HEAD /foo HTTP/1.0
|
|
s! HTTP/1.0 200 OK
|
Some: header
|
|
r6 r7 )r8 r5 rf r9 s$ HTTP/1.0 200 OK
|
Foo: a a a a a
|
|
)ÚFooz a a a a as HTTP/1.0 200 OK
|
Foo:
|
|
)ro Ú s HTTP/1.0 200 OK
|
Foo:
|
|
s HTTP/1.0 200
|
Foo: bar
|
|
)ro rb r> s# HTTP/1.1 200 OK
|
SomeHeader: val
|
|
)Z
|
SomeHeaderÚvalz1.1ÚOKs5 HTTP/1.1 200 OK
|
SomeHeader1: val1
|
SomeHeader2: val2
|
|
)Z SomeHeader1Zval1)Z SomeHeader2Zval2s7 HTTP/1.1 200 OK
|
SomeHeader1: val1
|
SomeHeader2: val2
|
|
s HEAD /foo HTTP/1.1
|
Host: example.com
|
Some: multi-line
|
header
|
nonsense
|
I guess
|
Connection: close
|
More-nonsense: in the
|
last header
|
|
©r- z example.com)rm z"multi-line header nonsense I guessr/ )z More-nonsensezin the last headerr2 s& HEAD /foo HTTP/1.1
|
folded: line
|
|
s# HEAD /foo HTTP/1.1
|
foo : line
|
|
s" HEAD /foo HTTP/1.1
|
foo : line
|
|
s HEAD /foo HTTP/1.1
|
: line
|
|
) rX r r r r r r r rg rh r! rA rA rA rB Útest_readers_unusual¨ s
|
üý
|
ý
|
üý
|
üý
|
ÿý
|
ÿý
|
ÿý
|
|
üý
|
üý
|
üý
|
üýõ
|
ý
|
ý
|
ý
|
ý rt c C s* t tddddgdtddgks&td S )Nó aaas bbbs cccs ddds bbb ccc)Úlistr Ú bytearrayrF rA rA rA rB Ú test__obsolete_line_fold_bytesH s
|
ýrx )rL rJ Údo_eofr= c c sB | |}|d krq(|V t |tkr q(q |r>|r4t| ¡ V d S rE )rN r rF Zread_eof)rL rJ ry r] rA rA rB Ú_run_reader_iterU s rz )Úargsr= c G s t t| }t|S rE )rv rz r* )r{ ÚeventsrA rA rB Ú _run_readerf s r} F)ÚthunkrI rD ry r= c C sÔ t d t|}t| |||ks&tt d | }t }g }tt|D ](}|t||d7 }||||d
7 }qJ|t|||7 }t||ksttdd |D }|rÐ|sÐt|d }t| |d|ksÐtd S )NzTest 1zTest 2Fr) c s s | ]}t |tkV qd S rE )rN r )Ú.0r] rA rA rB Ú <genexpr>| s z t_body_reader.<locals>.<genexpr>rT ) ÚprintrK r} rF r rU rV r* Úany) r~ rI rD ry rJ rL r| rW Z is_completerA rA rB Ú t_body_readerk s r c C s4 t dd dt g t dd dtddt g d S )Nc S s t dS )Nr ©r rA rA rA rB Ú<lambda> r> z*test_ContentLengthReader.<locals>.<lambda>r> c S s t dS )Né
|
r rA rA rA rB r
r> s
|
0123456789©rI )r r r rA rA rA rB Útest_ContentLengthReader s ýr c C sL t tdt gdd t tdtddgdd t tdtddt gdd d S )Nr> T)ry s asdfr F)r r r r rA rA rA rB Útest_Http10Reader s ÿr c C sô t tdt g t tdtdgdg t tdtddtdgdg t tdtddt g t td td
|
dt g t t¡ t td td dg W 5 Q R X t t¡ t td d W 5 Q R X t tdtddt g t tdtddt g d S )Nó 0
|
|
s 0
|
Some: header
|
|
rl ©r5 s3 5
|
01234
|
10
|
0123456789abcdef
|
0
|
Some: header
|
|
s 012340123456789abcdefr s% 5
|
01234
|
10
|
0123456789abcdef
|
0
|
|
sµ aA
|
xxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxx
|
0
|
|
sª xxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxsi 9999999999999999999999999999999999999999999999999999999999999999999999999999999999999999999999999999
|
xxxs xxxs 10
|
xxxsG 5; hello=there
|
xxxxx
|
0; random="junk"; some=more; canbe=lonnnnngg
|
|
s xxxxxs 5
|
01234
|
0
|
|
s 01234)r r r r rg rh r! rA rA rA rB Útest_ChunkedReader sJ ý
|
þú ýý ú ýr c C sN t d} t| tdddks tt| tdddks8tt| t dksLtt d} t t¡ t| tdd W 5 Q R X t d} t| tdd t t¡ t| tdd W 5 Q R X t d} t| tdd t t¡ t| t W 5 Q R X t d} t| tdddk t| tdddk t t¡ t| tdgd W 5 Q R X d S )
|
Né s 123r s 45r> s 123456s 456©ZEtagZasdfr )r# rC r rF r rg rh r! ©ÚwrA rA rB Útest_ContentLengthWriterÑ s( r c C s t } t| tdddks tt| tdddks6tt| tdddksNtt| t dksbtt| tdd gd
|
d ks~td S ) Nru r s 3
|
aaa
|
s aaaaaaaaaaaaaaaaaaaas 14
|
aaaaaaaaaaaaaaaaaaaa
|
r> r r )ÚaÚbr s 0
|
Etag: asdf
|
a: b
|
|
)r" rC r rF r r rA rA rB Útest_ChunkedWriterì s ÿÿr c C s^ t } t| tdddks tt| t dks2tt t¡ t| tdgd W 5 Q R X d S )Ns 1234r r> r r )r$ rC r rF r rg rh r! r rA rA rB Útest_Http10Writerû s
|
r c C s. t t¡ ttttf dd W 5 Q R X d S )Ns HTTP/1.0 200 OK xxxx
|
|
)rg rh r! rX r r r rA rA rA rB Ú&test_reject_garbage_after_request_line s r c C s. t t¡ ttttf dd W 5 Q R X d S )Ns& HEAD /foo HTTP/1.1 xxxxxx
|
Host: a
|
|
©rg rh r! rX r r r rA rA rA rB Ú'test_reject_garbage_after_response_line s
|
ýr c C s. t t¡ ttttf dd W 5 Q R X d S )Ns% HEAD /foo HTTP/1.1
|
Host: foo bar
|
|
r rA rA rA rB Ú"test_reject_garbage_in_header_line s
|
ýr c
|
C sT dD ]J} t d}| | ¡ | d¡ t t¡ tttt f |d W 5 Q R X qd S )Ns îs HEAD /s HTTP/1.1
|
Host: foobar
|
|
)
|
rw r? Úextendrg rh r! rX r r r )Zbad_charÚmessagerA rA rB Ú test_reject_non_vchar_in_path s
|
|
r c C s& t tttf dtddddgd d S )Ns\ HEAD /foo HTTP/1.1
|
Host: foo
|
Set-Cookie: ___utmvafIumyLc=kUdUpAt; path=/; Max-Age=900
|
|
rj rk r, )z
|
Set-Cookiez-___utmvafIumyLc=kUdUpAt; path=/; Max-Age=900r2 )rX r r r r rA rA rA rB Ú"test_allow_some_garbage_in_cookies% s
|
þýúr c C s t ttddgd d S )Nra rs s Host: example.com
|
foo: bar
|
|
)rH r& r rA rA rA rB Útest_host_comes_first7 s
|
|
ýr )F)KÚtypingr r r r rg Z_eventsr r r r
|
r r r Ú_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( Zhelpersr* rY rO rC rH rK rX r_ r` ri rt rx Úboolrz r} r r r r r r r r r r r r r rA rA rA rB Ú<module> s $ 0 $ ýù
|
ý ý ÿû ýå# ! þ
|
=
|
|
