Your IP : 216.73.216.213
�
f f.� � @ 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 e j Z Gd d � d � Z Gd d
� d
e � 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 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 d d � � Z# e$ d! k r
e# � n d S)"� N)�client)�supportz keycert.pemz keycert2.pemz selfsigned_pythontestdotnet.pemc @ s[ 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 S) �
FakeSocketNc C sd t | t � r! | j d � } n | | _ | | _ 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/python34/lib64/python3.4/test/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 �sendall$ s zFakeSocket.sendallc C sR | d k r' | d k r' t j � � n | 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 0 s zFakeSocket.file_closec C s d S)Nr )r r r r r 3 s zFakeSocket.close)
�__name__�
__module__� __qualname__�io�BytesIOr 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 8 s zEPipeSocket.__init__c C s7 | j | k r$ t t j d � � n | j | 7_ d S)NZgotcha)r% �OSError�errnoZEPIPEr )r r r r r r = s zEPipeSocket.sendallc C s d S)Nr )r r r r r B s zEPipeSocket.closeN)r r r! r r r r r r r r$ 6 s r$ c @ s4 e Z d Z d Z d d d � Z d d d � Z d S) � 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 s4 t j j | | � } | d k r0 t d � � n | S)Nr z caller tried to read past EOF)r" r# �read�AssertionError)r �nr r r r r) K s zNoEOFBytesIO.readNc C s4 t j j | | � } | d k r0 t d � � n | S)Nr z caller tried to read past EOF)r" r# �readliner* )r Zlengthr r r r r, Q s zNoEOFBytesIO.readline���)r r r! �__doc__r) r, r r r r r( E 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)� HeaderTestsc C s� Gd d � d t � } x� d D]� } x� d D]� } t j d � } t d � | _ | � | _ d
} i } | r� t t | � � | | <n | 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 su | j d � } t | � d k ra | d j d � j � } | j j | d � | j | d 7<n t j | | � d S)N� :r r r )�split�len�decode�lowerr0 �
setdefault�list�append)r �item�kvZlcKeyr r r r8 _ s zBHeaderTests.test_auto_headers.<locals>.HeaderCountingBuffer.appendN)r r r! r r8 r r r r �HeaderCountingBuffer\ s r; TF�Content-length�Host�Accept-encodingz example.comZ blahblahblahZ spamspamspam�POST�/r )TF)r< r= r> ) r7 r �HTTPConnectionr �sock�_bufferr r3 �request� assertEqualr0 r5 )r r; Zexplicit_header�header�conn�body�headersr r r �test_auto_headersX 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)r7 r �content_length)r r r r r x s
zHHeaderTests.test_content_length_0.<locals>.ContentLengthChecker.__init__c S sd | j d d � } t | � d k rP | d j � d k rP | d j � | _ n t j | | � d S)Nr1 r r s content-length)r2 r3 r5 �striprK r7 r8 )r r9 r: r r r r8 { s (zFHeaderTests.test_content_length_0.<locals>.ContentLengthChecker.appendN)r r r! r r8 r r r r �ContentLengthCheckerw s rM � �PUTr? �PATCHz example.comr@ � 0z%Header Content-Length incorrect on {}�GET�CONNECT�DELETE�HEAD�OPTIONS�TRACEz.Header Content-Length set for empty body on {}� � 1)NrN )rO zPOSTrP )rR rS rT rU rV rW )
r7 � itertools�productr rA r rB rC rD rE rK �format�chain)r rM ZbodiesZmethods_with_body�methodrH rG Zmethods_without_bodyr r r �test_content_length_0u 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.comrR r@ 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 Space�values Key Space: valuez KeySpace s KeySpace : values Nonbreak�Spaces Nonbreak�Space: values �NonbreakSpaces �NonbreakSpace: value)r rA r rB �
putrequest� putheader�assertInrC )r rG 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::]:81rN rR z/foosE GET /foo HTTP/1.1
Host: [2001:102A::]
Accept-Encoding: identity
z
[2001:102A::])r rA r rB rD �
assertTruer �
startswith)r �expectedrG rB 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�beginrE � getheader)r rH rB �respr r r �!test_malformed_headers_coped_with� s
z-HeaderTests.test_malformed_headers_coped_withc C s� t j d � } t d � | _ | j d d � d( } xY | D]Q \ } } | j | | f � �. | j t d � � | j | | � Wd QXWd QXq; Wd S))Nz example.comrN rR r@ �
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�rp rq �rr rq �rs rq �rt rq �ru rq �rv rq �rw rq �rx rq �ry rq �rz rq �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 rA r rB rb ZsubTest�assertRaisesRegex�
ValueErrorrc )r rG Zcases�namera r r r �test_invalid_headers� s. z HeaderTests.test_invalid_headersN) r r r! rJ r_ re ri ro r� r r r r r/ W 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? S)@� 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 rk rl rE r) � assertFalse�isclosed�closedrf r � assertRaises�
BadStatusLine)r rH rB rn r r r �test_status_lines s
zBasicTest.test_status_linesc C s) t j d � } | j t | � d � d S)NrN zBadStatusLine("''",))r r� rE �repr)r �excr r r �test_bad_status_repr+ 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 rk rl rE r) r� r� rf r� r )r rH rB rn r r r �test_partial_reads/ s
z BasicTest.test_partial_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 rk rl � bytearray�readintorE �bytesr� r� rf r� r )r rH rB rn �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 rk rl rE r) r� r� rf r� r )r rH rB rn r r r �$test_partial_reads_no_content_lengthR 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 rk rl r� r� rE r� r� r� rf )r rH rB rn r� r+ r r r �(test_partial_readintos_no_content_lengthb 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 rk rl rE r) r� r� rf )r rH rB rn r r r �"test_partial_reads_incomplete_bodyu 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 rk rl r� r� rE r� r� r� rf r� r )r rH rB rn 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 �
InvalidURLrA rE 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 rk rl rm rE )r r
Zhdr�sr Zcookiesr r r �test_response_headers� s
z BasicTest.test_response_headersc C sN t d t � } t j | d d �} | j � | j � rJ | j d � n d S)Nz*HTTP/1.1 200 OK
Content-Length: 14432
r^ rU z)Did not expect response from HEAD request)r r( r rk rl r) �fail)r rB rn r r r �test_read_head� s
zBasicTest.test_read_headc C sy t d t � } t j | d d �} | j � t d � } | j | � d k r_ | j d � n | j t | � d � d S) Nz*HTTP/1.1 200 OK
Content-Length: 14432
r^ rU � r z)Did not expect response from HEAD requests s )
r r( r rk rl r� r� r� rE r� )r rB rn 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) �join�ranger Z _MAXHEADERSr rk r� Z
HTTPExceptionrl )r rI 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 QXd S)NsP GET /foo HTTP/1.1
Host: example.com
Accept-Encoding: identity
Content-Length:r z example.comrR z/fooz%r != %r) �open�__file__r rA r rB rD rf r rg r3 )r rh rH rG rB r r r �test_send_file� 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 rA r rB �sendrE r �arrayr" r# )r rh rG rB r r r � test_send� 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 � 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 Z blocksizer r r r) s z<BasicTest.test_send_updating_file.<locals>.UpdatingFile.readNr- )r r r! r r� r) r )r r r � UpdatingFile s r� s dataz example.comrN )r rA r rB r� rE r )r r� rh rG rB r )r r �test_send_updating_file� s z!BasicTest.test_send_updating_filec C sm d } d d � } t j d � } t d � } | | _ | j d d | � i d d 6� | 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 rH s z&BasicTest.test_send_iter.<locals>.bodyz example.comrN rR z/fooZ11zContent-Length)r rA r rB rD rE r )r rh rH rG rB r r r �test_send_iter s zBasicTest.test_send_iterc C sK t j d � } t d � | _ | j t � � | j d d | � Wd QXd S)Nz example.comrN r? �test)r rA r rB r� � TypeErrorrD )r rG r r r �test_send_type_error s z BasicTest.test_send_type_errorc C s� d } d } t | d � } t j | d d �} | j � | j | j � | � | j � x� t d d � D]o } t | d � } t j | d d �} | j � | j | j | � | j | � | j � | � | j � qk Wx� d D]� } 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)
Nz{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 differentz0
r^ rR r � rN �foo
z IncompleteRead(%d bytes read)zIncompleteRead expected)rN r� )r r rk rl rE r) r r� �IncompleteRead�partialr3 r� r r� ) r �
chunked_startrh rB rn r+ �xr� �expected_messager r r � test_chunked& s4
0
)zBasicTest.test_chunkedc C sf d } d } t | � } t d � } t | d � } t j | d d �} | j � | j | � } | j | d | � | � | j | | � | j � x� t d d � D]� } t | d � } t j | d d �} | j � t
| � } | j | d | � � } | | j | | | | � � 7} | | j | | d � � 7} | j | d | � | � | j | | � | j � q� Wx� d D]� }
t | |
� } t j | d d �} | j � z� y | j | � } Wnw t j k
rA} zT | j | j | � d t | � } | j t
| � | � | j t | � | � WYd d } ~ Xn X| j d
� Wd | j � Xq}Wd S)Nz{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 different� z0
r^ rR r r� r rN �foo
z IncompleteRead(%d bytes read)zIncompleteRead expected)rN r� )r3 r� r r rk rl r� rE r r� �
memoryviewr� r� r� r r� ) r r� rh Z nexpectedr� rB rn r+ �mr� r� r� r r r �test_readinto_chunkedR sF
!
)z BasicTest.test_readinto_chunkedc C s� d } t | d � } 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
z0
r^ rU r �� �OK)
r r rk rl rE r) �status�reasonrf r� r� r� r )r r� rB rn r r r �test_chunked_head� s
zBasicTest.test_chunked_headc C s� d } t | d � } 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
z0
r^ rU r� r s r� r� s )r r rk rl r� r� rE r� r� r� rf r� r� r� r )r r� rB rn 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
r^ rR s Hello
)r r rk rl rE r) rf r� )r rB rn 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
r^ rR s Hello
z-IncompleteRead(7 bytes read, 3 more expected)zIncompleteRead expected)
r r rk rl r) r� rE r� r� r rf r� r� )r rB rn 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)NrO z/urlrH )rD r )rG r r �<lambda>� s z&BasicTest.test_epipe.<locals>.<lambda>i� zBasic realm="example"zwww-authenticate)
r$ r rA rB r� r& � getresponserE r� rm )r rB rn r )rG 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 rk r r� � LineTooLongr� rl )r rH rn 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 rk r r� r� rl )r rH rn 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
�0i za
hello world
0
)r rk r rl r� r� r) )r rH rn 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 rk rl rE r) rf r� r� r� r )r rH rB rn r r r �test_early_eof� s
zBasicTest.test_early_eofc C s� t j d � } t d � } | | _ d | j d } | j d d | � | j | j d � t j d � } t d � } | | _ d | j } | j d d | � | j | j d � d S)Nz example.com� xr r? r@ ) r rA r rB ZmssrD rE r
Z
assertGreater)r rG rB rH r r r �test_delayed_ack_opt s
z BasicTest.test_delayed_ack_optc 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 rk r )r �pos�kw)�responser r r s z4BasicTest.test_error_leak.<locals>.Response.__init__N)r r r! r r )r� r r �Response s r rN rR r@ )
r rA rk Zresponse_classr rB rD r� r� r� rf r� r )r rG r r )r� r �test_error_leak s zBasicTest.test_error_leakN)"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� s>
, 6 r� c @ s( e Z d Z d d � Z d d � Z d S)� OfflineTestc C s� d h } d d h } x] t t � D]O } | | k r: q" n t t | � } t | d d � d k r" | j | � q" q" W| j t j | � d S)N� responsesZ HTTPMessage�
parse_headersr z http.client)�dirr �getattr�addZassertCountEqual�__all__)r rh 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)rE r r Z NOT_FOUND)r r r r �test_responses7 s zOfflineTest.test_responsesN)r r r! r r
r r r r r ( s r 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 s\ t j t j t j � | _ t j | j � | _ t j � | _ | j j d � d | _
d S)Nr� ) �socket�AF_INET� SOCK_STREAM�servr � bind_portr Zfind_unused_port� source_port�listenrG )r r r r �setUp< s
zSourceAddressTest.setUpc C s< | j r" | j j � d | _ n | j j � d | _ d S)N)rG r r )r r r r �tearDownC 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_addressrN r )
r rA �HOSTr r rG �connectrE rB Z getsockname)r r r r � testHTTPConnectionSourceAddressJ s
z1SourceAddressTest.testHTTPConnectionSourceAddress�HTTPSConnectionz'http.client.HTTPSConnection not definedc C s+ t j t | j d d | j f �| _ d S)Nr rN )r r r r r rG )r r r r � testHTTPSConnectionSourceAddressP s z2SourceAddressTest.testHTTPSConnectionSourceAddressN) r r r! r r r �unittest�skipIf�hasattrr r r r r r r ; s
r c @ s: e Z d Z d Z d d � Z d d � Z d d � Z d S)� TimeoutTestNc C sD t j t j t j � | _ t j | j � t _ | j j d � d S)Nr� ) r r
r r r r r �PORTr )r r r r r ] s zTimeoutTest.setUpc C s | j j � d | _ d S)N)r r )r r r r r b 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)� assertIsNoner ZgetdefaulttimeoutZsetdefaulttimeoutr rA r r r r rE rB Z
gettimeoutr )r ZhttpConnr r r �testTimeoutAttributef s*
z TimeoutTest.testTimeoutAttribute)r r r! r r r r# r r r r r Z 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 � n d S)Nr zssl support required)r r ZskipTest)r r r r r � 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 rE r! )r r� r r r �test_attributes� s zHTTPSTest.test_attributesc C s� d d l } t j d � t j d � �X t j d d � } | j | j � � } | j d d � Wd QX| j | j
j d � Wd QXd S)Nr �networkzself-signed.pythontest.neti� rR r@ �CERTIFICATE_VERIFY_FAILED) �sslr �requires�transient_internetr r r� �SSLErrorrD rE � exceptionr� )r r+ r� �exc_infor r r �test_networked� s
zHTTPSTest.test_networkedc C s� d d l } t j d � t j d � �h | j � } t j d d d | �} | j d d � | j � } | j � | j
d | j d � � Wd QXd S)
Nr r) zself-signed.pythontest.neti� �contextrR r@ �nginx�server) r+ r r, r- Z_create_unverified_contextr r rD r� r rd rm )r r+ r2 r� rn r r r � test_networked_noverification� s
z'HTTPSTest.test_networked_noverificationc
C s~ t j d � t j d � �\ t j d d � } | j d d � | j � } | j d � } | j � | j d | � Wd QXd S)Nr) zwww.python.orgi� rR r@ z content-typez text/html)
r r, r- r r rD r� rm r rd )r r� rn 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 d | � Wd QXd S)
Nr r) zself-signed.pythontest.neti� r2 rR r@ r4 r3 )r+ r r, r- �
SSLContext�PROTOCOL_TLSv1�
CERT_REQUIRED� verify_mode�load_verify_locations� CERT_selfsigned_pythontestdotnetr r rD r� rm r rd )r r+ r2 r� rn 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 QX| j | j j d � Wd QXd S) Nr r) zself-signed.pythontest.neti� r2 rR r@ r* )r+ r r, r- r7 r8 r9 r: r; �CERT_localhostr r r� r. rD rE r/ r� )r r+ r2 r� r0 r r r �test_networked_bad_cert� s
z!HTTPSTest.test_networked_bad_certc
C su d d l } | j t � } t j d | j � } | j | j � � } | j d d � Wd QX| j | j
j d � d S)Nr � localhostrR r@ r* ) r+ r' r> r r r r� r. rD rE r/ r� )r r+ r4 r� r0 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 d d � | j � } | j
| j d � d S)Nr r@ r2 rR z /nonexistenti� )r+ r' r> r7 r8 r9 r: r; r r r rD r� rE r� )r r+ r4 r2 r� rn 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 QXt j
d | j d | d d �} | j | j
� � | j d d � Wd QXd | _ t j
d | j d | d d �} | j d d � | j � } | j | j d
� d | _ t j
d | j d | �} | j d d � | j | j � j d
� t j
d | j d | d d �} | j | j
� � | j d d � Wd QXd S) Nr Tr@ r2 rR r@ �check_hostnameFz /nonexistenti� )r+ r' �CERT_fakehostnamer7 r8 r9 r: rC r; r r r r� ZCertificateErrorrD r� rE r� )r r+ r4 r2 r� rn r r r �test_local_bad_hostname� s6
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]:)rF rG �rH rI rJ �rK rL rM �rN rL rM �rL rL rM �rO rI rM �rP rI rM )rQ rR rS rT rU rV )r� r r� r rE r r )r r� r� r� r� r r r r� s
zHTTPSTest.test_host_portN)r r r! r r' r( r1 r5 r Zsystem_must_validate_certr6 r= r? rA rB rE r r r r r� r r r r r$ � s
$r$ c @ sv e Z 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 S)�RequestBodyTestz3Test cases where a request includes a message body.c C s> t j d � | _ t d � | j _ | _ | j | j _ d S)Nz example.comrN )r rA rG r rB )r r r r r ' s zRequestBodyTest.setUpc C s8 t j | j j � } | j � t j | � } | | f S)N)r" r# rB r r, r r )r �f�messager r r �get_headers_and_fp, s
z"RequestBodyTest.get_headers_and_fpc C sk | j j d d d i d d 6� | j � \ } } | j d | j d � � | j d t | j � � � d S)NrO z/urlrH Z42zContent-Lengthzcontent-length� )rG rD rZ rE �getr3 r) )r rY rX r r r �test_manual_content_length2 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)NrO z/urlrH z
text/plain�4zcontent-lengths body) rG rD rZ rE �get_content_typer" � get_charsetr\ r) )r rY rX r r r �test_ascii_body; 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)NrO z/urlu bodyÁz
text/plain�5zcontent-lengths body�) rG rD rZ rE r_ r" r` r\ r) )r rY rX r r r �test_latin1_bodyC 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)NrO z/urls body�z
text/plainrb zcontent-length) rG rD rZ rE r_ r" r` r\ r) )r rY rX r r r �test_bytes_bodyK s z RequestBodyTest.test_bytes_bodyc C s� | j t j t j � t t j d � � } | j d � Wd QXt t j � �� } | j j d d | � | j � \ } } | j d | j
� � | j | j � � | j d | j
d � � | j d | j � � Wd QXd S) N�wrH rO z/urlz
text/plainr^ zcontent-lengths body)�
addCleanupr �unlink�TESTFNr� �writerG rD rZ rE r_ r" r` r\ r) )r rX rY r r r �test_file_bodyS s z RequestBodyTest.test_file_bodyc C s� | j t j t j � t t j d � � } | j d � Wd QXt t j d � �� } | j j d d | � | j � \ } } | j d | j
� � | j | j � � | j d | j
d � � | j d | j � � Wd QXd S) N�wbs body�r rO z/urlz
text/plainrb zcontent-length)rf r rg rh r� ri rG rD rZ rE r_ r" r` r\ r) )r rX rY r r r �test_binary_file_body_ s z%RequestBodyTest.test_binary_file_bodyN) r r r! r. r rZ r] ra rc rd rj rl r r r r rW $ s rW 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 rk rn rl )r rH rB r r r r n 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)rn rm rE )r rF r r r �test_getting_headeru s z$HTTPResponseTest.test_getting_headerc C s) | j j d d � } | j | d � d S)NzNo-Such-Headerz
default-value)rn rm rE )r rF 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)zdefaultzvalues)rn rm rE )r rF 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)rn rm rE )r rF 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-Headerrp r` )rn rm rE )r rF r r r � test_getting_header_defaultint� s z/HTTPResponseTest.test_getting_header_defaultintN) r r r! r rn ro rr rs rt r r r r rm l s rm 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)� TunnelTestsc sL d � d d � f d d � } d | _ t j | j � | _ | | j _ d S)Nz:HTTP/1.0 200 OK
HTTP/1.1 200 OK
Content-Length: 42
c s t � d | d d | d �S)Nr r r r )r )Zaddressr! r )�
response_textr r �create_connection� s z,TunnelTests.setUp.<locals>.create_connectionz proxy.com)r r rA rG Z_create_connection)r rw r )rv r r � s
zTunnelTests.setUpc C s | j j � d S)N)rG r )r r r r r � s zTunnelTests.tearDownc C s� d } d } i d d 6} | 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!Mozilla/5.0 (compatible, MSIE 11)z
User-Agentr rI rU r@ rN ) rG �
set_tunnelrD rE rB 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)rG r r� � RuntimeErrorrx )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.comrU r@ rN s CONNECT destination.coms Host: destination.com:Nones Host: destination.coms Host: proxy.com) rG rx rD rE rB r r r ry rd 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.comrO r@ rN s CONNECT destination.coms Host: destination.com) rG rx rD rE rB r r r ry rd r )r r r r �test_connect_put_request� s z$TunnelTests.test_connect_put_requestN) r r r! r r rz r| r} r~ r r r r ru � s
ru c
C s) t j t t t t t t t t t
� d S)N) r Z run_unittestr/ r r� r r$ rW r rm ru )�verboser r r � test_main� s r� �__main__)%r' Zhttpr r" rZ �osr� r r ZTestCaser� r �path�dirnamer� �herer� r> rD r< r r r$ r# r( r/ r� r r r r$ rW rm ru Z reap_threadsr� r r r r r �<module> s@ �� � .�H#<