Your IP : 216.73.216.213
a
XC?h�2 � @ s� d dl Z d dlmZ d dlmZ d dlmZ d dlZd dlZd dl Z
d dl Z
d dl Z e�
d� dd� Zdd � Ze �d
ejv d �Zee
jje
jj�ZG d d
� d
e j�ZG dd� de j�ZG dd� de j�ZG dd� de j�Z e dkr�e � � dS )� N)�support)�
socket_helper)�sanepathname2urlZnetworkc O s` t d�D ]N}z| |i |��W S |yT } z |}W Y d }~qW Y d }~qd }~0 0 q|�d S )N� )�range)�func�exc�args�kwargs�i�e�last_exc� r �9/opt/alt/python39/lib64/python3.9/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 ZdS )� AuthTestsN)�__name__�
__module__� __qualname__r r r r r * s r c @ s e Zd Zdd� ZdS )�CloseSocketTestc C sp | � tjj� tj}t�|��> t|�}|j }| �
|j � |� � | �
|j � W d � n1 sb0 Y d S r )�
addCleanup�urllib�request�
urlcleanupr �
TEST_HTTP_URLr �transient_internet�_urlopen_with_retry�fp� assertFalse�closed�close�
assertTrue)�self�urlZresponseZsockr r r �
test_closeS s zCloseSocketTest.test_closeN)r r r r( r r r r r Q 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 �setUpb s 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_ftpk s ��zOtherNetworkTests.test_ftpc C s� t j}t|d�}zV|�d� |�� dttj�|�� dd t j
j fg}| j || �
� dd� W t�|� n t�|� 0 | �tt 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�
ValueErrorr �urlopen)r& r4 �fr0 r r r � test_filet s
��zOtherNetworkTests.test_filec C sZ d}t �|��8 tj�|�}tj�|�}| �|�� d� W d � n1 sL0 Y d S )Nz)http://www.pythontest.net/index.html#frag�r r r r �Requestr< � assertEqualZgeturl)r& Z urlwith_frag�req�resr r r �test_urlwithfrag� s
�z"OtherNetworkTests.test_urlwithfragc C sZ d}t �|��8 tj�|�}tj�|�}| �|�� d� W d � n1 sL0 Y d S )Nz*http://www.pythontest.net/redir/with_frag/z)http://www.pythontest.net/elsewhere/#fragr? )r& Zredirect_url_with_fragrB rC r r r �test_redirect_url_withfrag� s
�z,OtherNetworkTests.test_redirect_url_withfragc C s� t j}t�|��� tj�� }tj�|�}| �|� � � |�
|� | � |� � � | � |� d�� |�
dd� |�
|� | �|�d�d� W d � n1 s�0 Y d S )Nz
User-agentz
User-Agentz
Test-Agent)r r r r r r � build_openerr@ r" Z header_itemsr5 r% Z
has_headerZ
add_headerrA Z
get_header)r& r' Zopenerr r r r �test_custom_headers� s
z%OtherNetworkTests.test_custom_headersz XXX: http://www.imdb.com is gonec C s� d}t �|��� z0tj�|��}W d � n1 s40 Y W n ty\ | �d� Y n 0 | �|� tj�|�}|�� }| �|� W d � n1 s�0 Y d S )Nzhttp://www.imdb.comzOurlopen failed for site not sending Connection:close) r r r r r< r; Zfailr% �read)r& ZURLrC rB 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
�rl|\}} }
nd } }
t�|���: z||| tj�} W nP ty� } z8|
r�d|
|| t| �| f }
| �| |
|
� n� W Y d } ~ n�d } ~ 0 0 z�tj�x tj�P tj�( | �� }|dt|� � W d � n1 �s.0 Y W d � n1 �sN0 Y W d � n1 �sn0 Y W n( tj�y� t d| t j d� Y n0 | � � W d � n1 �s�0 Y |� d� W d � q@1 �s�0 Y 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 r rF r5 r r, r- ZsubTest�
isinstance�tupler r r ZINTERNET_TIMEOUT�OSError�typeZassertIsInstanceZtime_outZsocket_peer_resetZioerror_peer_resetrH �len�socket�timeout�print�sys�stderrr$ �sleep)r& r0 �handlersr3 rK rL rM r< r' rB Z expected_errr= �err�msgZbufr r r r. � sB
���t(z OtherNetworkTests._test_urlsc C s2 g }t j�� }| �|j� |�d� |�|� |S )N� )r r ZCacheFTPHandlerr Z clear_cacheZ
setTimeout�append)r& rY Zcfhr r r r/ � s
z!OtherNetworkTests._extra_handlersN)T)r r r r+ �skip_ftp_test_on_travisr1 r> rD rE rG �unittest�skiprI r. r/ r r r r r) a 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 r+ s zTimeoutTest.setUpc C sn | � t�� � tj}tj|d d��8 t|�}| �|j � | � |j
j j �
� � W d � n1 s`0 Y d S �N�rT )� assertIsNonerS �getdefaulttimeoutr r r r r r r$ r! �raw�_sock�
gettimeout�r& r' �ur r r �test_http_basic
s zTimeoutTest.test_http_basicc
C s� | � t�� � tj}t�|��` t�d� z"t|�}| � |j
� W t�d � n t�d � 0 | � |j j
j�� d� W d � n1 s�0 Y d S �N�< )rd rS re r r r r �setdefaulttimeoutr r r$ rA r! rf rg rh ri r r r �test_http_default_timeout s
z%TimeoutTest.test_http_default_timeoutc
C s� | � t�� � tj}t�|��b t�d� z&t|d d�}| � |j
� W t�d � n t�d � 0 | � |j j j
�� � W d � n1 s�0 Y d S �Nrm rc )rd rS re r r r r rn r r r$ r! rf rg rh ri r r r �test_http_no_timeout s
z TimeoutTest.test_http_no_timeoutc C sb t j}t�|��> t|dd�}| �|j� | �|jj j
� � d� W d � n1 sT0 Y d S )N�x rc ) r r r r r r r$ rA r! rf rg rh ri r r r �test_http_timeout* s
z TimeoutTest.test_http_timeoutzftp://www.pythontest.net/c C sn | � t�� � tj| jd d��< t| j�}| �|j� | � |j j j
j � � � W d � n1 s`0 Y d S rb )
rd rS re r r �FTP_HOSTr r r$ r! rf rg rh �r& rj r r r �test_ftp_basic3 s
zTimeoutTest.test_ftp_basicc
C s� | � t�� � t�| j��d t�d� z$t| j�}| �|j � W t�d � n t�d � 0 | �
|j j j j
�� d� W d � n1 s�0 Y d S rl )rd rS re r r rt rn r r r$ rA r! rf rg rh ru r r r �test_ftp_default_timeout; s
z$TimeoutTest.test_ftp_default_timeoutc
C s� | � t�� � t�| j��f t�d� z(t| jd d�}| �|j � W t�d � n t�d � 0 | � |j
j
j j �
� � W d � n1 s�0 Y d S rp )rd rS re r r rt rn r r r$ r! rf rg rh ru r r r �test_ftp_no_timeoutG s
z TimeoutTest.test_ftp_no_timeoutc C sb t �| j��B t| jdd�}| �|j� | �|jjjj �
� d� W d � n1 sT0 Y d S rp ) r r rt r r r$ rA r! rf rg rh ru r r r �test_ftp_timeoutS s z TimeoutTest.test_ftp_timeoutN)r r r r+ rk ro rq rs rt r^ rv rw rx ry r r r r ra s
ra �__main__) r_ �testr Z test.supportr Ztest.test_urllib2r r7 rS Z urllib.errorr Zurllib.requestrV Zrequiresr r ZskipIf�environr^ r r< r, r- r ZTestCaser r r) ra r �mainr r r r �<module> s0
��' %V