Your IP : 216.73.216.196
o
0�iD8 � @ sL d dl Z d dlZd dlmZ d dlmZ d dlmZ d dlmZ d dlm Z d dl
Z
d dl Z d dl Z
d dlZ
d dlZe�d� dd � Zd
d � Ze�d e
jv d
�Zee
jje
jj�ZG dd� de�Z e e e j d�Z e e e j d�Z!e e e j d�Z"G dd� dej#�Z$G dd� dej#�Z%G dd� dej#�Z&G dd� dej#�Z'e(dkr�e�)� dS dS )� N)�support)� os_helper)�
socket_helper)�ResourceDenied)�sanepathname2urlZnetworkc O sN t d�D ] }z
| |i |��W S |y$ } z|}W Y d }~qd }~ww |�)N� )�range)�func�exc�args�kwargs�i�e�last_exc� r �;/opt/alt/python310/lib64/python3.10/test/test_urllib2net.py�
_retry_thrice s ��r c s � �fdd�}|S )Nc s t �� g| �R i |��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 ZdZdd� Zdd� Zd dd�ZdS )
�TransientResourcez�Raise ResourceDenied if an exception is raised while the context manager
is in effect that matches the specified exception and attributes.c K s || _ || _d S r )r
�attrs)�selfr
r r r r �__init__2 s
zTransientResource.__init__c C s | S r r �r r r r � __enter__6 � zTransientResource.__enter__Nc C s\ |dur*t | j|�r,| j�� D ]\}}t||�s dS t||�|kr% dS qtd��dS dS )z�If type_ is a subclass of self.exc and value has attributes matching
self.attrs, raise ResourceDenied. Otherwise let the exception
propagate (if any).Nz%an optional resource is not available)�
issubclassr
r �items�hasattr�getattrr )r Ztype_�value� traceback�attrZ
attr_valuer r r �__exit__9 s
��zTransientResource.__exit__)NNN)�__name__�
__module__� __qualname__�__doc__r r r% r r r r r - s
r )�errnoc @ s e Zd ZdZdS )� AuthTestsz&Tests urllib2 authentication features.N)r& r' r( r) r r r r r+ N s r+ c @ s e Zd Zdd� ZdS )�CloseSocketTestc C sr | � tjj� tj}t�|�� t|�}|j }| �
|j � |� � | �
|j � W d � d S 1 s2w Y d S r )�
addCleanup�urllib�request�
urlcleanupr �
TEST_HTTP_URLr �transient_internet�_urlopen_with_retry�fp� assertFalse�closed�close�
assertTrue)r �urlZresponseZsockr r r �
test_closew s "�zCloseSocketTest.test_closeN)r& r' r( r: r r r r r, u 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 )�logging� getLoggerZ
addHandlerZ
StreamHandler)r r< Zloggerr r r �setUp� r zOtherNetworkTests.setUpc C s& ddd t jjfg}| �|| �� � d S )Nz ftp://www.pythontest.net/READMEz*ftp://www.pythontest.net/non-existent-file)r. �error�URLError�
_test_urls�_extra_handlers)r �urlsr r r �test_ftp� s ��zOtherNetworkTests.test_ftpc C s� t j}t|d�}z+|�d� |�� dttj�|�� dd t j
j fg}| j || �
� dd� W t�|� nt�|� w | �tt jjd� d S )N�wz hi there
zfile:z file:///nonsensename/etc/passwdT)�retryz./relative_path/to/file)r �TESTFN�open�writer7 r �os�path�abspathr. r? r@ rA rB �removeZ assertRaises�
ValueErrorr/ �urlopen)r rG �frC r r r � test_file� s
��zOtherNetworkTests.test_filec C s\ d}t �|�� tj�|�}tj�|�}| �|�� d� W d � d S 1 s'w Y d S )Nz)http://www.pythontest.net/index.html#frag�r r2 r. r/ �RequestrO � assertEqualZgeturl)r Z urlwith_frag�req�resr r r �test_urlwithfrag� �
�"�z"OtherNetworkTests.test_urlwithfragc C s\ d}t �|�� tj�|�}tj�|�}| �|�� d� W d � d S 1 s'w Y d S )Nz*http://www.pythontest.net/redir/with_frag/z)http://www.pythontest.net/elsewhere/#fragrR )r Zredirect_url_with_fragrU rV r r r �test_redirect_url_withfrag� rX z,OtherNetworkTests.test_redirect_url_withfragc C s� t j}t�|��C tj�� }tj�|�}| �|� � � |�
|� | � |� � � | � |� d�� |�
dd� |�
|� | �|�d�d� W d � d S 1 sNw Y d S )Nz
User-agentz
User-Agentz
Test-Agent)r r1 r r2 r. r/ � build_openerrS r5 Z header_itemsrH r8 Z
has_headerZ
add_headerrT Z
get_header)r r9 Zopenerr/ r r r �test_custom_headers� s
"�z%OtherNetworkTests.test_custom_headersz XXX: http://www.imdb.com is gonec C s� d}t �|��D ztj�|��}W d � n1 sw Y W n ty. | �d� Y nw | �|� tj�|�}|�� }| �|� W d � d S 1 sNw Y d S )Nzhttp://www.imdb.comzOurlopen failed for site not sending Connection:close) r r2 r. r/ rO rN Zfailr8 �read)r ZURLrV rU 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 ]�}| j |d��� t |t
�r4|\}} }
nd } }
t�|��� z ||| tj�} W n( tyo } z |
rdd|
|| t| �| f }
| �| |
|
� n� W Y d } ~ n`d } ~ ww zDt�8 t�& t� | �� }|dt|� � W d � n1 s�w Y W d � n1 s�w Y W d � n1 s�w Y W n ty� td| t j d� Y nw | � � W d � n1 s�w Y |� d� W d � n1 s�w Y q d S ) Nr Z test_urllib2)r9 z5Didn't get expected error(s) %s for %s %s, got %s: %sz
read %d bytesz
<timeout: %s>)�fileg�������?) �timer< r= �debugr. r/ rZ rH r r? r@ ZsubTest�
isinstance�tupler r2 r ZINTERNET_TIMEOUT�OSError�typeZassertIsInstance�time_out�socket_peer_reset�ioerror_peer_resetr\ �len� TimeoutError�print�sys�stderrr7 �sleep)r rC �handlersrF r_ r< r` rO r9 rU Z expected_errrP �err�msgZbufr r r rA � s`
� ����� �� � � �� � ���z OtherNetworkTests._test_urlsc C s2 g }t j�� }| �|j� |�d� |�|� |S )N� )r. r/ ZCacheFTPHandlerr- Z clear_cacheZ
setTimeout�append)r rn Zcfhr r r rB s
z!OtherNetworkTests._extra_handlersN)T)r& r' r( r> �skip_ftp_test_on_travisrD rQ rW rY r[ �unittest�skipr] rA rB r r r r r; � s
-
'r; c @ sh e Zd Zdd� Zdd� Zdd� Zdd� Zd d
� Zd Ze d d
� �Z
e dd� �Z e dd� �Z e dd� �Z
dS )� TimeoutTestc C s | � tjj� d S r )r- r. r/ r0 r r r r r> * s zTimeoutTest.setUpc C sp | � t�� � tj}tj|d d�� t|�}| �|j � | � |j
j j �
� � W d � d S 1 s1w Y d S �N�Ztimeout)� assertIsNone�socket�getdefaulttimeoutr r1 r r2 r3 r- r7 r4 �raw�_sock�
gettimeout�r r9 �ur r r �test_http_basic. s "�zTimeoutTest.test_http_basicc
C s� | � t�� � tj}t�|��1 t�d� zt|�}| � |j
� W t�d � nt�d � w | � |j j
j�� d� W d � d S 1 sCw Y d S �N�< )ry rz r{ r r1 r r2 �setdefaulttimeoutr3 r- r7 rT r4 r| r} r~ r r r r �test_http_default_timeout6 s
"�z%TimeoutTest.test_http_default_timeoutc
C s� | � t�� � tj}t�|��2 t�d� zt|d d�}| � |j
� W t�d � nt�d � w | � |j j j
�� � W d � d S 1 sDw Y d S �Nr� rx )ry rz r{ r r1 r r2 r� r3 r- r7 r4 r| r} r~ r r r r �test_http_no_timeoutB s
"�z TimeoutTest.test_http_no_timeoutc C sd t j}t�|�� t|dd�}| �|j� | �|jj j
� � d� W d � d S 1 s+w Y d S )N�x rx ) r r1 r r2 r3 r- r7 rT r4 r| r} r~ r r r r �test_http_timeoutN s "�z TimeoutTest.test_http_timeoutzftp://www.pythontest.net/c C sp | � t�� � tj| jd d�� t| j�}| �|j� | � |j j j
j � � � W d � d S 1 s1w Y d S rw )
ry rz r{ r r2 �FTP_HOSTr3 r- r7 r4 r| r} r~ �r r� r r r �test_ftp_basicW s
"�zTimeoutTest.test_ftp_basicc
C s� | � t�� � t�| j��3 t�d� zt| j�}| �|j � W t�d � nt�d � w | �
|j j j j
�� d� W d � d S 1 sCw Y d S r� )ry rz r{ r r2 r� r� r3 r- r7 rT r4 r| r} r~ r� r r r �test_ftp_default_timeout_ s
"�z$TimeoutTest.test_ftp_default_timeoutc
C s� | � t�� � t�| j��4 t�d� zt| jd d�}| �|j � W t�d � nt�d � w | � |j
j
j j �
� � W d � d S 1 sDw Y d S r� )ry rz r{ r r2 r� r� r3 r- r7 r4 r| r} r~ r� r r r �test_ftp_no_timeoutk s
"�z TimeoutTest.test_ftp_no_timeoutc C sd t �| j��" t| jdd�}| �|j� | �|jjjj �
� d� W d � d S 1 s+w Y d S r� ) r r2 r� r3 r- r7 rT r4 r| r} r~ r� r r r �test_ftp_timeoutw s
"�z TimeoutTest.test_ftp_timeoutN)r& r' r( r> r� r� r� r� r� rs r� r� r� r� r r r r rv ) s
rv �__main__)*r* rt �testr Z test.supportr r r Ztest.test_urllib2r rJ rz Z urllib.errorr. Zurllib.requestrk Zrequiresr r ZskipIf�environrs r/ rO r? r@ r3 �objectr rc Z ETIMEDOUTre Z
ECONNRESETrf rg ZTestCaser+ r, r; rv r& �mainr r r r �<module> sB
�� ' %V �