a ((a< @sddlZddlTddlmZmZddlmZmZmZm Z m Z ddl m Z ddl TddlmZddlmZmZmZmZmZmZmZd d lmZeefed d d dgddfeefeddgdddfeefedgdddfeefe ddgdddfeefe dgdddfgZ!ddZ"ddZ#dd Z$d!d"Z%d#d$Z&d%d&Z'd'd(Z(d)d*Z)d+d,Z*d-d.Z+d/d0Z,dLd2d3Z-d4d5Z.d6d7Z/d8d9Z0d:d;Z1dd?Z3d@dAZ4dBdCZ5dDdEZ6dFdGZ7dHdIZ8dJdKZ9dS)MN)*)Headersnormalize_and_validate)_obsolete_line_fold ChunkedReaderContentLengthReader Http10ReaderREADERS ReceiveBuffer)LocalProtocolError) ChunkedWriterContentLengthWriter Http10Writerwrite_any_response write_headers write_requestWRITERS)normalize_data_eventsGETz/aHostfoo Connectionclosemethodtargetheaderss1GET /a HTTP/1.1 Host: foo Connection: close OK) status_coder!reasons&HTTP/1.1 200 OK Connection: close sHTTP/1.1 200 OK e)ZUpgradeZ websocketsUpgrades,HTTP/1.1 101 Upgrade Upgrade: websocket sHTTP/1.1 101 Upgrade cCsg}|||jd|S)N)appendjoin)writerobjZgot_listr,X/Users/bob/PycharmProjects/fastapi/venv/lib/python3.9/site-packages/h11/tests/test_io.pydowrite=s r.cCst||}||ksJdSN)r.)r*r+expectedgotr,r,r-twCs r2cCst}||7}|Sr/r )databufr,r,r-makebufHsr5csfdd}t|}||||r(Jt}tt|D](}||dusNJ||||d7}q:|||t|}|d7}|||t|dksJdS)NcsP|ks Jt|dgD]2\}}t||t|tus:Jt|tusJqdS)Nr!)getattrprinttypebytes)r1namevaluer0r,r-checkOs   ztr..checkrtrailing)r5r rangelenr9)readerr3r0r=r4ir,r<r-trNs    rCcCs,tD]"\\}}}}tt||f||qdSr/) SIMPLE_CASESr2rrolestateeventbinaryr,r,r-test_writers_simpleksrJcCs,tD]"\\}}}}tt||f||qdSr/)rDrCr rEr,r,r-test_readers_simplepsrKc Cstttddgdtttgdtt,tttddddgd d dWdn1s`0Ytt(tt t d dgd d dWdn1s0YdS) Nrbar)ZbazZquuxsfoo: bar baz: quux s r/rr1.0rr r! http_versionr")r$r!rQ) r2rrrpytestraisesr rRequestrResponser,r,r,r-test_writers_unusualus2  " rVc Csttttfdtdddgddttttfdtddgddttttfdtd dgdd d ttttfd td d gdd d ttttfdtd dgdd d ttttfdtd dgdd d ttttfdtd dgddd ttttfdtd dgddd ttttfdtd ddgddd ttttfdtd ddgddd ttttfdtddgddt t $ttttfddWdn1s0Yt t $ttttfd dWdn1s0Yt t $ttttfd!dWdn1s.0Yt t $ttttfd!dWdn1sn0Yt t $ttttfd"dWdn1s0YdS)#Ns$HEAD /foo HTTP/1.0 Some: header HEAD/fooSomeheaderrOrPsHEAD /foo HTTP/1.0 s!HTTP/1.0 200 OK Some: header r"r#)r$r!rQr%s$HTTP/1.0 200 OK Foo: a a a a a )Fooz a a a a asHTTP/1.0 200 OK Foo: )r\sHTTP/1.0 200 OK Foo: sHTTP/1.0 200 Foo: bar )r\rMr's#HTTP/1.1 200 OK SomeHeader: val )Z SomeHeadervalz1.1OKs5HTTP/1.1 200 OK SomeHeader1: val1 SomeHeader2: val2 )Z SomeHeader1Zval1)Z SomeHeader2Zval2s7HTTP/1.1 200 OK SomeHeader1: val1 SomeHeader2: val2 sHEAD /foo HTTP/1.1 Host: example.com Some: multi-line header nonsense I guess Connection: close More-nonsense: in the last header )rz example.com)rZz"multi-line header nonsense I guessr)z More-nonsensezin the last headerrs&HEAD /foo HTTP/1.1 folded: line s#HEAD /foo HTTP/1.1 foo : line s"HEAD /foo HTTP/1.1 foo : line sHEAD /foo HTTP/1.1 : line ) rCr CLIENTIDLErTSERVER SEND_RESPONSErUrRrSr r,r,r,r-test_readers_unusuals                         $  $  $  $ recCs&ttgddtddgks"JdS)N)aaasbbbs cccdddrfsbbb cccrg)listr bytearrayr,r,r,r-test__obsolete_line_fold_bytes4s rjccsB||}|durq(|Vt|turq(q|r>|r4J|VdSr/)r8 EndOfMessageZread_eof)rAr4do_eofrHr,r,r-_run_reader_iterAs rmcGstt|}t|Sr/)rhrmr)argseventsr,r,r- _run_readerPs rpFc Cstdt|}t||||ks&Jtd|}t}g}tt|D](}|t||d7}||||d7}qJ|t|||7}t||ksJtdd|D}|r|st|d}t||d|ksJdS)NzTest 1zTest 2Frcss|]}t|tuVqdSr/)r8rk).0rHr,r,r- fr'z t_body_reader..r>)r7r5rpr r?r@rany) Zthunkr3r0rlr4rArorBZ is_completer,r,r- t_body_readerUs  rtcCs4tdddtgtdddtddtgdS)NcSstdS)Nrrr,r,r,r-mr'z*test_ContentLengthReader..r'cSstdS)N rur,r,r,r-rvpr's 0123456789r3)rtrkDatar,r,r,r-test_ContentLengthReaderls rzcCsLttdtgddttdtddgddttdtddtgdddS)Nr'T)rlsasdfrxF)rtr rkryr,r,r,r-test_Http10Readervs r{cCsttdtgttdtdgdgttdtddtdgdgttdtddtgttd td dtgtt$ttd td dgWdn1s0Yttttd dWdn1s0YttdtddtgdS)N0 s0 Some: header rYr!s35 01234 10 0123456789abcdef 0 Some: header s012340123456789abcdefrxs%5 01234 10 0123456789abcdef 0 saA xxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxx 0 sxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxsi9999999999999999999999999999999999999999999999999999999999999999999999999999999999999999999999999999 xxxsxxxs10 xxxsG5; hello=there xxxxx 0; random="junk"; some=more; canbe=lonnnnngg sxxxxx)rtrrkryrRrSr r,r,r,r-test_ChunkedReader~s@    2 *r~cCstd}t|tdddks Jt|tdddks8Jt|tdksLJtd}tt t|tddWdn1s0Ytd}t|tddtt t|tddWdn1s0Ytd}t|tddttt|tWdn1s&0Ytd}t|tdddkt|tdddktt"t|tdgd Wdn1s0YdS) Ns123rxs45r's123456s456ZEtagZasdfr})rr.ryrkrRrSr wr,r,r-test_ContentLengthWriters( . . , rcCst}t|tdddksJt|tdddks6Jt|tdddksNJt|tdksbJt|tdd gd d ks~JdS) Nrfrxs3 aaa saaaaaaaaaaaaaaaaaaaas14 aaaaaaaaaaaaaaaaaaaa r'r|r)abr}s0 Etag: asdf a: b )rr.ryrkrr,r,r-test_ChunkedWritersrcCsrt}t|tdddksJt|tdks2Jtt"t|tdgdWdn1sd0YdS)Ns1234rxr'rr})rr.ryrkrRrSr rr,r,r-test_Http10Writers  rcCsBtt$ttttfddWdn1s40YdS)NsHTTP/1.0 200 OKxxxx )rRrSr rCr rcrdr,r,r,r-&test_reject_garbage_after_request_lines rcCsBtt$ttttfddWdn1s40YdS)Ns&HEAD /foo HTTP/1.1 xxxxxx Host: a rRrSr rCr rarbr,r,r,r-'test_reject_garbage_after_response_lines   rcCsBtt$ttttfddWdn1s40YdS)Ns%HEAD /foo HTTP/1.1 Host: foobar rr,r,r,r-"test_reject_garbage_in_header_lines   rc CshdD]^}td}|||dtt$tttt f|dWdq1sX0YqdS)Ns sHEAD /s HTTP/1.1 Host: foobar ) rir(extendrRrSr rCr rarb)bad_charmessager,r,r-test_reject_non_vchar_in_paths    rcCs&ttttfdtddddgddS)Ns\HEAD /foo HTTP/1.1 Host: foo Set-Cookie: ___utmvafIumyLc=kUdUpAt; path=/; Max-Age=900 rWrXr)z Set-Cookiez-___utmvafIumyLc=kUdUpAt; path=/; Max-Age=900r)rCr rarbrTr,r,r,r-"test_allow_some_garbage_in_cookies s rcCstttddgddS)NrLr`sHost: example.com foo: bar )r2rrr,r,r,r-test_host_comes_firsts  r)F):rRZ_events_headersrr_readersrrrr r Z_receivebufferr _stateZ_utilr _writersrrrrrrrZhelpersrrarbrTrcrdrUZInformationalResponserDr.r2r5rCrJrKrVrerjrmrprtrzr{r~rrrrrrrrrr,r,r,r-sx  $    #!   7