Your IP : 216.73.216.85
U
i�f�1 � @ s� d dl Z d dlmZ d dlmZ d dlZd dlZd dlZd dl Zd dl
Z
e� d� dZ dd� Z
dd � Ze �d
ejkd �Zeejjejj�ZG d d
� d
e j�ZG dd� de j�ZG dd� de j�ZG dd� de j�Ze dkr�e � � dS )� N)�support)�sanepathname2urlZnetwork�< c O sV t d�D ]D}z| ||�W S |k
rJ } z|}W Y �qW 5 d }~X Y qX q|�d S )N� )�range)�func�exc�args�kwargs�i�e�last_exc� r �9/opt/alt/python38/lib64/python3.8/test/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 �_wrap_with_retry_thrice s r ZTRAVISz%bpo-35411: skip FTP test on Travis CIc @ s e Zd ZdZdS )� AuthTestsz&Tests urllib2 authentication features.N)�__name__�
__module__� __qualname__�__doc__r r r r r + s r c @ s e Zd Zdd� ZdS )�CloseSocketTestc C sN t j}t �|��4 t|�}|j}| �|j� |�� | �|j� W 5 Q R X d S r ) r �
TEST_HTTP_URL�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
� d
�dd� �Z ddd�Z
dd� ZdS )�OtherNetworkTestsc C s d S r r )r# r r r �setUp` s zOtherNetworkTests.setUpc C s& ddd t jjfg}| �|| �� � 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|�d� |�� dttj� |�� dd t
j j fg}| j
|| �� dd� W 5 t�|� X | �tt
jjd� d S )N�wz hi there
zfile:z file:///nonsensename/etc/passwdT)�retryz./relative_path/to/file)r �TESTFN�open�os�remove�writer! r �path�abspathr( r) r* r+ r, Z assertRaises�
ValueError�request�urlopen)r# r1 �fr- r r r � test_filer s
�� zOtherNetworkTests.test_filec C sF d}t �|��. tj�|�}tj�|�}| �|�� d� W 5 Q R X d S )Nz)http://www.pythontest.net/index.html#frag�r r r( r9 �Requestr: � assertEqualZgeturl)r# Z urlwith_frag�req�resr r r �test_urlwithfrag� s
�z"OtherNetworkTests.test_urlwithfragc C sF d}t �|��. tj�|�}tj�|�}| �|�� d� W 5 Q R X d S )Nz*http://www.pythontest.net/redir/with_frag/z)http://www.pythontest.net/elsewhere/#fragr= )r# Zredirect_url_with_fragr@ rA r r r �test_redirect_url_withfrag� s
�z,OtherNetworkTests.test_redirect_url_withfragc C s� t j}t �|��z tj�� }tj�|�}| �|�� � |� |� | �
|�� � | �
|� d�� |� dd� |� |� | �
|�d�d� W 5 Q R X d S )Nz
User-agentz
User-Agentz
Test-Agent)r r r r( r9 � build_openerr> r Z header_itemsr2 r" Z
has_headerZ
add_headerr? Z
get_header)r# r$ Zopenerr9 r r r �test_custom_headers� s
z%OtherNetworkTests.test_custom_headersz XXX: http://www.imdb.com is gonec C s� d}t �|��x z tj�|��}W 5 Q R X W n, tk
rX } z| �d� W 5 d }~X Y n X | �|� tj�|�}|�� }| �|� W 5 Q R X d S )Nzhttp://www.imdb.comzOurlopen failed for site not sending Connection:close) r r r( r9 r: r8 Zfailr" �read)r# ZURLrA r r@ r r r � test_sites_no_connection_close� s
z0OtherNetworkTests.test_sites_no_connection_closeTc C s� dd l }dd l}|�d�j}tjj|� j}|r<t|tj j
�}|D �]B}| j |d���* t |t
�rl|\}} }
nd } }
t�|��� z||| t�} W nN tk
r� } z0|
r�d|
|| t| �| f }
| �| |
|
� n� W 5 d } ~ X Y n�X zRtj�B tj�0 tj� | �� }|dt|� � W 5 Q R X W 5 Q R X W 5 Q R X W n* tjk
�r\ td| t j d� Y nX | � � W 5 Q R X |� d� W 5 Q R X q@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( r9 rD r2 r r) r* ZsubTest�
isinstance�tupler r �TIMEOUT�OSError�typeZassertIsInstanceZtime_outZsocket_peer_resetZioerror_peer_resetrF �len�socket�timeout�print�sys�stderrr! �sleep)r# r- �handlersr0 rI rJ rK r: r$ r@ Z expected_errr; �err�msgZbufr r r r+ � sB
���2z OtherNetworkTests._test_urlsc C s2 g }t j�� }| �|j� |�d� |�|� |S )N� )r( r9 ZCacheFTPHandler�
addCleanupZ clear_cacheZ
setTimeout�append)r# rX Zcfhr r r r, � s
z!OtherNetworkTests._extra_handlersN)T)r r r r' �skip_ftp_test_on_travisr. r< rB rC rE �unittest�skiprG 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 sZ | � t�� � tj}tj|d d��. t|�}| �|j� | � |j j
j � � � W 5 Q R X d S �N�rS )
� assertIsNonerR �getdefaulttimeoutr r r r r\ r! r �raw�_sock�
gettimeout�r# r$ �ur r r �test_http_basic s zTimeoutTest.test_http_basicc
C st | � t�� � tj}t�|��L t�d� zt|�}| �|j � W 5 t�d � X | �
|j j j
�� d� W 5 Q R X d S �Nr )rd rR re r r r �setdefaulttimeoutr r\ r! r? r rf rg rh ri r r r �test_http_default_timeout s
z%TimeoutTest.test_http_default_timeoutc
C sv | � t�� � tj}t�|��N t�d� z t|d d�}| �|j � W 5 t�d � X | � |j
j j �
� � W 5 Q R X d S �Nr rc )rd rR re r r r rm r r\ r! r rf rg rh ri r r r �test_http_no_timeout s
z TimeoutTest.test_http_no_timeoutc C sN t j}t �|��4 t|dd�}| �|j� | �|jjj �
� d� W 5 Q R X d S )N�x rc ) r r r r r\ r! r? r rf rg rh ri r r r �test_http_timeout$ s
z TimeoutTest.test_http_timeoutzftp://www.pythontest.net/c C sZ | � t�� � tj| jd d��2 t| j�}| �|j� | � |j j j
j � � � W 5 Q R X d S rb )
rd rR re r r �FTP_HOSTr r\ r! r rf rg rh �r# rj r r r �test_ftp_basic- s
zTimeoutTest.test_ftp_basicc
C st | � t�� � t�| j��P t�d� zt| j�}| �|j � W 5 t�d � X | �
|j j j j
�� d� W 5 Q R X d S rl )rd rR re r r rs rm r r\ r! r? r rf rg rh rt r r r �test_ftp_default_timeout5 s
z$TimeoutTest.test_ftp_default_timeoutc
C sv | � t�� � t�| j��R t�d� z t| jd d�}| �|j � W 5 t�d � X | � |j
j
j j �
� � W 5 Q R X d S ro )rd rR re r r rs rm r r\ r! r rf rg rh rt r r r �test_ftp_no_timeoutA s
z TimeoutTest.test_ftp_no_timeoutc C sN t �| j��8 t| jdd�}| �|j� | �|jjjj �
� d� W 5 Q R X d S ro ) r r rs r r\ r! r? r rf rg rh rt r r r �test_ftp_timeoutM s z TimeoutTest.test_ftp_timeoutN)
r r r rk rn rp rr rs r^ ru rv rw rx r r r r ra s
ra �__main__) r_ Ztestr Ztest.test_urllib2r r3 rR Z urllib.errorr( Zurllib.requestrU ZrequiresrN r r ZskipIf�environr^ r9 r: r) r* r ZTestCaser r r& ra r �mainr r r r �<module> s0
��'
%R