File: //usr/local/lib/python3.7/test/__pycache__/test_httplib.cpython-37.opt-1.pyc
B
��g�2 � @ s d dl Z d dlmZ d dlZd dlZd dlZd dlZd dlZd dlZd dl Z d dl
Z
e
j Z d dl m
Z
ej�e�Zej�ed�Zej�ed�Zej�ed�ZdZdZd Zd
Zd e d Zd
Zd Z e
j Z G dd� d�Z G dd� de �Z G dd� dej �Z!G dd� dej"�Z#G dd� de �Z$G dd� de �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#�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 �Z/G d.d/� d/e �Z0G d0d1� d1e �Z1G d2d3� d3e �Z2e3d4k�r e
j4d5d6� dS )7� N)�client)�supportz 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
�0z
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 �-/usr/local/lib/python3.7/test/test_httplib.py�__init__/ 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|dkrt �� �| �| j�| _| j| j_| jS )N�r�rb)r ZUnimplementedFileModer r �file�
file_close�close)r �mode�bufsizer r r �makefile> s
zFakeSocket.makefilec C s
d| _ d S )NT)r )r r r r r F s zFakeSocket.file_closec C s d S )Nr )r r r r r I s zFakeSocket.closec C s d S )Nr )r �levelZoptname�valuer r r �
setsockoptL s zFakeSocket.setsockopt)N) �__name__�
__module__� __qualname__�io�BytesIOr r r r r r" r r r r r . s
r c @ s$ e Zd Zdd� Zdd� Zdd� ZdS )� EPipeSocketc C s t �| |� || _d S )N)r r � pipe_trigger)r r r) r r r r Q s zEPipeSocket.__init__c C s( | j |krttjd��| j|7 _d S )NZgotcha)r) �OSError�errnoZEPIPEr
)r r
r r r r V s
zEPipeSocket.sendallc C s d S )Nr )r r r r r [ s zEPipeSocket.closeN)r# r$ r% r r r r r r r r( O s r( 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 s" t j�| |�}|dkr td��|S )Nr z caller tried to read past EOF)r& r' �read�AssertionError)r �nr
r r r r. d s zNoEOFBytesIO.readNc C s" t j�| |�}|dkr td��|S )Nr z caller tried to read past EOF)r& r' �readliner/ )r �lengthr
r r r r1 j s zNoEOFBytesIO.readline)r- )N)r# r$ r% �__doc__r. r1 r r r r r, ^ s
r, 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 z example.com)� connections�superr �fake_socket_args�create_connection�_create_connection)r �args)� __class__r r r s s z!FakeSocketHTTPConnection.__init__c s | j d7 _ t� �� S )z.Count the number of times connect() is invokedr )r5 r6 �connect)r )r; r r r< y s z FakeSocketHTTPConnection.connectc O s
t | j� S )N)r r7 )r �pos�kwr r r r8 ~ s z*FakeSocketHTTPConnection.create_connection)r# r$ r% r3 r r< r8 �
__classcell__r r )r; r r4 p s r4 c @ sL e Zd Zdd� Zdd� Zdd� Zdd� Zd d
� Zd d � Zd
d� Z dd� Z
dS )� HeaderTestsc C s� G dd� dt �}xzdD ]r}xldD ]d}t�d�}td�|_|� |_d}i }|r\tt|��||< |�dd ||� | � |jj
|� � d
� q W qW d S ) Nc @ s e Zd Zdd� Zdd� ZdS )z;HeaderTests.test_auto_headers.<locals>.HeaderCountingBufferc S s
i | _ d S )N)�count)r r r r r � s zDHeaderTests.test_auto_headers.<locals>.HeaderCountingBuffer.__init__c S sX |� d�}t|�dkrH|d �d��� }| j�|d� | j| d7 < t�| |� d S )N� :r r r )�split�len�decode�lowerrA �
setdefault�list�append)r �item�kvZlcKeyr r r rI � s
zBHeaderTests.test_auto_headers.<locals>.HeaderCountingBuffer.appendN)r# r$ r% r rI r r r r �HeaderCountingBuffer� s rL )TF)zContent-lengthZHostzAccept-encodingz example.comZ blahblahblahZ spamspamspam�POST�/r ) rH r �HTTPConnectionr �sock�_bufferr rD �request� assertEqualrA rF )r rL Zexplicit_header�header�conn�body�headersr r r �test_auto_headers� s
z HeaderTests.test_auto_headersc C sv G dd� dt �}d}d}xZt�||�D ]J\}}t�d�}td �|_|� |_|�|d|� | � |jj
dd� |�� q&W d }xN|D ]F}t�d�}td �|_|� |_|�|dd � | � |jj
d d
� |�� q~W xN|D ]F}t�d�}td �|_|� |_|�|dd � | � |jj
dd� |�� q�W xXt� ||�D ]H}t�d�}td �|_|� |_|�|dd � | � |jj
d
d� |�� �q&W d S )Nc @ s e Zd Zdd� Zdd� ZdS )z?HeaderTests.test_content_length_0.<locals>.ContentLengthCheckerc S s t �| � d | _d S )N)rH r �content_length)r r r r r � s
zHHeaderTests.test_content_length_0.<locals>.ContentLengthChecker.__init__c S sF |� dd�}t|�dkr6|d �� dkr6|d �� | _t�| |� d S )NrB r r s content-length)rC rD rF �striprY rH rI )r rJ rK r r r rI � s zFHeaderTests.test_content_length_0.<locals>.ContentLengthChecker.appendN)r# r$ r% r rI r r r r �ContentLengthChecker� s r[ )N� )�PUTrM ZPATCHz example.comrN � 0z%Header Content-Length incorrect on {})�GET�CONNECTZDELETE�HEADZOPTIONSZTRACEz.Header Content-Length set for empty body on {}r\ � � 1)
rH � itertools�productr rO r rP rQ rR rS rY �format�chain)r r[ ZbodiesZmethods_with_body�methodrV rU Zmethods_without_bodyr r r �test_content_length_0� sH
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 ),Nz example.comr_ rN zContent-length�* 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 rO r rP �
putrequest� putheader�assertInrQ )r rU 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::]:81r\ r_ z/foosE GET /foo HTTP/1.1
Host: [2001:102A::]
Accept-Encoding: identity
z
[2001:102A::])r rO r rP rR �
assertTruer
�
startswith)r �expectedrU rP 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�beginrS � getheader)r rV rP �respr r r �!test_malformed_headers_coped_with s
z-HeaderTests.test_malformed_headers_coped_withc C s0 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� xH|�d�|jd fD ]0}| �|�d�� | �
d|� | �|�d�� q�W 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
zContent-Lengthr z!#$%&'*+-.^_`|~r! r\ ZVCHARzobs-textzobs-foldr z folded with spacezfolded with tab)�bytes�ranger r ru rv rS rw �msg�join�map�chrZassertIsNotNonern rp rq �endswith)r rV rP rx ZvcharZfoldedr r r �test_parse_all_octets s" (
z!HeaderTests.test_parse_all_octetsc C sx t �d�}td�|_|�dd� d}xN|D ]F\}}| �||f��* | �td�� |�||� W d Q R X W d Q R X q*W d S )Nz example.comr\ r_ rN ))s
Invalid
Names
ValidValue)s Invalid
Names
ValidValue)s Invalid
Names
ValidValue)s
InvalidNames
ValidValue)s
InvalidNames
ValidValue)s
InvalidNames
ValidValue)s InvalidNames
ValidValue)s InvalidNames
ValidValue)s Invalid:Names
ValidValue)s :InvalidNames
ValidValue)s ValidNames Invalid
Value)s ValidNames
Invalid
Value)s ValidNames
Invalid
Value)s ValidNames InvalidValue
)s ValidNames
InvalidValue
)s ValidNames
InvalidValue
zInvalid header) r rO r rP rl �subTest�assertRaisesRegex�
ValueErrorrm )r rU �cases�namer! r r r �test_invalid_headers> s
z HeaderTests.test_invalid_headersc C s� d}t |�}tj|dd�}t�� �}|�� W d Q R X |�� �� }| �|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 ru r �captured_stdoutrv �getvalue�
splitlinesrS )r rV rP rx �output�linesr r r �test_headers_debuglevel\ s
z#HeaderTests.test_headers_debuglevelN) r# r$ r% rX ri ro rs ry r� r� r� r r r r r@ � s B"
r@ c @ s e Zd Zdd� ZdS )�HttpMethodTestsc
C sP d}xF|D ]>}| � td��( t�d�}td �|_|j|dd� W d Q R X q
W d 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 charactersz example.comrN )rh �url)r� r� r rO r rP rR )r �methodsrh rU r r r �test_invalid_method_nameso s
z)HttpMethodTests.test_invalid_method_namesN)r# r$ r% r� r r r r r� n s 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 )Nz example.comr rM rN T)�encode_chunked) r rO r rP rl Z
endheaders�
_make_body�_parse_requestr
�_parse_chunkedrS �
expected_body)r rU �_rV 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 )Nz example.comr rM rN zTransfer-Encoding�chunkedzcontent-lengthc S s g | ] }|� � �qS r )rF )�.0�kr r r �
<listcomp>� s z>TransferEncodingTest.test_explicit_headers.<locals>.<listcomp>zlatin-1c S s g | ] }|� � �qS r )rF )r� r� r r r r� � s z
gzip, chunkedT)rW r� rV c S s g | ] }|� � �qS r )rF )r� r� r r r r� � s )r rO r rP rR r� r� r
� assertNotIn�keysrS r� rE r� )r rU r� rW rV r r r �test_explicit_headers� s6
z*TransferEncodingTest.test_explicit_headersc C s� x�dD ]�}t �d�}td�|_|�dd| j|d�� | �|jj�\}}}| �|�}| � || j
� | � |d d� | � d d
d � |D �� qW d S ) N)FTz example.comr rM rN )� empty_lineszTransfer-Encodingr� zcontent-lengthc S s g | ] }|� � �qS r )rF )r� r� r r r r� � s z5TransferEncodingTest.test_request.<locals>.<listcomp>) r rO r rP rR r� r� r
r� rS r� r� )r r� rU r� rW rV 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 ) Nz example.comr rM rN r zTransfer-Encodingr� zcontent-lengthc S s g | ] }|� � �qS r )rF )r� r� r r r r� � s z8TransferEncodingTest.test_empty_body.<locals>.<listcomp>s 0
) r rO r rP rR r� r
rS r� )r rU r� rW rV r r r �test_empty_body� s
z$TransferEncodingTest.test_empty_bodyFc c sZ | j �d�}xHt|�D ]<\}}|r0|d r0dV |t|�d k rL|d V q|V qW d S )N� r� r r )r� rC � enumeraterD )r r� r� �idx�liner r r r� � s z TransferEncodingTest._make_bodyc C s� |� d�}|d }i }d}xZ|t|�k rtt|| �dkrt|| � d�\}}|�d��� }|�d��� ||< |d7 }q W ||d�||d d � �fS )Ns
r r rB zlatin-1)rC rD rE rZ r� )r r
r� rR rW r0 �keyrt r r r r� � s
z#TransferEncodingTest._parse_requestc C s� g }i }d}|� d�}xd|||d � \}}t|d�}|dkrH|d7 }P | �|t|�� |�|� |d7 }|t|�krP qW d�|�S )Nr s
r� � r r )rC �intrS rD rI r� )r r
rV �trailersr0 r� �size�chunkr r r r� � s
z#TransferEncodingTest._parse_chunkedN)F) r# r$ r% 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_S )`� BasicTestc C s� d}t |�}t�|�}|�� | �|�d�d� | �|�� � | �|j� | �|�� d� | � |�� � | �|j� |�
� | � |j� d}t |�}t�|�}| � tj |j� d S )NzHTTP/1.1 200 Ok
Textr r s Textz HTTP/1.1 400.100 Not Ok
Text)
r r ru rv rS r. � assertFalse�isclosed�closedrp r � assertRaises�
BadStatusLine)r rV rP rx r r r �test_status_lines s
zBasicTest.test_status_linesc C s t �d�}| �t|�d� d S )Nr\ zBadStatusLine("''"))r r� rS �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 )Nz*HTTP/1.1 200 Ok
Content-Length: 4
Textr� s Tes xt) r r ru rv rS r. r� r� rp r� r )r rV rP rx 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 ru rv rS r1 r� r� r. rp r� r )r rV rP rx r r r �test_mixed_reads- s
zBasicTest.test_mixed_readsc C s� d}t |�}t�|�}|�� td�}|�|�}| �|d� | �t|�d� | �|� � � |�|�}| �|d� | �t|�d� | �
|� � � | �|j � |� � | �
|j � d S )Nz*HTTP/1.1 200 Ok
Content-Length: 4
Textr� s Tes xt)
r r ru rv � bytearray�readintorS r~ r� r� rp r� r )r rV rP rx �br0 r r r �test_partial_readintos< s
z BasicTest.test_partial_readintosc C s� d}t |�}t�|�}|�� | �|�d�d� | �|�� � | �|�d�d� | �|�d�d� | �|�� � | �|j � |�
� | �|j � d S )NzHTTP/1.1 200 Ok
Textr� s Tes xtr r ) r r ru rv rS r. r� r� rp r� r )r rV rP rx r r r �$test_partial_reads_no_content_lengthP s
z.BasicTest.test_partial_reads_no_content_lengthc C s� d}t |�}t�|�}|�� td�}|�|�}| �|d� | �t|�d� | �|� � � |�|�}| �|d� | �t|�d� |�|�}| �|d� | �
|� � � d S )NzHTTP/1.1 200 Ok
Textr� s Tes xtr ) r r ru rv r� r� rS r~ r� r� rp )r rV rP rx r� r0 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 )Nz+HTTP/1.1 200 Ok
Content-Length: 10
Textr� s Tes xtr r ) r r ru rv rS r. r� r� rp )r rV rP rx r r r �"test_partial_reads_incomplete_bodys 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 )Nz+HTTP/1.1 200 Ok
Content-Length: 10
Textr� s Tes xtr )
r r ru rv r� r� rS r~ r� r� rp r� r )r rV rP rx r� r0 r r r �&test_partial_readintos_incomplete_body� s$
z0BasicTest.test_partial_readintos_incomplete_bodyc C s^ x dD ]}| � tjtj|� qW x8dD ]0\}}}t�|�}| �||j� | �||j� q&W d S )N)zwww.python.org:abcz user:password@www.python.org))z[fe80::207:e9ff:fe9b]:8000zfe80::207:e9ff:fe9bi@ )zwww.python.org:80zwww.python.org�P )zwww.python.org:zwww.python.orgr� )zwww.python.orgzwww.python.orgr� )z[fe80::207:e9ff:fe9b]zfe80::207:e9ff:fe9br� )z[fe80::207:e9ff:fe9b]:zfe80::207:e9ff:fe9br� )r� r �
InvalidURLrO rS r r )r �hp�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 ru rv rw rS )r r Zhdr�sr Zcookiesr r r �test_response_headers� s
z BasicTest.test_response_headersc C s6 t dt�}tj|dd�}|�� |�� r2| �d� d S )Nz*HTTP/1.1 200 OK
Content-Length: 14432
ra )rh z)Did not expect response from HEAD request)r r, r ru rv r. �fail)r rP rx r r r �test_read_head� s zBasicTest.test_read_headc C sT t dt�}tj|dd�}|�� td�}|�|�dkr@| �d� | �t |�d� d S )Nz*HTTP/1.1 200 OK
Content-Length: 14432
ra )rh � r z)Did not expect response from HEAD requests )
r r, r ru rv r� r� r� rS r~ )r rP rx 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 )Nz
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 ru r� �
HTTPExceptionrv )r rW r r� r r r r �test_too_many_headers� s
z BasicTest.test_too_many_headersc C sn d}t td��V}t�d�}t|�}||_|�dd|� | �|j� |�d|jd t
|�� |f � W d Q R X d S )Ns_ GET /foo HTTP/1.1
Host: example.com
Accept-Encoding: identity
Transfer-Encoding: chunked
r z example.comr_ z/fooz%r != %r) �open�__file__r rO r rP rR rp r
rq rD )r rr rV rU rP 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 testz example.comr r� )
r rO r rP �sendrS r
�arrayr& r' )r rr rU rP 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
� s 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 )N)�next�d)r � blocksizer r r r. s z<BasicTest.test_send_updating_file.<locals>.UpdatingFile.readN)r- )r# r$ r% r r� r. r )r
r r � UpdatingFile s r� s dataz example.comr\ ) r& �
TextIOBaser rO r rP r� rS r
)r r� rr rU rP 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 rV s z&BasicTest.test_send_iter.<locals>.bodyz example.comr\ r_ z/foozContent-LengthZ11)r rO r rP rR rS r
)r rr rV rU rP r r r �test_send_iter 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.� z example.com)r� N� a� br] rN zContent-Length�9r� s
r ) r rO r rP rR r& r' rS r r
rC )r r� rU rP rr rV r r r �test_blocksize_request 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.r� z example.com)r� Nr� r� r� )
r rO r rP r� r& r' rS r r
)r r� rU rP rr r r r �test_blocksize_send+ s z BasicTest.test_blocksize_sendc C s< t �d�}td�|_| �t�� |�dd|� W d Q R X d S )Nz example.comr\ rM �test)r rO r rP r� � TypeErrorrR )r rU r r r �test_send_type_error6 s
z BasicTest.test_send_type_errorc C sd t }ttt t �}tj|dd�}|�� | �|� � |� |�
� xdt dd�D ]V}ttt t �}tj|dd�}|�� | �|� |�|� |� |� � |� |�
� qNW x�dD ]�}tt| �}tj|dd�}|�� z|y |� � W n` tj k
�rB } z>| �|j
|� dt|� }| �t|�|� | �t|�|� W d d }~X Y n X | �d� W d |�
� X q�W d S )Nr_ )rh r � )r\ zfoo
z IncompleteRead(%d bytes read)zIncompleteRead expected)�chunked_expectedr �
chunked_start�
last_chunk� chunked_endr ru rv rS r. r r �IncompleteRead�partialrD r� r r� )r rr rP rx r0 �xr� �expected_messager r r � test_chunked= s2 $
"zBasicTest.test_chunkedc C s� t }t|�}td�}ttt t �}tj|dd�}|� � |�
|�}| � |d |� |� | � ||� |� � x�t
dd�D ]�}ttt t �}tj|dd�}|� � t|�}|�
|d|� �}||�
|||| � �7 }||�
||d � �7 }| � |d |� |� | � ||� |� � qxW x�dD ]�} tt| �}tj|dd�}|� � z~y|�
|�}W n` tjk
�r� } z>| � |j|� dt|� }
| � t|�|
� | � t|�|
� W d d }~X Y n X | �d � W d |� � X �q W d S )
Nr| r_ )rh r r� r )r\ zfoo
z IncompleteRead(%d bytes read)zIncompleteRead expected)r� rD r� r r� r� r� r ru rv r� rS r r �
memoryviewr� r� r� r r� ) r rr Z nexpectedr� rP rx r0 �mr� r� r� r r r �test_readinto_chunked] sD
"z BasicTest.test_readinto_chunkedc C s� d}t |t t �}tj|dd�}|�� | �|�� d� | �|jd� | �|j d� | �
|� � � | � |j
� |�� | �
|j
� d S )NzEHTTP/1.1 200 OK
Transfer-Encoding: chunked
a
hello world
1
d
ra )rh r �� �OK)r r� r� r ru rv rS r. �status�reasonrp r� r� r� r )r r� rP rx 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 ) NzEHTTP/1.1 200 OK
Transfer-Encoding: chunked
a
hello world
1
d
ra )rh r� r s r� r� )r r� r� r ru rv r� r� rS r~ r� r� rp r� r� r� r )r r� rP rx r� r0 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
r_ )rh s Hello
)r r ru rv rS r. rp r� )r rP rx r r r � test_negative_content_length� s z&BasicTest.test_negative_content_lengthc
C s� t d�}tj|dd�}|�� y |�� W n` tjk
r� } z@| �|jd� | �t|�d� | �t |�d� | �
|� � � W d d }~X Y n X | � d� d S )Nz.HTTP/1.1 200 OK
Content-Length: 10
Hello
r_ )rh s Hello
z-IncompleteRead(7 bytes read, 3 more expected)zIncompleteRead expected)
r r ru rv r. r� rS r� r� r rp r� r� )r rP rx 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-Lengthz example.comc s � � ddd�S )Nr] z/urlrV )rR r )rU r r �<lambda>� r z&BasicTest.test_epipe.<locals>.<lambda>i� zBasic realm="example"zwww-authenticate)
r( r rO rP r� r* � getresponserS r� rw )r rP rx r )rU r �
test_epipe� s
zBasicTest.test_epipec C s8 ddd d }t �t|��}| �t jt jf|j� d S )NzHTTP/1.1 200 Okr� i z
)r ru r r� � LineTooLongr� rv )r rV rx 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 i z
)r ru r r� r rv )r rV rx r r r � test_overflowing_header_line� s z&BasicTest.test_overflowing_header_linec C s^ dd }t �t|��}| �t j��}|�� W d Q R X | �dt|j�� | �dt|j�� d S )NzHTTP/1.1 100 OK
r
i � zgot more than rW ) r ru r r� r� rv rn r � exception)r rV rx �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 i za
hello world
0
)r ru r rv r� r r. )r rV rx r r r � test_overflowing_chunked_line� s z'BasicTest.test_overflowing_chunked_linec C s` d}t |�}t�|�}|�� | �|�� d� | �|�� � | �|j � |�
� | �|j � d S )NzHTTP/1.1 200 Okr ) r r ru rv rS r. rp r� r� r� r )r rV rP rx 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 )Nz example.comc s e Zd Z� fdd�ZdS )z+BasicTest.test_error_leak.<locals>.Responsec s | � t jj| f|�|� d S )N)r ru r )r r= r> )�responser r r s z4BasicTest.test_error_leak.<locals>.Response.__init__N)r# r$ r% r r )r r r �Response s r zInvalid status liner_ rN )
r rO ru Zresponse_classr rP rR r� r� r rp r� r )r rU r r )r r �test_error_leak s
zBasicTest.test_error_leakc C sR d}t d }tt| t t �}tj|dd�}|�� | �|� � |� |�
� d S )Nz3;foo=bar
abc
s abcr_ )rh ) r� r r� �last_chunk_extendedr� r ru rv rS r. r )r Zextrarr rP rx r r r �test_chunked_extension s z BasicTest.test_chunked_extensionc C sB t }ttt �}tj|dd�}|�� | �|�� |� |� � dS )z9some servers may serve up a short chunked encoding streamr_ )rh N)
r� r r� r� r ru rv rS r. r )r rr rP rx r r r �test_chunked_missing_end) 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 ignoredr_ )rh r N)
r� r r� r� r� r� r ru rv rS r. r r )r rr rP rx r r r �test_chunked_trailers2 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� extradatar_ )rh r N)r� r r� r� r� r� r ru rv rS r. r r
r )r rr r rP rx r r r �test_chunked_sync= 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 streams extradatas
Hello123
s' HTTP/1.1 200 OK
Content-Length: 10
r_ )rh N)r r ru rv rS r. r r )r r rr rP rx r r r �test_content_length_syncI s z"BasicTest.test_content_length_syncc C s` d}d}t d| | �}tj|dd�}|�� | �|�d�|g� | �|j�� |� |�� d S )Ns extradatas
Hello123
s' HTTP/1.1 200 OK
Content-Length: 10
r_ )rh i� ) r r ru rv rS � readlinesr r. r )r r rr rP rx r r r � test_readlines_content_lengthU s z'BasicTest.test_readlines_content_lengthc C s^ d}d}t d| | �}tj|dd�}|�� | �|�d�|� | �|j�� |� |�� d S )Ns extradatas
Hello123
s' HTTP/1.1 200 OK
Content-Length: 10
r_ )rh i� ) r r ru rv rS �read1r r. r )r r rr rP rx 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 )Ns extradatas
Hello123
s' HTTP/1.1 200 OK
Content-Length: 10
r_ )rh �
r ) r r ru rv rS r1 r r. r )r r rr rP rx r r r �"test_readline_bound_content_lengthk s z,BasicTest.test_readline_bound_content_lengthc C sv d}d}t d|d | �}tj|dd�}|�� | �|�d�|d � | �|�� |� | �|j�� |� |�� d S ) Ns extradatas
Hello123
s' HTTP/1.1 200 OK
Content-Length: 30
r� r_ )rh � r� ) r r ru rv rS r r. r r )r r rr rP rx r r r � test_read1_bound_content_lengthw s z)BasicTest.test_read1_bound_content_lengthc s� t � t jt jt j��| ��j� ��tdf� ��� d � � �fdd�}t j
|d�}|� � | �|j t
d�� tj��� � }|�dd� |�� }z>| �|jtj� t j |�� d�}z|�d � W d |�� X W d |�� |�� X |� � | �� d � d S )
Nr c
sb �� � \} }| �H | �d��2}x|�� }|�d�s P q W | �d� |�� � W d Q R X W d Q R X d S )Nr s
s' HTTP/1.1 200 Connection established
)Zacceptr r1 �rstripr r. )rU �address�readerr� )�result�servr r �
run_server� s
z2BasicTest.test_response_fileno.<locals>.run_server)�targetr r` z
dummy:1234)�filenos
proxied data
)�socket�AF_INET� SOCK_STREAMZ IPPROTO_TCP�
addCleanupr Zbind�HOST�listen� threadingZThread�startr� �floatr rO � getsocknamerR r rS r� r� r( r �detach)r r& ZthreadrU r r� r )r$ r% r �test_response_fileno� s.
z BasicTest.test_response_filenoc C s4 G dd� dt j�}|d�}td�|_|�dd� dS )zv
It should be possible to override the default validation
behavior in putrequest (bpo-38216).
c @ s e Zd Zdd� ZdS )zRBasicTest.test_putrequest_override_domain_validation.<locals>.UnsafeHTTPConnectionc S s d S )Nr )r r� r r r �_validate_path� s zaBasicTest.test_putrequest_override_domain_validation.<locals>.UnsafeHTTPConnection._validate_pathN)r# r$ r% r5 r r r r �UnsafeHTTPConnection� s r6 z example.comr\ r_ z/ N)r rO r rP rl )r r6 rU 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 @ s e Zd Zdd� ZdS )zPBasicTest.test_putrequest_override_host_validation.<locals>.UnsafeHTTPConnectionc S s d S )Nr )r r� r r r �_validate_host� s z_BasicTest.test_putrequest_override_host_validation.<locals>.UnsafeHTTPConnection._validate_hostN)r# r$ r% r8 r r r r r6 � s r6 z
example.com
r\ r_ rN r )Z skip_host)r rO r rP rl )r r6 rU r r r �(test_putrequest_override_host_validation� s
z2BasicTest.test_putrequest_override_host_validationc C s4 G dd� dt j�}|d�}td�|_|�dd� dS )z�
It should be possible to override the default encoding
to transmit bytes in another encoding even if invalid
(bpo-36274).
c @ s e Zd Zdd� ZdS )zIBasicTest.test_putrequest_override_encoding.<locals>.UnsafeHTTPConnectionc S s
|� d�S )Nzutf-8)r
)r Zstr_urlr r r �_encode_request� s zYBasicTest.test_putrequest_override_encoding.<locals>.UnsafeHTTPConnection._encode_requestN)r# r$ r% r: r r r r r6 � s r6 z example.comr\ r_ u /☃N)r rO r rP rl )r r6 rU r r r �!test_putrequest_override_encoding� s
z+BasicTest.test_putrequest_override_encodingN)2r# r$ r% r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r r r r r r r r r
r r r r r r r r r r r r4 r7 r9 r; r r r r r� s^
+
(
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 z�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 )Nr_ )rh ) r r� r ru rv r& �BufferedReader�fprx )r rP rx r r r �setUp� s
zExtendedReadTest.setUpc s� | j }|jj� d� fdd� }||j_g }x�|�d�}|r�| �t|�d� |�� }| �t|�t|�� | �|�|�� |�t|��}| � ||� n|�� }| �
|� |� |� |s*P q*W | � d� |�| j
� d S )Nr- c s( � | �}| dkr |d | � S |d d� S )Nr r r )r0 r� )�oldpeekr r �mypeek� s z*ExtendedReadTest.test_peek.<locals>.mypeekr� r r )r- )rx r> �peekZ
assertGreaterrD ZassertGreaterEqualrp rq r. rS r� rI r� �lines_expected)r rx rA �allr� Zp2r� r )r@ r � test_peek� s(
zExtendedReadTest.test_peekc C s | j }| �| j j| j� d S )N)rx �_verify_readliner1 rC )r rx r r r �
test_readline s z ExtendedReadTest.test_readlinec C s` g }xD|d�}|r6|dkr6t |�dk r6| �|�d�� |�|� |sP qW | �d�|�|� d S )Nr� s foo�
r )rD rp r� rI rS r� )r r1 rr rD r� r r r rF s
z!ExtendedReadTest._verify_readlinec s0 �j � � �fdd�}t|�}��|j�j� d S )Nc s � � d�} ��t| �d� | S )N� )r �assertLessEqualrD )Zres)rx r r r r + s
z&ExtendedReadTest.test_read1.<locals>.r)rx � ReadlinerrF r1 rC )r r Z readlinerr )rx r r �
test_read1) s zExtendedReadTest.test_read1c C s@ | j }g }x |�� }|sP |�|� q W | �d�|�| j� d S )Nr )rx r rI rS r� rC )r rx rD r
r r r �test_read1_unbounded2 s z%ExtendedReadTest.test_read1_unboundedc C sR | j }g }x.|�d�}|s P | �t|�d� |�|� q W | �d�|�| j� d S )Nr r )rx r rJ rD rI rS r� rC )r rx rD r
r r r �test_read1_bounded< s
z#ExtendedReadTest.test_read1_boundedc C s | � | j�d�d� d S )Nr r )rS rx r )r r r r � test_read1_0G s z ExtendedReadTest.test_read1_0c C s | j �d�}| �dt|�� d S )Nr )rx rB rJ rD )r r� r r r � test_peek_0J s z ExtendedReadTest.test_peek_0)r# r$ r% r3 r� �findr
rC Z
lines_chunkedr? rE rG rF rL rM rN rO rP r r r r r< � s
r< c @ s e Zd ZdZdZdS )�ExtendedReadTestChunkedz:
Test peek(), read1(), readline() in chunked mode
z�HTTP/1.1 200 OK
Transfer-Encoding: chunked
a
hello worl
3
d!
9
and now
23
for something completely different
3
foo
0
N)r# r$ r% r3 r� r r r r rR O s rR c @ s e Zd ZdZdd� Zdd� ZdS )rK zT
a simple readline class that uses an arbitrary read function and buffering
c C s || _ d| _d S )Nr )�readfunc� remainder)r rS r r r r i s zReadliner.__init__c C s� g }d}| j }y�xP|�d�}|dkr&P |t|� |krB|| d }|�|� | �� }|sd}P qW |d7 }|�|d |� � ||d � | _ d�|�S d�|�| _ � Y nX d S )Nr rH r- r r )rT rQ rD rI rS r� )r �limitr
Zdatalenr. r� r r r r1 m s,
zReadliner.readlineN)r# r$ r% r3 r r1 r r r r rK e s rK c @ s$ e Zd Zdd� Zdd� Zdd� ZdS )� OfflineTestc C sj dh}ddh}xHt t�D ]<}|�d�s||kr0qtt|�}t|dd �dkr|�|� qW | �tj|� d S )N� responsesZ HTTPMessage�
parse_headersr� r$ z http.client)�dirr rq �getattr�addZassertCountEqual�__all__)r rr Z blacklistr� Z
module_objectr r r �test_all� s
zOfflineTest.test_allc C s | � tjtj d� d S )Nz Not Found)rS r rW � NOT_FOUND)r r r r �test_responses� s zOfflineTest.test_responsesc 6 C s� ddddddddd d
d d d
ddddddddddddddd d d d d d!d"d#d$d%d&d'd(d)d*d+d,d-d.d/d0d1d2d3d4d5d6g6}x4|D ],}| j |d7�� | �tt|�� W d Q R X qvW d S )8NZCONTINUEZSWITCHING_PROTOCOLSZ
PROCESSINGr� 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_FAILEDZMISDIRECTED_REQUESTZUNPROCESSABLE_ENTITYZLOCKEDZFAILED_DEPENDENCYZUPGRADE_REQUIREDZPRECONDITION_REQUIREDZTOO_MANY_REQUESTSZ REQUEST_HEADER_FIELDS_TOO_LARGEZINTERNAL_SERVER_ERRORZNOT_IMPLEMENTEDZ BAD_GATEWAYZSERVICE_UNAVAILABLEZGATEWAY_TIMEOUTZHTTP_VERSION_NOT_SUPPORTEDZINSUFFICIENT_STORAGEZ NOT_EXTENDEDZ NETWORK_AUTHENTICATION_REQUIRED)Zconstant)r� rp �hasattrr )r rr �constr r r �test_client_constants� sr
z!OfflineTest.test_client_constantsN)r# r$ r% r] r_ rb r r r r rV � s rV 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 )N) r) r* r+ r% r � bind_portr Zfind_unused_port� source_portr. rU )r r r r r? � s
zSourceAddressTest.setUpc C s* | j r| j �� d | _ | j�� d | _d S )N)rU r r% )r r r r �tearDown� s
zSourceAddressTest.tearDownc C sB t jt| jd| jfd�| _| j�� | �| jj� � d | j� d S )Nr\ )�source_addressr )
r rO r- r re rU r< rS rP r2 )r r r r � testHTTPConnectionSourceAddress� s
z1SourceAddressTest.testHTTPConnectionSourceAddress�HTTPSConnectionz'http.client.HTTPSConnection not definedc C s t jt| jd| jfd�| _d S )Nr\ )rg )r ri r- r re rU )r r r r � testHTTPSConnectionSourceAddress� s
z2SourceAddressTest.testHTTPSConnectionSourceAddressN) r# r$ r% r? rf rh �unittest�skipIfr` r rj r r r r rc � s
rc c @ s( e Zd ZdZdd� Zdd� Zdd� ZdS )� TimeoutTestNc C s. t � t jt j�| _t�| j�t_| j�� d S )N) r) r* r+ r% r rd rm �PORTr. )r r r r r? � s zTimeoutTest.setUpc C s | j �� d | _ d S )N)r% r )r r r r rf � s
zTimeoutTest.tearDownc C s� | � t�� � t�d� zt�ttj�}|� � W d t�d � X | �
|j � � d� |�
� | � t�� � t�d� z tjttjd d�}|� � W d t�d � X | �
|j � � d � |�
� tjttjdd�}|� � | �
|j � � d� |�
� d S )N� )�timeout)� assertIsNoner) ZgetdefaulttimeoutZsetdefaulttimeoutr rO r- rm rn r< rS rP Z
gettimeoutr )r ZhttpConnr r r �testTimeoutAttribute s*
z TimeoutTest.testTimeoutAttribute)r# r$ r% rn r? rf rr r r r r rm � s rm c @ s$ e Zd Zdd� Zdd� Zdd� ZdS )�PersistenceTestc C s� d}x�|D ]�\}}}| j ||d��� d�||�}t|�}| �|j� |�dd� |�� �"}| �|jd k| � |�� W d Q R X | �|jd k| � | �|j d� |�dd� | �|j |r�dnd� W d Q R X q
W d S ) N))z1.0r\ F)z1.0zConnection: keep-alive
T)z1.1r\ T)z1.1zConnection: close
F)z1.0zConnection: keep-ALIVE
T)z1.1zConnection: cloSE
F)�versionrT z4HTTP/{} 200 OK
{}Content-Length: 12
Dummy body
r_ z/open-connectionr z/subsequent-requestr� )
r� rf r4 rq rP rR r rS r. r5 )r �testsrt rT Zreuser� rU r r r r �test_reuse_reconnect' s
z$PersistenceTest.test_reuse_reconnectc
C s� dd� }t jtjf|tff}xr|D ]j\}}| j|d��P td|�}|�dd� | �||j � | �
|j � |�dd� | � |j
d� W d Q R X q"W d S ) Nc s& t �| �� � fdd�}|� _t �� �S )z3Return BufferedReader that raises ECONNRESET at EOFc s t j�� | �}|dkr t� �|S )Nr )r& r' r� �ConnectionResetError)�bufferr� )�streamr r r� J s zNPersistenceTest.test_disconnected.<locals>.make_reset_reader.<locals>.readinto)r& r' r� r= )r r� r )ry r �make_reset_readerG s
z<PersistenceTest.test_disconnected.<locals>.make_reset_reader)r
r r_ z
/eof-responsez
/reconnectr� )r& r' r �RemoteDisconnectedrw r� r4 rR r� r rq rP rS r5 )r rz ru Zstream_factoryr
rU r r r �test_disconnectedE s
z!PersistenceTest.test_disconnectedc C sV t d�}|jddddid� | �tj|j� | �|j� |�dd� | �|j d� d S ) Ns HTTP/1.1 100 Continue
r_ rN ZExpectz 100-continue)rW z
/reconnectr� )
r4 rR r� r r{ r rq rP rS r5 )r rU r r r �test_100_close` s z PersistenceTest.test_100_closeN)r# r$ r% rv r| r} r r r r rs % s rs 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 )Nri zssl support required)r` r �skipTest)r r r r r? o s
zHTTPSTest.setUpc C s ddl m} || |d�S )Nr )�make_https_server)�certfile)Ztest.ssl_serversr� )r r� r� r r r � make_servers s zHTTPSTest.make_serverc C s$ t jttjdd�}| �|jd� d S )Nro )rp )r ri r- rm rn rS rp )r r� r r r �test_attributesw s zHTTPSTest.test_attributesc
C sl dd l }t�d� t�d��F t�dd�}| �|j��}|�dd� W d Q R X | � |j
j d� W d Q R X d S )Nr �networkzself-signed.pythontest.neti� r_ rN �CERTIFICATE_VERIFY_FAILED) �sslr �requires�transient_internetr ri r� �SSLErrorrR rS r
r� )r r� r� �exc_infor r r �test_networked| s
zHTTPSTest.test_networkedc C sz dd l }t�d� t�d��T |�� }tjdd|d�}|�dd� |�� }|� � | �
d|� d �� |� � W d Q R X d S )
Nr r� zself-signed.pythontest.neti� )�contextr_ rN �nginx�server) r� r r� r� Z_create_unverified_contextr ri rR r r rn rw )r r� r� r� rx r r r � test_networked_noverification� s
z'HTTPSTest.test_networked_noverificationc C sj t �d� t �d��L t�dd�}|�dd� |�� }|�d�}|�� |�� | � d|� W d Q R X d S )Nr� zwww.python.orgi� r_ rN z content-typez text/html)
r r� r� r ri rR r rw r rn )r r� rx 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�|��� |�|j�}| �|j|j� | �|j d� |�
t � y(t j
|d|d�}|�dd� |�� }W nR |jk
r� } z2t|�}t�d |�r�t�d
|� d |� d ���� W d d }~X Y nX |�d
�}|�� |�� | �d|� W d Q R X d S )Nr r� zself-signed.pythontest.netTi� )r� r_ rN z(?i)key.too.weakzGot z trying to connect to z). See https://bugs.python.org/issue36816.r� r� )r� r r� r� �
SSLContext�PROTOCOL_TLS_CLIENTrS Z verify_modeZ
CERT_REQUIRED�check_hostname�load_verify_locations� CERT_selfsigned_pythontestdotnetr ri rR r r� r �re�searchrk ZSkipTestrw r rn ) r r� Zselfsigned_pythontestdotnetr� r� rx 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� tj dd|d�}| �
|j ��}|� dd� W d Q R X | �
|jjd� W d Q R X d S ) Nr r� zself-signed.pythontest.neti� )r� r_ rN r� )r� r r� r� r� r� r� �CERT_localhostr ri r� r� rR rS r
r� )r r� r� r� r� r r r �test_networked_bad_cert� s
z!HTTPSTest.test_networked_bad_certc C sX dd l }| �t�}t�d|j�}| �|j��}|�dd� W d Q R X | � |j
j d� d S )Nr � localhostr_ rN r� ) r� r� r� r ri r r� r� rR rS r
r� )r r� r� r� r� r r r �test_local_unknown_cert� 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� r_ z /nonexistenti� )r� r� r� r� r� r� r ri r r, r rR r rS r� )r r� r� r� r� rx r r r �test_local_good_hostname� s
z"HTTPSTest.test_local_good_hostnamec C s� dd l }| �t�}|�|j�}|�t� tjd|j|d�}| � |j
�� |� dd� W d Q R X t �
dtf�� tjd|j|dd�}W d Q R X | � |j
�� |� dd� W d Q R X d |_t �
dtf�� tjd|j|d d�}W d Q R X |� dd
� |�� }|�� |�� | �|jd � d |_tjd|j|d�}|� dd
� |�� }| �|jd � |�� |�� t �
dtf�� tjd|j|dd�}W d Q R X | � |j
�� |� dd� W d Q R X d S ) Nr r� )r� r_ rN r\ T)r� r� Fz /nonexistenti� )r� r� �CERT_fakehostnamer� r� r� r ri r r� ZCertificateErrorrR r Zcheck_warnings�DeprecationWarningr� r r rS r� )r r� r� r� r� rx r r r �test_local_bad_hostname� sB
z!HTTPSTest.test_local_bad_hostnameri z)http.client.HTTPSConnection not availablec C s^ x dD ]}| � tjtj|� qW x8dD ]0\}}}t�|�}| �||j� | �||j� q&W d S )N)zwww.python.org:abcz user:password@www.python.org))z[fe80::207:e9ff:fe9b]:8000zfe80::207:e9ff:fe9bi@ )zwww.python.org:443zwww.python.orgi� )zwww.python.org:zwww.python.orgi� )zwww.python.orgzwww.python.orgi� )z[fe80::207:e9ff:fe9b]zfe80::207:e9ff:fe9bi� )z[fe80::207:e9ff:fe9b]:zfe80::207:e9ff:fe9bi� )r� r r� ri rS r r )r r� r� r� r� r r r r� s
zHTTPSTest.test_host_portc C s� dd l }|js| �d� t�dd�}| �|jj� |�|j �}| �
|j� tjdd|d�}| � |j|� | �
|jj� tjdd|t d�}| �|jj� d S )Nr zTLS 1.3 support requiredr� i� )r� )r� Z cert_file)
r� Z HAS_TLSv1_3r r ri rp Z_contextZpost_handshake_authr� r� r� ZassertIsr� )r r� r� r� r r r �test_tls13_pha" s
zHTTPSTest.test_tls13_phaN)r# r$ r% r? r� r� r� r� r Zsystem_must_validate_certr� r� r� r� r� r� rk rl r` r r� r� r r r r r~ m 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 )Nz example.comr\ )r rO rU r rP )r r r r r? 8 s zRequestBodyTest.setUpc C s( t �| jj�}|�� t�|�}||fS )N)r& r' rP r
r1 r rX )r �f�messager r r �get_headers_and_fp= s
z"RequestBodyTest.get_headers_and_fpc
C s� ddgdfdf}x�|D ]�\}}| � |��z t�d�| _td� | j_| _| j�dd|� | �� \}}| �d |� | �d
|� | � |�
d �d � | � ||� � � W d Q R X qW d S )
Ns foos bars 3
foo
3
bar
0
))s foos bars 3
foo
3
bar
0
z example.comr\ r] z/urlz Content-TypezContent-LengthzTransfer-Encodingr� ) r� r rO rU r rP rR r� r� rS �getr. )r r� rV rr r� r� r r r �test_list_bodyC s
z RequestBodyTest.test_list_bodyc C sL | j �dddddi� | �� \}}| �d|�d�� | �dt|�� �� d S )Nr] z/urlrV zContent-LengthZ42zcontent-lengthrI )rU rR r� rS r� rD r. )r r� r� r r r �test_manual_content_lengthW s
z*RequestBodyTest.test_manual_content_lengthc C s` | j �ddd� | �� \}}| �d|�� � | �|�� � | �d|�d�� | �d|�� � d S )Nr] z/urlrV z
text/plain�4zcontent-lengths body) rU rR r� rS �get_content_typerq � get_charsetr� r. )r r� r� r r r �test_ascii_body` s z RequestBodyTest.test_ascii_bodyc C s` | j �ddd� | �� \}}| �d|�� � | �|�� � | �d|�d�� | �d|�� � d S )Nr] z/urlu bodyÁz
text/plain�5zcontent-lengths body�) rU rR r� rS r� rq r� r� r. )r r� r� r r r �test_latin1_bodyh s z RequestBodyTest.test_latin1_bodyc C s` | j �ddd� | �� \}}| �d|�� � | �|�� � | �d|�d�� | �d|�� � d S )Nr] z/urls body�z
text/plainr� zcontent-length) rU rR r� rS r� rq r� r� r. )r r� r� r r r �test_bytes_bodyp s z RequestBodyTest.test_bytes_bodyc C s� | � tjtj� ttjd��}|�d� W d Q R X ttj��r}| j�dd|� | �� \}}| � d|�
� � | � |� � � | � |�
d�� | � d|�
d�� | � d |�� � W d Q R X d S )
N�wrV r] z/urlz
text/plainzcontent-lengthr� ztransfer-encodings 4
body
0
)r, r �unlink�TESTFNr� �writerU rR r� rS r� rq r� r� r. )r r� r� r r r �test_text_file_bodyx s z#RequestBodyTest.test_text_file_bodyc C s� | � tjtj� ttjd��}|�d� W d Q R X ttjd��n}| j�dd|� | �� \}}| � d|�
� � | � |� � � | � d|�
d�� | �d |� | � d
|�� � W d Q R X d S ) N�wbs body�r r] z/urlz
text/plainr� zTransfer-EncodingzContent-Lengths 5
body�
0
)r, r r� r� r� r� rU rR r� rS r� rq r� r� r� r. )r r� r� r r r �test_binary_file_body� s z%RequestBodyTest.test_binary_file_bodyN)
r# r$ r% r3 r? r� r� r� r� r� r� r� r� r r r r r� 5 s r� 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 ru rx rv )r rV rP 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)rx rw rS )r rT r r r �test_getting_header� s z$HTTPResponseTest.test_getting_headerc C s | j �dd�}| �|d� d S )NzNo-Such-Headerz
default-value)rx rw rS )r rT 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 )NzNo-Such-Header�default�valueszdefault, values)r� r� )rx rw rS )r rT 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 )NzNo-Such-Header)rx rw rS )r rT 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 )NzNo-Such-Headerrj )r� )rx rw rS )r rT r r r � test_getting_header_defaultint� s z/HTTPResponseTest.test_getting_header_defaultintN) r# r$ r% r? r� r� r� r� r� r r r r r� � s r� c @ sL e Zd Zdd� Zdd� Zdd� Zdd� Zd d
� Zd d � Zd
d� 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 rO rU r9 )r �
response_textr r r r? � s zTunnelTests.setUpc C s | j �� d S )N)rU r )r r r r rf � s zTunnelTests.tearDownc s d� fdd� }|S )Nc s t � | d | d d�S )Nr r )r r )r )r" rp rg )r� r r r8 � s z9TunnelTests._create_connection.<locals>.create_connection)NNr )r r� r8 r )r� r r9 � 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 ) Nzdestination.comi�" z
User-Agentz!Mozilla/5.0 (compatible, MSIE 11))r rW ra rN r\ ) rU �
set_tunnelrR rS rP 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_headers� s
z-TunnelTests.test_set_tunnel_host_port_headersc C s | j �� | �t| j jd� d S )Nzdestination.com)rU r< r� � RuntimeErrorr� )r r r r �&test_disallow_set_tunnel_after_connect� 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 ) Nzdestination.comra rN r\ s CONNECT destination.coms Host: destination.com:Nones Host: destination.coms Host: proxy.com) rU r� rR rS rP r r r r� rn r
r� )r r r r �test_connect_with_tunnel� s z$TunnelTests.test_connect_with_tunnelc C sl | j �d� | j �ddd� | �| j jj| j� | �| j jjtj� | � d| j jj
� | � d| j jj
� d S )Nzdestination.comr] rN r\ s CONNECT destination.coms Host: destination.com) rU r� rR rS rP r r r r� rn r
)r r r r �test_connect_put_request� s z$TunnelTests.test_connect_put_requestc C sz d}d� |�}| j�d� | �|�| j_| j�d� t�� �}| j�ddd� W d Q R X |�� � � }| �
d� |�|� d S ) Nz
X-Dummy: 1zHTTP/1.0 200 OK
{}
r zdestination.comr] rN r\ z
header: {}) rf rU Zset_debuglevelr9 r� r r� rR r� r� rn )r Zexpected_headerr� r� r� r r r �test_tunnel_debuglog� s
z TunnelTests.test_tunnel_debuglogN) r# r$ r% r? rf r9 r� r� r� r� r� r r r r r� � s
r� �__main__r� )� verbosity)5r+ Zhttpr r& rd �osr� r� r) r/ rk ZTestCaser� r �path�dirnamer� �herer� r� r� r� r� r� Zchunk_extensionr� r r� r� r- r r( r' r, rO r4 r@ r� r� r� r< rR rK rV rc rm rs r~ r� r� r� r# �mainr r r r �<module> sh ! n S}"Q .H I`#I