Your IP : 216.73.216.213
3
��2a�1 � @ s� d dl Z d dlmZ d dlmZ d dlZd dlZd dlZd dl Zd dl
Z
ej d� dZ dd� Z
dd � Ze jd
ejkd �Zeejjejj�ZG d d
� d
e j�ZG dd� de j�ZG dd� de j�ZG dd� de j�Ze dkr�e j � dS )� N)�support)�sanepathname2urlZnetwork�< c O sN xDt d�D ]8}y
| ||�S |k
r@ } z |}w
W Y d d }~X q
X q
W |�d S )N� )�range)�func�exc�args�kwargs�i�e�last_exc� r �4/opt/alt/python36/lib64/python3.6/test_urllib2net.py�
_retry_thrice s
r c s � �fdd�}|S )Nc s t �� f| �|�S )N)r )r r
)r r r r �wrapped s z(_wrap_with_retry_thrice.<locals>.wrappedr )r r r r )r r r �_wrap_with_retry_thrice s r ZTRAVISz%bpo-35411: skip FTP test on Travis CIc @ s e Zd ZdS )� AuthTestsN)�__name__�
__module__� __qualname__r r r r r + s r c @ s e Zd Zdd� ZdS )�CloseSocketTestc C sL d}t j|��4 t|�}|j}| j|j� |j� | j|j� W d Q R X d S )Nzhttp://www.example.com/)r �transient_internet�_urlopen_with_retry�fp� assertFalse�closed�close�
assertTrue)�self�urlZresponseZsockr r r �
test_closeT s zCloseSocketTest.test_closeN)r r r r! r r r r r R s r c @ sd e Zd Zdd� Zedd� �Zdd� Zdd� Zd d
� Zd d � Z e
j d
�dd� �Z ddd�Z
dd� ZdS )�OtherNetworkTestsc C s d S )Nr )r r r r �setUp` s zOtherNetworkTests.setUpc C s& ddd t jjfg}| j|| j� � d S )Nz ftp://www.pythontest.net/READMEz*ftp://www.pythontest.net/non-existent-file)�urllib�error�URLError�
_test_urls�_extra_handlers)r �urlsr r r �test_ftpi s zOtherNetworkTests.test_ftpc C s� t j}t|d�}zL|jd� |j� dttjj|�� dd t j
j fg}| j || j
� dd� W d tj|� X | jtt jjd� d S )N�wz hi there
zfile:z file:///nonsensename/etc/passwdT)�retryz./relative_path/to/file)r �TESTFN�open�writer r �os�path�abspathr$ r% r& r' r( �removeZ assertRaises�
ValueError�request�urlopen)r r- �fr) r r r � test_filer s
zOtherNetworkTests.test_filec
C sF d}t j|��. tjj|�}tjj|�}| j|j� d� W d Q R X d S )Nz)http://www.pythontest.net/index.html#frag)r r r$ r5 �Requestr6 � assertEqual�geturl)r Z urlwith_frag�req�resr r r �test_urlwithfrag� s
z"OtherNetworkTests.test_urlwithfragc
C sF d}t j|��. tjj|�}tjj|�}| j|j� d� W d Q R X d S )Nz*http://www.pythontest.net/redir/with_frag/z)http://www.pythontest.net/elsewhere/#frag)r r r$ r5 r9 r6 r: r; )r Zredirect_url_with_fragr< r= r r r �test_redirect_url_withfrag� s
z,OtherNetworkTests.test_redirect_url_withfragc
C s� d}t j|��z tjj� }tjj|�}| j|j� � |j|� | j |j� � | j |j
d�� |j dd� |j|� | j |j
d�d� W d Q R X d S )Nzhttp://www.example.comz
User-agentz
User-Agentz
Test-Agent)r r r$ r5 � build_openerr9 r Z header_itemsr. r Z
has_headerZ
add_headerr: Z
get_header)r r �openerr5 r r r �test_custom_headers� s
z%OtherNetworkTests.test_custom_headersz XXX: http://www.imdb.com is gonec C s� d}t j|��x y tjj|��}W d Q R X W n, tk
rX } z| jd� W Y d d }~X n X | j|� tjj|�}|j� }| j|� W d Q R X d S )Nzhttp://www.imdb.comzOurlopen failed for site not sending Connection:close) r r r$ r5 r6 r4 Zfailr �read)r ZURLr= r r< r r r � test_sites_no_connection_close� s
z0OtherNetworkTests.test_sites_no_connection_closeTc 6 C s� dd l }dd l}|jd�j}tjj|� j}|r<t|tj j
�}�xL|D �]B}| j |d���* t |t
�rp|\}} }
nd } }
tj|��� y||| t�} W nN tk
r� } z2|
r�d|
|| t| �| f }
| j| |
|
� n� W Y d d } ~ X n�X yRtj�B tj�0 tj� | j� }|dt|� � W d Q R X W d Q R X W d Q R X W n* tjk
�r` td| t j d� Y nX | j � W d Q R X |j d� W d Q R X qDW d S ) Nr Z test_urllib2)r z5Didn't get expected error(s) %s for %s %s, got %s: %sz
read %d bytesz
<timeout: %s>)�fileg�������?) �time�loggingZ getLogger�debugr$ r5 r@ r. r r% r& ZsubTest�
isinstance�tupler r �TIMEOUT�OSError�typeZassertIsInstanceZtime_outZsocket_peer_resetZioerror_peer_resetrC �len�socket�timeout�print�sys�stderrr Zsleep)r r) �handlersr, rF rG rH r6 r r< Z expected_errr7 �err�msg�bufr r r r' � s<
2z OtherNetworkTests._test_urlsc C s2 g }t jj� }| j|j� |jd� |j|� |S )N� )r$ r5 ZCacheFTPHandler�
addCleanupZ clear_cacheZ
setTimeout�append)r rT Zcfhr r r r( � s
z!OtherNetworkTests._extra_handlersN)T)r r r r# �skip_ftp_test_on_travisr* r8 r> r? rB �unittest�skiprD r' r( r r r r r" _ s -
'r" c @ s` e Zd Zdd� Zdd� Zdd� Zdd� Zd Zed
d � �Z ed d
� �Z
edd� �Z edd� �Z dS )� TimeoutTestc C sX | j tj� � d}tj|d d��. t|�}| j|j� | j |jj j
j � � W d Q R X d S )Nzhttp://www.example.com)rP ) � assertIsNonerO �getdefaulttimeoutr r r rY r r �raw�_sock�
gettimeout)r r �ur r r �test_http_basic s zTimeoutTest.test_http_basicc C sr | j tj� � d}tj|��L tjd� zt|�}| j|j� W d tjd � X | j |j
j j j
� d� W d Q R X d S )Nzhttp://www.example.comr )r_ rO r` r r �setdefaulttimeoutr rY r r: r ra rb rc )r r rd r r r �test_http_default_timeout s
z%TimeoutTest.test_http_default_timeoutc C st | j tj� � d}tj|��N tjd� z t|d d�}| j|j� W d tjd � X | j |j j
j j � � W d Q R X d S )Nzhttp://www.example.comr )rP )
r_ rO r` r r rf r rY r r ra rb rc )r r rd r r r �test_http_no_timeout s
z TimeoutTest.test_http_no_timeoutc C sL d}t j|��4 t|dd�}| j|j� | j|jjjj � d� W d Q R X d S )Nzhttp://www.example.com�x )rP )
r r r rY r r: r ra rb rc )r r rd r r r �test_http_timeout$ s
z TimeoutTest.test_http_timeoutzftp://www.pythontest.net/c C sZ | j tj� � tj| jd d��2 t| j�}| j|j� | j |j j j
j j � � W d Q R X d S )N)rP )
r_ rO r` r r �FTP_HOSTr rY r r ra rb rc )r rd r r r �test_ftp_basic- s
zTimeoutTest.test_ftp_basicc C st | j tj� � tj| j��P tjd� zt| j�}| j|j � W d tjd � X | j
|j j j j
j� d� W d Q R X d S )Nr )r_ rO r` r r rk rf r rY r r: r ra rb rc )r rd r r r �test_ftp_default_timeout5 s
z$TimeoutTest.test_ftp_default_timeoutc C sv | j tj� � tj| j��R tjd� z t| jd d�}| j|j � W d tjd � X | j |j
j
j j j
� � W d Q R X d S )Nr )rP )r_ rO r` r r rk rf r rY r r ra rb rc )r rd r r r �test_ftp_no_timeoutA s
z TimeoutTest.test_ftp_no_timeoutc C sN t j| j��8 t| jdd�}| j|j� | j|jjjj j
� d� W d Q R X d S )Nr )rP ) r r rk r rY r r: r ra rb rc )r rd r r r �test_ftp_timeoutM s z TimeoutTest.test_ftp_timeoutN)
r r r re rg rh rj rk r[ rl rm rn ro r r r r r^ s r^ �__main__) r\ Ztestr Ztest.test_urllib2r r0 rO Z urllib.errorr$ Zurllib.requestrR ZrequiresrK r r ZskipIf�environr[ r5 r6 r% r& r ZTestCaser r r" r^ r �mainr r r r �<module> s,
'
%R