Your IP : 216.73.216.213
3
��2aQ# � @ s� d dl Z d dlmZ d dlZd dlZd dlZd dlZd dlZ d dl
Z
ej d� G dd� de j �Z
G dd� de j �ZG dd � d e j �Zed
kr�e j� dS ) � N)�supportZnetworkc @ s( e Zd ZdZdd� Zdd� Zdd� ZdS ) �URLTimeoutTestg >@c C s t j| j� d S )N)�socket�setdefaulttimeout�TIMEOUT)�self� r �3/opt/alt/python36/lib64/python3.6/test_urllibnet.py�setUp s zURLTimeoutTest.setUpc C s t jd � d S )N)r r )r r r r �tearDown s zURLTimeoutTest.tearDownc C s. t jd�� tjjd�}|j� W d Q R X d S )Nzwww.example.comzhttp://www.example.com/)r �transient_internet�urllib�request�urlopen�read)r �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 j|��* tjj||�}z
|V W d |j� X W d Q R X d S )Nr )r r r
r r �close)r �args�kwargs�resource�rr r r r 1 s
zurlopenNetworkTests.urlopenc C sP | j | j��:}x"d
D ]}| jt||�d| � qW | j|j� d � W d Q R X d S ) Nr �readline� readlines�filenor �info�geturlz3object returned from urlopen lacks the %s attributezcalling 'read' failed)r r r r r r r )r �url�
assertTrue�hasattrr )r �open_url�attrr r r �
test_basic; s z urlopenNetworkTests.test_basicc C s@ | j | j��*}| j|j� td� | j|j� 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_readlinesD s
z"urlopenNetworkTests.test_readlinesc C sF | j | j��0}|j� }| j|tjjd� | j|j� 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_infoL s
z urlopenNetworkTests.test_infoc
C s2 | j | j�� }|j� }| j|| j� W d Q R X d S )N)r r! r r. )r r$ Z
gotten_urlr r r � test_geturlU s z urlopenNetworkTests.test_geturlc C sn | j d }tj|��P | jt�� tjj� j|�}W d Q R X z |j � }W d |j
� X | j |d� W d Q R X d S )NZ
XXXinvalidXXXi� ) r! r r Z assertWarns�DeprecationWarningr
r ZFancyURLopener�openZgetcoder r. )r ZURLr$ �coder r r � test_getcode[ s
z urlopenNetworkTests.test_getcodec C sj d}yt j|� W n tk
r& Y nX | jd| � d}| jt|d�� tjjdj|�� 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_addressg s z$urlopenNetworkTests.test_bad_addressN)
r r r r! �
contextlib�contextmanagerr r& r* r/ r0 r4 r8 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 j|��4 tjj||�\}}z||fV W d t j|� 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 | j��D\}}| jtjj|�d� t|d��}| j|j� 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�existsr2 r )r r> r r r r r r& � s z"urlretrieveNetworkTests.test_basicc C sj | j | jtj��P\}}| j|tj� | jtjj|�� t |d��}| j|j
� d� W d Q R X W d Q R X d S )Nr? z"reading from temporary file failed) r< r@ r ZTESTFNr. r" rA rB rC r2 r )r r> r r r r r �test_specified_path� s z+urlretrieveNetworkTests.test_specified_pathc C s2 | j | j�� \}}| j|tjjd� W d Q R X d S )Nz0info is not an instance of email.message.Message)r< r@ r' r+ r, r- )r r> r r r r � test_header� s z#urlretrieveNetworkTests.test_headerzhttp://www.pythontest.net/c C sb | j | j��L\}}|jd�}d}ytj||� W n" tk
rR | jd| � Y nX W d Q R X d S )NZDatez%a, %d %b %Y %H:%M:%S GMTzDate value not in %r format)r< r@ �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� | j� d d d� | j� d d dd | d� | j� d d
|� | j� d d
|� d d � � D �}| j� d d h|d
| d� | j� d d � d d |d| d� d S )Nc s � j | ||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.)r5 r zblock size can't be 0 in %s� c S s h | ]\}}}|�qS r r )�.0�_rL 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���rT )r< r@ �int�reprZ
assertGreater�lenr. ZassertGreaterEqual)r rN r> rI Z
expected_sizeZ records_reprZ block_sizesr )rM r �test_reporthook� s$ z'urlretrieveNetworkTests.test_reporthookN) r r r r9 r: r< r& rD rE r@ rJ rX r r r r r; � s
r; �__main__)ZunittestZtestr r9 r Zurllib.requestr
rA Z
email.messager+ rG ZrequiresZTestCaser r r; r �mainr r r r �<module> s
nK