U
|
M ±d ã @ s d dl Z d dlZd dlZd dlZd dlZd dlmZmZ d dl m
|
Z
|
d dl m Z m Z d dlmZ d dlZee dejf e ejddf ddd Zej ej e¡d
|
¡Zeed Ze ¡ Z W 5 Q R X G d d d e
|
Z ddddZ G dd dejZ ddddZ dS )é N)ÚclosingÚcontextmanager)ÚSimpleHTTPRequestHandler)ÚCallableÚ Generator)Úurlopen.)ÚhandlerÚreturnc c sH t d| ¡}tj|jddid}d|_z| ¡ |V W 5 | ¡ X d S )N)z 127.0.0.1r Ú poll_intervalg{®Gáz?)ÚtargetÚkwargsT)Ú socketserverÚ TCPServerÚ threadingÚThreadÚ serve_foreverÚdaemonÚshutdownÚstart)r ÚhttpdÚthread© r úYd:\z\workplace\vscode\pyvenv\venv\Lib\site-packages\h11/tests/test_against_stdlib_http.pyÚ socket_server s ÿ
|
r zdata/test-fileÚrbc @ s e Zd ZeedddZdS )ÚSingleMindedRequestHandler)Úpathr c C s t S )N)Útest_file_path)Úselfr r r r Útranslate_path$ s z)SingleMindedRequestHandler.translate_pathN)Ú__name__Ú
|
__module__Ú __qualname__Ústrr r r r r r # s r ©r c C sú t tè} tt | j¡Ì}t tj¡}| |
|
tj dddgd¡¡ | |
|
t ¡ ¡¡ t }| ¡ }t| |tjkr| | d¡¡ q`t|tjkr¨|jdks¨tt|tjkrÀ||j7 }t|tj kr`qÒq`t|tksâtW 5 Q R X W 5 Q R X d S )NÚGETz/foo)ÚHostÚ localhost©Úmethodr Úheadersé
|
éÈ )r r r ÚsocketÚcreate_connectionÚserver_addressÚh11Ú
|
ConnectionZCLIENTÚsendallÚsendÚRequestÚ EndOfMessageÚ bytearrayÚ
|
next_eventÚprintÚ NEED_DATAÚ receive_dataÚrecvÚtypeÚResponseÚ status_codeÚAssertionErrorÚDataÚdataÚbytesÚtest_file_data)r ÚsÚcrA Úeventr r r Útest_h11_as_client( s4
|
ÿÿÿ
|
|
rG c @ s e Zd ZddddZdS )ÚH11RequestHandlerNr$ c C sø t | jä}t tj¡}d }| ¡ }|tjkr@| | d¡¡ q t |tj
|
krR|}t |tj kr qdq |d k spt t |j d¡|j d¡dd |jD d¡}| | tjdg d¡¡ | | tj| d¡d¡¡ | | t ¡ ¡¡ W 5 Q R X d S ) Nr+ Úasciic S s" i | ]\}}| d ¡| d ¡qS )rI )Údecode)Ú.0ÚnameÚvaluer r r Ú
|
<dictcomp>] s ÿ z,H11RequestHandler.handle.<locals>.<dictcomp>r( r, )r> r* )rA )r Úrequestr0 r1 ZSERVERr7 r9 r: r; r< r4 r5 r? ÚjsonÚdumpsr) rJ r r* r2 r3 r= r@ Úencode)r rD rE rO rF Úinfor r r ÚhandleI s.
|
|
|
þýÿ
|
zH11RequestHandler.handle)r r! r" rT r r r r rH H s rH c
|
C sª t tL} | j\}}d ||¡}tt| }| ¡ dks>t| ¡ }W 5 Q R X W 5 Q R X t
|
| d¡¡}t | |d dkst|d dkstd|d d
|
ks¦td S ) Nzhttp://{}:{}/some-pathr, rI r) r% r z
|
/some-pathÚurllibr* z
|
user-agent) r rH r/ Úformatr r Úgetcoder? ÚreadrP ÚloadsrJ r8 )r ÚhostÚportÚurlÚfrA rS r r r Útest_h11_as_serverh s
|
|
r^ )!rP Úos.pathÚosr- r r Ú
|
contextlibr r Z http.serverr Útypingr r Úurllib.requestr r0 ÚBaseRequestHandlerr r r ÚjoinÚdirnameÚ__file__r Úopenr] rX rC r rG rH r^ r r r r Ú<module> s( þ
|
