File: //usr/local/lib/python3.7/distutils/tests/__pycache__/test_upload.cpython-37.pyc
B
!��g� � @ s� d Z ddlZddlZddlmZ ddlmZ ddlmZ ddl m
Z ddl m
Z
ddl
mZ ddlmZ ddlmZmZ dd lmZmZ d
Zd ZG d d
� d
e�ZG dd� de�Zdd� Z e dkr�ee � � dS )z#Tests for distutils.command.upload.� N)� HTTPError)� run_unittest)�upload)� Distribution)�DistutilsError)�ERROR�INFO)�PYPIRC�BasePyPIRCCommandTestCasez�[distutils]
index-servers =
server1
server2
[server1]
username:me
password:aaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaa
[server2]
username:meagain
password: secret
realm:acme
repository:http://another.pypi/
z@[distutils]
index-servers =
server1
[server1]
username:me
c @ s0 e Zd Zd
dd�Zd dd�Zdd� Zdd � ZdS ) �FakeOpenNc C s6 || _ t|t�s|| _nd | _|p$d| _|p.d| _d S )NZOK�� )�url�
isinstance�str�req�msg�code)�selfr
r r � r �7/usr/local/lib/python3.7/distutils/tests/test_upload.py�__init__0 s
zFakeOpen.__init__c C s ddi� |�� |�S )Nz content-typeztext/plain; charset=utf-8)�get�lower)r �name�defaultr r r � getheader9 s zFakeOpen.getheaderc C s dS )Ns xyzzyr )r r r r �read> s z
FakeOpen.readc C s | j S )N)r )r r r r �getcodeA s zFakeOpen.getcode)NN)N)�__name__�
__module__� __qualname__r r r r r r r r r . s
r c s` e Zd Z� fdd�Z� fdd�Zdd� Zdd� Zd d
� Zd d � Zd
d� Z dd� Z
dd� Z � Z S )�uploadTestCasec s4 t t| ��� tj| _| jt_d | _d | _d | _ d S )N)
�superr! �setUp�
upload_mod�urlopen�old_open�_urlopen� last_open�next_msg� next_code)r )� __class__r r r# G s zuploadTestCase.setUpc s | j t_tt| ��� d S )N)r& r$ r% r" r! �tearDown)r )r+ r r r, O s zuploadTestCase.tearDownc C s t || j| jd�| _| jS )N)r r )r r) r* r( )r r
r r r r' S s zuploadTestCase._urlopenc C sL | � | jt� t� }t|�}|�� x"dD ]\}}| �t||�|� q*W d S )N))Zusername�me)�passwordZsecret)ZrealmZpypi)Z
repositoryz https://upload.pypi.org/legacy/)�
write_file�rcr r r �finalize_options� assertEqual�getattr)r �dist�cmd�attrZwaitedr r r �test_finalize_optionsW s z$uploadTestCase.test_finalize_optionsc C sZ | � | jt� t� }t|�}|�� | �|jd � d|_t|�}|�� | �|jd� d S )N�xxx)r/ r0 �PYPIRC_NOPASSWORDr r r1 r2 r. )r r4 r5 r r r �test_saved_passwordc s z"uploadTestCase.test_saved_passwordc C s | � � }tj�|d�}| �|� dd| }}}|||fg}| �| jt� | j|d�\}}t|�} d| _ | �
� | � � t | j
jj�}
| �|
d d� |
d } | �| �d�� | �| j
j�� d � d
} | �| j
j�� | � | �d | j
jjk� | �d | j
jj� | �t�}
| �|
d
d� d S )Nr8 z2.6)�
dist_files� zContent-lengthZ2162z Content-typezmultipart/form-dataZPOSTz https://upload.pypi.org/legacy/s xxxs protocol_version���z�---------------------------------------------------------------------------
xyzzy
---------------------------------------------------------------------------)�mkdtemp�os�path�joinr/ r0 �PYPIRC_LONG_PASSWORD� create_distr �
show_response�ensure_finalized�run�dictr( r �headersr2 Z
assertTrue�
startswithZ
get_methodZ get_full_url�data�assertIn�get_logsr )r �tmpr@ �command� pyversion�filenamer; �pkg_dirr4 r5 rH Z content_typeZ expected_url�resultsr r r � test_uploadt s,
zuploadTestCase.test_uploadc C s� | � � }tj�|d�}| j|dd� dd| }}}|||fg}| �| jt� | j|dd�\}}t|�} d| _ | �
� | � � t | j
jj�}
| �|
d d � | �d
| j
jj� d S ) Nr8 zyy
)Zcontentz2.6zlong description
)r; � descriptionr< zContent-lengthZ2172s long description
)r> r? r@ rA r/ r0 rB rC r rD rE rF rG r( r rH r2 rK rJ ) r rM r@ rN rO rP r; rQ r4 r5 rH r r r �test_upload_correct_cr� s z%uploadTestCase.test_upload_correct_crc C s d| _ d| _| �t| j� d S )Nz Not Foundi� )r) r* � assertRaisesr rS )r r r r �test_upload_fails� s z uploadTestCase.test_upload_failsc C s | � � }tj�|d�}| �|� dd|fg}| �| jt� | j|d�\}}td�dtft dddi d �dt
fg}x�|D ]�\}}} | j t |�j
d ��x tjd
tj|d �d ��V | �| �� t|�}
|
�� |
�� W d Q R X | �t�} | �|| d
� | �� W d Q R X W d Q R X qtW d S )Nr8 z2.6)r; Zoserrorr
i� Z httperrorz Upload failed (400): httperror)� exceptionz distutils.command.upload.urlopen)Z side_effect)�newr= )r> r? r@ rA r/ r0 rB rC �OSErrorr r ZsubTest�typer �mockZpatchZMockrV r rE rF rL r rK Z
clear_logs) r rM r@ r; rQ r4 ZtestsrX ZexpectedZraised_exceptionr5 rR r r r �test_wrong_exception_order� s(
z)uploadTestCase.test_wrong_exception_order)
r r r r# r, r' r7 r: rS rU rW r] �
__classcell__r r )r+ r r! E s r! c C s
t �t�S )N)�unittestZ makeSuiter! r r r r �
test_suite� s r` �__main__) �__doc__r? r_ Z
unittest.mockr\ Zurllib.requestr Z test.supportr Zdistutils.commandr r$ Zdistutils.command.uploadZdistutils.corer Zdistutils.errorsr Z
distutils.logr r Zdistutils.tests.test_configr r
rB r9 �objectr r! r` r r r r r �<module> s&