File: //usr/local/lib/python3.10/test/__pycache__/test_urllib2net.cpython-310.opt-2.pyc
o
�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 �1/usr/local/lib/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 Z dd� Zdd� Zddd�ZdS ) �TransientResourcec 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^ |d ur+t | j|�r-| j�� D ]\}}t||�s d S t||�|kr& d S qtd��d S d S )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__r r r% r r r r r - s
r )�errnoc @ s e Zd ZdS )� AuthTestsN)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 �url�response�sockr 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� getLogger�
addHandler�
StreamHandler)r r= �loggerr 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�writer6 r �os�path�abspathr- rC rD rE rF �removeZ assertRaises�
ValueErrorr. �urlopen)r rK �frG 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 r1 r- r. �RequestrS � assertEqual�geturl)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/#fragrV )r Zredirect_url_with_fragrZ r[ r r r �test_redirect_url_withfrag� r] 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 r0 r r1 r- r. � build_openerrW r4 Z header_itemsrL r7 Z
has_headerZ
add_headerrX Z
get_header)r r8 �openerr. 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 r1 r- r. rS rR Zfailr7 �read)r ZURLr[ rZ 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)r8 z5Didn't get expected error(s) %s for %s %s, got %s: %sz
read %d bytesz
<timeout: %s>)�fileg�������?) �timer= r> �debugr- r. r_ rL r rC rD ZsubTest�
isinstance�tupler r1 r ZINTERNET_TIMEOUT�OSError�typeZassertIsInstance�time_out�socket_peer_reset�ioerror_peer_resetrb �len� TimeoutError�print�sys�stderrr6 �sleep)r rG �handlersrJ re r= rf rS r8 rZ Z expected_errrT �err�msg�bufr r r rE � s`
� ����� �� � � �� � ���z OtherNetworkTests._test_urlsc C s2 g }t j�� }| �|j� |�d� |�|� |S )N� )r- r. ZCacheFTPHandlerr, � clear_cacheZ
setTimeout�append)r rt Zcfhr r r rF s
z!OtherNetworkTests._extra_handlersN)T)r& r' r( rB �skip_ftp_test_on_travisrH rU r\ r^ ra �unittest�skiprc rE rF 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. r/ r r r r rB * 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��timeout)� assertIsNone�socket�getdefaulttimeoutr r0 r r1 r2 r, r6 r3 �raw�_sock�
gettimeout�r r8 �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�< )r� r� r� r r0 r r1 �setdefaulttimeoutr2 r, r6 rX r3 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� r� )r� r� r� r r0 r r1 r� r2 r, r6 r3 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 r� ) r r0 r r1 r2 r, r6 rX r3 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 r )
r� r� r� r r1 �FTP_HOSTr2 r, r6 r3 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� )r� r� r� r r1 r� r� r2 r, r6 rX r3 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� )r� r� r� r r1 r� r� r2 r, r6 r3 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 r1 r� r2 r, r6 rX r3 r� r� r� r� r r r �test_ftp_timeoutw s
"�z TimeoutTest.test_ftp_timeoutN)r& r' r( rB r� r� r� r� r� r{ r� r� r� r� r r r r r~ ) s
r~ �__main__)*r) r| �testr Z test.supportr r r Ztest.test_urllib2r rN r� Z urllib.errorr- Zurllib.requestrq Zrequiresr r ZskipIf�environr{ r. rS rC rD r2 �objectr ri Z ETIMEDOUTrk Z
ECONNRESETrl rm ZTestCaser* r+ r< r~ r& �mainr r r r �<module> sB
�� ' %V �