Your IP : 216.73.216.213
���]� � @ s d d l Z d d l Z d d l j Z d d l m Z d d l m Z d d l m Z
d d l m Z d d l m
Z
d d l m Z d d l m Z m Z d d l m Z m Z d Z d
Z Gd d � d e � Z Gd
d � d e � Z d d � Z e d k re e � � d S)� 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 @ sI e Z d Z d d d d � Z d d d � Z d d � Z d d � Z d S)
�FakeOpenNc C sO | | _ t | t � s$ | | _ n d | _ | p6 d | _ | pE d | _ d S)NZOK�� )�url�
isinstance�str�req�msg�code)�selfr
r r � r �0/opt/alt/python35/lib64/python3.5/test_upload.py�__init__0 s zFakeOpen.__init__c C s d d i j | j � | � S)Nz content-typeztext/plain; charset=utf-8)�get�lower)r �name�defaultr r r � getheader9 s zFakeOpen.getheaderc C s d S)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)�__name__�
__module__� __qualname__r r r r r r r r r . s r c s| e Z d Z � f d d � Z � f d d � Z d d � Z d d � Z d d
� Z d d � Z d
d � Z d d � Z
� S)�uploadTestCasec sJ t t | � j � t j | _ | j t _ 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 _ t t | � j � d S)N)r& r$ r% r" r! �tearDown)r )r+ r r r, O s zuploadTestCase.tearDownc C s( t | d | j d | j �| _ | j S)Nr r )r r) r* r( )r r
r r r r' S s !zuploadTestCase._urlopenc C sf | j | j t � t � } t | � } | j � x- d
D]% \ } } | j t | | � | � q9 Wd S)N�username�me�password�secret�realm�pypi�
repository� https://upload.pypi.org/legacy/�r- r. �r/ r0 �r1 r2 �r3 r4 )r5 r6 r7 r8 )�
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 s{ | j | j t � t � } t | � } | j � | j | j d � d | _ t | � } | j � | j | j d � d S)N�xxx)r9 r: �PYPIRC_NOPASSWORDr r r; r< r/ )r r> r? r r r �test_saved_passwordc s
z"uploadTestCase.test_saved_passwordc C s� | j � } t j j | d � } | j | � d d | } } } | | | f g } | j | j t � | j d | � \ } } t | � } d | _ | j
� | j � t | j
j j � }
| j |
d d � |
d } | j | j d � � | j | j
j j � d � d
} | j | j
j j � | � | j d | j
j j k � | j d | j
j j � | j t � }
| j |
d d
d d d
d � d S)NrB z2.6�
dist_files� zContent-lengthZ2162z Content-typezmultipart/form-dataZPOSTz https://upload.pypi.org/legacy/s xxxs protocol_version�K �-z
xyzzy
���)�mkdtemp�os�path�joinr9 r: �PYPIRC_LONG_PASSWORD� create_distr Z
show_response�ensure_finalized�run�dictr( r �headersr< Z
assertTrue�
startswithZ
get_methodZ get_full_url�data�assertIn�get_logsr )r �tmprL ZcommandZ pyversion�filenamerE �pkg_dirr> r? rS Z content_typeZ expected_url�resultsr r r � test_uploadt s,
zuploadTestCase.test_uploadc C s) d | _ d | _ | j t | j � d S)Nz Not Foundi� )r) r* � assertRaisesr r\ )r r r r �test_upload_fails� s z uploadTestCase.test_upload_failsc C so | j � } t j j | d � } | j | � d d | f g } | j | j t � | j d | � \ } } t d � d t f t d d d i d � d t
f g } x� | D]� \ } } } | j d t | � j
� �� t j d
d t j d | � ��i | j | � �% t | � }
|
j � |
j � Wd QRX| j t � } | j | | d � | j � Wd QRXWd QRXq� Wd S)NrB z2.6rE Zoserrorr
i� Z httperrorz Upload failed (400): httperror� exceptionz distutils.command.upload.urlopen�newZ side_effectrF rI )rJ rK rL rM r9 r: rN rO �OSErrorr r ZsubTest�typer �mockZpatchZMockr] r rP rQ rW r rV Z
clear_logs) r rX rL rE rZ r> Ztestsr_ ZexpectedZraised_exceptionr? r[ r r r �test_wrong_exception_order� s(
z)uploadTestCase.test_wrong_exception_order) r r r r# r, r' rA rD r\ r^ rd r r )r+ r r! E s r! c C s
t j t � S)N)�unittestZ makeSuiter! r r r r �
test_suite� s rf �__main__) rK re Z
unittest.mockrc Zurllib.requestr Z test.supportr Zdistutils.commandr r$ Zdistutils.command.uploadZdistutils.corer Zdistutils.errorsr Z
distutils.logr r Zdistutils.tests.test_configr r
rN rC �objectr r! rf r r r r r �<module> s" l