| Server IP : 118.27.122.248 / Your IP : 216.73.216.15 Web Server : Apache System : Linux web0264.sh.tyo1 4.18.0-553.79.1.lve.el7h.x86_64 #1 SMP Wed Oct 15 16:34:46 UTC 2025 x86_64 User : c9415830 ( 11735) PHP Version : 8.4.17 Disable Function : NONE MySQL : OFF | cURL : ON | WGET : ON | Perl : ON | Python : ON | Sudo : OFF | Pkexec : OFF Directory : /opt/alt/python36/lib64/python3.6/test/__pycache__/ |
Upload File : |
3
�w2_x# � @ s� d dl Z d dlmZ d dlZd dlZd dlZd dlZd dlZd dlZej Z d dl
m Z ej j
e�Zej jed�Zej jed�Zej jed�ZdZdZd Zd
Zd e d Zd
Zd Ze jZG 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!�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 �Z-G d.d/� d/e �Z.G d0d1� d1e �Z/e0d2k�r�ej1d3d4� dS )5� 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|jd�}|| _|| _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 �6/opt/alt/python36/lib64/python3.6/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 �sendall8 s zFakeSocket.sendallc C s6 |dkr|dkrt j� �| j| 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 D s zFakeSocket.file_closec C s d S )Nr )r r r r r G s zFakeSocket.closec C s d S )Nr )r �levelZoptname�valuer r r �
setsockoptJ 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 j| |� || _d S )N)r r � pipe_trigger)r r r) r r r r O s zEPipeSocket.__init__c C s( | j |krttjd��| j|7 _d S )NZgotcha)r) �OSError�errnoZEPIPEr
)r r
r r r r T s
zEPipeSocket.sendallc C s d S )Nr )r r r r r Y s zEPipeSocket.closeN)r# r$ r% r r r r r r r r( M s r( c @ s$ e Zd ZdZd dd�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.
r c C s" t jj| |�}|dkr td��|S )Nr z caller tried to read past EOF)r& r' �read�AssertionError)r �nr
r r r r- b s zNoEOFBytesIO.readNc C s" t jj| |�}|dkr td��|S )Nr z caller tried to read past EOF)r& r' �readliner. )r Zlengthr
r r r r0 h s zNoEOFBytesIO.readline���)r1 )N)r# r$ r% �__doc__r- r0 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� jd� || _| j| _d S )Nr z example.com)� connections�superr �fake_socket_args�create_connection�_create_connection)r �args)� __class__r r r q s z!FakeSocketHTTPConnection.__init__c s | j d7 _ t� j� S )z.Count the number of times connect() is invokedr )r4 r5 �connect)r )r: r r r; w s z FakeSocketHTTPConnection.connectc O s
t | j� S )N)r r6 )r �pos�kwr r r r7 | s z*FakeSocketHTTPConnection.create_connection)r# r$ r% r2 r r; r7 �
__classcell__r r )r: r r3 n s r3 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}tjd�}td �|_|� |_d
}i }|r\tt|��||< |jd d ||� | j |jj
|j � 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 |j d�}t|�dkrH|d jd�j� }| jj|d� | j| d7 < tj| |� d S )N� :r r r )�split�len�decode�lowerr@ �
setdefault�list�append)r �item�kvZlcKeyr r r rH � s
zBHeaderTests.test_auto_headers.<locals>.HeaderCountingBuffer.appendN)r# r$ r% r rH r r r r �HeaderCountingBuffer� s rK TF�Content-length�Host�Accept-encodingz example.comZ blahblahblahZ spamspamspam�POST�/r )TF)rL rM rN ) rG r �HTTPConnectionr �sock�_bufferr rC �request� assertEqualr@ rE )r rK Zexplicit_header�header�conn�body�headersr r r �test_auto_headers� s
z HeaderTests.test_auto_headersc C sv G dd� dt �}d}d}xZtj||�D ]J\}}tjd�}td �|_|� |_|j|d|� | j |jj
d d
j |�� q&W d}xN|D ]F}tjd�}td �|_|� |_|j|dd � | j |jj
d dj |�� q~W xN|D ]F}tjd�}td �|_|� |_|j|dd� | j |jj
d d
j |�� q�W xXtj ||�D ]H}tjd�}td �|_|� |_|j|dd� | j |jj
dd
j |�� �q&W d S )Nc @ s e Zd Zdd� Zdd� ZdS )z?HeaderTests.test_content_length_0.<locals>.ContentLengthCheckerc S s t j| � d | _d S )N)rG r �content_length)r r r r r � s
zHHeaderTests.test_content_length_0.<locals>.ContentLengthChecker.__init__c S sF |j dd�}t|�dkr6|d j� dkr6|d j� | _tj| |� d S )NrA r r s content-length)rB rC rE �stripr[ rG rH )r rI rJ r r r rH � s zFHeaderTests.test_content_length_0.<locals>.ContentLengthChecker.appendN)r# r$ r% r rH r r r r �ContentLengthChecker� s r] � �PUTrO �PATCHz example.comrP � 0z%Header Content-Length incorrect on {}�GET�CONNECT�DELETE�HEAD�OPTIONS�TRACEz.Header Content-Length set for empty body on {}� � 1)Nr^ )r_ rO r` )rb rc rd re rf rg )
rG � itertools�productr rQ r rR rS rT rU r[ �format�chain)r r] ZbodiesZmethods_with_body�methodrX rW Zmethods_without_bodyr r r �test_content_length_0� sH
z!HeaderTests.test_content_length_0c C s� t jd�}td �|_|jdd� |jdd� | jd|j� |jdd� | jd |j� |jd
d � | jd |j� |jd
d� | jd|j� |jddd� | jd|j� |jdd� | jd|j� |jdd� | jd|j� |jdd� | jd |j� |jd d � | jd |j� |jd d!� | jd"|j� |jd#d$� | jd%|j� |jd&d$� | jd'|j� |jd(d$� | jd)|j� |jd*d$� | jd+|j� d S ),Nz example.comrb rP 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 rQ r rR �
putrequest� putheader�assertInrS )r rW r r r �test_putheader� s>
zHeaderTests.test_putheaderc C sx d}t jd�}td�}||_|jdd� | j|jj|�� d}t jd�}td�}||_|jdd� | j|jj|�� d S )NsC GET /foo HTTP/1.1
Host: [2001::]:81
Accept-Encoding: identity
z [2001::]:81r^ rb z/foosE GET /foo HTTP/1.1
Host: [2001:102A::]
Accept-Encoding: identity
z
[2001:102A::])r rQ r rR rT �
assertTruer
�
startswith)r �expectedrW rR r r r �test_ipv6host_header s
z HeaderTests.test_ipv6host_headerc C sF d}t |�}tj|�}|j� | j|jd�d� | j|jd�d� d S )Nz4HTTP/1.1 200 OK
First: val
: nval
Second: val
ZFirst�valZSecond)r r � HTTPResponse�beginrU � getheader)r rX rR �respr r r �!test_malformed_headers_coped_with s
z-HeaderTests.test_malformed_headers_coped_withc C s2 dt tdd�� d t tdd�� d }t|�}tj|�}|j� | j|jd �d
� | j|jd d
� | j|jd �d � | j|jd d � d
j t
t tdd���}| j|jd�|� | j|jd |� | j |jd�� | j
d|j� xJ|jd�|jd fD ]2}| j|jd�� | j
d|� | j|jd�� � q�W d S )Ns0 HTTP/1.1 200 OK
!#$%&'*+-.^_`|~: value
VCHAR: �! �~ r 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� � r� )�bytes�ranger r r{ r| rU r} �msg�join�map�chrZassertIsNotNonert rv rw �endswith)r rX rR r~ ZvcharZfoldedr r r �test_parse_all_octets s" (
z!HeaderTests.test_parse_all_octetsc C sx t jd�}td�|_|jdd� d(}xN|D ]F\}}| j||f��* | jtd�� |j||� W d Q R X W d Q R X q*W d S ))Nz example.comr^ rb rP �
Invalid
Name�
ValidValue� Invalid
Name� Invalid
Name�
InvalidName�
InvalidName�
InvalidName� InvalidName� InvalidName� Invalid:Name� :InvalidName� ValidName� Invalid
Value�
Invalid
Value�
Invalid
Value� InvalidValue
�
InvalidValue
�
InvalidValue
zInvalid header�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� r� r� r� ) r rQ r rR rr �subTest�assertRaisesRegex�
ValueErrorrs )r rW �cases�namer! r r r �test_invalid_headers< s.
z HeaderTests.test_invalid_headersc
C sv d}t |�}tj|dd�}tj� �}|j� W d Q R X |j� j� }| j|d d� | j|d d� | j|d d� d S ) Ns* HTTP/1.1 200 OK
First: val
Second: val
r )Z
debuglevelr z reply: 'HTTP/1.1 200 OK\r\n'zheader: First: val� zheader: Second: val) r r r{ r �captured_stdoutr| �getvalue�
splitlinesrU )r rX rR r~ �output�linesr r r �test_headers_debuglevelZ s
z#HeaderTests.test_headers_debuglevelN) r# r$ r% rZ ro ru ry r r� r� r� r r r r r? s B"
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 jd�}td�|_|jdd� |j| j� dd� | j|jj�\}}}| j |�}| j
|| j � d S )Nz example.comr rO rP T)�encode_chunked) r rQ r rR rr Z
endheaders�
_make_body�_parse_requestr
�_parse_chunkedrU �
expected_body)r rW �_rX r r r �test_endheaders_chunkedm s
z,TransferEncodingTest.test_endheaders_chunkedc C sv t jd�}td�|_|jdd| j� ddi� | j|jj�\}}}| jddd � |j � D �� | j
|d d� | j
|| j � t jd�}td�|_|jdd| j j d
�ddi� | j|jj�\}}}| jdd d � |j � D �� | j
|d d� | j
|| j � t jd�}td�|_|jdddd id
| j� d� | j|jj�\}}}| jddd � |D �� | j
|d d � | j
| j
|�| j � d S )Nz example.comr rO rP zTransfer-Encoding�chunkedzcontent-lengthc S s g | ] }|j � �qS r )rE )�.0�kr r r �
<listcomp>� s z>TransferEncodingTest.test_explicit_headers.<locals>.<listcomp>zlatin-1c S s g | ] }|j � �qS r )rE )r� r� r r r r� � s z
gzip, chunkedT)rY r� rX c S s g | ] }|j � �qS r )rE )r� r� r r r r� � s )r rQ r rR rT r� r� r
� assertNotIn�keysrU r� rD r� )r rW r� rY rX r r r �test_explicit_headersw s6
z*TransferEncodingTest.test_explicit_headersc C s� x�d
D ]�}t jd�}td�|_|jdd| j|d�� | j|jj�\}}}| j|�}| j || j
� | j |d d � | j d
d d � |D �� qW d S )NFTz example.comr rO rP )� empty_lineszTransfer-Encodingr� zcontent-lengthc S s g | ] }|j � �qS r )rE )r� r� r r r r� � s z5TransferEncodingTest.test_request.<locals>.<listcomp>)FT) r rQ r rR rT r� r� r
r� rU r� r� )r r� rW r� rY rX r r r � test_request� s
z!TransferEncodingTest.test_requestc C sl t jd�}td�|_|jddf � | j|jj�\}}}| j|d d� | jddd � |D �� | j|d
� d S ) Nz example.comr rO rP zTransfer-Encodingr� zcontent-lengthc S s g | ] }|j � �qS r )rE )r� r� r r r r� � s z8TransferEncodingTest.test_empty_body.<locals>.<listcomp>s 0
) r rQ r rR rT r� r
rU r� )r rW r� rY rX r r r �test_empty_body� s
z$TransferEncodingTest.test_empty_bodyFc c sZ | j jd�}xHt|�D ]<\}}|r0|d r0dV |t|�d k rL|d V q|V qW d S )N� r� r r )r� rB � enumeraterC )r r� r� �idx�liner r r r� � s z TransferEncodingTest._make_bodyc C s� |j d�}|d }i }d}xZ|t|�k rtt|| �dkrt|| j d�\}}|jd�j� }|jd�j� ||< |d7 }q W ||dj||d d � �fS )Ns
r r rA zlatin-1)rB rC rD r\ r� )r r
r� rT rY r/ �keyrz r r r r� � s
z#TransferEncodingTest._parse_requestc C s� g }i }d}|j d�}xd|||d � \}}t|d�}|dkrH|d7 }P | j|t|�� |j|� |d7 }|t|�krP qW dj|�S )Nr s
r� � r r )rB �intrU rC rH r� )r r
rX �trailersr/ 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� j s
&
r� c @ sl 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.dYS )Z� BasicTestc C s� d}t |�}tj|�}|j� | j|jd�d� | j|j� � | j|j� | j|j� d� | j |j� � | j|j� |j
� | j |j� d}t |�}tj|�}| j tj |j� d S )NzHTTP/1.1 200 Ok
Textr r s Textz HTTP/1.1 400.100 Not Ok
Text)
r r r{ r| rU r- � assertFalse�isclosed�closedrv r � assertRaises�
BadStatusLine)r rX rR r~ r r r �test_status_lines� s
zBasicTest.test_status_linesc C s t jd�}| jt|�d� d S )Nr^ zBadStatusLine("''",))r r� rU �repr)r �excr r r �test_bad_status_repr s
z BasicTest.test_bad_status_reprc C s� d}t |�}tj|�}|j� | j|jd�d� | j|j� � | j|jd�d� | j|j� � | j|j � |j
� | j|j � d S )Nz*HTTP/1.1 200 Ok
Content-Length: 4
Textr� s Tes xt) r r r{ r| rU r- r� r� rv r� r )r rX rR r~ r r r �test_partial_reads s
z BasicTest.test_partial_readsc C s~ d}t |�}tj|�}|j� | j|j� d� | j|j� � | j|j� d� | j |j� � | j|j
� |j � | j |j
� d S )Nz4HTTP/1.1 200 Ok
Content-Length: 13
Text
Anothers Text
s Another) r r r{ r| rU r0 r� r� r- rv r� r )r rX rR r~ r r r �test_mixed_reads s
zBasicTest.test_mixed_readsc C s� d}t |�}tj|�}|j� td�}|j|�}| j|d� | jt|�d� | j|j � � |j|�}| j|d� | jt|�d� | j
|j � � | j|j � |j � | j
|j � d S )Nz*HTTP/1.1 200 Ok
Content-Length: 4
Textr� s Tes xt)
r r r{ r| � bytearray�readintorU r� r� r� rv r� r )r rX rR r~ �br/ r r r �test_partial_readintos" s
z BasicTest.test_partial_readintosc C s� d}t |�}tj|�}|j� | j|jd�d� | j|j� � | j|jd�d� | j|jd�d� | j|j� � | j|j � |j
� | j|j � d S )NzHTTP/1.1 200 Ok
Textr� s Tes xtr r ) r r r{ r| rU r- r� r� rv r� r )r rX rR r~ r r r �$test_partial_reads_no_content_length6 s
z.BasicTest.test_partial_reads_no_content_lengthc C s� d}t |�}tj|�}|j� td�}|j|�}| j|d� | jt|�d� | j|j � � |j|�}| j|d� | jt|�d� |j|�}| j|d� | j
|j � � d S )NzHTTP/1.1 200 Ok
Textr� s Tes xtr ) r r r{ r| r� r� rU r� r� r� rv )r rX rR r~ r� r/ r r r �(test_partial_readintos_no_content_lengthF s
z2BasicTest.test_partial_readintos_no_content_lengthc C st d}t |�}tj|�}|j� | j|jd�d� | j|j� � | j|jd�d� | j|jd�d� | j|j� � d S )Nz+HTTP/1.1 200 Ok
Content-Length: 10
Textr� s Tes xtr r ) r r r{ r| rU r- r� r� rv )r rX rR r~ r r r �"test_partial_reads_incomplete_bodyY s
z,BasicTest.test_partial_reads_incomplete_bodyc C s� d}t |�}tj|�}|j� td�}|j|�}| j|d� | jt|�d� | j|j � � |j|�}| j|d� | jt|�d� |j|�}| j|d� | j
|j � � | j|j � |j � | j
|j � d S )Nz+HTTP/1.1 200 Ok
Content-Length: 10
Textr� s Tes xtr )
r r r{ r| r� r� rU r� r� r� rv r� r )r rX rR r~ r� r/ r r r �&test_partial_readintos_incomplete_bodyf s$
z0BasicTest.test_partial_readintos_incomplete_bodyc C s^ x d D ]}| j tjtj|� qW x8dD ]0\}}}tj|�}| j||j� | j||j� q&W d S )N�www.python.org:abc� user:password@www.python.org�[fe80::207:e9ff:fe9b]:8000�fe80::207:e9ff:fe9b�@ �www.python.org:80�www.python.org�P �www.python.org:�[fe80::207:e9ff:fe9b]�[fe80::207:e9ff:fe9b]:)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 �
InvalidURLrQ rU r r )r �hp�h�p�cr r r �test_host_port| s
zBasicTest.test_host_portc C s< d}d}t |�}tj|�}|j� |jd�}| j||� 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} rU )r r Zhdr�sr Zcookiesr r r �test_response_headers� s
z BasicTest.test_response_headersc C s6 t dt�}tj|dd�}|j� |j� r2| jd� d S )Nz*HTTP/1.1 200 OK
Content-Length: 14432
re )rn z)Did not expect response from HEAD request)r r, r r{ r| r- �fail)r rR r~ r r r �test_read_head� s zBasicTest.test_read_headc C sT t dt�}tj|dd�}|j� td�}|j|�dkr@| jd� | jt |�d� d S ) Nz*HTTP/1.1 200 OK
Content-Length: 14432
re )rn � r z)Did not expect response from HEAD request� s )
r r, r r{ r| r� r� r rU r� )r rR r~ r� r r r �test_readinto_head� s
z BasicTest.test_readinto_headc C sR dj dd� ttjd �D ��d }d| }t|�}tj|�}| jtjd|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 r{ r� Z
HTTPExceptionr| )r rY r r r r r r �test_too_many_headers� s
z BasicTest.test_too_many_headersc C sn d}t td��V}tjd�}t|�}||_|jdd|� | j|jj |�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.comrb z/fooz%r != %r) �open�__file__r rQ r rR rT rv r
rw rC )r rx rX rW rR r r r �test_send_file� s
zBasicTest.test_send_filec C s� d}t jd�}td �}||_|j|� | j||j� d|_|jtjd|�� | j||j� d|_|jtj |�� | j||j� d S )Ns" this is a test this is only a testz example.comr r� )
r rQ r rR �sendrU r
�arrayr& r' )r rx rW rR r r r � test_send� s
zBasicTest.test_sendc sX dd� � G � fdd�dt j�}d}tjd�}td�}||_|j|� � | j|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 Z blocksizer r r r- � s z<BasicTest.test_send_updating_file.<locals>.UpdatingFile.readNr1 )r1 )r# r$ r% r r r- r )r
r r � UpdatingFile� s r s dataz example.comr^ ) r& �
TextIOBaser rQ r rR r rU r
)r r rx rW rR r )r
r �test_send_updating_file� s
z!BasicTest.test_send_updating_filec C sL d}dd� }t jd�}td�}||_|jdd|� dd i� | j|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 rX � s z&BasicTest.test_send_iter.<locals>.bodyz example.comr^ rb z/foozContent-LengthZ11)r rQ r rR rT rU r
)r rx rX rW rR r r r �test_send_iter� s
zBasicTest.test_send_iterc C s< t jd�}td�|_| jt�� |jdd|� W d Q R X d S )Nz example.comr^ rO �test)r rQ r rR r� � TypeErrorrT )r rW r r r �test_send_type_error s
z BasicTest.test_send_type_errorc C sd t }ttt t �}tj|dd�}|j� | j|j � |� |j
� xdt dd�D ]V}ttt t �}tj|dd�}|j� | j|j |�|j |� |j � |� |j
� qNW x�d D ]�}tt| �}tj|dd�}|j� z|y |j � W n` tj k
�rB } z@| j|j
|� dt|� }| jt|�|� | jt|�|� W Y d d }~X n X | jd� W d |j
� X q�W d S )
Nrb )rn r � r^ �foo
z IncompleteRead(%d bytes read)zIncompleteRead expected)r^ r )�chunked_expectedr �
chunked_start�
last_chunk� chunked_endr r{ r| rU r- r r� �IncompleteRead�partialrC r� r r )r rx rR r~ r/ �xr �expected_messager r r � test_chunked s2 $
"zBasicTest.test_chunkedc C s� t }t|�}td�}ttt t �}tj|dd�}|j � |j
|�}| j |d |� |� | j ||� |j � x�t
dd�D ]�}ttt t �}tj|dd�}|j � t|�}|j
|d|� �}||j
|||| � �7 }||j
||d � �7 }| j |d |� |� | j ||� |j � qxW x�d D ]�} tt| �}tj|dd�}|j � z~y|j
|�}W n` tjk
�r� } z@| j |j|� d t|� }
| j t|�|
� | j t|�|
� W Y d d }~X n X | jd
� W d |j � X �q W d S ) Nr� rb )rn r r r r^ �foo
z IncompleteRead(%d bytes read)zIncompleteRead expected)r^ r( )r rC r� r r r! r" r r{ r| r� rU r r� �
memoryviewr# r$ r� r r ) r rx Z nexpectedr� rR r~ r/ �mr r% r& r r r �test_readinto_chunked, sD
"z BasicTest.test_readinto_chunkedc C s� d}t |t t �}tj|dd�}|j� | j|j� d� | j|jd� | j|j d� | j
|j � � | j |j
� |j� | j
|j
� d S )NzEHTTP/1.1 200 OK
Transfer-Encoding: chunked
a
hello world
1
d
re )rn r �� �OK)r r! r" r r{ r| rU r- �status�reasonrv r� r� r� r )r r rR r~ r r r �test_chunked_headW s zBasicTest.test_chunked_headc C s� d}t |t t �}tj|dd�}|j� td�}|j|�}| j|d� | jt |�d � | j|j
d� | j|j d� | j |j
� � | j|j� |j� | j |j� d S )
NzEHTTP/1.1 200 OK
Transfer-Encoding: chunked
a
hello world
1
d
re )rn r r r r, r- s )r r! r" r r{ r| r� r� rU r� r. r/ rv r� r� r� r )r r rR r~ r� r/ r r r �test_readinto_chunked_headk s
z$BasicTest.test_readinto_chunked_headc C s@ t d�}tj|dd�}|j� | j|j� d� | j|j� � d S )Nz.HTTP/1.1 200 OK
Content-Length: -1
Hello
rb )rn s Hello
)r r r{ r| rU r- rv r� )r rR r~ r r r � test_negative_content_length� s z&BasicTest.test_negative_content_lengthc C s� t d�}tj|dd�}|j� y |j� W n` tjk
r� } zB| j|jd� | jt|�d� | jt |�d� | j
|j � � W Y d d }~X n X | j d� d S )Nz.HTTP/1.1 200 OK
Content-Length: 10
Hello
rb )rn s Hello
z-IncompleteRead(7 bytes read, 3 more expected)zIncompleteRead expected)
r r r{ r| r- r# rU r$ r� r rv r� r )r rR r~ r r r r �test_incomplete_read� s
z BasicTest.test_incomplete_readc sZ t dd�}tjd�� |� _| jt� fdd�� � j� }| jd|j� | jd|j d�� d S ) NzgHTTP/1.0 401 Authorization Required
Content-type: text/html
WWW-Authenticate: Basic realm="example"
s Content-Lengthz example.comc s � j ddd�S )Nr_ z/urlrX )rT r )rW r r �<lambda>� s z&BasicTest.test_epipe.<locals>.<lambda>i� zBasic realm="example"zwww-authenticate)
r( r rQ rR r� r* � getresponserU r. r} )r rR r~ r )rW r �
test_epipe� s
zBasicTest.test_epipec C s8 ddd d }t jt|��}| jt jt jf|j� d S )NzHTTP/1.1 200 Okr� i z
)r r{ r r� � LineTooLongr� r| )r rX r~ r r r � test_overflowing_status_line� s z&BasicTest.test_overflowing_status_linec C s2 ddd d }t jt|��}| jt j|j� d S )NzHTTP/1.1 200 OK
X-Foo: barr i z
)r r{ r r� r7 r| )r rX r~ r r r � test_overflowing_header_line� s z&BasicTest.test_overflowing_header_linec C s: ddd d }t jt|��}|j� | jt j|j� d S )Nz/HTTP/1.1 200 OK
Transfer-Encoding: chunked
r i za
hello world
0
)r r{ r r| r� r7 r- )r rX r~ r r r � test_overflowing_chunked_line� s z'BasicTest.test_overflowing_chunked_linec C s` d}t |�}tj|�}|j� | j|j� d� | j|j� � | j|j � |j
� | j|j � d S )NzHTTP/1.1 200 Okr ) r r r{ r| rU r- rv r� r� r� r )r rX rR r~ r r r �test_early_eof� s
zBasicTest.test_early_eofc sn t jd�}d � G � fdd�dt j�}||_td�|_|jdd� | jt j|j � | 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 r{ 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 linerb rP )
r rQ r{ Zresponse_classr rR rT r� r� r5 rv r� r )r rW r= r )r< r �test_error_leak� s
zBasicTest.test_error_leakc C sR d}t d }tt| t t �}tj|dd�}|j� | j|j � |� |j
� d S )Nz 3;foo=bar
zabc
s abcrb )rn z3;foo=bar
abc
) r r r �last_chunk_extendedr" r r{ r| rU r- r )r Zextrarx rR r~ r r r �test_chunked_extension� s z BasicTest.test_chunked_extensionc C sB t }ttt �}tj|dd�}|j� | j|j� |� |j � dS )z9some servers may serve up a short chunked encoding streamrb )rn N)
r r r r! r r{ r| rU r- r )r rx rR r~ r r r �test_chunked_missing_end� s z"BasicTest.test_chunked_missing_endc C s\ t }ttt t t �}tj|dd�}|j� | j |j
� |� | j |j j
� d� |j � dS )z&See that trailers are read and ignoredrb )rn r N)
r r r r! r� r" r r{ r| rU r- r r )r rx rR r~ r r r �test_chunked_trailers� s z BasicTest.test_chunked_trailersc C sj t }d}ttt t t | �}tj|dd�}|j� | j |j
� |� | j |j j
� |j d�� |j
� dS )zDCheck that we don't read past the end of the chunked-encoding stream� extradatarb )rn r N)r r r r! r� r" r r{ r| rU r- r r
r )r rx rC rR r~ r r r �test_chunked_sync� s zBasicTest.test_chunked_syncc C s\ d}d}t d| | �}tj|dd�}|j� | j|j� |� | j|jj� |� |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
rb )rn N)r r r{ r| rU r- r r )r rC rx rR r~ r r r �test_content_length_sync s z"BasicTest.test_content_length_syncc C s` d}d}t d| | �}tj|dd�}|j� | j|jd�|g� | j|jj� |� |j� d S )Ns extradatas
Hello123
s' HTTP/1.1 200 OK
Content-Length: 10
rb )rn i� ) r r r{ r| rU � readlinesr r- r )r rC rx rR r~ r r r � test_readlines_content_length s z'BasicTest.test_readlines_content_lengthc C s^ d}d}t d| | �}tj|dd�}|j� | j|jd�|� | j|jj� |� |j� d S )Ns extradatas
Hello123
s' HTTP/1.1 200 OK
Content-Length: 10
rb )rn i� ) r r r{ r| rU �read1r r- r )r rC rx rR r~ r r r �test_read1_content_length" s z#BasicTest.test_read1_content_lengthc C sp d}d}t d| | �}tj|dd�}|j� | j|jd�|� | j|jd�d� | j|jj� |� |j� d S )Ns extradatas
Hello123
s' HTTP/1.1 200 OK
Content-Length: 10
rb )rn �
r ) r r r{ r| rU r0 r r- r )r rC rx rR r~ 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�}|j� | j|jd�|d � | j|j� |� | j|jj� |� |j� d S ) Ns extradatas
Hello123
s' HTTP/1.1 200 OK
Content-Length: 30
� rb )rn � r� ) r r r{ r| rU rH r- r r )r rC rx rR r~ r r r � test_read1_bound_content_length9 s z)BasicTest.test_read1_bound_content_lengthc s t jd�}tjtjtjtj��| j�j� �jt df� �j
� d � � �fdd�}|j |d�}|j � | j|j
td�� tj�j� � }|jdd� |j� }z>| j|jtj� tj|j� d �}z|jd
� W d |j� X W d |j� |j� X |j
� | j� d
� d S ) N� threadingr c sb �j � \} }| �H | jd��2}x|j� }|jd�s P q W | jd� |j� � W d Q R X W d Q R X d S )Nr s
s' HTTP/1.1 200 Connection established
)Zacceptr r0 �rstripr r- )rW �address�readerr� )�result�servr r �
run_serverP s
z2BasicTest.test_response_fileno.<locals>.run_server)�targetr rc z
dummy:1234)�filenos
proxied data
)r �
import_module�socket�AF_INET� SOCK_STREAMZ IPPROTO_TCP�
addCleanupr Zbind�HOST�listenZThread�startr� �floatr rQ � getsocknamerT r5 rU r. r- rW r �detach)r rO rU ZthreadrW r< r r )rS rT r �test_response_filenoE s0
z BasicTest.test_response_filenoc C s4 G dd� dt j�}|d�}td�|_|jdd� 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 �urlr r r �_validate_pathu s zaBasicTest.test_putrequest_override_domain_validation.<locals>.UnsafeHTTPConnection._validate_pathN)r# r$ r% re r r r r �UnsafeHTTPConnectiont s rf z example.comr^ rb z/ N)r rQ r rR rr )r rf rW r r r �*test_putrequest_override_domain_validationo 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 rd r r r �_validate_host~ s z_BasicTest.test_putrequest_override_host_validation.<locals>.UnsafeHTTPConnection._validate_hostN)r# r$ r% rh r r r r rf } s rf z
example.com
r^ rb rP r )Z skip_host)r rQ r rR rr )r rf rW r r r �(test_putrequest_override_host_validation| s
z2BasicTest.test_putrequest_override_host_validationc C s4 G dd� dt j�}|d�}td�|_|jdd� 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
|j 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% rj r r r r rf � s rf z example.comr^ rb u /☃N)r rQ r rR rr )r rf rW r r r �!test_putrequest_override_encoding� s
z+BasicTest.test_putrequest_override_encodingN)/r# r$ r% r� r� r� r� r� r� r� r� r� r r r r
r
r r r r r r' r+ r0 r1 r2 r3 r6 r8 r9 r: r; r> r@ rA rB rD rE rG rI rK rN rc rg ri rk r r r r r� � sX
+
*
r� c @ sx e Zd ZdZdZeejd�d� jd�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�}|j� tj|j�|_|| _d S )Nrb )rn ) r r� r r{ r| r& �BufferedReader�fpr~ )r rR r~ r r r �setUp� s
zExtendedReadTest.setUpc s� | j }|jj� d� fdd� }||j_g }x�|jd�}|r�| jt|�d� |j� }| jt|�t|�� | j|j|�� |jt|��}| j ||� n|j� }| j
|� |j |� |s*P q*W | j dj |�| j
� d S ) Nr c s( � | �}| dkr |d | � S |d d� S )Nr rJ r )r/ r )�oldpeekr r �mypeek� s z*ExtendedReadTest.test_peek.<locals>.mypeekrL r r r1 )r1 )r~ rn �peekZ
assertGreaterrC ZassertGreaterEqualrv rw r- rU r� rH r� �lines_expected)r r~ rq �allr Zp2r r )rp r � test_peek� s(
zExtendedReadTest.test_peekc C s | j }| j| j j| j� d S )N)r~ �_verify_readliner0 rs )r r~ r r r �
test_readline� s z ExtendedReadTest.test_readlinec C s` g }xD|d�}|r6|dkr6t |�dk r6| j|jd�� |j|� |sP qW | jdj|�|� d S )Nr s foo�
r )rC rv r� rH rU r� )r r0 rx rt r� r r r rv � s
z!ExtendedReadTest._verify_readlinec s0 �j � � �fdd�}t|�}�j|j�j� d S )Nc s � j d�} �jt| �d� | S )N� )rH �assertLessEqualrC )�res)r~ r r r r � s
z&ExtendedReadTest.test_read1.<locals>.r)r~ � Readlinerrv r0 rs )r r Z readlinerr )r~ r r �
test_read1� s zExtendedReadTest.test_read1c C s@ | j }g }x |j� }|sP |j|� q W | jdj|�| j� d S )Nr )r~ rH rH rU r� rs )r r~ rt r
r r r �test_read1_unbounded� s z%ExtendedReadTest.test_read1_unboundedc C sR | j }g }x.|jd�}|s P | jt|�d� |j|� q W | jdj|�| j� d S )NrJ r )r~ rH rz rC rH rU r� rs )r r~ rt r
r r r �test_read1_bounded s
z#ExtendedReadTest.test_read1_boundedc C s | j | jjd�d� d S )Nr r )rU r~ rH )r r r r � test_read1_0 s z ExtendedReadTest.test_read1_0c C s | j jd�}| jdt|�� d S )Nr )r~ rr rz rC )r r r r r � test_peek_0 s z ExtendedReadTest.test_peek_0)r# r$ r% r2 r� �findr
rs Z
lines_chunkedro ru rw rv r} r~ r r� r� r r r r rl � s
rl 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% r2 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 }y�xP|jd�}|dkr&P |t|� |krB|| d }|j|� | j� }|sd}P qW |d7 }|j|d |� � ||d � | _ dj|�S dj|�| _ � Y nX d S )Nr rx r r r1 )r� r� rC rH r� r� )r �limitr
Zdatalenr- r� r r r r0 1 s,
zReadliner.readlineN)r# r$ r% r2 r r0 r r r r r| ) s r| c @ s$ e Zd Zdd� Zdd� Zdd� ZdS )� OfflineTestc C sj dh}ddh}xHt t�D ]<}|jd�s||kr0qtt|�}t|dd �dkr|j|� qW | jtj|� d S )N� responsesZ HTTPMessage�
parse_headersr� r$ z http.client)�dirr rw �getattr�addZassertCountEqual�__all__)r rx Z blacklistr� Z
module_objectr r r �test_allL s
zOfflineTest.test_allc C s | j tjtj d� d S )Nz Not Found)rU r r� � NOT_FOUND)r r r r �test_responsesZ s zOfflineTest.test_responsesc 5 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/d0d1d2d3d4d5g5}x4|D ],}| j |d6�� | jtt|�� W d Q R X qtW d S )7NZCONTINUEZSWITCHING_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_FAILEDZUNPROCESSABLE_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� rv �hasattrr )r rx Zconstr r r �test_client_constants] sp
z!OfflineTest.test_client_constantsN)r# r$ r% r� r� r� r r r r r� K s r� c @ s@ e Zd Zdd� Zdd� Zdd� Zejee d� d�d d
� �Z
d S ) �SourceAddressTestc C s> t j t jt j�| _tj| j�| _tj� | _| jj � d | _
d S )N) rY rZ r[ rT r � bind_portr Zfind_unused_port� source_portr^ rW )r r r r ro � s
zSourceAddressTest.setUpc C s* | j r| j j� d | _ | jj� d | _d S )N)rW r rT )r r r r �tearDown� s
zSourceAddressTest.tearDownc C sB t jt| jd| jfd�| _| jj� | j| jjj � d | j� d S )Nr^ )�source_addressr )
r rQ r] r r� rW r; rU rR ra )r r r r � testHTTPConnectionSourceAddress� s
z1SourceAddressTest.testHTTPConnectionSourceAddress�HTTPSConnectionz'http.client.HTTPSConnection not definedc C s t jt| jd| jfd�| _d S )Nr^ )r� )r r� r] r r� rW )r r r r � testHTTPSConnectionSourceAddress� s
z2SourceAddressTest.testHTTPSConnectionSourceAddressN) r# r$ r% ro 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 j t jt j�| _tj| j�t_| jj� d S )N) rY rZ r[ rT r r� r� �PORTr^ )r r r r ro � s zTimeoutTest.setUpc C s | j j� d | _ d S )N)rT r )r r r r r� � s
zTimeoutTest.tearDownc C s� | j tj� � tjd� ztjttj�}|j � W d tjd � X | j
|j j � d� |j
� | j tj� � tjd� z tjttjd d�}|j � W d tjd � X | j
|j j � d � |j
� tjttjdd�}|j � | j
|j j � d� |j
� d S )N� )�timeout)� assertIsNonerY ZgetdefaulttimeoutZsetdefaulttimeoutr rQ r] r� r� r; rU rR Z
gettimeoutr )r ZhttpConnr r r �testTimeoutAttribute� s*
z TimeoutTest.testTimeoutAttribute)r# r$ r% r� ro r� r� r r r r r� � s r� c @ s$ e Zd Zdd� Zdd� Zdd� ZdS )�PersistenceTestc C s� d}x�|D ]�\}}}| j ||d
��� d j||�}t|�}| j|j� |jd d
� |j� �"}| j|jd k| � |j� W d Q R X | j|jd k| � | j|j d� |jd d� | j|j |r�dnd� W d Q R X q
W d S )N�1.0r^ F�Connection: keep-alive
T�1.1�Connection: close
�Connection: keep-ALIVE
�Connection: cloSE
)�versionrV z4HTTP/{} 200 OK
{}Content-Length: 12
Dummy body
rb z/open-connectionr z/subsequent-requestr� �r� r^ F�r� r� T�r� r^ T�r� r� F�r� r� T�r� r� F)r� r� r� r� r� r� )
r� rl r3 r� rR rT r5 rU r- r4 )r �testsr� rV Zreuser� rW 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|�}|jdd� | j||j � | j
|j � |jdd� | j |j
d� W d Q R X q"W d S ) Nc s& t j| �� � fdd�}|� _t j� �S )z3Return BufferedReader that raises ECONNRESET at EOFc s t jj� | �}|dkr t� �|S )Nr )r& r' r� �ConnectionResetError)�bufferr� )�streamr r r�
s zNPersistenceTest.test_disconnected.<locals>.make_reset_reader.<locals>.readinto)r& r' r� rm )r r� r )r� r �make_reset_reader
s
z<PersistenceTest.test_disconnected.<locals>.make_reset_reader)� exceptionr rb z
/eof-responsez
/reconnectr� )r& r' r �RemoteDisconnectedr� r� r3 rT r� r5 r� rR rU r4 )r r� r� Zstream_factoryr� rW r r r �test_disconnected s
z!PersistenceTest.test_disconnectedc C sV t d�}|jddddid� | jtj|j� | j|j� |jdd� | j|j d� d S ) Ns HTTP/1.1 100 Continue
rb rP ZExpectz 100-continue)rY z
/reconnectr� )
r3 rT r� r r� r5 r� rR rU r4 )r rW r r r �test_100_close# s z PersistenceTest.test_100_closeN)r# r$ r% 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� Zejeed� d�dd� �ZdS ) � HTTPSTestc C s t td�s| jd� d S )Nr� zssl support required)r� r ZskipTest)r r r r ro 2 s
zHTTPSTest.setUpc C s ddl m} || |d�S )Nr )�make_https_server)�certfile)Ztest.ssl_serversr� )r r� r� r r r � make_server6 s zHTTPSTest.make_serverc C s$ t jttjdd�}| j|jd� d S )Nr� )r� )r r� r] r� r� rU r� )r r r r r �test_attributes: s zHTTPSTest.test_attributesc C sl dd l }tjd� tjd��F tjdd�}| j|j��}|jdd� W d Q R X | j |j
j d� W d Q R X d S )Nr �networkzself-signed.pythontest.neti� rb rP �CERTIFICATE_VERIFY_FAILED) �sslr �requires�transient_internetr r� r� �SSLErrorrT rU r� r/ )r r� r �exc_infor r r �test_networked? s
zHTTPSTest.test_networkedc C sz dd l }tjd� tjd��T |j� }tjdd|d�}|jdd� |j� }|j � | j
d|j d �� |j � W d Q R X d S )
Nr r� zself-signed.pythontest.neti� )�contextrb rP �nginx�server) r� r r� r� Z_create_unverified_contextr r� rT r5 r rt r} )r r� r� r r~ r r r � test_networked_noverificationI s
z'HTTPSTest.test_networked_noverificationc
C sj t jd� t jd��L tjdd�}|jdd� |j� }|jd�}|j� |j� | j d|� W d Q R X d S )Nr� zwww.python.orgi� rb rP z content-typez text/html)
r r� r� r r� rT r5 r} r rt )r r r~ Z content_typer r r �&test_networked_trusted_by_default_certW s
z0HTTPSTest.test_networked_trusted_by_default_certc C s� dd l }tjd� tjd��n |j|j�}|j|_|jt � t
j dd|d�}|j dd� |j
� }|jd�}|j� |j� | jd |� W d Q R X d S )
Nr r� zself-signed.pythontest.neti� )r� rb rP r� r� )r� r r� r� �
SSLContext� PROTOCOL_TLS�
CERT_REQUIRED� verify_mode�load_verify_locations� CERT_selfsigned_pythontestdotnetr r� rT r5 r} r rt )r r� r� r r~ Z
server_stringr r r �test_networked_good_certd s
z"HTTPSTest.test_networked_good_certc C s� dd l }tjd� tjd��h |j|j�}|j|_|jt � t
j dd|d�}| j |j
��}|jdd� W d Q R X | j|jjd� W d Q R X d S ) Nr r� zself-signed.pythontest.neti� )r� rb rP r� )r� r r� r� r� r� r� r� r� �CERT_localhostr r� r� r� rT rU r� r/ )r r� r� r r� r r r �test_networked_bad_certt s
z!HTTPSTest.test_networked_bad_certc
C sX dd l }| jt�}tjd|j�}| j|j��}|jdd� W d Q R X | j |j
j d� d S )Nr � localhostrb rP r� ) r� r� r� r r� r r� r� rT rU r� r/ )r r� r� r r� r r r �test_local_unknown_cert� s
z!HTTPSTest.test_local_unknown_certc C s� dd l }| jt�}|j|j�}|j|_|jt� tj d|j
|d�}| j |j � |j
dd� |j� }| j |j � | j|jd� d S )Nr r� )r� rb z /nonexistenti� )r� r� r� r� r� r� r� r� r r� r r\ r rT r5 rU r. )r r� r� r� r r~ r r r �test_local_good_hostname� s
z"HTTPSTest.test_local_good_hostnamec - C s� dd l }| jt�}|j|j�}|j|_d|_|jt� t j
d|j |d�}| j |j
�� |jdd� W d Q R X tjdtf�� t j
d|j |dd�}W d Q R X | j |j
�� |jdd� W d Q R X d |_tjdtf�� t j
d|j |d d�}W d Q R X |jdd
� |j� }|j� |j� | j|jd � d |_t j
d|j |d�}|jdd
� |j� }| j|jd � |j� |j� tjdtf�� t j
d|j |dd�}W d Q R X | j |j
�� |jdd� W d Q R X d S ) Nr Tr� )r� rb rP r^ )r� �check_hostnameFz /nonexistenti� )r� r� �CERT_fakehostnamer� r� r� r� r� r� r r� r r� ZCertificateErrorrT r Zcheck_warnings�DeprecationWarningr5 r rU r. )r r� r� r� r r~ r r r �test_local_bad_hostname� sF
z!HTTPSTest.test_local_bad_hostnamer� z)http.client.HTTPSConnection not availablec C s^ x d D ]}| j tjtj|� qW x8dD ]0\}}}tj|�}| j||j� | j||j� q&W d S )N�www.python.org:abc� user:password@www.python.org�[fe80::207:e9ff:fe9b]:8000�fe80::207:e9ff:fe9b�@ �www.python.org:443�www.python.org� �www.python.org:�[fe80::207:e9ff:fe9b]�[fe80::207:e9ff:fe9b]:)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� rU r r )r r� r r r r r r r � s
zHTTPSTest.test_host_portN)r# r$ r% ro r� r� r� r� r Zsystem_must_validate_certr� r� r� r� r� r� r� r� r� r r r r r r r� 0 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 jd�| _td� | j_| _| j| j_d S )Nz example.comr^ )r rQ rW r rR )r r r r ro � s zRequestBodyTest.setUpc C s( t j| jj�}|j� tj|�}||fS )N)r& r' rR r
r0 r r� )r �f�messager r r �get_headers_and_fp� s
z"RequestBodyTest.get_headers_and_fpc C s� ddgdfd
f}x�|D ]�\}}| j |��z tjd�| _td� | j_| _| jjdd|� | j� \}}| jd|� | jd |� | j |j
d
�d � | j ||j � � W d Q R X qW d S )N� foo� bar� 3
foo
3
bar
0
z example.comr^ r_ z/urlz Content-TypezContent-LengthzTransfer-Encodingr� �r� r� )r� r� ) r� r rQ rW r rR rT r� r� rU �getr- )r r� rX rx r� r� r r r �test_list_body� s
z RequestBodyTest.test_list_bodyc C sL | j jdddddi� | j� \}}| jd|jd�� | jdt|j� �� d S )Nr_ z/urlrX zContent-LengthZ42zcontent-lengthry )rW rT r� rU r� rC r- )r r� r� r r r �test_manual_content_length� s
z*RequestBodyTest.test_manual_content_lengthc C s` | j jddd� | j� \}}| jd|j� � | j|j� � | jd|jd�� | jd|j� � d S )Nr_ z/urlrX z
text/plain�4zcontent-lengths body) rW rT r� rU �get_content_typer� � get_charsetr� r- )r r� r� r r r �test_ascii_body s z RequestBodyTest.test_ascii_bodyc C s` | j jddd� | j� \}}| jd|j� � | j|j� � | jd|jd�� | jd|j� � d S )Nr_ z/urlu bodyÁz
text/plain�5zcontent-lengths body�) rW rT r� rU r� r� r� r� r- )r r� r� r r r �test_latin1_body s z RequestBodyTest.test_latin1_bodyc C s` | j jddd� | j� \}}| jd|j� � | j|j� � | jd|jd�� | jd|j� � d S )Nr_ z/urls body�z
text/plainr� zcontent-length) rW rT r� rU r� r� r� r� r- )r r� r� r r r �test_bytes_body s z RequestBodyTest.test_bytes_bodyc C s� | j tjtj� ttjd��}|jd� W d Q R X ttj��r}| jjdd|� | j� \}}| j d|j
� � | j |j � � | j |j
d�� | j d|j
d�� | j d |j� � W d Q R X d S )
N�wrX r_ z/urlz
text/plainzcontent-lengthr� ztransfer-encodings 4
body
0
)r\ r �unlink�TESTFNr �writerW rT r� rU r� r� r� r� r- )r r� r� r r r �test_text_file_body s z#RequestBodyTest.test_text_file_bodyc C s� | j tjtj� ttjd��}|jd� W d Q R X ttjd��n}| jjdd|� | j� \}}| j d|j
� � | j |j � � | j d|j
d�� | jd |� | j d
|j� � 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 rW rT r� rU r� r� r� r� r� r- )r r� r� r r r �test_binary_file_body+ s z%RequestBodyTest.test_binary_file_bodyN)
r# r$ r% r2 ro 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
S )�HTTPResponseTestc C s& d}t |�}tj|�| _| jj� d S )NzXHTTP/1.1 200 Ok
My-Header: first-value
My-Header: second-value
Text)r r r{ r~ r| )r rX rR r r r ro ; s zHTTPResponseTest.setUpc C s6 | j jd�}| j|d� | j jdd�}| j|d� d S )Nz My-Headerzfirst-value, second-valuez some default)r~ r} rU )r rV r r r �test_getting_headerB s z$HTTPResponseTest.test_getting_headerc C s | j jdd�}| j|d� d S )NzNo-Such-Headerz
default-value)r~ r} rU )r rV r r r �3test_getting_nonexistent_header_with_string_defaultI s zDHTTPResponseTest.test_getting_nonexistent_header_with_string_defaultc C s< | j jdddg�}| j|d� | j jdd�}| j|d� d S )NzNo-Such-Header�default�valueszdefault, values)r
r )r~ r} rU )r rV r r r �5test_getting_nonexistent_header_with_iterable_defaultM s zFHTTPResponseTest.test_getting_nonexistent_header_with_iterable_defaultc C s | j jd�}| j|d � d S )NzNo-Such-Header)r~ r} rU )r rV r r r �/test_getting_nonexistent_header_without_defaultT s z@HTTPResponseTest.test_getting_nonexistent_header_without_defaultc C s | j jddd�}| j|d� d S )NzNo-Such-Headerrp )r
)r~ r} rU )r rV r r r � test_getting_header_defaultintX s z/HTTPResponseTest.test_getting_header_defaultintN) r# r$ r% ro r r r r
r r r r r r 9 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| _ tj| j �| _| j|�| j_d S )Nz:HTTP/1.0 200 OK
HTTP/1.1 200 OK
Content-Length: 42
z proxy.com)r r rQ rW r8 )r �
response_textr r r ro ] s zTunnelTests.setUpc C s | j j� d S )N)rW r )r r r r r� g s zTunnelTests.tearDownc s d� fdd� }|S )Nc s t � | d | d d�S )Nr r )r r )r )rQ r� r� )r r r r7 k s z9TunnelTests._create_connection.<locals>.create_connection)NNr )r r r7 r )r r r8 j s z TunnelTests._create_connectionc C s� d}d}ddi}| j j|||d� | j jddd� | j| j jj| j� | j| j jjtj� | j| j j |� | j| j j
|� | j| j j |� d S ) Nzdestination.comi�" z
User-Agentz!Mozilla/5.0 (compatible, MSIE 11))r rY re rP r^ ) rW �
set_tunnelrT rU rR 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_headerso s
z-TunnelTests.test_set_tunnel_host_port_headersc C s | j j� | jt| j jd� d S )Nzdestination.com)rW r; r� � RuntimeErrorr )r r r r �&test_disallow_set_tunnel_after_connect| s
z2TunnelTests.test_disallow_set_tunnel_after_connectc C s� | j jd� | j jddd� | j| j jj| j� | j| j jjtj� | j d| j jj
� | j d| j jj
� | j d| j jj
� | j d| j jj
� d S ) Nzdestination.comre rP r^ s CONNECT destination.coms Host: destination.com:Nones Host: destination.coms Host: proxy.com) rW r rT rU rR r r r r rt r
r� )r r r r �test_connect_with_tunnel� s z$TunnelTests.test_connect_with_tunnelc C sl | j jd� | j jddd� | j| j jj| j� | j| j jjtj� | j d| j jj
� | j d| j jj
� d S )Nzdestination.comr_ rP r^ s CONNECT destination.coms Host: destination.com) rW r rT rU rR r r r r rt r
)r r r r �test_connect_put_request� s z$TunnelTests.test_connect_put_requestc C sz d}dj |�}| jjd� | j|�| j_| jjd� tj� �}| jjddd� W d Q R X |j� j � }| j
dj |�|� d S ) Nz
X-Dummy: 1zHTTP/1.0 200 OK
{}
r zdestination.comr_ rP r^ z
header: {}) rl rW Zset_debuglevelr8 r r r� rT r� r� rt )r Zexpected_headerr r� r� r r r �test_tunnel_debuglog� s
z TunnelTests.test_tunnel_debuglogN) r# r$ r% ro r� r8 r r r r r r r r r r \ s
r �__main__r� )� verbosity)2r+ Zhttpr r& rj �osr rY r� ZTestCaser r �path�dirnamer �herer� r� r� r� r r Zchunk_extensionr! r? r� r" r] r r( r' r, rQ r3 r? r� r� rl r� r| r� r� r� r� r� r� r r r# �mainr r r r �<module> sb ! l 1}"P .H *`#I