Your IP : 216.73.216.85
���]� � @ s� d d l Z d d l m Z d d l Z d d l Z d d l Z d d l Z d d l Z d d l Z e j Z d d l
m Z e j j
e � Z e j j e d � Z e j j e d � Z e j j e d � Z d Z d Z d Z d
Z d e d Z d
Z d Z e j Z Gd d � d � Z Gd d � d e � Z Gd d � d e j � Z Gd d � d e j � Z! Gd d � d e � Z" Gd d � d e � Z# Gd d � d e � Z$ Gd d � d e$ � Z% Gd d � d � Z& Gd d! � d! e � Z' Gd" d# � d# e � Z( Gd$ d% � d% e � Z) Gd& d' � d' e � Z* Gd( d) � d) e � Z+ Gd* d+ � d+ e � Z, Gd, d- � d- e � Z- Gd. d/ � d/ e � Z. e j/ d d0 d1 � � Z0 e1 d2 k r�e0 � d S)3� 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 @ sg e Z d Z e j d d d d � Z d d � Z d d d � Z d d � Z d
d � Z d d
� Z
d S)�
FakeSocketNc C sa t | t � r | j 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 �1/opt/alt/python35/lib64/python3.5/test_httplib.py�__init__- s zFakeSocket.__init__c C s" | j d 7_ | j | 7_ d S)N� )r r
)r r
r r r �sendall8 s zFakeSocket.sendallc C sO | d k r$ | d k r$ t j � � | j | j � | _ | j | j _ | j S)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) �__name__�
__module__� __qualname__�io�BytesIOr r r r r r" r r r r r , s r c @ s4 e Z d Z d d � Z d d � Z d d � Z d S)� 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 s4 | j | k r! t t j d � � | 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 Z d Z d d d � Z d d d � Z d S)� NoEOFBytesIOr c C s1 t j j | | � } | d k r- t d � � | S)Nr z caller tried to read past EOF)r& r' �read�AssertionError)r �nr
r r r r- b s zNoEOFBytesIO.readNc C s1 t j j | | � } | d k r- t d � � | S)Nr z caller tried to read past EOF)r& r' �readliner. )r �lengthr
r r r r0 h s zNoEOFBytesIO.readline���)r# r$ r% r- r0 r r r r r, \ s r, c s@ e Z d Z � f d d � Z � f d d � Z d d � Z � S)�FakeSocketHTTPConnectionc s2 d | _ t � j d � | | _ | 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 d 7_ t � j � S)Nr )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% r r; r7 r r )r: r r3 n s r3 c @ sd e Z d Z d d � Z d d � Z d d � Z d d � Z d d
� Z d d � Z d
d � Z d S)� HeaderTestsc C s� Gd d � d t � } x� d D]� } x� d D]� } t j d � } t d � | _ | � | _ d
} i } | r� t t | � � | | <| j d d | | � | j | j j
| j � d
� q* Wq Wd S)Nc @ s( e Z d Z d d � Z d d � Z d S)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 sr | j d � } t | � d k r^ | d j d � j � } | j j | d � | j | d 7<t j | | � d S)N� :r r r )�split�len�decode�lowerr? �
setdefault�list�append)r �item�kvZlcKeyr r r rG � s zBHeaderTests.test_auto_headers.<locals>.HeaderCountingBuffer.appendN)r# r$ r% r rG r r r r �HeaderCountingBuffer� s rJ TF�Content-length�Host�Accept-encodingz example.comZ blahblahblahZ spamspamspam�POST�/r )TF)rK rL rM ) rF r �HTTPConnectionr �sock�_bufferr rB �request� assertEqualr? rD )r rJ Zexplicit_header�header�conn�body�headersr r r �test_auto_headers� s
z HeaderTests.test_auto_headersc C s
Gd d � d t � } d } d } x t j | | � D]k \ } } t j d � } t d � | _ | � | _ | j | d | � | j | j j
d d
j | � � q5 Wd } xm | D]e } t j d � } t d � | _ | � | _ | j | d d � | j | j j
d d j | � � q� Wxm | D]e } t j d � } t d � | _ | � | _ | j | d d � | j | j j
d d
j | � � q!Wxy t j | | � D]e } t j d � } t d � | _ | � | _ | j | d d � | j | j j
d d
j | � � q�Wd S)Nc @ s( e Z d Z d d � Z d d � Z d S)z?HeaderTests.test_content_length_0.<locals>.ContentLengthCheckerc S s t j | � d | _ d S)N)rF r �content_length)r r r r r � s
zHHeaderTests.test_content_length_0.<locals>.ContentLengthChecker.__init__c S sa | j d d � } t | � d k rM | d j � d k rM | d j � | _ t j | | � d S)Nr@ r r s content-length)rA rB rD �striprZ rF rG )r rH rI r r r rG � s (zFHeaderTests.test_content_length_0.<locals>.ContentLengthChecker.appendN)r# r$ r% r rG r r r r �ContentLengthChecker� s r\ � �PUTrN �PATCHz example.comrO � 0z%Header Content-Length incorrect on {}�GET�CONNECT�DELETE�HEAD�OPTIONS�TRACEz.Header Content-Length set for empty body on {}� � 1)Nr] )r^ rN r_ )ra rb rc rd re rf )
rF � itertools�productr rP r rQ rR rS rT rZ �format�chain)r r\ ZbodiesZmethods_with_body�methodrW rV Zmethods_without_bodyr r r �test_content_length_0� sH
z!HeaderTests.test_content_length_0c C s t j d � } t d � | _ | j d d � | j d d � | j d | j � | j d d � | j d | j � | j d
d � | j d | j � | j d
d � | j d | j � | j d d d � | j d | j � | j d d � | j d | j � | j d d � | j d | j � | j d d � | j d | j � | j d d � | j d | j � | j d d! � | j d" | j � | j d# d$ � | j d% | j � | j d& d$ � | j d' | j � | j d( d$ � | j d) | j � | j d* d$ � | j d+ | j � d S),Nz example.comra rO 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 LatinHeaders �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 rP r rQ �
putrequest� putheader�assertInrR )r rV r r r �test_putheader� s> zHeaderTests.test_putheaderc C s� d } t j d � } t d � } | | _ | j d d � | j | j j | � � d } t j d � } t d � } | | _ | j d d � | j | j j | � � d S)NsC GET /foo HTTP/1.1
Host: [2001::]:81
Accept-Encoding: identity
z [2001::]:81r] ra z/foosE GET /foo HTTP/1.1
Host: [2001:102A::]
Accept-Encoding: identity
z
[2001:102A::])r rP r rQ rS �
assertTruer
�
startswith)r �expectedrV rQ r r r �test_ipv6host_header s z HeaderTests.test_ipv6host_headerc C sa d } t | � } t j | � } | j � | j | j d � d � | j | j d � d � d S)Nz4HTTP/1.1 200 OK
First: val
: nval
Second: val
ZFirst�valZSecond)r r � HTTPResponse�beginrT � getheader)r rW rQ �respr r r �!test_malformed_headers_coped_with s
z-HeaderTests.test_malformed_headers_coped_withc C s� d t t d d � � d t t d d � � d } t | � } t j | � } | j � | j | j d � d
� | j | j d d
� | j | j d � d � | j | j d d � d
j t
t t d d � � � } | j | j d � | � | j | j d | � | j | j d � � | j
d | j � x` | j d � | j d f D]B } | j | j d � � | j
d | � | j | j d � � qPWd 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 ry rz rT r{ �msg�join�map�chrZassertIsNotNonerr rt ru �endswith)r rW rQ r| ZvcharZfoldedr r r �test_parse_all_octets s" 4
!#z!HeaderTests.test_parse_all_octetsc C s� t j d � } t d � | _ | j d d � d( } x[ | D]S \ } } | j | | f � �/ | j t d � � | j | | � Wd QRXWd QRXq; Wd S))Nz example.comr] ra rO �
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 rP r rQ rp �subTest�assertRaisesRegex�
ValueErrorrq )r rV Zcases�namer! r r r �test_invalid_headers< s. z HeaderTests.test_invalid_headersN)
r# r$ r% rY rn rs rw r} r� r� r r r r r> s B "
r> c @ s e Z d Z d d � Z d d � Z d d � Z d d � Z d d
� Z d d � Z d
d � Z d d � Z
d d � Z d d � Z d d � Z
d d � Z d d � Z d d � Z d d � Z d d � Z d! d" � Z d# d$ � Z d% d& � Z d' d( � Z d) d* � Z d+ d, � Z d- d. � Z d/ d0 � Z d1 d2 � Z d3 d4 � Z d5 d6 � Z d7 d8 � Z d9 d: � Z d; d< � Z d= d> � Z! d? d@ � Z" dA dB � Z# dC dD � Z$ dE dF � Z% dG dH � Z& dI dJ � Z' dK dL � Z( dM dN � Z) dO dP � Z* dQ dR � Z+ dS dT � Z, dU dV � Z- dW S)X� BasicTestc C s� d } t | � } t j | � } | j � | j | j d � d � | j | j � � | j | j � | j | j � d � | j | j � � | j | j � | j
� | j | j � d } t | � } t j | � } | j t j | j � d S)NzHTTP/1.1 200 Ok
Textr r s Textz HTTP/1.1 400.100 Not Ok
Text)
r r ry rz rT r- � assertFalse�isclosed�closedrt r � assertRaises�
BadStatusLine)r rW rQ r| r r r �test_status_lines\ s
zBasicTest.test_status_linesc C s) t j d � } | j t | � d � d S)Nr] zBadStatusLine("''",))r r� rT �repr)r �excr r r �test_bad_status_reprq s z BasicTest.test_bad_status_reprc C s� d } t | � } t j | � } | j � | j | j d � d � | j | j � � | j | j d � d � | j | j � � | j | j � | j
� | j | j � d S)Nz*HTTP/1.1 200 Ok
Content-Length: 4
Text� s Tes xt) r r ry rz rT r- r� r� rt r� r )r rW rQ r| r r r �test_partial_readsu s
z BasicTest.test_partial_readsc C s� d } t | � } t j | � } | 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 ry rz rT r0 r� r� r- rt r� r )r rW rQ r| r r r �test_mixed_reads� s
zBasicTest.test_mixed_readsc C s� d } t | � } t j | � } | j � t d � } | j | � } | j | d � | j t | � d � | j | j � � | j | � } | j | d � | j t | � 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 ry rz � bytearray�readintorT r� r� r� rt r� r )r rW rQ r| �br/ r r r �test_partial_readintos� s
z BasicTest.test_partial_readintosc C s� d } t | � } t j | � } | j � | j | j d � d � | j | j � � | j | j d � d � | j | j d � d � | j | j � � | j | j � | j
� | j | j � d S)NzHTTP/1.1 200 Ok
Textr� s Tes xtr r ) r r ry rz rT r- r� r� rt r� r )r rW rQ r| r r r �$test_partial_reads_no_content_length� s
z.BasicTest.test_partial_reads_no_content_lengthc C s� d } t | � } t j | � } | j � t d � } | j | � } | j | d � | j t | � d � | j | j � � | j | � } | j | d � | j t | � d � | j | � } | j | d � | j
| j � � d S)NzHTTP/1.1 200 Ok
Textr� s Tes xtr ) r r ry rz r� r� rT r� r� r� rt )r rW rQ r| r� r/ r r r �(test_partial_readintos_no_content_length� s
z2BasicTest.test_partial_readintos_no_content_lengthc C s� d } t | � } t j | � } | j � | j | j d � d � | j | j � � | j | j d � d � | j | j d � d � | j | j � � d S)Nz+HTTP/1.1 200 Ok
Content-Length: 10
Textr� s Tes xtr r ) r r ry rz rT r- r� r� rt )r rW rQ r| r r r �"test_partial_reads_incomplete_body� s
z,BasicTest.test_partial_reads_incomplete_bodyc C s d } t | � } t j | � } | j � t d � } | j | � } | j | d � | j t | � d � | j | j � � | j | � } | j | d � | j t | � 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 ry rz r� r� rT r� r� r� rt r� r )r rW rQ r| r� r/ r r r �&test_partial_readintos_incomplete_body� s$
z0BasicTest.test_partial_readintos_incomplete_bodyc C s} x' d D] } | j t j t j | � q WxL d D]D \ } } } t j | � } | j | | j � | j | | j � q1 Wd 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 �
InvalidURLrP rT r r )r �hp�h�p�cr r r �test_host_port� s
zBasicTest.test_host_portc C sT d } d } t | � } t j | � } | j � | j d � } | 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 ry rz r{ rT )r r Zhdr�sr Zcookiesr r r �test_response_headers� s
z BasicTest.test_response_headersc C sK t d t � } t j | d d �} | j � | j � rG | j d � d S)Nz*HTTP/1.1 200 OK
Content-Length: 14432
rm rd z)Did not expect response from HEAD request)r r, r ry rz r- �fail)r rQ r| r r r �test_read_head s
zBasicTest.test_read_headc C sv t d t � } t j | d d �} | j � t d � } | j | � d k r\ | j d � | j t | � d � d S) Nz*HTTP/1.1 200 OK
Content-Length: 14432
rm rd � r z)Did not expect response from HEAD requests s )
r r, r ry rz r� r� r� rT r� )r rQ r| r� r r r �test_readinto_head s
z BasicTest.test_readinto_headc C so d j d d � t t j d � D� � d } d | } t | � } t j | � } | j t j d | j � d S)Nz
c s s | ] } d | Vq d S)z
Header%d: fooNr )�.0�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 ry r� Z
HTTPExceptionrz )r rX r r� r r r r �test_too_many_headers, s
z BasicTest.test_too_many_headersc C s� d } t t d � �w } t j d � } t | � } | | _ | j d d | � | j | j j | � d | j d t
| � � | f � Wd QRXd S)NsP GET /foo HTTP/1.1
Host: example.com
Accept-Encoding: identity
Content-Length:r z example.comra z/fooz%r != %r) �open�__file__r rP r rQ rS rt r
ru rB )r rv rW rV rQ r r r �test_send_file5 s zBasicTest.test_send_filec C s� d } t j d � } t d � } | | _ | j | � | j | | j � d | _ | j t j d | � � | j | | j � d | _ | j t j | � � | j | | j � d S)Ns" this is a test this is only a testz example.comr r� )
r rP r rQ �sendrT r
�arrayr& r' )r rv rV rQ r r r � test_sendA s
zBasicTest.test_sendc sv d d � � G� f d d � d � } d } t j d � } t d � } | | _ | j | � � | j | j | � d S)Nc s s d Vd Vd Vd S)Nr
Zdata_twor r r r r r
P s z/BasicTest.test_send_updating_file.<locals>.datac s. e Z d Z d Z � � Z d d d � Z d S)z7BasicTest.test_send_updating_file.<locals>.UpdatingFiler r c S s
| j j � S)N)�d�__next__)r � blocksizer r r r- X s z<BasicTest.test_send_updating_file.<locals>.UpdatingFile.readNr2 )r# r$ r% r r� r- r )r
r r � UpdatingFileU s r� s dataz example.comr] )r rP r rQ r� rT r
)r r� rv rV rQ r )r
r �test_send_updating_fileO s z!BasicTest.test_send_updating_filec C sl d } d d � } t j d � } t d � } | | _ | j d d | � d d i � | j | j | � d S)
Nsb GET /foo HTTP/1.1
Host: example.com
Accept-Encoding: identity
Content-Length: 11
onetwothreec s s d Vd Vd Vd S)Ns ones twos threer r r r r rW i s z&BasicTest.test_send_iter.<locals>.bodyz example.comr] ra z/foozContent-LengthZ11)r rP r rQ rS rT r
)r rv rW rV rQ r r r �test_send_iterd s zBasicTest.test_send_iterc C sL t j d � } t d � | _ | j t � � | j d d | � Wd QRXd S)Nz example.comr] rN �test)r rP r rQ r� � TypeErrorrS )r rV r r r �test_send_type_errort s z BasicTest.test_send_type_errorc C s� t } t t t t � } t j | d d �} | j � | j | j � | � | j
� x� t d d � D]s } t t t t � } t j | d d �} | j � | j | j | � | j | � | j � | � | j
� qi Wx� d D]� } t t | � } t j | d d �} | j � z� y | j � Wnw t j k
r�} zT | j | j
| � d t | � } | j t | � | � | j t | � | � WYd d } ~ Xn X| j d � Wd | j
� Xq� Wd S)
Nrm ra r � r] �foo
z IncompleteRead(%d bytes read)zIncompleteRead expected)r] r� )�chunked_expectedr �
chunked_start�
last_chunk� chunked_endr ry rz rT r- r r� �IncompleteRead�partialrB r� r r� )r rv rQ r| r/ �xr� �expected_messager r r � test_chunked{ s2
0
)zBasicTest.test_chunkedc C sh t } t | � } t d � } t t t t � } t j | d d �} | j � | j
| � } | j | d | � | � | j | | � | j � x� t
d d � D]� } t t t t � } t j | d d �} | j � t | � } | j
| d | � � } | | j
| | | | � � 7} | | j
| | d � � 7} | j | d | � | � | j | | � | j � q� Wx� d D]� } t t | � } t j | d d �} | j � z� y | j
| � } Wnw t j k
rC} zT | j | j | � d t | � }
| j t | � |
� | j t | � |
� WYd d } ~ Xn X| j d
� Wd | j � XqWd S) Nr� rm ra r r� r r] �foo
z IncompleteRead(%d bytes read)zIncompleteRead expected)r] r )r� rB r� r r� r� r� r ry rz r� rT r r� �
memoryviewr� r� r� r r� ) r rv Z nexpectedr� rQ r| r/ �mr� r r r r r �test_readinto_chunked� sD
!
)z BasicTest.test_readinto_chunkedc C s� d } t | t t � } t j | d d �} | j � | j | j � 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
rm rd r �� �OK)r r� r� r ry rz rT r- �status�reasonrt r� r� r� r )r r� rQ r| r r r �test_chunked_head� s
zBasicTest.test_chunked_headc C s� d } t | t t � } t j | d d �} | j � t d � } | j | � } | j | d � | j t | � 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
rm rd r� r s r r s )r r� r� r ry rz r� r� rT r� r r
rt r� r� r� r )r r� rQ r| r� r/ r r r �test_readinto_chunked_head� s
z$BasicTest.test_readinto_chunked_headc C sX t d � } t j | d d �} | j � | j | j � d � | j | j � � d S)Nz.HTTP/1.1 200 OK
Content-Length: -1
Hello
rm ra s Hello
)r r ry rz rT r- rt r� )r rQ r| r r r � test_negative_content_length� s
z&BasicTest.test_negative_content_lengthc C s� t d � } t j | d d �} | j � y | j � Wnz t j k
r� } zW | j | j d � | j t | � d � | j t | � d � | j
| j � � WYd d } ~ Xn X| j d � d S)Nz.HTTP/1.1 200 OK
Content-Length: 10
Hello
rm ra s Hello
z-IncompleteRead(7 bytes read, 3 more expected)zIncompleteRead expected)
r r ry rz r- r� rT r� r� r rt r� r� )r rQ r| r� r r r �test_incomplete_read� s
&z BasicTest.test_incomplete_readc s t d d � } t j d � � | � _ | j t � f d d � � � j � } | j d | j � | j d | 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 d d d � S)Nr^ z/urlrW )rS r )rV r r �<lambda> s z&BasicTest.test_epipe.<locals>.<lambda>i� zBasic realm="example"zwww-authenticate)
r( r rP rQ r� r* � getresponserT r r{ )r rQ r| r )rV r �
test_epipe s zBasicTest.test_epipec C sJ d d d d } t j t | � � } | j t j t j f | j � d S)NzHTTP/1.1 200 Ok�ki z
)r ry r r� � LineTooLongr� rz )r rW r| r r r � test_overflowing_status_line s z&BasicTest.test_overflowing_status_linec C sA d d d d } t j t | � � } | j t j | j � d S)NzHTTP/1.1 200 OK
X-Foo: barr i z
)r ry r r� r rz )r rW r| r r r � test_overflowing_header_line s z&BasicTest.test_overflowing_header_linec C sK d d d d } t j t | � � } | j � | j t j | j � d S)Nz/HTTP/1.1 200 OK
Transfer-Encoding: chunked
r i za
hello world
0
)r ry r rz r� r r- )r rW r| r r r � test_overflowing_chunked_line' s
z'BasicTest.test_overflowing_chunked_linec C s� d } t | � } t j | � } | j � | j | j � d � | j | j � � | j | j � | j
� | j | j � d S)NzHTTP/1.1 200 Okr ) r r ry rz rT r- rt r� r� r� r )r rW rQ r| r r r �test_early_eof4 s
zBasicTest.test_early_eofc s� t j d � } d � G� f d d � d t j � } | | _ t d � | _ | j d d � | j t j | j � | j
� j � | j
| j j � d S)Nz example.comc s" e Z d Z � f d d � Z d S)z+BasicTest.test_error_leak.<locals>.Responsec s | � t j j | | | � d S)N)r ry r )r r< r= )�responser r r E s z4BasicTest.test_error_leak.<locals>.Response.__init__N)r# r$ r% r r )r r r �ResponseD s r zInvalid status linera rO )
r rP ry Zresponse_classr rQ rS r� r� r rt r� r )r rV r r )r r �test_error_leak@ s zBasicTest.test_error_leakc C sk d } t d } t t | t t � } t j | d d �} | j � | j | j � | � | j
� d S)Nz 3;foo=bar
zabc
s abcrm ra z3;foo=bar
abc
) r� r r� �last_chunk_extendedr� r ry rz rT r- r )r �extrarv rQ r| r r r �test_chunked_extensionP s
z BasicTest.test_chunked_extensionc C sY t } t t t � } t j | d d �} | j � | j | j � | � | j � d S)Nrm ra )
r� r r� r� r ry rz rT r- r )r rv rQ r| r r r �test_chunked_missing_endZ s
z"BasicTest.test_chunked_missing_endc C sz t } t t t t t � } t j | d d �} | j � | j | j
� | � | j | j j
� d � | j � d S)Nrm ra r )
r� r r� r� �trailersr� r ry rz rT r- r r )r rv rQ r| r r r �test_chunked_trailersc s
z BasicTest.test_chunked_trailersc C s� t } d } t t t t t | � } t j | d d �} | j � | j | j
� | � | j | j j
� | j d � � | j
� d S)N� extradatarm ra r )r� r r� r� r r� r ry rz rT r- r r
r )r rv r! rQ r| r r r �test_chunked_syncn s
"zBasicTest.test_chunked_syncc C s| d } d } t d | | � } t j | d d �} | j � | j | j � | � | j | j j � | � | j � d S)Ns extradatas
Hello123
s' HTTP/1.1 200 OK
Content-Length: 10
rm ra )r r ry rz rT r- r r )r r! rv rQ r| r r r �test_content_length_syncz s
z"BasicTest.test_content_length_syncc C s� d } d } t d | | � } t j | d d �} | j � | j | j d � | g � | j | j j � | � | j � d S)Ns extradatas
Hello123
s' HTTP/1.1 200 OK
Content-Length: 10
rm ra i� ) r r ry rz rT � readlinesr r- r )r r! rv rQ r| r r r � test_readlines_content_length� s
z'BasicTest.test_readlines_content_lengthc C s d } d } t d | | � } t j | d d �} | j � | j | j d � | � | j | j j � | � | j � d S)Ns extradatas
Hello123
s' HTTP/1.1 200 OK
Content-Length: 10
rm ra i� ) r r ry rz rT �read1r r- r )r r! rv rQ r| r r r �test_read1_content_length� s
z#BasicTest.test_read1_content_lengthc C s� d } d } t d | | � } t j | d d �} | j � | j | j d � | � | j | j d � d � | j | j j � | � | j � d S)Ns extradatas
Hello123
s' HTTP/1.1 200 OK
Content-Length: 10
rm ra �
r ) r r ry rz rT r0 r r- r )r r! rv rQ r| r r r �"test_readline_bound_content_length� s
z,BasicTest.test_readline_bound_content_lengthc C s� d } d } t d | d | � } t j | d d �} | j � | j | j d � | d � | j | j � | � | j | j j � | � | j � d S) Ns extradatas
Hello123
s' HTTP/1.1 200 OK
Content-Length: 30
� rm ra � r� ) r r ry rz rT r& r- r r )r r! rv rQ r| r r r � test_read1_bound_content_length� s
z)BasicTest.test_read1_bound_content_lengthc sI t j d � } t j t j t j t j � � | j � j � � j t d f � � j
� d � � � f d d � } | j d | � } | j � t
j � j � � } | j d d � | j � } zQ | j | j t
j � t j d | j � � } z | j d � Wd | j � XWd | j � | j � | j � X| j � d � d S)
N� threadingr c sy � j � \ } } | �Z | j d � �C } x | j � } | j d � s. Pq. W| j d � | j � � Wd QRXWd QRXd S)Nr s
s' HTTP/1.1 200 Connection established
)�acceptr r0 �rstripr r- )rV �address�reader�line)�result�servr r �
run_server� s
z2BasicTest.test_response_fileno.<locals>.run_server�targetrb z
dummy:1234�filenos
proxied data
)r �
import_module�socket�AF_INET� SOCK_STREAM� IPPROTO_TCP�
addCleanupr �bind�HOST�listen�Thread�startr rP � getsocknamerS r rT r r r7 r �detachr� )r r- r5 �threadrV r r� r )r3 r4 r �test_response_fileno� s.
z BasicTest.test_response_filenoc C sH Gd d � d t j � } | d � } t d � | _ | j d d � d S)Nc @ s e Z d Z d d � Z d S)zKBasicTest.test_putrequest_override_validation.<locals>.UnsafeHTTPConnectionc S s d S)Nr )r Zurlr r r �_validate_path� s zZBasicTest.test_putrequest_override_validation.<locals>.UnsafeHTTPConnection._validate_pathN)r# r$ r% rG r r r r �UnsafeHTTPConnection� s rH z example.comr] ra z/ )r rP r rQ rp )r rH rV r r r �#test_putrequest_override_validation� s z-BasicTest.test_putrequest_override_validationc C sH Gd d � d t j � } | d � } t d � | _ | j d d � d S)Nc @ s e Z d Z d d � Z d S)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 rH � s rH z example.comr] ra u /☃)r rP r rQ rp )r rH rV 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 r r r
r r r r r r r r r r r" r# r% r' r) r, rF rI rK r r r r r� [ sV
+
)
r� c @ s� e Z d Z d Z e e j d � d � j d � Z d Z d d � Z d d � Z d
d � Z
d d
� Z d d � Z d d � Z
d d � Z d d � Z d d � Z d S)�ExtendedReadTestzOHTTP/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 sP t | j � } t j | d d �} | j � t j | j � | _ | | _ d S)Nrm ra ) r �linesr ry rz r& �BufferedReader�fpr| )r rQ r| r r r �setUp s
zExtendedReadTest.setUpc s! | j } | j j � d � f d d � } | | j _ g } x� | j d � } | r� | j t | � d � | j � } | j t | � t | � � | j | j | � � | j t | � � } | j | | � n | j � } | j
| � | j | � | s? Pq? W| j d j | � | j
� d S)Nr c s4 � | � } | d k r& | d | � S| d d � S)Nr r( r )r/ r� )�oldpeekr r �mypeek$ s z*ExtendedReadTest.test_peek.<locals>.mypeekr* r r r2 )r| rO �peekZ
assertGreaterrB ZassertGreaterEqualrt ru r- rT r� rG r� �lines_expected)r r| rR �allr� Zp2�nextr )rQ r � test_peek s(
zExtendedReadTest.test_peekc C s& | j } | j | j j | j � d S)N)r| �_verify_readliner0 rT )r r| r r r �
test_readline? s z ExtendedReadTest.test_readlinec C s� g } x^ | d � } | rO | d k rO t | � d k rO | j | j d � � | j | � | s Pq W| j d j | � | � d S)Nr� s foos
r )rB rt r� rG rT r� )r r0 rv rU r2 r r r rX C s
z!ExtendedReadTest._verify_readlinec sD � j � � � f d d � } t | � } � j | j � j � d S)Nc s) � j d � } � j t | � d � | S)N� )r& �assertLessEqualrB )�res)r| r r r r R s z&ExtendedReadTest.test_read1.<locals>.r)r| � ReadlinerrX r0 rT )r r Z readlinerr )r| r r �
test_read1P s zExtendedReadTest.test_read1c C sV | j } g } x$ | j � } | s% P| j | � q W| j d j | � | j � d S)Nr )r| r& rG rT r� rT )r r| rU r
r r r �test_read1_unboundedY s z%ExtendedReadTest.test_read1_unboundedc C so | j } g } x= | j d � } | s( P| j t | � d � | j | � q W| j d j | � | j � d S)Nr( r )r| r& r[ rB rG rT r� rT )r r| rU r
r r r �test_read1_boundedc s z#ExtendedReadTest.test_read1_boundedc C s | j | j j d � d � d S)Nr r )rT r| r& )r r r r � test_read1_0n s z ExtendedReadTest.test_read1_0c C s, | j j d � } | j d t | � � d S)Nr )r| rS r[ rB )r r� r r r � test_peek_0q s z ExtendedReadTest.test_peek_0)r# r$ r% rM �findr
rT Z
lines_chunkedrP rW rY rX r^ r_ r` ra rb r r r r rL � s "
rL c @ s e Z d Z d Z d S)�ExtendedReadTestChunkedz�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% rM r r r r rd v s rd c @ s( e Z d Z d d � Z d d � Z d S)r] c C s | | _ d | _ d S)Nr )�readfunc� remainder)r re r r r r � s zReadliner.__init__c
C s� g } d } | j } y� xj | j d � } | d k r7 P| t | � | k r[ | | d } | j | � | j � } | s d } Pq W| d 7} | j | d | � � | | d � | _ d j | � SWn d j | � | _ � Yn Xd S)Nr s
r r r2 )rf rc rB rG re r� )r �limitr
Zdatalenr- �idxr r r r0 � s,
zReadliner.readlineN)r# r$ r% r r0 r r r r r] � s r] c @ s4 e Z d Z d d � Z d d � Z d d � Z d S)� OfflineTestc C s� d h } d d h } xf t t � D]X } | j d � s" | | k rF q" t t | � } t | d d � d k r" | j | � q" W| j t j | � d S)N� responsesZ HTTPMessage�
parse_headers�_r$ z http.client)�dirr ru �getattr�addZassertCountEqual�__all__)r rv Z blacklistr� Z
module_objectr r r �test_all� s zOfflineTest.test_allc C s | j t j t j d � d S)Nz Not Found)rT r rj � NOT_FOUND)r r r r �test_responses� s zOfflineTest.test_responsesc 5 C s� d d d d d d d d d d
d d d
d d d d d d d d d d d d d d d d d d d d! d" d# d$ d% d& d' d( d) d* d+ d, d- d. d/ d0 d1 d2 d3 d4 d5 g5 } x> | D]6 } | j d6 | � � | j t t | � � Wd QRXq� Wd 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 FORBIDDENrr 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_REQUIREDZconstant)r� rt �hasattrr )r rv �constr r r �test_client_constants� sp
z!OfflineTest.test_client_constantsN)r# r$ r% rq rs rv r r r r ri � s ri c @ s\ e Z d Z d d � Z d d � Z d d � Z e j e e d � d � d d
� � Z
d S) �SourceAddressTestc C sY t j t j t j � | _ t j | j � | _ t j � | _ | j j � d | _
d S)N) r9 r: r; r4 r � bind_portr Zfind_unused_port� source_portr@ rV )r r r r rP � s
zSourceAddressTest.setUpc C s9 | j r | j j � d | _ | j j � d | _ d S)N)rV r r4 )r r r r �tearDown s
zSourceAddressTest.tearDownc C s[ t j t | j d d | j f �| _ | j j � | j | j j j � d | j � d S)N�source_addressr] r )
r rP r? r ry rV r; rT rQ rC )r r r r � testHTTPConnectionSourceAddress
s
z1SourceAddressTest.testHTTPConnectionSourceAddress�HTTPSConnectionz'http.client.HTTPSConnection not definedc C s+ t j t | j d d | j f �| _ d S)Nr{ r] )r r} r? r ry rV )r r r r � testHTTPSConnectionSourceAddress s z2SourceAddressTest.testHTTPSConnectionSourceAddressN) r# r$ r% rP rz r| �unittest�skipIfrt r r~ r r r r rw � s
rw c @ s: e Z d Z d Z d d � Z d d � Z d d � Z d S)� TimeoutTestNc C sA t j t j t j � | _ t j | j � t _ | j j � d S)N) r9 r: r; r4 r rx r� �PORTr@ )r r r r rP s zTimeoutTest.setUpc C s | j j � d | _ d S)N)r4 r )r r r r rz % s
zTimeoutTest.tearDownc C s@ | j t j � � t j d � z# t j t t j � } | j � Wd t j d � X| j
| j j � d � | j
� | j t j � � t j d � z) t j t t j d d �} | j � Wd t j d � X| j
| j j � d � | j
� t j t t j d d �} | j � | j
| j j � d � | j
� d S)N� �timeout)� assertIsNoner9 �getdefaulttimeout�setdefaulttimeoutr rP r? r� r� r; rT rQ �
gettimeoutr )r ZhttpConnr r r �testTimeoutAttribute) s*
z TimeoutTest.testTimeoutAttribute)r# r$ r% r� rP rz r� r r r r r� s r� c @ s4 e Z d Z d d � Z d d � Z d d � Z d S)�PersistenceTestc C s d } x | D]\ } } } | j d
| d | � �� d j | | � } t | � } | j | j � | j d
d � | j � �+ } | j | j d k | � | j � Wd QRX| j | j d k | � | j | j d � | j d
d � | j | j | rd n d � Wd QRXq
Wd S)N�1.0r] F�Connection: keep-alive
T�1.1�Connection: close
�Connection: keep-ALIVE
�Connection: cloSE
�versionrU z4HTTP/{} 200 OK
{}Content-Length: 12
Dummy body
ra 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� rk r3 r� rQ rS r rT r- r4 )r �testsr� rU Zreuser� rV r r r r �test_reuse_reconnectM s( z$PersistenceTest.test_reuse_reconnectc C s� d d � } t j t j f | t f f } x� | D]� \ } } | j d | � �j t d | � } | j d d � | j | | j � | j
| j � | j d d � | j | j
d � Wd QRXq1 Wd S) Nc s7 t j | � � � f d d � } | � _ t j � � S)Nc s. t j j � | � } | d k r* t � � | S)Nr )r& r' r� �ConnectionResetError)�buffer�size)�streamr r r� p s zNPersistenceTest.test_disconnected.<locals>.make_reset_reader.<locals>.readinto)r& r' r� rN )r r� r )r� r �make_reset_readerm s z<PersistenceTest.test_disconnected.<locals>.make_reset_reader� exceptionr ra z
/eof-responsez
/reconnectr� )r& r' r �RemoteDisconnectedr� r� r3 rS r� r r� rQ rT r4 )r r� r� Zstream_factoryr� rV r r r �test_disconnectedk s z!PersistenceTest.test_disconnectedc C su t d � } | j d d d d d i �| j t j | j � | j | j � | j d d � | j | j d � d S) Ns HTTP/1.1 100 Continue
ra rO rX ZExpectz 100-continuez
/reconnectr� )
r3 rS r� r r� r r� rQ rT r4 )r rV r r r �test_100_close� s z PersistenceTest.test_100_closeN)r# r$ r% r� r� r� r r r r r� K s r� c @ s� e Z d Z d d � Z d d � Z d d � Z d d � Z d d
� Z e j d d � � Z
d
d � Z d d � Z d d � Z
d d � Z d d � Z e j e e d � d � d d � � Z d S) � HTTPSTestc C s t t d � s | j d � d S)Nr} zssl support required)rt r ZskipTest)r r r r rP � s zHTTPSTest.setUpc C s d d l m } | | d | �S)Nr )�make_https_server�certfile)Ztest.ssl_serversr� )r r� r� r r r � make_server� s zHTTPSTest.make_serverc C s2 t j t t j d d �} | j | j d � d S)Nr� r� )r r} r? r� r� rT r� )r r� r r r �test_attributes� s zHTTPSTest.test_attributesc C s� d d l } t j d � t j d � �Y t j d d � } | j | j � � } | j d d � Wd QRX| j | j
j d � Wd QRXd S)Nr �networkzself-signed.pythontest.neti� ra rO �CERTIFICATE_VERIFY_FAILED) �sslr �requires�transient_internetr r} r� �SSLErrorrS rT r� r
)r r� r� �exc_infor r r �test_networked� s
zHTTPSTest.test_networkedc C s� d d l } t j d � t j d � �r | j � } t j d d d | �} | j d d � | j � } | j � | j
d | j d � � | j � Wd QRXd S)
Nr r� zself-signed.pythontest.neti� �contextra rO �nginx�server) r� r r� r� Z_create_unverified_contextr r} rS r r rr r{ )r r� r� r� r| r r r � test_networked_noverification� s
z'HTTPSTest.test_networked_noverificationc
C s� t j d � t j d � �f t j d d � } | j d d � | j � } | j d � } | j � | j � | j d | � Wd QRXd S)Nr� zwww.python.orgi� ra rO z content-typez text/html)
r r� r� r r} rS r r{ r rr )r r� r| Z content_typer r r �&test_networked_trusted_by_default_cert� s
z0HTTPSTest.test_networked_trusted_by_default_certc C s� d d l } t j d � t j d � �� | j | j � } | j | _ | j t � t
j d d d | �} | j d d � | j
� } | j d � } | j � | j � | j d | � Wd QRXd S)
Nr r� zself-signed.pythontest.neti� r� ra rO r� r� )r� r r� r� �
SSLContext�PROTOCOL_TLSv1�
CERT_REQUIRED� verify_mode�load_verify_locations� CERT_selfsigned_pythontestdotnetr r} rS r r{ r rr )r r� r� r� r| Z
server_stringr r r �test_networked_good_cert� s
z"HTTPSTest.test_networked_good_certc C s� d d l } t j d � t j d � �� | j | j � } | j | _ | j t � t
j d d d | �} | j | j
� � } | j d d � Wd QRX| j | j j d � Wd QRXd S) Nr r� zself-signed.pythontest.neti� r� ra rO r� )r� r r� r� r� r� r� r� r� �CERT_localhostr r} r� r� rS rT r� r
)r r� r� r� r� r r r �test_networked_bad_cert� s
z!HTTPSTest.test_networked_bad_certc
C sv d d l } | j t � } t j d | j � } | j | j � � } | j d d � Wd QRX| j | j
j d � d S)Nr � localhostra rO r� ) r� r� r� r r} r r� r� rS rT r� r
)r r� r� r� r� r r r �test_local_unknown_cert� s z!HTTPSTest.test_local_unknown_certc C s� d d l } | j t � } | j | j � } | j | _ | j t � t j d | j
d | �} | j | j � | j
d d � | j � } | j | j � | j | j d � d S)Nr r� r� ra z /nonexistenti� )r� r� r� r� r� r� r� r� r r} r r= r rS r rT r )r r� r� r� r� r| r r r �test_local_good_hostname� s
z"HTTPSTest.test_local_good_hostnamec C s d d l } | j t � } | j | j � } | j | _ d | _ | j t � t j
d | j d | �} | j | j
� � | j d d � Wd QRXt j
d | j d | d d �} | j | j
� � | j d d � Wd QRXd | _ t j
d | j d | d d �} | j d d � | j � } | j � | j � | j | j d
� d | _ t j
d | j d | �} | j d d � | j � } | j | j d
� | j � | j � t j
d | j d | d d �} | j | j
� � | j d d � Wd QRXd S) Nr Tr� r� ra rO �check_hostnameFz /nonexistenti� )r� r� �CERT_fakehostnamer� r� r� r� r� r� r r} r r� ZCertificateErrorrS r r rT r )r r� r� r� r� r| r r r �test_local_bad_hostname� s@
z!HTTPSTest.test_local_bad_hostnamer} z)http.client.HTTPSConnection not availablec C s} x' d D] } | j t j t j | � q WxL d D]D \ } } } t j | � } | j | | j � | j | | j � q1 Wd 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} rT r r )r r� r� r� r� r r r r� $ s
zHTTPSTest.test_host_portN)r# r$ r% rP r� r� r� r� r Zsystem_must_validate_certr� r� r� r� r� r� r r� rt r r� r r r r r� � s
)r� c @ sp e Z d Z d d � Z d d � Z d d � Z d d � Z d d
� Z d d � Z d
d � Z d d � Z
d S)�RequestBodyTestc C s> t j d � | _ t d � | j _ | _ | j | j _ d S)Nz example.comr] )r rP rV r rQ )r r r r rP < s zRequestBodyTest.setUpc C s8 t j | j j � } | j � t j | � } | | f S)N)r& r' rQ r
r0 r rk )r �f�messager r r �get_headers_and_fpA s
z"RequestBodyTest.get_headers_and_fpc C sj | j j d d d d d i � | j � \ } } | j d | j d � � | j d t | j � � � d S)Nr^ z/urlrW zContent-LengthZ42zcontent-lengthrZ )rV rS r� rT �getrB r- )r r� r� r r r �test_manual_content_lengthG s
z*RequestBodyTest.test_manual_content_lengthc C s� | j j d d d � | j � \ } } | j d | j � � | j | j � � | j d | j d � � | j d | j � � d S)Nr^ z/urlrW z
text/plain�4zcontent-lengths body) rV rS r� rT �get_content_typer� � get_charsetr� r- )r r� r� r r r �test_ascii_bodyP s z RequestBodyTest.test_ascii_bodyc C s� | j j d d d � | j � \ } } | j d | j � � | j | j � � | j d | j d � � | j d | j � � d S)Nr^ z/urlu bodyÁz
text/plain�5zcontent-lengths body�) rV rS r� rT r� r� r� r� r- )r r� r� r r r �test_latin1_bodyX s z RequestBodyTest.test_latin1_bodyc C s� | j j d d d � | j � \ } } | j d | j � � | j | j � � | j d | j d � � | j d | j � � d S)Nr^ z/urls body�z
text/plainr� zcontent-length) rV rS r� rT r� r� r� r� r- )r r� r� r r r �test_bytes_body` s z RequestBodyTest.test_bytes_bodyc C s� | j t j t j � t t j d � � } | j d � Wd QRXt t j � �� } | j j d d | � | j � \ } } | j d | j
� � | j | j � � | j d | j
d � � | j d | j � � Wd QRXd S) N�wrW r^ z/urlz
text/plainr� zcontent-lengths body)r= r �unlink�TESTFNr� �writerV rS r� rT r� r� r� r� r- )r r� r� r r r �test_file_bodyh s z RequestBodyTest.test_file_bodyc C s� | j t j t j � t t j d � � } | j d � Wd QRXt t j d � �� } | j j d d | � | j � \ } } | j d | j
� � | j | j � � | j d | j
d � � | j d | j � � Wd QRXd S) N�wbs body�r r^ z/urlz
text/plainr� zcontent-length)r= r r� r� r� r� rV rS r� rT r� r� r� r� r- )r r� r� r r r �test_binary_file_bodyt s z%RequestBodyTest.test_binary_file_bodyN) r# r$ r% rP r� r� r� r� r� r� r� r r r r r� 9 s r� c @ sX e Z d Z d d � Z d d � Z d d � Z d d � Z d d
� Z d d � Z d
S)�HTTPResponseTestc C s5 d } t | � } t j | � | _ | j j � d S)NzXHTTP/1.1 200 Ok
My-Header: first-value
My-Header: second-value
Text)r r ry r| rz )r rW rQ r r r rP � s zHTTPResponseTest.setUpc C sK | j j d � } | j | d � | j j d d � } | j | d � d S)Nz My-Headerzfirst-value, second-valuez some default)r| r{ rT )r rU r r r �test_getting_header� s z$HTTPResponseTest.test_getting_headerc C s) | j j d d � } | j | d � d S)NzNo-Such-Headerz
default-value)r| r{ rT )r rU r r r �3test_getting_nonexistent_header_with_string_default� s zDHTTPResponseTest.test_getting_nonexistent_header_with_string_defaultc C sT | j j d d d g � } | j | d � | j j d d � } | j | d � d S)NzNo-Such-Header�default�valueszdefault, values)r� r� )r| r{ rT )r rU r r r �5test_getting_nonexistent_header_with_iterable_default� s zFHTTPResponseTest.test_getting_nonexistent_header_with_iterable_defaultc C s& | j j d � } | j | d � d S)NzNo-Such-Header)r| r{ rT )r rU r r r �/test_getting_nonexistent_header_without_default� s z@HTTPResponseTest.test_getting_nonexistent_header_without_defaultc C s, | j j d d d �} | j | d � d S)NzNo-Such-Headerr� ro )r| r{ rT )r rU r r r � test_getting_header_defaultint� s z/HTTPResponseTest.test_getting_header_defaultintN) r# r$ r% rP r� r� r� r� r� r r r r r� � s r� c @ sp e Z d Z d d � Z d d � Z d d � Z d d � Z d d
� Z d d � Z d
d � Z d d � Z
d S)� TunnelTestsc C s= d } d | _ t j | j � | _ | j | � | j _ d S)Nz:HTTP/1.0 200 OK
HTTP/1.1 200 OK
Content-Length: 42
z proxy.com)r r rP rV r8 )r �
response_textr r r rP � s zTunnelTests.setUpc C s | j j � d S)N)rV r )r r r r rz � s zTunnelTests.tearDownc s d d � f d d � } | S)Nc s t � d | d d | d �S)Nr r r r )r )r0 r� r{ )r� r r r7 � s z9TunnelTests._create_connection.<locals>.create_connectionr )r r� r7 r )r� r r8 � s z TunnelTests._create_connectionc C s� d } d } d d i } | j j | d | d | �| j j d d d � | j | j j j | j � | j | j j j t j � | j | j j | � | j | j j
| � | j | j j | � d S)
Nzdestination.comi�" z
User-Agentz!Mozilla/5.0 (compatible, MSIE 11)r rX rd rO r] ) rV �
set_tunnelrS rT rQ 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 j � | j t | j j d � d S)Nzdestination.com)rV r; r� � RuntimeErrorr� )r r r r �&test_disallow_set_tunnel_after_connect� s
z2TunnelTests.test_disallow_set_tunnel_after_connectc C s� | j j d � | j j d d d � | j | j j j | j � | j | j j j t j � | j d | j j j
� | j d | j j j
� | j d | j j j
� | j d | j j j
� d S) Nzdestination.comrd rO r] s CONNECT destination.coms Host: destination.com:Nones Host: destination.coms Host: proxy.com) rV r� rS rT rQ r r r r� rr r
Z assertNotIn)r r r r �test_connect_with_tunnel� s z$TunnelTests.test_connect_with_tunnelc C s� | j j d � | j j d d d � | j | j j j | j � | j | j j j t j � | j d | j j j
� | j d | j j j
� d S)Nzdestination.comr^ rO r] s CONNECT destination.coms Host: destination.com) rV r� rS rT rQ r r r r� rr r
)r r r r �test_connect_put_request� s z$TunnelTests.test_connect_put_requestc C s� d } d j | � } | j j d � | j | � | j _ | j j d � t j � � } | j j d d d � Wd QRX| j � j � } | j
d j | � | � d S) Nz
X-Dummy: 1zHTTP/1.0 200 OK
{}
r zdestination.comr^ rO r] z
header: {}) rk rV Zset_debuglevelr8 r� r Zcaptured_stdoutrS �getvalue�
splitlinesrr )r Zexpected_headerr� �outputrM r r r �test_tunnel_debuglog� s z TunnelTests.test_tunnel_debuglogN) r# r$ r% rP rz r8 r� r� r� r� r� r r r r r� � s
r� c
C s2 t j t t t t t t t t t
t t t
� d S)N)r Z run_unittestr> ri r� r� r� r� r� rw r� rL rd r� )�verboser r r � test_main� s
r� �__main__)2r+ Zhttpr r& ri �osr� r9 r ZTestCaser� r �path�dirnamer� �herer� r� r� r� r� r� Zchunk_extensionr� r r r� r? r r( r' r, rP r3 r> r� rL rd r] ri rw r� r� r� r� r� r� Z reap_threadsr� r# r r r r �<module> sX !�� � �}"P .H�H#I