U
    g@                  
   @   s  d dl mZmZmZmZ d dlZddlmZmZm	Z	m
Z
mZmZmZ ddlmZmZ ddlmZmZmZmZmZ ddlmZ ddlmZmZmZmZmZmZm Z m!Z!m"Z"m#Z# dd	l$m%Z% dd
l&m'Z'm(Z(m)Z)m*Z*m+Z+m,Z,m-Z- ddl.m/Z/ eefeddddgddfe"e!feddgdddfe"e!fedg dddfe"e!feddgdddfe"e!fedg dddfgZ0ed ee1ddd Z2eeedd!d"d#Z3e1ed$d%d&Z4ee1edd'd(d)Z5dd*d+d,Z6dd*d-d.Z7dd*d/d0Z8dd*d1d2Z9dd*d3d4Z:ee1e;eeddf d5d6d7Z<eee
 d8d9d:Z=dWee1ee;dd<d=d>Z>dd*d?d@Z?dd*dAdBZ@dd*dCdDZAdd*dEdFZBdd*dGdHZCdd*dIdJZDdd*dKdLZEdd*dMdNZFdd*dOdPZGdd*dQdRZHdd*dSdTZIdd*dUdVZJdS )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   )ZUpgradeZ	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<   Zgot_list rA   5/tmp/pip-unpacked-wheel-ib4s2i_e/h11/tests/test_io.pydowriteR   s    rC   )r;   r<   expectedr=   c                 C   s   t | |}||kstd 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 fdd}t|}|| | |r0tt }tt|D ](}| |d ksVt||||d  7 }qB|| | t|}|d7 }|| | t|dkstd S )N)rG   r=   c                    sF   |  kst t| dg D ](\}}t|tks0t t|tkst qd S )Nr5   )rF   getattrtypebytes)rG   namevaluerD   rA   rB   checkd   s    ztr.<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 ]"\\} }}}tt| |f || qd S rE   )SIMPLE_CASESrH   r(   rolestateeventbinaryrA   rA   rB   test_writers_simple   s    r_   c                  C   s,   t D ]"\\} }}}tt| |f || qd S rE   )rY   rX   r   rZ   rA   rA   rB   test_readers_simple   s    r`   c                	   C   s   t ttddgd t ttg d tt" t ttddddgd	d
d  W 5 Q R X tt t t	t
ddgd	dd  W 5 Q R X d S )Nr.   bar)ZbazZquuxs   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 tttf dtdddgdd t tttf dtddg dd t tttf dtd	dgdd
d t tttf dtd	dgdd
d t tttf dtd	dgdd
d t tttf dtd	dgdd
d t tttf dtd	dgddd t tttf dtd	dgddd t tttf dtd	ddgddd t tttf dtd	ddgddd t tttf dtdddddd gd! t	t
 t tttf d"d  W 5 Q R X t	t
 t tttf d#d  W 5 Q R X t	t
 t tttf d$d  W 5 Q R X t	t
 t tttf d$d  W 5 Q R X t	t
 t tttf 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valz1.1OKs5   HTTP/1.1 200 OK
SomeHeader1: val1
SomeHeader2: val2

)ZSomeHeader1Zval1)ZSomeHeader2Zval2s7   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-nonsensezin 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 tddddgdtddgks&td 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 krq(|V  t |tkr q(q |r>|r4t|  V  d S rE   )rN   r	   rF   Zread_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|  |||ks&tt d |  }t }g }tt|D ](}|t||d7 }||||d  7 }qJ|t|||7 }t||ksttdd |D }|r|st|d }t|  |d|kstd S )NzTest 1zTest 2Fr)   c                 s   s   | ]}t |tkV  qd 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 dd dt g t dd dtddt g d S )Nc                   S   s   t dS )Nr   r   rA   rA   rA   rB   <lambda>  r>   z*test_ContentLengthReader.<locals>.<lambda>r>   c                   S   s   t dS )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 tdt gdd t tdtddgdd t tdtddt gdd d S )Nr>   T)ry   s   asdfr   F)r   r   r	   r   rA   rA   rA   rB   test_Http10Reader  s       r   c                	   C   s   t tdt g t tdtdgdg t tdtddtdgdg t tdtddt g t td	td
dt g tt t tdtddg W 5 Q R X tt t tdd  W 5 Q R X t tdtddt g t tdtddt 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| tdddks tt| tdddks8tt| t dksLtt d} tt t| tdd W 5 Q R X t d} t| tdd tt t| tdd W 5 Q R X t d} t| tdd tt t| t  W 5 Q R X t d} t| tdddk t| tdddk tt t| tdgd	 W 5 Q R X d S )
N   s   123r   s   45r>   s   123456s   456ZEtagZasdfr   )r#   rC   r   rF   r	   rg   rh   r!   wrA   rA   rB   test_ContentLengthWriter  s(    r   c                  C   s   t  } t| tdddkstt| tdddks6tt| tdddksNtt| t dksbtt| tdd	gd
dks~td 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| tdddkstt| t dks2ttt t| tdgd 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 ttttf dd  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 ttttf dd  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 ttttf dd  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   dD ]J} t d}||  |d tt tttt	f |d  W 5 Q R X qd S )Ns     s   HEAD /s    HTTP/1.1
Host: foobar

)
rw   r?   extendrg   rh   r!   rX   r   r   r   )Zbad_charmessagerA   rA   rB   test_reject_non_vchar_in_path  s    

r   c                   C   s&   t tttf dtddddgd d S )Ns\   HEAD /foo HTTP/1.1
Host: foo
Set-Cookie: ___utmvafIumyLc=kUdUpAt; path=/; Max-Age=900

rj   rk   r,   )z
Set-Cookiez-___utmvafIumyLc=kUdUpAt; 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 ttddgd 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(   Zhelpersr*   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$	
  # !  
=			
