File: //usr/local/lib/python3.7/test/__pycache__/test_urllibnet.cpython-37.pyc
B
��g�# � @ s� d dl Z d dlmZ d dlZd dlZd dlZd dlZd dlZd dl Z
d dl Z e� d� G dd� de j
�ZG dd� de j
�ZG dd � d e j
�Zed
kr�e �� dS ) � N)�supportZnetworkc @ s( e Zd ZdZdd� Zdd� Zdd� ZdS ) �URLTimeoutTestg >@c C s t �| j� d S )N)�socket�setdefaulttimeout�TIMEOUT)�self� r �//usr/local/lib/python3.7/test/test_urllibnet.py�setUp s zURLTimeoutTest.setUpc C s t �d � d S )N)r r )r r r r �tearDown s zURLTimeoutTest.tearDownc C s@ t j�tj�j}t�|�� t j�tj�}|� � W d Q R X d S )N)
�urllib�parseZurlparser Z
TEST_HTTP_URLZnetloc�transient_internet�request�urlopen�read)r �domain�fr r r � testURLread s zURLTimeoutTest.testURLreadN)�__name__�
__module__� __qualname__r r
r r r r r r r s r c @ sR e Zd ZdZdZejdd� �Zdd� Zdd� Z d d
� Z
d d � Z d
d� Z dd� Z
dS )�urlopenNetworkTestsaA Tests urllib.request.urlopen using the network.
These tests are not exhaustive. Assuming that testing using files does a
good job overall of some of the basic interface features. There are no
tests exercising the optional 'data' and 'proxies' arguments. No tests
for transparent redirection have been written.
setUp is not used for always constructing a connection to
http://www.pythontest.net/ since there a few tests that don't use that address
and making a connection is expensive enough to warrant minimizing unneeded
connections.
zhttp://www.pythontest.net/c o sF |d }t �|��* tjj||�}z
|V W d |�� X W d Q R X d S )Nr )r r r r r �close)r �args�kwargs�resource�rr r r r 3 s
zurlopenNetworkTests.urlopenc C sP | � | j��:}x"dD ]}| �t||�d| � qW | �|�� d� W d Q R X d S )N)r �readline� readlines�filenor �info�geturlz3object returned from urlopen lacks the %s attributezcalling 'read' failed)r �url�
assertTrue�hasattrr )r �open_url�attrr r r �
test_basic= s
z urlopenNetworkTests.test_basicc C s@ | � | j��*}| �|�� td� | �|�� td� W d Q R X d S )Nz readline did not return a stringz readlines did not return a list)r r# �assertIsInstancer �bytesr �list)r r& r r r �test_readlinesF s
z"urlopenNetworkTests.test_readlinesc C sF | � | j��0}|�� }| �|tjjd� | �|�� d� W d Q R X d S )NzEobject returned by 'info' is not an instance of email.message.MessageZhtml) r r# r! r) �email�message�Message� assertEqualZget_content_subtype)r r& Zinfo_objr r r � test_infoN s
z urlopenNetworkTests.test_infoc C s2 | � | j�� }|�� }| �|| j� W d Q R X d S )N)r r# r" r0 )r r& Z
gotten_urlr r r � test_geturlW s z urlopenNetworkTests.test_geturlc
C sn | j d }t�|��P | �t�� tj�� �|�}W d Q R X z |� � }W d |�
� X | � |d� W d Q R X d S )NZ
XXXinvalidXXXi� ) r# r r Z assertWarns�DeprecationWarningr r ZFancyURLopener�openZgetcoder r0 )r ZURLr& �coder r r � test_getcode] s
z urlopenNetworkTests.test_getcodec C sj d}yt �|� W n tk
r& Y nX | �d| � d}| jt|d�� tj�d�|�� W d Q R X d S )Nzsadflkjsasf.i.nvali.d.z&%r should not resolve for test to workzuopening an invalid URL did not raise OSError; can be caused by a broken DNS server (e.g. returns 404 or hijacks page))�msgz
http://{}/) r Z
gethostbyname�OSErrorZskipTestZ assertRaisesr r r �format)r Z bogus_domainZfailure_explanationr r r �test_bad_addressi s z$urlopenNetworkTests.test_bad_addressN)r r r �__doc__r# �
contextlib�contextmanagerr r( r, r1 r2 r6 r: r r r r r " s
r c @ sJ e Zd ZdZejdd� �Zdd� Zdd� Zdd � Z d
Z
d d � Z d
d� Z dS )�urlretrieveNetworkTestsz3Tests urllib.request.urlretrieve using the network.c
o sP |d }t �|��4 tjj||�\}}z||fV W d t �|� X W d Q R X d S )Nr )r r r r � urlretrieve�unlink)r r r r �
file_locationr! r r r r? � s z#urlretrieveNetworkTests.urlretrievec
C sZ | � | j��D\}}| �tj�|�d� t|d��}| �|�� d� W d Q R X W d Q R X d S )Nz9file location returned by urlretrieve is not a valid path�rbz=reading from the file location returned by urlretrieve failed)r? �logor$ �os�path�existsr4 r )r rA r! r r r r r( � s z"urlretrieveNetworkTests.test_basicc
C sj | � | jtj��P\}}| �|tj� | �tj�|�� t |d��}| �|�
� d� W d Q R X W d Q R X d S )NrB z"reading from temporary file failed) r? rC r ZTESTFNr0 r$ rD rE rF r4 r )r rA r! r r r r �test_specified_path� s z+urlretrieveNetworkTests.test_specified_pathc C s2 | � | j�� \}}| �|tjjd� W d Q R X d S )Nz0info is not an instance of email.message.Message)r? rC r) r- r. r/ )r rA r! r r r � test_header� s z#urlretrieveNetworkTests.test_headerzhttp://www.pythontest.net/c C sb | � | j��L\}}|�d�}d}yt�||� W n" tk
rR | �d| � Y nX W d Q R X d S )NZDatez%a, %d %b %Y %H:%M:%S GMTzDate value not in %r format)r? rC �get�timeZstrptime�
ValueErrorZfail)r rA � fileheadersZ datevalueZ
dateformatr r r �test_data_header� s
z(urlretrieveNetworkTests.test_data_headerc s g � � fdd�}| j | j|d��\}}t|d �}W d Q R X t� �}| jt� �ddd� | �� d d d� | j� d d dd | d� | �� d d
|� | �� d d
|� d d
� � D �}| j� d d h|d| d� | j� d d � d d |d| d� d S )Nc s � � | ||f� d S )N)�append)Zblocks�
block_sizeZ
total_size)�recordsr r �recording_reporthook� s zEurlretrieveNetworkTests.test_reporthook.<locals>.recording_reporthook)Z
reporthookzContent-Length� zKThere should always be two calls; the first one before the transfer starts.)r7 r zblock size can't be 0 in %s� ���c S s h | ]\}}}|�qS r r )�.0�_rO r r r � <setcomp>� s z:urlretrieveNetworkTests.test_reporthook.<locals>.<setcomp>z block sizes in %s must be equalz9number of blocks * block size must be >= total size in %s)r? rC �int�reprZ
assertGreater�lenr0 ZassertGreaterEqual)r rQ rA rL Z
expected_sizeZ records_reprZ block_sizesr )rP r �test_reporthook� s$ z'urlretrieveNetworkTests.test_reporthookN)
r r r r; r< r= r? r( rG rH rC rM r[ r r r r r> � s
r> �__main__)ZunittestZtestr r< r Z urllib.parser Zurllib.requestrD Z
email.messager- rJ ZrequiresZTestCaser r r> r �mainr r r r �<module> s
nK