✘✘ GRAYBYTE WORDPRESS FILE MANAGER ✘✘

3

� f��@s�dZd
dl
Z
d
dlZd
dljZd
dlmZ
d
dlmZ
d
dl	m
Z
d
dlm
Z

d
dl
mZ
d
dlmZ
d
dlmZmZ
d
d	lmZmZ
d
ZdZGdd
�d
e�ZGdd�de�Zdd�Zedkr�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@s0eZ
dZd
dd�
Zddd�
Zdd�Zdd	�Zd
S)�FakeOpenNcCs6|
|_t
|
t�s|
|_nd|_|p$d
|_|p.d|_dS)NZOK��)�url�
isinstance�str�req�msg�code)�selfr
rr�r�@/opt/alt/python36/lib64/python3.6/distutils/tests/test_upload.py�__init__0s






zFakeOpen.__init__cCsd
di
j|
j
�|�S)Nzcontent-typeztext/plain; charset=utf-8)�get�lower)r�name�defaultrrr�	getheader9s
zFakeOpen.getheaderc


Csd
S)Nsxyzzyr)
rrrr�read>s
z
FakeOpen.readc


Cs|jS)
N)
r)
rrrr�getcodeAs
zFakeOpen.getcode)NN)
N)�__name__�
__module__�__qualname__rrrrrrrrr.s
	
rcs`eZ
dZ�f
d
d�Z�f
dd�Zdd�Zdd�Zd	d
�Zdd�Zd
d�Z	dd�Z
dd�Z�ZS)�uploadTestCasec

s4tt
|�j�
tj|_|jt_d|_d|_d|_	dS)
N)
�superr!�setUp�
upload_mod�urlopen�old_open�_urlopen�	last_open�next_msg�	next_code)
r)
�	__class__rrr#Gs





zuploadTestCase.setUpc

s|jt
_tt|�j�
dS)
N)r&r$r%r"r!�tearDown)
r)
r+rrr,Os

zuploadTestCase.tearDowncCst|
|j
|jd
�|_|jS)N)rr)rr)r*r()rr
rrrr'Ss

zuploadTestCase._urlopenc
CsL|j|j
t�
t�}
t|
�
}|j�
x"d
D]\}}|jt||�|�
q*WdS)N�username�me�password�secret�realm�pypi�
repository�https://upload.pypi.org/legacy/�r-r.�r/r0�r1r2�r3r4)r5r6r7r8)�
write_file�rcr	rr�finalize_options�assertEqual�getattr)r�dist�cmd�attrZwaitedrrr�test_finalize_optionsWs






z$uploadTestCase.test_finalize_optionsc
CsZ|j|j
t�
t�}
t|
�
}|j�
|j|jd�
d
|
_t|
�
}|j�
|j|jd
�
dS)N�xxx)r9r:�PYPIRC_NOPASSWORDrrr;r<r/)rr>r?rrr�test_saved_passwordcs





z"uploadTestCase.test_saved_passwordc
Cs.
|j�}
t
jj|
d
�}|j|�

d
d|}}}|||fg
}|j|jt�
|j|d�
\}}t|�
}	d|	_	|	j
�
|	j�
t|j
jj�
}
|j|
dd�
|
d}|j|jd�
�

|j|j
jj�d	�
d
}|j|j
jj�|�
|jd|j
jjk�

|jd|j
jj�
|jt�
}
|j|
dd
ddd
d�
dS)NrBz2.6)
�
dist_files�
zContent-lengthZ2162zContent-typezmultipart/form-dataZPOSTzhttps://upload.pypi.org/legacy/sxxxsprotocol_version�K�
-z
xyzzy
���)�mkdtemp�os�path�joinr9r:�PYPIRC_LONG_PASSWORD�create_distr�
show_response�ensure_finalized�run�dictr(r�headersr<Z
assertTrue�
startswithZ
get_methodZget_full_url�data�assertIn�get_logsr)r�tmprL�command�	pyversion�filenamerE�pkg_dirr>r?rTZcontent_typeZexpected_url�resultsrrr�test_uploadts,




















zuploadTestCase.test_uploadc
Cs�|j�}
t
jj|
d
�}|j|dd�
d
d|}}}|||fg
}|j|jt�
|j|dd�\}}t|�
}	d|	_	|	j
�
|	j�
t|j
jj�
}
|j|
dd	�
|jd
|j
jj�
dS)NrBzyy
)
Zcontentz2.6zlong description
)rEZdescriptionrFzContent-lengthZ2172slong description
)rJrKrLrMr9r:rNrOrrPrQrRrSr(rrTr<rWrV)rrYrLrZr[r\rEr]r>r?rTrrr�test_upload_correct_cr�s 











z%uploadTestCase.test_upload_correct_crc

Csd
|_d|_
|jt|j�
dS)Nz	Not Foundi�
)r)r*�assertRaisesrr_)
rrrr�test_upload_fails�s


z uploadTestCase.test_upload_failsc
Cs
|j�}
t
jj|
d
�}|j|�

d
d|fg
}|j|jt�
|j|d�
\}}td�
dtft	dddid�dt
fg}x�|D]�\}}}	|jt|�
j
d	�
�x
tjd
tj|d�
d��V
|j|	�
�
t|�
}
|
j�
|
j�
WdQRX|jt�
}|j||d�
|j�
WdQRXWdQRXqtWdS)NrBz2.6)
rEZoserrorr
i�
Z	httperrorzUpload failed (400): httperror)
�	exceptionz distutils.command.upload.urlopen)
Zside_effect)
�newrFrI)rJrKrLrMr9r:rNrO�OSErrorrrZsubTest�typer�mockZpatchZMockrarrQrRrXrrWZ
clear_logs)rrYrLrEr]r>ZtestsrcZexpectedZraised_exceptionr?r^rrr�test_wrong_exception_order�s(



















z)uploadTestCase.test_wrong_exception_order)
rrr r#r,r'rArDr_r`rbrh�
__classcell__rr)
r+rr!Es r!cCs
tj
t�
S)
N)�unittestZ	makeSuiter!rrrr�
test_suite�s
rk�__main__)�__doc__rKrjZ
unittest.mockrgZurllib.requestrZtest.supportrZdistutils.commandrr$Zdistutils.command.uploadZdistutils.corerZdistutils.errorsrZ
distutils.logrrZdistutils.tests.test_configr	r
rNrC�objectrr!rkrrrrr�<module>
s&