Format: application/x-python-code
1 o
2
3 .'h� � @ s d dl Z d dlZd dlZd dlZd dlZd dlmZmZ d dl m
4 Z
5 d dl m Z m
6 Z
7 d dlmZ d dlZede dejf de
8 ejddf fd d
9 ��Zej�ej�e�d �Zeed �� Ze�� Z W d � n1 siw Y G d
10 d� de
11 �Z ddd�Z G dd� dej�Z ddd�Z dS )� N)�closing�contextmanager)�SimpleHTTPRequestHandler)�Callable� Generator)�urlopen�handler.�returnc c sP � t �d| �}tj|jddid�}d|_z|�� |V W |�� d S |�� w )N)z 127.0.0.1r �
12 poll_intervalg{�G�z�?)�target�kwargsT)� socketserver� TCPServer� threading�Thread�
13 serve_forever�daemon�start�shutdown)r �httpd�thread� r �XC:\Users\shubh\Videos\agent\nova\Lib\site-packages\h11/tests/test_against_stdlib_http.py�
14 socket_server s �
15 �r zdata/test-file�rbc @ s e Zd Zdedefdd�ZdS )�SingleMindedRequestHandler�pathr c C s t S )N)�test_file_path)�selfr r r r �translate_path$ s z)SingleMindedRequestHandler.translate_pathN)�__name__�
16 __module__� __qualname__�strr r r r r r # s r c C s6 t t���} tt�| j���l}t�tj�}|� |�
17 tj dddgd��� |� |�
18 t� � �� t
19 � } |�� }t|� |tju rG|�|�d�� q0t|�tju rU|jdksUJ �t|�tju ra||j7 }t|�tj u rinq1t|�tksrJ �W d � n1 s|w Y W d � d S W d � d S 1 s�w Y d S )N�GETz/foo)�Host� localhost��methodr �headersT�
20 �� )r r r �socket�create_connection�server_address�h11�
21 ConnectionZCLIENT�sendall�send�Request� EndOfMessage� bytearray�
22 next_event�print� NEED_DATA� receive_data�recv�type�Response� status_code�Data�data�bytes�test_file_data)r �s�cr? �eventr r r �test_h11_as_client( s<
23 ���
24
25 � ��"�rE c @ s e Zd Zddd�ZdS )�H11RequestHandlerr Nc C s t | j��y}t�tj�}d } |�� }|tju r!|�|�d�� qt |�tj
26 u r*|}t |�tj u r2nq|d us9J �t �
27 |j�d�|j�d�dd� |jD �d��}|�|�tjdg d��� |�|�tj|�d�d ��� |�|�t� � �� W d � d S 1 s�w Y d S )
28 NTr* �asciic S s" i | ]
29 \}}|� d �|� d ��qS )rG )�decode)�.0�name�valuer r r �
30 <dictcomp>] s ��z,H11RequestHandler.handle.<locals>.<dictcomp>r' r+ )r= r) )r? )r �requestr/ r0 ZSERVERr6 r8 r9 r: r; r3 r4 �json�dumpsr( rH r r) r1 r2 r< r> �encode)r rB rC rM rD �infor r r �handleI s4
31 �
32
33 ���
34 "�zH11RequestHandler.handle�r N)r r! r" rR r r r r rF H s rF c C s� t t��5} | j\}}d�||�}tt|���}|�� dks J �|�� }W d � n1 s-w Y W d � n1 s<w Y t� |�
35 d��}t |� |d dksUJ �|d dks]J �d|d d
36 v sgJ �d S ) Nzhttp://{}:{}/some-pathr+ rG r( r$ r z
37 /some-path�urllibr) z
38 user-agent) r rF r. �formatr r �getcode�readrN �loadsrH r7 )r �host�port�url�fr? rQ r r r �test_h11_as_serverh s
39
40
41 �� �r] rS )!rN Zos.path�osr, r
42 r �
43 contextlibr r Z http.serverr �typingr r �urllib.requestr r/ �BaseRequestHandlerr r r �join�dirname�__file__r �openr\ rW rA r rE rF r] r r r r �<module> s0 � �
44 �
45