File: //usr/local/lib/python3.7/test/__pycache__/test_urllibnet.cpython-37.opt-2.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 @ sN e Zd ZdZejdd� �Zdd� Zdd� Zdd � Z d
d � Z
d d
� Z dd� Z dS )�urlopenNetworkTestszhttp://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 r# �
contextlib�contextmanagerr r( r, r1 r2 r6 r: r r r r r " s
r c @ sF e Zd Zejdd� �Zdd� Zdd� Zdd� Zd Z d
d � Z
d d
� Z dS )�urlretrieveNetworkTestsc
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 r@ 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 )NrA z"reading from temporary file failed) r> rB r ZTESTFNr0 r$ rC rD rE r4 r )r r@ 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> rB r) r- r. r/ )r r@ 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> rB �get�timeZstrptime�
ValueErrorZfail)r r@ � 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�_rN 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> rB �int�reprZ
assertGreater�lenr0 ZassertGreaterEqual)r rP r@ rK Z
expected_sizeZ records_reprZ block_sizesr )rO r �test_reporthook� s$ z'urlretrieveNetworkTests.test_reporthookN) r r r r; r< r> r( rF rG rB rL rZ r r r r r= � s
r= �__main__)ZunittestZtestr r; r Z urllib.parser Zurllib.requestrC Z
email.messager- rI ZrequiresZTestCaser r r= r �mainr r r r �<module> s
nK