Your IP : 216.73.216.196
o
0�i#E � @ sP d dl Z d dlmZmZ d dlZd dlZd dlZd dlZd dlZd dl Z d dl
Z
d dl Z d dl Z d dl m
Z
e jZd dlmZ d dlmZ d dlmZ d dlmZ ej�e�Zej�ed�Zej�ed �Zej�ed
�Z d Z d Z d
Z dZ de d Z!dZ"dZ#ej$Z$G dd� d�Z%G dd� de%�Z&G dd� dej'�Z(G dd� dej)�Z*G dd� de�Z+G d d � d e�Z,G d d � d e�Z-G d d!� d!e�Z.G d"d#� d#e�Z/G d$d%� d%e/�Z0G d&d'� d'�Z1G d(d)� d)e�Z2G d*d+� d+e�Z3G d,d-� d-e�Z4G d.d/� d/e�Z5G d0d1� d1e�Z6G d2d3� d3e�Z7G d4d5� d5e�Z8G d6d7� d7e�Z9e:d8k�r&e j;d9d:� dS dS );� N)�client�
HTTPStatus)�mock)�support)� os_helper)�
socket_helper)�warnings_helperz keycert.pemz keycert2.pemz selfsigned_pythontestdotnet.pemz{HTTP/1.1 200 OK
Transfer-Encoding: chunked
a
hello worl
3
d!
8
and now
22
for something completely different
s7 hello world! and now for something completely differentz;foo=barz0
�0�
z X-Dummy: foo
X-Dumm2: bar
c @ sH e Zd Zejddfdd�Zdd� Zddd�Zdd � Zd
d � Z d d
� Z
dS )�
FakeSocketNc C sB t |t�r
|�d�}|| _|| _d| _d| _d| _|| _|| _ d S )N�ascii� r F)
�
isinstance�str�encode�text� fileclass�data�
sendall_calls� file_closed�host�port)�selfr r r r � r �8/opt/alt/python310/lib64/python3.10/test/test_httplib.py�__init__5 s
zFakeSocket.__init__c C s | j d7 _ | j|7 _d S )N� )r r �r r r r r �sendall@ s zFakeSocket.sendallc C s6 |dkr |dkr t �� �| �| j�| _| j| j_| jS )N�r�rb)r ZUnimplementedFileModer r �file�
file_close�close)r �mode�bufsizer r r �makefileD s
zFakeSocket.makefilec C s
d| _ d S )NT)r �r r r r r" L �
zFakeSocket.file_closec C � d S �Nr r' r r r r# O � zFakeSocket.closec C r) r* r )r �levelZoptname�valuer r r �
setsockoptR r+ zFakeSocket.setsockoptr* ) �__name__�
__module__� __qualname__�io�BytesIOr r r& r" r# r. r r r r r 4 s
r c @ �$ e Zd Zdd� Zdd� Zdd� ZdS )� EPipeSocketc C s t �| |� || _d S r* )r r � pipe_trigger)r r r6 r r r r W s
zEPipeSocket.__init__c C s( | j |v r ttjd��| j|7 _d S )NZgotcha)r6 �OSError�errnoZEPIPEr r r r r r \ s
zEPipeSocket.sendallc C r) r* r r' r r r r# a r+ zEPipeSocket.closeN)r/ r0 r1 r r r# r r r r r5 U s r5 c @ s$ e Zd ZdZddd�Zd dd�ZdS )
� NoEOFBytesIOz�Like BytesIO, but raises AssertionError on EOF.
This is used below to test that http.client doesn't try to read
more from the underlying file than it should.
���c C �" t j�| |�}|dkrtd��|S �Nr
z caller tried to read past EOF)r2 r3 �read�AssertionError)r �nr r r r r= j � zNoEOFBytesIO.readNc C r; r< )r2 r3 �readliner> )r Zlengthr r r r rA p r@ zNoEOFBytesIO.readline�r: r* )r/ r0 r1 �__doc__r= rA r r r r r9 d s
r9 c s4 e Zd ZdZ� fdd�Z� fdd�Zdd� Z� ZS )�FakeSocketHTTPConnectionz@HTTPConnection subclass using FakeSocket; counts connect() callsc s$ d| _ t� �d� || _| j| _d S )Nr � example.com)� connections�superr �fake_socket_args�create_connection�_create_connection)r �args�� __class__r r r y s z!FakeSocketHTTPConnection.__init__c s | j d7 _ t� �� S )z.Count the number of times connect() is invokedr )rF rG �connectr' rL r r rN s
z FakeSocketHTTPConnection.connectc O s
t | j� S r* )r rH �r �pos�kwr r r rI � r( z*FakeSocketHTTPConnection.create_connection)r/ r0 r1 rC r rN rI �
__classcell__r r rL r rD v s
rD c @ sL e Zd Zdd� Zdd� Zdd� Zdd� Zd d
� Zd d � Zd
d� Z dd� Z
dS )� HeaderTestsc C s� G dd� dt �}dD ]7}dD ]2}t�d�}td�|_|� |_d}i }|r,tt|��||< |�dd ||� | � |jj
|� � d
� qq
d S ) Nc @ � e Zd Zdd� Zdd� ZdS )z;HeaderTests.test_auto_headers.<locals>.HeaderCountingBufferc S s
i | _ d S r* )�countr' r r r r � r( zDHeaderTests.test_auto_headers.<locals>.HeaderCountingBuffer.__init__c S sX |� d�}t|�dkr$|d �d��� }| j�|d� | j| d7 < t�| |� d S )N� :r r r )�split�len�decode�lowerrU �
setdefault�list�append)r �item�kvZlcKeyr r r r] � s
zBHeaderTests.test_auto_headers.<locals>.HeaderCountingBuffer.appendN�r/ r0 r1 r r] r r r r �HeaderCountingBuffer� s ra )TF)�Content-lengthZHostzAccept-encodingrE Z blahblahblahZ spamspamspam�POST�/r ) r\ r �HTTPConnectionr �sock�_bufferr rX �request� assertEqualrU rZ )r ra Zexplicit_header�header�conn�body�headersr r r �test_auto_headers� s
��z HeaderTests.test_auto_headersc C sd G dd� dt �}d}d}t�||�D ]%\}}t�d�}td �|_|� |_|�|d|� | � |jj
dd� |�� qd }|D ]#}t�d�}td �|_|� |_|�|dd � | � |jj
d d
� |�� q<|D ]#}t�d�}td �|_|� |_|�|dd � | � |jj
dd� |�� qbt� ||�D ]#}t�d�}td �|_|� |_|�|dd � | � |jj
d
d� |�� q�d S )Nc @ rT )z?HeaderTests.test_content_length_0.<locals>.ContentLengthCheckerc S s t �| � d | _d S r* )r\ r �content_lengthr' r r r r � �
zHHeaderTests.test_content_length_0.<locals>.ContentLengthChecker.__init__c S sF |� dd�}t|�dkr|d �� dkr|d �� | _t�| |� d S )NrV r r s content-length)rW rX rZ �stripro r\ r] )r r^ r_ r r r r] � s zFHeaderTests.test_content_length_0.<locals>.ContentLengthChecker.appendNr` r r r r �ContentLengthChecker� s rr )N� )�PUTrc ZPATCHrE rd � 0z%Header Content-Length incorrect on {})�GET�CONNECTZDELETE�HEADZOPTIONSZTRACEz.Header Content-Length set for empty body on {}rs � � 1)
r\ � itertools�productr re r rf rg rh ri ro �format�chain)r rr ZbodiesZmethods_with_body�methodrl rk Zmethods_without_bodyr r r �test_content_length_0� sR
�
�
�
��z!HeaderTests.test_content_length_0c C s� t �d�}td �|_|�dd� |�dd� | �d|j� |�dd� | �d |j� |�d
d � | �d |j� |�d
d� | �d|j� |�ddd� | �d|j� |�dd� | �d|j� |�dd� | �d|j� |�dd� | �d |j� |�d d � | �d |j� |�d d!� | �d"|j� |�d#d$� | �d%|j� |�d&d$� | �d'|j� |�d(d$� | �d)|j� |�d*d$� | �d+|j� d S ),NrE rv rd rb �* s Content-length: 42ZFooz bar s
Foo: bar ZBarz baz s
Bar: baz Z
AuthorizationzBearer mytokens Authorization: Bearer mytokenZ
IterHeaderZIterAZIterBs IterHeader: IterA
IterBZ LatinHeader� �s LatinHeader: �Z
Utf8Headers Às Utf8Header: Àz
C1-Controls next�lines C1-Control: next�linezEmbedded-Fold-Spacez is
alloweds! Embedded-Fold-Space: is
allowedzEmbedded-Fold-Tabz is
alloweds Embedded-Fold-Tab: is
allowedz Key Spacer- s Key Space: valuez KeySpace s KeySpace : values Nonbreak�Spaces Nonbreak�Space: values �NonbreakSpaces �NonbreakSpace: value)r re r rf �
putrequest� putheader�assertInrg �r rk r r r �test_putheader� s>
zHeaderTests.test_putheaderc C sx d}t �d�}td�}||_|�dd� | �|j�|�� d}t �d�}td�}||_|�dd� | �|j�|�� d S )NsC GET /foo HTTP/1.1
Host: [2001::]:81
Accept-Encoding: identity
z [2001::]:81rs rv �/foosE GET /foo HTTP/1.1
Host: [2001:102A::]
Accept-Encoding: identity
z
[2001:102A::])r re r rf rh �
assertTruer �
startswith�r �expectedrk rf r r r �test_ipv6host_header s
z HeaderTests.test_ipv6host_headerc C sF d}t |�}t�|�}|�� | �|�d�d� | �|�d�d� d S )Nz4HTTP/1.1 200 OK
First: val
: nval
Second: val
ZFirst�valZSecond)r r � HTTPResponse�beginri � getheader�r rl rf �respr r r �!test_malformed_headers_coped_with s
z-HeaderTests.test_malformed_headers_coped_withc C s, dt tdd�� d t tdd�� d }t|�}t�|�}|�� | �|�d�d � | �|jd d � | �|�d
�d � | �|jd
d � d � t
t tdd���}| �|�d
�|� | �|jd
|� | � |�d�� | �
d|j� |�d�|jd fD ]}| �|�d�� | �
d|� | �|�d�� q{d S )Ns0 HTTP/1.1 200 OK
!#$%&'*+-.^_`|~: value
VCHAR: �! � s
obs-text: � � sM
obs-fold: text
folded with space
folded with tab
Content-Length: 0
�Content-Lengthr z!#$%&'*+-.^_`|~r- rs ZVCHARzobs-textzobs-foldr z folded with spacezfolded with tab)�bytes�ranger r r� r� ri r� �msg�join�map�chrZassertIsNotNoner� r� r� �endswith)r rl rf r� ZvcharZfoldedr r r �test_parse_all_octets& s6 �� ���
�z!HeaderTests.test_parse_all_octetsc
C s� t �d�}td�|_|�dd� d}|D ]7\}}| �||f��$ | �td�� |�||� W d � n1 s7w Y W d � n1 sFw Y qd S )NrE rs rv rd ))s
Invalid
Name�
ValidValue)s Invalid
Namer� )s Invalid
Namer� )s
InvalidNamer� )s
InvalidNamer� )s
InvalidNamer� )s InvalidNamer� )s InvalidNamer� )s Invalid:Namer� )s :InvalidNamer� )� ValidNames Invalid
Value)r� s
Invalid
Value)r� s
Invalid
Value)r� s InvalidValue
)r� s
InvalidValue
)r� s
InvalidValue
zInvalid header) r re r rf r� �subTest�assertRaisesRegex�
ValueErrorr� )r rk �cases�namer- r r r �test_invalid_headersD s
�� ���z HeaderTests.test_invalid_headersc C s� d}t |�}tj|dd�}t�� � }|�� W d � n1 s w Y |�� �� }| �|d d� | �|d d� | �|d d� | �|d d
� d S ) Ns9 HTTP/1.1 200 OK
First: val
Second: val1
Second: val2
r )Z
debuglevelr z reply: 'HTTP/1.1 200 OK\r\n'zheader: First: val� zheader: Second: val1� zheader: Second: val2) r r r� r �captured_stdoutr� �getvalue�
splitlinesri )r rl rf r� �output�linesr r r �test_headers_debuglevelb s �
� z#HeaderTests.test_headers_debuglevelN) r/ r0 r1 rn r� r� r� r� r� r� r� r r r r rS � s B"
rS c @ � e Zd Zdd� ZdS )�HttpMethodTestsc C s` d}|D ])}| � td�� t�d�}td �|_|j|dd� W d � n1 s(w Y qd S )N) zGET
zPOST
zPUT
z
POST
Valuez
POST
HOST:abczGET
rHost:abc
zPOST
Remainder:
z
GET
HOST:
z
PUTz'method can't contain control charactersrE rd )r �url)r� r� r re r rf rh )r �methodsr rk r r r �test_invalid_method_namesu s �
���z)HttpMethodTests.test_invalid_method_namesN)r/ r0 r1 r� r r r r r� t � r� c @ sJ e Zd ZdZdd� Zdd� Zdd� Zdd � Zdd d �Zd
d� Z dd� Z
dS )�TransferEncodingTests It's just a flesh woundc C sb t �d�}td�|_|�dd� |j| �� dd� | �|jj�\}}}| � |�}| �
|| j � d S )NrE r
rc rd T)�encode_chunked) r re r rf r� Z
endheaders�
_make_body�_parse_requestr �_parse_chunkedri �
expected_body)r rk �_rl r r r �test_endheaders_chunked� s
z,TransferEncodingTest.test_endheaders_chunkedc C sv t �d�}td�|_|�dd| �� ddi� | �|jj�\}}}| �ddd � |� � D �� | �
|d d� | �
|| j � t �d�}td�|_|�dd| j � d
�ddi� | �|jj�\}}}| �dd d � |� � D �� | �
|d d� | �
|| j � t �d�}td�|_|jdddd id
| �� d� | �|jj�\}}}| �ddd � |D �� | �
|d d � | �
| �
|�| j � d S )NrE r
rc rd �Transfer-Encoding�chunked�content-lengthc S � g | ]}|� � �qS r �rZ ��.0�kr r r �
<listcomp>� � z>TransferEncodingTest.test_explicit_headers.<locals>.<listcomp>�latin-1c S r� r r� r� r r r r� � r� z
gzip, chunkedT)rm r� rl c S r� r r� r� r r r r� � r� )r re r rf rh r� r� r � assertNotIn�keysri r� rY r� �r rk r� rm rl r r r �test_explicit_headers� s<
�
�
�z*TransferEncodingTest.test_explicit_headersc C s� dD ]@}t �d�}td�|_|�dd| j|d�� | �|jj�\}}}| �|�}| � || j
� | � |d d� | � d d
d � |D �� qd S ) N)FTrE r
rc rd )� empty_linesr� r� r� c S r� r r� r� r r r r� � r� z5TransferEncodingTest.test_request.<locals>.<listcomp>) r re r rf rh r� r� r r� ri r� r� )r r� rk r� rm rl r r r � test_request� s
�
�z!TransferEncodingTest.test_requestc C sl t �d�}td�|_|�ddd� | �|jj�\}}}| �|d d� | �dd d
� |D �� | �|d � d S ) NrE r
rc rd r r� r� r� c S r� r r� r� r r r r� � r� z8TransferEncodingTest.test_empty_body.<locals>.<listcomp>s 0
) r re r rf rh r� r ri r� r� r r r �test_empty_body� s
z$TransferEncodingTest.test_empty_bodyFc c sX � | j �d�}t|�D ] \}}|r|d rdV |t|�d k r&|d V q |V q d S )N� r� r
r )r� rW � enumeraterX )r r� r� �idx�liner r r r� � s � �z TransferEncodingTest._make_bodyc C s� |� d�}|d }i }d}|t|�k rFt|| �dkrF|| � d�\}}|�d��� }|�d��� ||< |d7 }|t|�k rFt|| �dks||d�||d d � �fS )N�
r r rV r� )rW rX rY rq r� )r r r� rh rm r? �keyr� r r r r� � s
� z#TransferEncodingTest._parse_requestc C s� g }i }d}|� d�} |||d � \}}t|d�}|dkr$|d7 }n| �|t|�� |�|� |d7 }|t|�kr<nq d�|�S )Nr r� Tr� � r r
)rW �intri rX r] r� )r r rl �trailersr? r� �size�chunkr r r r� � s"
�
z#TransferEncodingTest._parse_chunkedN)F) r/ r0 r1 r� r� r� r� r� r� r� r� r r r r r� � s
&
r� c @ s� e Zd Zdd� Zdd� Zdd� Zdd� Zd d
� Zd d � Zd
d� Z dd� Z
dd� Z dd� Z dd� Z
dd� Zdd� Zdd � Zd d � Zd d � Zd!d"� Zd#d$� Zd%d&� Zd'd(� Zd)d*� Zd+d,� Zd-d.� Zd/d0� Zd1d2� Zd3d4� Z d5d6� Z d7d8� Z d9d:� Z d;d<� Z d=d>� Z!d?d@� Z"dAdB� Z#dCdD� Z$dEdF� Z%dGdH� Z&dIdJ� Z'dKdL� Z(dMdN� Z)dOdP� Z*dQdR� Z+dSdT� Z,dUdV� Z-dWdX� Z.dYdZ� Z/d[d\� Z0d]d^� Z1d_d`� Z2dadb� Z3dcdd� Z4dedf� Z5dgdh� Z6diS )j� BasicTestc C s" | � h d�tttd���k� d S )N> � descriptionr� �phraser- � )r� �set�dirr r' r r r �&test_dir_with_added_behavior_on_status s "z0BasicTest.test_dir_with_added_behavior_on_statusc C s� d}t |�}t�|�}|�� | �|�d�d� | �|�� � | �|j� | �|�� d� | � |�� � | �|j� |�
� | � |j� d}t |�}t�|�}| � tj |j� d S )N�HTTP/1.1 200 Ok
Textr r
� Textz HTTP/1.1 400.100 Not Ok
Text)
r r r� r� ri r= � assertFalse�isclosed�closedr� r# � assertRaises�
BadStatusLiner� r r r �test_status_lines s
zBasicTest.test_status_linesc C s t �d�}| �t|�d� d S )Nrs zBadStatusLine("''"))r r� ri �repr)r �excr r r �test_bad_status_repr$ s
z BasicTest.test_bad_status_reprc C s� d}t |�}t�|�}|�� | �|�d�d� | �|�� � | �|�d�d� | �|�� � | �|j � |�
� | �|j � d S �N�*HTTP/1.1 200 Ok
Content-Length: 4
Textr� � Te� xt� r r r� r� ri r= r� r� r� r� r# r� r r r �test_partial_reads( s
z BasicTest.test_partial_readsc C s~ d}t |�}t�|�}|�� | �|�� d� | �|�� � | �|�� d� | � |�� � | �|j
� |� � | � |j
� d S )Nz4HTTP/1.1 200 Ok
Content-Length: 13
Text
Anothers Text
s Another) r r r� r� ri rA r� r� r= r� r� r# r� r r r �test_mixed_reads7 s
zBasicTest.test_mixed_readsc C s� d}t |�}t�|�}|�� td�}|�|�}| �|d� | �t|�d� | �|� � � |�|�}| �|d� | �t|�d� | �
|� � � | �|j � |� � | �
|j � d S r� �
r r r� r� � bytearray�readintori r� r� r� r� r� r# �r rl rf r� �br? r r r �test_partial_readintosF s
z BasicTest.test_partial_readintosc C sb d}t |�}t�|�}|�� | �|�d�d� | �|�� � | �|j � |�
� | �|j � d S )Nr� �
r� � r r r� r� ri r= r� r� r� r� r# r� r r r �test_partial_reads_past_endZ s
z%BasicTest.test_partial_reads_past_endc C s� d}t |�}t�|�}|�� td�}|�|�}| �|d� | �t|�d d� d� | �|� � � | �
|j � |� � | �|j � d S )Nr� r� � r� )
r r r� r� r� r� ri r� r� r� r� r� r# r� r r r � test_partial_readintos_past_endf s
z)BasicTest.test_partial_readintos_past_endc C s� d}t |�}t�|�}|�� | �|�d�d� | �|�� � | �|�d�d� | �|�d�d� | �|�� � | �|j � |�
� | �|j � d S )Nr� r� r� r� r r
r� r� r r r �$test_partial_reads_no_content_lengthu s
z.BasicTest.test_partial_reads_no_content_lengthc C s� d}t |�}t�|�}|�� td�}|�|�}| �|d� | �t|�d� | �|� � � |�|�}| �|d� | �t|�d� |�|�}| �|d� | �
|� � � d S )Nr� r� r� r� r ) r r r� r� r� r� ri r� r� r� r� r� r r r �(test_partial_readintos_no_content_length� s
z2BasicTest.test_partial_readintos_no_content_lengthc C st d}t |�}t�|�}|�� | �|�d�d� | �|�� � | �|�d�d� | �|�d�d� | �|�� � d S )N�+HTTP/1.1 200 Ok
Content-Length: 10
Textr� r� r� r r
) r r r� r� ri r= r� r� r� r� r r r �"test_partial_reads_incomplete_body� s
z,BasicTest.test_partial_reads_incomplete_bodyc C s� d}t |�}t�|�}|�� td�}|�|�}| �|d� | �t|�d� | �|� � � |�|�}| �|d� | �t|�d� |�|�}| �|d� | �
|� � � | �|j � |� � | �
|j � d S )Nr r� r� r� r r� r� r r r �&test_partial_readintos_incomplete_body� s$
z0BasicTest.test_partial_readintos_incomplete_bodyc C �V dD ] }| � tjtj|� qdD ]\}}}t�|�}| �||j� | �||j� qd S )N�zwww.python.org:abcz user:password@www.python.org)�z[fe80::207:e9ff:fe9b]:8000�fe80::207:e9ff:fe9bi@ )zwww.python.org:80�www.python.org�P )�www.python.org:r
r )r
r
r )�[fe80::207:e9ff:fe9b]r r )�[fe80::207:e9ff:fe9b]:r r )r� r �
InvalidURLre ri r r �r Zhp�h�p�cr r r �test_host_port� s
�zBasicTest.test_host_portc C s< d}d}t |�}t�|�}|�� |�d�}| �||� d S )Nz�HTTP/1.1 200 OK
Set-Cookie: Customer="WILE_E_COYOTE"; Version="1"; Path="/acme"
Set-Cookie: Part_Number="Rocket_Launcher_0001"; Version="1"; Path="/acme"
No body
zrCustomer="WILE_E_COYOTE"; Version="1"; Path="/acme", Part_Number="Rocket_Launcher_0001"; Version="1"; Path="/acme"z
Set-Cookie)r r r� r� r� ri )r r Zhdr�sr Zcookiesr r r �test_response_headers� s
z BasicTest.test_response_headersc C s: t dt�}tj|dd�}|�� |�� r| �d� d S d S )N�*HTTP/1.1 200 OK
Content-Length: 14432
rx �r �)Did not expect response from HEAD request)r r9 r r� r� r= �fail�r rf r� r r r �test_read_head� s ��zBasicTest.test_read_headc C sT t dt�}tj|dd�}|�� td�}|�|�dkr | �d� | �t |�d� d S )Nr rx r � r r � )
r r9 r r� r� r� r� r ri r� )r rf r� r� r r r �test_readinto_head� s �
z BasicTest.test_readinto_headc C sR d� dd� ttjd �D ��d }d| }t|�}t�|�}| �tjd|j� d S )Nr
c s s � | ]}d | V qdS )z
Header%d: fooNr )r� �ir r r � <genexpr>� s � �z2BasicTest.test_too_many_headers.<locals>.<genexpr>r zHTTP/1.1 200 OK
zgot more than \d+ headers) r� r� r Z _MAXHEADERSr r� r� �
HTTPExceptionr� )r rm r r r r r r �test_too_many_headers� s
��
�z BasicTest.test_too_many_headersc C s� d}t td��1}t�d�}t|�}||_|�dd|� | �|j� |�d|jd t
|�� |f � W d � d S 1 s;w Y d S )Ns_ GET /foo HTTP/1.1
Host: example.com
Accept-Encoding: identity
Transfer-Encoding: chunked
r rE rv r� z%r != %r) �open�__file__r re r rf rh r� r r� rX �r r� rl rk rf r r r �test_send_file s
�"�zBasicTest.test_send_filec C s� d}t �d�}td �}||_|�|� | �||j� d|_|�t�d|�� | �||j� d|_|�t� |�� | �||j� d S )Ns" this is a test this is only a testrE r
r� )
r re r rf �sendri r �arrayr2 r3 r� r r r � test_send s
zBasicTest.test_sendc sX dd� � G � fdd�dt j�}d}t�d�}td�}||_|�|� � | �|j|� d S )Nc s s � dV d V dV d S )Nr Zdata_twor r r r r r � �
z/BasicTest.test_send_updating_file.<locals>.datac s e Zd ZdZ� � Zddd�ZdS )z7BasicTest.test_send_updating_file.<locals>.UpdatingFiler r: c S s
t | j�S r* )�next�d)r � blocksizer r r r= ( r( z<BasicTest.test_send_updating_file.<locals>.UpdatingFile.readNrB )r/ r0 r1 r$ r- r= r �r r r � UpdatingFile% s r0 s datarE rs ) r2 �
TextIOBaser re r rf r( ri r )r r0 r� rk rf r r/ r �test_send_updating_file s
z!BasicTest.test_send_updating_filec C sL d}dd� }t �d�}td�}||_|�dd|� dd i� | �|j|� d S )
Nsb GET /foo HTTP/1.1
Host: example.com
Accept-Encoding: identity
Content-Length: 11
onetwothreec s s � dV dV dV d S )Ns ones twos threer r r r r rl 9 r+ z&BasicTest.test_send_iter.<locals>.bodyrE rs rv r� r� Z11)r re r rf rh ri r r& r r r �test_send_iter4 s
zBasicTest.test_send_iterc C sv d}t jd|d�}td�}||_d| d }|�ddt�|�d d
i� | �|jd � |j �
d d
�d
}| �||� dS )z8Check that request() respects the configured block size.� rE �r. N� a� brt rd r� �9r� �
r ) r re r rf rh r2 r3 ri r r rW )r r. rk rf r� rl r r r �test_blocksize_requestD s z BasicTest.test_blocksize_requestc C s\ d}t jd|d�}td�}||_d| d }|�t�|�� | �|jd� | �|j |� dS )z5Check that send() respects the configured block size.r4 rE r5 Nr6 r7 r� )
r re r rf r( r2 r3 ri r r )r r. rk rf r� r r r �test_blocksize_sendP s z BasicTest.test_blocksize_sendc C sR t �d�}td�|_| �t�� |�dd|� W d � d S 1 s"w Y d S )NrE rs rc �test)r re r rf r� � TypeErrorrh r� r r r �test_send_type_error[ s
"�z BasicTest.test_send_type_errorc C sb t }ttt t �}tj|dd�}|�� | �|� � |� |�
� t dd�D ]+}ttt t �}tj|dd�}|�� | �|� |�|� |� |� � |� |�
� q&dD ]Z}tt| �}tj|dd�}|�� zBz|� � W n0 tj y� } z#| �|j
|� dt|� }| �t|�|� | �t|�|� W Y d }~n
d }~ww | �d� W |�
� qT|�
� w d S )Nrv r r � �rs zfoo
� IncompleteRead(%d bytes read)�IncompleteRead expected)�chunked_expectedr �
chunked_start�
last_chunk� chunked_endr r� r� ri r= r# r� �IncompleteRead�partialrX r� r r )r r� rf r� r? �xr �expected_messager r r � test_chunkedb s: $
��
��zBasicTest.test_chunkedc C s� t }t|�}td�}ttt t �}tj|dd�}|� � |�
|�}| � |d |� |� | � ||� |� � t
dd�D ]N}ttt t �}tj|dd�}|� � t|�}|�
|d|� �}||�
|||| � �7 }||�
||d � �7 }| � |d |� |� | � ||� |� � q;dD ][} tt| �}tj|dd�}|� � zCz|�
|�}W n0 tjy� } z#| � |j|� dt|� }
| � t|�|
� | � t|�|
� W Y d }~n
d }~ww | �d � W |� � q�|� � w d S )
Nr� rv r r r? r r@ rA rB )rC rX r� r rD rE rF r r� r� r� ri r# r� �
memoryviewrG rH r� r r ) r r� Z nexpectedr� rf r� r? �mr rI rJ r r r �test_readinto_chunked� sL
��
��z BasicTest.test_readinto_chunkedc C s� d}t |t t �}tj|dd�}|�� | �|�� d� | �|jd� | �|j d� | �
|� � � | � |j
� |�� | �
|j
� d S )N�EHTTP/1.1 200 OK
Transfer-Encoding: chunked
a
hello world
1
d
rx r r
�� �OK)r rE rF r r� r� ri r= �status�reasonr� r� r� r� r# )r rD rf r� r r r �test_chunked_head� s � zBasicTest.test_chunked_headc C s� d}t |t t �}tj|dd�}|�� td�}|�|�}| �|d� | �t |�d� | �|j
d� | �|j d� | � |�
� � | �|j� |�� | � |j� d S ) NrO rx r r r r rP rQ )r rE rF r r� r� r� r� ri r� rR rS r� r� r� r� r# )r rD rf r� r� r? r r r �test_readinto_chunked_head� s �
z$BasicTest.test_readinto_chunked_headc C s@ t d�}tj|dd�}|�� | �|�� d� | �|�� � d S )Nz.HTTP/1.1 200 OK
Content-Length: -1
Hello
rv r � Hello
)r r r� r� ri r= r� r� r r r r � test_negative_content_length� s �z&BasicTest.test_negative_content_lengthc
C s� t d�}tj|dd�}|�� z|�� W n2 tjyG } z%| �|jd� | �t|�d� | �t |�d� | �
|� � � W Y d }~d S d }~ww | � d� d S )Nz.HTTP/1.1 200 OK
Content-Length: 10
Hello
rv r rV z-IncompleteRead(7 bytes read, 3 more expected)rB )
r r r� r� r= rG ri rH r� r r� r� r )r rf r� r r r r �test_incomplete_read� s"
�
� ��z BasicTest.test_incomplete_readc sZ t dd�}t�d�� |� _| �t� fdd�� � �� }| �d|j� | �d|� d�� d S ) NzgHTTP/1.0 401 Authorization Required
Content-type: text/html
WWW-Authenticate: Basic realm="example"
s Content-LengthrE c s � � ddd�S )Nrt �/urlrl )rh r �rk r r �<lambda>� s z&BasicTest.test_epipe.<locals>.<lambda>i� zBasic realm="example"zwww-authenticate)
r5 r re rf r� r7 � getresponseri rR r� r r rZ r �
test_epipe� s �
��zBasicTest.test_epipec C s8 ddd d }t �t|��}| �t jt jf|j� d S )N�HTTP/1.1 200 Okr� � r
)r r� r r� � LineTooLongr� r� �r rl r� r r r � test_overflowing_status_line s z&BasicTest.test_overflowing_status_linec C s2 ddd d }t �t|��}| �t j|j� d S )NzHTTP/1.1 200 OK
X-Foo: barr r_ z
)r r� r r� r` r� ra r r r � test_overflowing_header_line s ���z&BasicTest.test_overflowing_header_linec C sr dd }t �t|��}| �t j�� }|�� W d � n1 s w Y | �dt|j�� | �dt|j�� d S )NzHTTP/1.1 100 OK
r
i � zgot more than rm ) r r� r r� r" r� r� r � exception)r rl r� �cmr r r �'test_overflowing_header_limit_after_100 s ��
�z1BasicTest.test_overflowing_header_limit_after_100c C s: ddd d }t �t|��}|�� | �t j|j� d S )Nz/HTTP/1.1 200 OK
Transfer-Encoding: chunked
r r_ za
hello world
0
)r r� r r� r� r` r= ra r r r � test_overflowing_chunked_line s ���z'BasicTest.test_overflowing_chunked_linec C s` d}t |�}t�|�}|�� | �|�� d� | �|�� � | �|j � |�
� | �|j � d S )Nr^ r
r� r� r r r �test_early_eof( s
zBasicTest.test_early_eofc sn t �d�}d � G � fdd�dt j�}||_td�|_|�dd� | �t j|j � | �
� j � | �
|jj � d S )NrE c s e Zd Z� fdd�ZdS )z+BasicTest.test_error_leak.<locals>.Responsec s" | � t jj| g|�R i |�� d S r* )r r� r rO ��responser r r 9 s z4BasicTest.test_error_leak.<locals>.Response.__init__N)r/ r0 r1 r r ri r r �Response8 s rk zInvalid status linerv rd )
r re r� Zresponse_classr rf rh r� r� r\ r� r� r )r rk rk r ri r �test_error_leak4 s
zBasicTest.test_error_leakc C sR d}t d }tt| t t �}tj|dd�}|�� | �|� � |� |�
� d S )Nz3;foo=bar
abc
s abcrv r ) rC r rD �last_chunk_extendedrF r r� r� ri r= r# )r Zextrar� rf r� r r r �test_chunked_extensionD s z BasicTest.test_chunked_extensionc C sB t }ttt �}tj|dd�}|�� | �|�� |� |� � dS )z9some servers may serve up a short chunked encoding streamrv r N)
rC r rD rE r r� r� ri r= r# �r r� rf r� r r r �test_chunked_missing_endN s z"BasicTest.test_chunked_missing_endc C s\ t }ttt t t �}tj|dd�}|�� | � |�
� |� | � |j �
� d� |� � dS )z&See that trailers are read and ignoredrv r r
N)
rC r rD rE r� rF r r� r� ri r= r! r# ro r r r �test_chunked_trailersW s z BasicTest.test_chunked_trailersc C sj t }d}ttt t t | �}tj|dd�}|�� | � |�
� |� | � |j �
� |� d�� |�
� dS )zDCheck that we don't read past the end of the chunked-encoding stream� extradatarv r r N)rC r rD rE r� rF r r� r� ri r= r! r r# )r r� rr rf r� r r r �test_chunked_syncb s zBasicTest.test_chunked_syncc C s\ d}d}t d| | �}tj|dd�}|�� | �|�� |� | �|j�� |� |�� dS )zBCheck that we don't read past the end of the Content-Length stream� extradata�
Hello123
�' HTTP/1.1 200 OK
Content-Length: 10
rv r N)r r r� r� ri r= r! r# �r rr r� rf r� r r r �test_content_length_syncn s z"BasicTest.test_content_length_syncc C s` d}d}t d| | �}tj|dd�}|�� | �|�d�|g� | �|j�� |� |�� d S �Nrt ru rv rv r i� ) r r r� r� ri � readlinesr! r= r# rw r r r � test_readlines_content_lengthz s z'BasicTest.test_readlines_content_lengthc C s^ d}d}t d| | �}tj|dd�}|�� | �|�d�|� | �|j�� |� |�� d S ry ) r r r� r� ri �read1r! r= r# rw r r r �test_read1_content_length� s z#BasicTest.test_read1_content_lengthc C sp d}d}t d| | �}tj|dd�}|�� | �|�d�|� | �|�d�d� | �|j�� |� |�� d S )Nrt ru rv rv r r� r
) r r r� r� ri rA r! r= r# rw r r r �"test_readline_bound_content_length� s z,BasicTest.test_readline_bound_content_lengthc C sv d}d}t d|d | �}tj|dd�}|�� | �|�d�|d � | �|�� |� | �|j�� |� |�� d S ) Nrt ru s' HTTP/1.1 200 OK
Content-Length: 30
r� rv r � r� ) r r r� r� ri r| r= r! r# rw r r r � test_read1_bound_content_length� s z)BasicTest.test_read1_bound_content_lengthc s� t �tdf��| ��j� d � � �fdd�}tj|d�}|�� | �|jt d�� t
j �� � � }|�
dd� |�� }z+| �|jt
j� t j |�� d�}z |�d � W |�� n|�� w W |�� |�� n |�� |�� w |�� | �� d � d S )
Nr c s� �� � \} }| �; | �d�� } |�� }|�d�snq| �d� |�� � W d � n1 s.w Y W d � d S W d � d S 1 sFw Y d S )Nr Tr� s' HTTP/1.1 200 Connection established
)�acceptr& rA �rstripr r= )rk �address�readerr� ��result�servr r �
run_server� s
�
P�z2BasicTest.test_response_fileno.<locals>.run_server��targetr rw z
dummy:1234)�filenos
proxied data
)�socket�
create_server�HOST�
addCleanupr# � threading�Thread�startr� �floatr re � getsocknamerh r\ ri rR rQ r� r �detach)r r� �threadrk rj r r r� r �test_response_fileno� s,
�
z BasicTest.test_response_filenoc s� t �tdf�� | �� j� � �fdd�}tj|d�}|�� | �|jd� t j
� � � � }z<t dd�D ]*}d|> �|�
d d
� |�� �}| �t|�� ��� W d � n1 sWw Y q2W |�� |�d� d S |�� |�d� w ) Nr c s� � � � \} }| �6 | �d�r6| �d� � | �d�d � | �d��d � | �d�sW d � d S W d � d S 1 sAw Y d S )N� s' HTTP/1.1 200 Ok
Content-Length: %d
� Ar� � B�r� Zrecvr �rk r� �r� r� r r r� � s
�� ��"�z7BasicTest.test_large_content_length.<locals>.run_serverr� � �?� � r rv rd )r� r� r� r� r# r� r� r� r� r re r� r� rh r\ ri rX r= �r r� r� rk �wrj r r� r �test_large_content_length� s(
���� z#BasicTest.test_large_content_lengthc
s t �tdf�� | �� j� � �fdd�}tj|d�}|�� | �|jd� t j
� � � � }zJt dd�D ],}d|> �|�
d d
� |�� �}| �t j|j� W d � n1 sUw Y |�� q2W |�� d�|�
d d
� |�� |�d� d S |�� d�|�
d d
� |�� |�d� w ) Nr c sb � � � \} }| � | �d� �s W d � d S | �d� � W d � n1 s+w Y q)NTr� s+ HTTP/1.1 200 Ok
Content-Length: %d
Textr� r� r� r r r� � s
��� ��zABasicTest.test_large_content_length_truncated.<locals>.run_serverr� r� � �A r rv rd )r� r� r� r� r# r� r� r� r� r re r� r� rh r\ r� rG r= r� r r� r �#test_large_content_length_truncated� s4
�
� � z-BasicTest.test_large_content_length_truncatedc C �4 G dd� dt j�}|d�}td�|_|�dd� dS )zv
It should be possible to override the default validation
behavior in putrequest (bpo-38216).
c @ r� )zRBasicTest.test_putrequest_override_domain_validation.<locals>.UnsafeHTTPConnectionc S r) r* r �r r� r r r �_validate_path r+ zaBasicTest.test_putrequest_override_domain_validation.<locals>.UnsafeHTTPConnection._validate_pathN)r/ r0 r1 r� r r r r �UnsafeHTTPConnection r� r� rE rs rv z/ N�r re r rf r� �r r� rk r r r �*test_putrequest_override_domain_validation s
z4BasicTest.test_putrequest_override_domain_validationc C s8 G dd� dt j�}|d�}td�|_|jdddd� d S ) Nc @ r� )zPBasicTest.test_putrequest_override_host_validation.<locals>.UnsafeHTTPConnectionc S r) r* r r� r r r �_validate_host r+ z_BasicTest.test_putrequest_override_host_validation.<locals>.UnsafeHTTPConnection._validate_hostN)r/ r0 r1 r� r r r r r� r� r� z
example.com
rs rv rd r )Z skip_hostr� r� r r r �(test_putrequest_override_host_validation s
z2BasicTest.test_putrequest_override_host_validationc C r� )z�
It should be possible to override the default encoding
to transmit bytes in another encoding even if invalid
(bpo-36274).
c @ r� )zIBasicTest.test_putrequest_override_encoding.<locals>.UnsafeHTTPConnectionc S s
|� d�S )N�utf-8)r )r Zstr_urlr r r �_encode_request. r( zYBasicTest.test_putrequest_override_encoding.<locals>.UnsafeHTTPConnection._encode_requestN)r/ r0 r1 r� r r r r r� - r� r� rE rs rv u /☃Nr� r� r r r �!test_putrequest_override_encoding' s
z+BasicTest.test_putrequest_override_encodingN)7r/ r0 r1 r� r� r� r� r� r� r� r r r r r r r r r r# r' r* r2 r3 r: r; r> rK rN rT rU rW rX r] rb rc rf rg rh rl rn rp rq rs rx r{ r} r~ r� r� r� r� r� r� r� r r r r r�
sj
+
% $
r� c @ sx e Zd ZdZdZee�d�d� �d�ZdZdd� Z d d
� Z
d d � Z d
d� Z dd� Z
dd� Zdd� Zdd� Zdd� ZdS )�ExtendedReadTestz*
Test peek(), read1(), readline()
zOHTTP/1.1 200 OK
hello world!
and now
for something completely different
fooZhelloNr ��HTTP/1.1 200 OK
Transfer-Encoding: chunked
a
hello worl
3
d!
9
and now
23
for something completely different
3
foo
0
c C s8 t | j�}tj|dd�}|�� t�|j�|_|| _d S )Nrv r ) r r� r r� r� r2 �BufferedReader�fpr� r r r r �setUpT s
zExtendedReadTest.setUpc s� | j }|jj� d� fdd� }||j_g } |�d�}|rH| �t|�d� |�� }| �t|�t|�� | �|�|�� |�t|��}| � ||� n |�� }| �
|� |� |� |sYnq| � d� |�| j
� d S ) Nr: c s( � | �}| dkr|d | � S |d d� S )Nr r� r )r? r �Zoldpeekr r �mypeeka s z*ExtendedReadTest.test_peek.<locals>.mypeekTr� r r
rB )r� r� �peek�
assertGreaterrX ZassertGreaterEqualr� r� r= ri r� r] r� �lines_expected)r r� r� �allr Zp2r, r r� r � test_peek] s*
�zExtendedReadTest.test_peekc C s | j }| �| j j| j� d S r* )r� �_verify_readlinerA r� )r r� r r r �
test_readline| s z ExtendedReadTest.test_readlinec C s^ g } |d�}|r|dkrt |�dk r| �|�d�� |�|� |s#nq| �d�|�|� d S )NTr � foo�
r
)rX r� r� r] ri r� )r rA r� r� r� r r r r� � s
� z!ExtendedReadTest._verify_readlinec s0 �j � � �fdd�}t|�}��|j�j� d S )Nc s � � d�} ��t| �d� | S )Nr� )r| �assertLessEqualrX )�res�r� r r r r � s
z&ExtendedReadTest.test_read1.<locals>.r)r� � Readlinerr� rA r� )r r Z readlinerr r� r �
test_read1� s zExtendedReadTest.test_read1c C s> | j }g } |�� }|s
n|�|� q| �d�|�| j� d S )NTr
)r� r| r] ri r� r� �r r� r� r r r r �test_read1_unbounded� s
�z%ExtendedReadTest.test_read1_unboundedc C sP | j }g } |�d�}|sn| �t|�d� |�|� q| �d�|�| j� d S )NTr� r
)r� r| r� rX r] ri r� r� r� r r r �test_read1_bounded� s
�z#ExtendedReadTest.test_read1_boundedc C s | � | j�d�d� d S )Nr r
)ri r� r| r' r r r � test_read1_0� � z ExtendedReadTest.test_read1_0c C s | j �d�}| �dt|�� d S �Nr )r� r� r� rX )r r r r r � test_peek_0� s z ExtendedReadTest.test_peek_0)r/ r0 r1 rC r� �findr r� Z
lines_chunkedr� r� r� r� r� r� r� r� r� r r r r r� 6 s ��
r� c @ s e Zd ZdZdZdS )�ExtendedReadTestChunkedz:
Test peek(), read1(), readline() in chunked mode
r� N)r/ r0 r1 rC r� r r r r r� � s �r� c @ s e Zd ZdZdd� Zdd� ZdS )r� zT
a simple readline class that uses an arbitrary read function and buffering
c C s || _ d| _d S )Nr
)�readfunc� remainder)r r� r r r r � s
zReadliner.__init__c C s� g }d}| j }zB |�d�}|dkrn |t|� |kr!|| d }|�|� | �� }|s/d}nq |d7 }|�|d |� � ||d � | _ d�|�W S d�|�| _ � )Nr Tr� r: r r
)r� r� rX r] r� r� )r �limitr Zdatalenr= r� r r r rA � s.
� zReadliner.readlineN)r/ r0 r1 rC r rA r r r r r� � s r� c @ r4 )� OfflineTestc C sf dh}ddh}t t�D ] }|�d�s||v rq tt|�}t|dd �dkr)|�|� q | �tj|� d S )N� responsesZ HTTPMessage�
parse_headersr� r0 z http.client)r� r r� �getattr�addZassertCountEqual�__all__)r r� Zdenylistr� Z
module_objectr r r �test_all� s
�zOfflineTest.test_allc C s | � tjtj d� d S )Nz Not Found)ri r r� � NOT_FOUNDr' r r r �test_responses� r� zOfflineTest.test_responsesc C sR g d�}|D ] }| j |d�� | �tt|�� W d � n1 s!w Y qd S )N):ZCONTINUEZSWITCHING_PROTOCOLSZ
PROCESSINGrQ ZCREATEDZACCEPTEDZ NON_AUTHORITATIVE_INFORMATIONZ
NO_CONTENTZ
RESET_CONTENTZPARTIAL_CONTENTZ MULTI_STATUSZIM_USEDZMULTIPLE_CHOICESZMOVED_PERMANENTLYZFOUNDZ SEE_OTHERZ NOT_MODIFIEDZ USE_PROXYZTEMPORARY_REDIRECTZ BAD_REQUESTZ UNAUTHORIZEDZPAYMENT_REQUIREDZ FORBIDDENr� ZMETHOD_NOT_ALLOWEDZNOT_ACCEPTABLEZ PROXY_AUTHENTICATION_REQUIREDZREQUEST_TIMEOUTZCONFLICTZGONEZLENGTH_REQUIREDZPRECONDITION_FAILEDZREQUEST_ENTITY_TOO_LARGEZREQUEST_URI_TOO_LONGZUNSUPPORTED_MEDIA_TYPEZ REQUESTED_RANGE_NOT_SATISFIABLEZEXPECTATION_FAILEDZ IM_A_TEAPOTZMISDIRECTED_REQUESTZUNPROCESSABLE_ENTITYZLOCKEDZFAILED_DEPENDENCYZUPGRADE_REQUIREDZPRECONDITION_REQUIREDZTOO_MANY_REQUESTSZ REQUEST_HEADER_FIELDS_TOO_LARGEZ UNAVAILABLE_FOR_LEGAL_REASONSZINTERNAL_SERVER_ERRORZNOT_IMPLEMENTEDZ BAD_GATEWAYZSERVICE_UNAVAILABLEZGATEWAY_TIMEOUTZHTTP_VERSION_NOT_SUPPORTEDZINSUFFICIENT_STORAGEZ NOT_EXTENDEDZ NETWORK_AUTHENTICATION_REQUIREDZ EARLY_HINTSZ TOO_EARLY)Zconstant)r� r� �hasattrr )r r� Zconstr r r �test_client_constants� s < ���z!OfflineTest.test_client_constantsN)r/ r0 r1 r� r� r� r r r r r� � s r� c @ s@ e Zd Zdd� Zdd� Zdd� Ze�ee d� d�d d
� �Z
d S ) �SourceAddressTestc C s> t � t jt j�| _t�| j�| _t�� | _| j� � d | _
d S r* ) r� �AF_INET� SOCK_STREAMr� r � bind_portr Zfind_unused_port� source_port�listenrk r' r r r r� A s
zSourceAddressTest.setUpc C s* | j r | j �� d | _ | j�� d | _d S r* )rk r# r� r' r r r �tearDownH s
zSourceAddressTest.tearDownc C sB t jt| jd| jfd�| _| j�� | �| jj� � d | j� d S )Nrs ��source_addressr )
r re r� r r� rk rN ri rf r� r' r r r � testHTTPConnectionSourceAddressO s
�
z1SourceAddressTest.testHTTPConnectionSourceAddress�HTTPSConnectionz'http.client.HTTPSConnection not definedc C s t jt| jd| jfd�| _d S )Nrs r� )r r� r� r r� rk r' r r r � testHTTPSConnectionSourceAddressU s
�z2SourceAddressTest.testHTTPSConnectionSourceAddressN) r/ r0 r1 r� r� r� �unittest�skipIfr� r r� r r r r r� @ s �r� c @ s( e Zd ZdZdd� Zdd� Zdd� ZdS )� TimeoutTestNc C s. t � t jt j�| _t�| j�t_| j�� d S r* ) r� r� r� r� r r� r� �PORTr� r' r r r r� b s zTimeoutTest.setUpc C s | j �� d | _ d S r* )r� r# r' r r r r� g rp zTimeoutTest.tearDownc C s | � t�� � t�d� zt�ttj�}|� � W t�d � nt�d � w | �
|j � � d� |�
� | � t�� � t�d� ztjttjd d�}|� � W t�d � nt�d � w | �
|j � � d � |�
� tjttjdd�}|� � | �
|j � � d� |�
� d S �N� )�timeout)� assertIsNoner� ZgetdefaulttimeoutZsetdefaulttimeoutr re r� r� r� rN ri rf Z
gettimeoutr# )r ZhttpConnr r r �testTimeoutAttributek s,
�
z TimeoutTest.testTimeoutAttribute)r/ r0 r1 r� r� r� r� r r r r r� _ s
r� c @ r4 )�PersistenceTestc
C s� d}|D ]v\}}}| j ||d��b d�||�}t|�}| �|j� |�dd� |�� �}| �|jd u | � |�� W d � n1 sDw Y | �|jd u | � | �|j d� |�dd� | �|j |rhdnd� W d � n1 suw Y qd S ) N))�1.0rs F)r� zConnection: keep-alive
T)�1.1rs T)r� zConnection: close
F)r� zConnection: keep-ALIVE
T)r� zConnection: cloSE
F)�versionrj z4HTTP/{} 200 OK
{}Content-Length: 12
Dummy body
rv z/open-connectionr z/subsequent-requestr� )
r� r} rD r� rf rh r\ ri r= rF )r �testsr� rj Zreuser� rk rj r r r �test_reuse_reconnect� s( �
� ���z$PersistenceTest.test_reuse_reconnectc C s� dd� }t jtjf|tff}|D ]?\}}| j|d��- td|�}|�dd� | �||j � | �
|j � |�dd� | � |j
d� W d � n1 sJw Y qd S ) Nc s& t �| �� � fdd�}|� _t �� �S )z3Return BufferedReader that raises ECONNRESET at EOFc s t j�� | �}|dkrt� �|S r� )r2 r3 r� �ConnectionResetError)�bufferr� ��streamr r r� � s zNPersistenceTest.test_disconnected.<locals>.make_reset_reader.<locals>.readinto)r2 r3 r� r� )r r� r r� r �make_reset_reader� s
z<PersistenceTest.test_disconnected.<locals>.make_reset_reader)rd r
rv z
/eof-response�
/reconnectr� )r2 r3 r �RemoteDisconnectedr� r� rD rh r� r\ r� rf ri rF )r r� r� Zstream_factoryrd rk r r r �test_disconnected� s
�
���z!PersistenceTest.test_disconnectedc C sV t d�}|jddddid� | �tj|j� | �|j� |�dd� | �|j d� d S ) Ns HTTP/1.1 100 Continue
rv rd ZExpectz 100-continue)rm r� r� )
rD rh r� r r r\ r� rf ri rF r� r r r �test_100_close� s � z PersistenceTest.test_100_closeN)r/ r0 r1 r� r r r r r r r� � s r� c @ s� e Zd Zdd� Zdd� Zdd� Zdd� Zd d
� Zej d d � �Z
d
d� Z dd� Z dd� Z
dd� Zdd� Ze�eed� d�dd� �Zdd � Zd S ) � HTTPSTestc C s t td�s | �d� d S d S )Nr� zssl support required)r� r �skipTestr' r r r r� � s
�zHTTPSTest.setUpc C s ddl m} || |d�S )Nr )�make_https_server)�certfile)Ztest.ssl_serversr )r r r r r r � make_server� s zHTTPSTest.make_serverc C s$ t jttjdd�}| �|jd� d S r� )r r� r� r� r� ri r� )r r r r r �test_attributes� s zHTTPSTest.test_attributesc C s� dd l }t�d� t�d��3 t�dd�}| �|j��}|� dd� W d � n1 s,w Y | �
|j j d� W d � d S 1 sDw Y d S )Nr �network�self-signed.pythontest.net� rv rd �CERTIFICATE_VERIFY_FAILED)
�sslr �requiresr �transient_internetr r� r� �SSLErrorrh ri rd rS )r r
r �exc_infor r r �test_networked� s
�"�zHTTPSTest.test_networkedc C s� dd l }t�d� t�d��0 |�� }tjdd|d�}|�dd� |� � }|�
� | � d|� d �� |�
� W d � d S 1 sAw Y d S )
Nr r r
r ��contextrv rd �nginx�server)
r
r r r r Z_create_unverified_contextr r� rh r\ r# r� r� )r r
r r r� r r r � test_networked_noverification� s
�
"�z'HTTPSTest.test_networked_noverificationc C s� t �d� t�d��, t�dd�}|�dd� |�� }|�d�}|� � |� � | �
d|� W d � d S 1 s9w Y d S )Nr r
r rv rd z content-typez text/html) r r r r r r� rh r\ r� r# r� )r r r� Z content_typer r r �&test_networked_trusted_by_default_cert� s
"�z0HTTPSTest.test_networked_trusted_by_default_certc C s dd l }t�d� d}t�|��o |�|j�}| �|j|j � | �|j
d� |� t � zt
j|d|d�}|�dd� |�� }W n$ |jyc } zt|�}t�d |�r^t�d
|� d |� d ���� d }~ww |�d
�}|�� |�� | �d|� W d � d S 1 s�w Y d S )Nr r r
Tr r rv rd z(?i)key.too.weakzGot z trying to connect to z). See https://bugs.python.org/issue36816.r r )r
r r r r �
SSLContext�PROTOCOL_TLS_CLIENTri Z verify_modeZ
CERT_REQUIRED�check_hostname�load_verify_locations� CERT_selfsigned_pythontestdotnetr r� rh r\ r r �re�searchr� ZSkipTestr� r# r� ) r r
Zselfsigned_pythontestdotnetr r r� Zssl_errZ ssl_err_strZ
server_stringr r r �test_networked_good_cert s<
� ����
"�z"HTTPSTest.test_networked_good_certc C s� dd l }t�d� t�d��@ |�|j�}|�t� t j
dd|d�}| � |j ��}|�
dd� W d � n1 s9w Y | �|jjd� W d � d S 1 sQw Y d S ) Nr r r
r r rv rd r )r
r r r r r r r �CERT_localhostr r� r� r rh ri rd rS )r r
r r r r r r �test_networked_bad_cert* s
�"�z!HTTPSTest.test_networked_bad_certc C sl dd l }| �t�}t�d|j�}| �|j��}|�dd� W d � n1 s'w Y | � |j
j d� d S )Nr � localhostrv rd r ) r
r r! r r� r r� r rh ri rd rS )r r
r r r r r r �test_local_unknown_cert6 s
�z!HTTPSTest.test_local_unknown_certc C sx dd l }| �t�}|�|j�}|�t� tjd|j|d�}| � |j
� |� dd� |� � }| � |j
� | �
|jd� d S )Nr r# r rv � /nonexistentr� )r
r r! r r r r r� r r� r# rh r\ ri rR �r r
r r r r� r r r �test_local_good_hostname? s
z"HTTPSTest.test_local_good_hostnamec C s2 dd l }| �t�}|�|j�}|�t� tjd|j|d�}| � |j
�� |� dd� W d � n1 s4w Y t �
dtf�� tjd|j|dd�}W d � n1 sUw Y | � |j
�� |� dd� W d � n1 sqw Y d |_t �
dtf�� tjd|j|d d�}W d � n1 s�w Y |� dd
� |�� }|�� |�� | �|jd � d |_tjd|j|d�}|� dd
� |�� }| �|jd � |�� |�� t �
dtf�� tjd|j|dd�}W d � n1 s�w Y | � |j
�� |� dd� W d � d S 1 �sw Y d S ) Nr r# r rv rd rs T)r r Fr% r� )r
r �CERT_fakehostnamer r r r r� r r� ZCertificateErrorrh r Zcheck_warnings�DeprecationWarningr r\ r# ri rR r& r r r �test_local_bad_hostnameL sT
�
� � �
� �
� �$�z!HTTPSTest.test_local_bad_hostnamer� z)http.client.HTTPSConnection not availablec C r )Nr )r )zwww.python.org:443r
r )r r
r )r
r
r )r
r r )r r r )r� r r r� ri r r r r r r r v s
�zHTTPSTest.test_host_portc C s� dd l }|js | �d� t�dd�}| �|jj� |�|j �}| �
|j� tjdd|d�}| � |j|� | �
|jj� t �
� � t �ddt� tjdd|td�}W d � n1 sZw Y | �|jj� d S ) Nr zTLS 1.3 support requiredr# r r �ignorez5key_file, cert_file and check_hostname are deprecated)r Z cert_file)r
Z HAS_TLSv1_3r r r� r� Z_contextZpost_handshake_authr r r� ZassertIs�warnings�catch_warnings�filterwarningsr) r! )r r
r r r r r �test_tls13_pha� s&
�
� �zHTTPSTest.test_tls13_phaN)r/ r0 r1 r� r r r r r Zsystem_must_validate_certr r r" r$ r' r* r� r� r� r r r/ r r r r r � s$
!
*�
r c @ sX e Zd ZdZdd� Zdd� Zdd� Zdd � Zd
d � Zd d
� Z dd� Z
dd� Z dd� Z dS )�RequestBodyTestz3Test cases where a request includes a message body.c C s, t �d�| _td� | j_| _| j| j_d S )NrE rs )r re rk r rf r' r r r r� � s zRequestBodyTest.setUpc C s( t �| jj�}|�� t�|�}||fS r* )r2 r3 rf r rA r r� �r �f�messager r r �get_headers_and_fp� s
z"RequestBodyTest.get_headers_and_fpc C s� ddgdfdf}|D ]S\}}| � |��B t�d�| _td� | j_| _| j�dd|� | �� \}}| �d |� | �d
|� | � |�
d �d � | � ||� � � W d � n1 sXw Y q
d S )
Nr� � bar� 3
foo
3
bar
0
))r� r5 r6 rE rs rt rY z Content-Typer� r� r� ) r� r re rk r rf rh r4 r� ri �getr= )r r� rl r� r� r2 r r r �test_list_body� s
� ���z RequestBodyTest.test_list_bodyc C sL | j �dddddi� | �� \}}| �d|�d�� | �dt|�� �� d S )Nrt rY rl r� Z42r� r� )rk rh r4 ri r7 rX r= �r r3 r2 r r r �test_manual_content_length� s � z*RequestBodyTest.test_manual_content_lengthc C �` | j �ddd� | �� \}}| �d|�� � | �|�� � | �d|�d�� | �d|�� � d S )Nrt rY rl �
text/plain�4r� s body� rk rh r4 ri �get_content_typer� � get_charsetr7 r= r9 r r r �test_ascii_body� � z RequestBodyTest.test_ascii_bodyc C r; )Nrt rY u bodyÁr< �5r� � body�r> r9 r r r �test_latin1_body� rB z RequestBodyTest.test_latin1_bodyc C s` | j �ddd� | �� \}}| �d|�� � | �|�� � | �d|�d�� | �d|�� � d S )Nrt rY rD r< rC r� r> r9 r r r �test_bytes_body� rB z RequestBodyTest.test_bytes_bodyc C s� | � tjtj� ttjddd��
}|�d� W d � n1 s w Y ttjdd��?}| j�dd|� | �� \}}| � d|�
� � | � |� � � | � |�
d�� | � d |�
d
�� | � d |�� � W d � d S 1 snw Y d S ) Nr� r� )�encodingrl rt rY r< r� r� ztransfer-encodings 4
body
0
)r� r �unlink�TESTFNr$ �writerk rh r4 ri r? r� r@ r7 r= r1 r r r �test_text_file_body� s � "�z#RequestBodyTest.test_text_file_bodyc C s� | � tjtj� ttjd��
}|�d� W d � n1 s w Y ttjd��=}| j�dd|� | �� \}}| � d|�
� � | � |� � � | � d|�
d�� | �d |� | � d
|�� � W d � d S 1 siw Y d S ) N�wbrD r rt rY r< r� r� r� s 5
body�
0
)r� r rH rI r$ rJ rk rh r4 ri r? r� r@ r7 r� r= r1 r r r �test_binary_file_body� s � "�z%RequestBodyTest.test_binary_file_bodyN)
r/ r0 r1 rC r� r4 r8 r: rA rE rF rK rM r r r r r0 � s r0 c @ s< e Zd Zdd� Zdd� Zdd� Zdd� Zd d
� Zd d � Zd
S )�HTTPResponseTestc C s& d}t |�}t�|�| _| j�� d S )NzXHTTP/1.1 200 Ok
My-Header: first-value
My-Header: second-value
Text)r r r� r� r� )r rl rf r r r r� s zHTTPResponseTest.setUpc C s6 | j �d�}| �|d� | j �dd�}| �|d� d S )Nz My-Headerzfirst-value, second-valuez some default�r� r� ri �r rj r r r �test_getting_header s z$HTTPResponseTest.test_getting_headerc C s | j �dd�}| �|d� d S )N�No-Such-Headerz
default-valuerO rP r r r �3test_getting_nonexistent_header_with_string_default s zDHTTPResponseTest.test_getting_nonexistent_header_with_string_defaultc C s< | j �dddg�}| �|d� | j �dd�}| �|d� d S )NrR �default�valueszdefault, values)rT rU rO rP r r r �5test_getting_nonexistent_header_with_iterable_default s zFHTTPResponseTest.test_getting_nonexistent_header_with_iterable_defaultc C s | j �d�}| �|d � d S )NrR rO rP r r r �/test_getting_nonexistent_header_without_default s z@HTTPResponseTest.test_getting_nonexistent_header_without_defaultc C s | j jddd�}| �|d� d S )NrR r� )rT rO rP r r r � test_getting_header_defaultint s z/HTTPResponseTest.test_getting_header_defaultintN) r/ r0 r1 r� rQ rS rV rW rX r r r r rN s rN c @ sT e Zd Zdd� Zdd� Zdd� Zdd� Zd d
� Zd d � Zd
d� Z dd� Z
dd� Z dS )� TunnelTestsc C s* d}d| _ t�| j �| _| �|�| j_d S )Nz:HTTP/1.0 200 OK
HTTP/1.1 200 OK
Content-Length: 42
z proxy.com)r r re rk rJ )r �
response_textr r r r� $ s
�zTunnelTests.setUpc C s | j �� d S r* )rk r# r' r r r r� . s zTunnelTests.tearDownc s d� fdd� }|S )Nc s t � | d | d d�S )Nr r )r r )r )r� r� r� �rZ r r rI 2 s z9TunnelTests._create_connection.<locals>.create_connection)NNr )r rZ rI r r[ r rJ 1 s z TunnelTests._create_connectionc C s� d}d}ddi}| j j|||d� | j �ddd� | �| j jj| j� | �| j jjtj� | �| j j |� | �| j j
|� | �| j j |� d S ) N�destination.comi�" z
User-Agentz!Mozilla/5.0 (compatible, MSIE 11))r rm rx rd rs ) rk �
set_tunnelrh ri rf r r r � HTTP_PORTZ _tunnel_hostZ _tunnel_portZ_tunnel_headers)r Z tunnel_hostZ tunnel_portZtunnel_headersr r r �!test_set_tunnel_host_port_headers6 s
�z-TunnelTests.test_set_tunnel_host_port_headersc C s | j �� | �t| j jd� d S )Nr\ )rk rN r� � RuntimeErrorr] r' r r r �&test_disallow_set_tunnel_after_connectC s
�z2TunnelTests.test_disallow_set_tunnel_after_connectc C s� | j �d� | j �ddd� | �| j jj| j� | �| j jjtj� | � d| j jj
� | � d| j jj
� | � d| j jj
� | � d| j jj
� d S ) Nr\ rx rd rs � CONNECT destination.coms Host: destination.com:None� Host: destination.coms Host: proxy.com) rk r] rh ri rf r r r r^ r� r r� r' r r r �test_connect_with_tunnelI s z$TunnelTests.test_connect_with_tunnelc C s� t j�| jd��}| j�d� | j�� | j�dd� W d � n1 s%w Y |�� | jt |j
�dd|j
� �d� |j
d d d }| � d
|� | j |�
d �d |��d� dS )
z8Regresstion test for https://bugs.python.org/issue43332.r( r\ rv rd Nr z!unexpected number of send calls: )r� r rb r9 zunexpected proxy data sent )r Zpatch�objectrk r] rN rh Z
assert_calledr� rX Z
mock_callsr� r� r� )r Z mock_sendZproxy_setup_data_sentr r r �0test_tunnel_connect_single_send_connection_setupV s
�
�
�z<TunnelTests.test_tunnel_connect_single_send_connection_setupc C sl | j �d� | j �ddd� | �| j jj| j� | �| j jjtj� | � d| j jj
� | � d| j jj
� d S )Nr\ rt rd rs rb rc ) rk r] rh ri rf r r r r^ r� r r' r r r �test_connect_put_requestg s z$TunnelTests.test_connect_put_requestc C s� d}d� |�}| j�d� | �|�| j_| j�d� t�� �}| j�ddd� W d � n1 s1w Y |�� � � }| �
d� |�|� d S ) Nz
X-Dummy: 1zHTTP/1.0 200 OK
{}
r r\ rt rd rs z
header: {}) r} rk Zset_debuglevelrJ r] r r� rh r� r� r� )r Zexpected_headerrZ r� r� r r r �test_tunnel_debuglogo s
� z TunnelTests.test_tunnel_debuglogN) r/ r0 r1 r� r� rJ r_ ra rd rf rg rh r r r r rY # s
rY �__main__r� )� verbosity)<r8 Zhttpr r r2 r{ �osr) r r� r� r, r� r ZTestCaser< r Z test.supportr r r �path�dirnamer% �herer� r! r( r rD rC Zchunk_extensionrE rm r� rF r� r r5 r3 r9 re rD rS r� r� r� r� r� r� r� r� r� r� r r0 rN rY r/ �mainr r r r �<module> sz � ! n 2}"U .H L`#
Z�