Your IP : 216.73.216.213
�
i fm � @ s� d Z d d l Z d d l 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 Z d
Z Gd d � d e � Z Gd
d � d e � Z d d � Z e d k r� e e � � n d S)z#Tests for distutils.command.upload.� N)� run_unittest)�upload)� Distribution)�DistutilsError)�INFO)�PYPIRC�PyPIRCCommandTestCasez�[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 �@/opt/alt/python34/lib64/python3.4/distutils/tests/test_upload.py�__init__- s zFakeOpen.__init__c C s i d d 6j | j � | � S)Nztext/plain; charset=utf-8z content-type)�get�lower)r �name�defaultr r r � getheader6 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 �getcode> s zFakeOpen.getcode)�__name__�
__module__� __qualname__r r r r r r r r r + s r c sp 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 � 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! D s zuploadTestCase.setUpc s# | j t _ t t | � j � d S)N)r$ r" r# r r �tearDown)r )r) r r r* L s zuploadTestCase.tearDownc C s( t | d | j d | j �| _ | j S)Nr r )r r' r( r& )r r r r r r% P 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+ zme�zpasswordr. �r/ r0 �r1 r2 )r3 r4 r5 r6 )�
write_file�rcr r r �finalize_options� assertEqual�getattr)r �dist�cmd�attrZwaitedr r r �test_finalize_optionsT 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)r7 r8 �PYPIRC_NOPASSWORDr r r9 r: r- )r r< r= r r r �test_saved_password` s
z"uploadTestCase.test_saved_passwordc C sr | 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 t � }
| j d |
d
� d S)Nr@ z2.6�
dist_files� zContent-lengthZ2161z Content-typezmultipart/form-dataZPOSTz https://upload.pypi.org/legacy/s xxxzxyzzy
���)Zmkdtemp�os�path�joinr7 r8 �PYPIRC_LONG_PASSWORDZ create_distr Z
show_responseZensure_finalizedZrun�dictr& r �headersr: Z
assertTrue�
startswithZ
get_methodZ get_full_url�dataZget_logsr ZassertIn)r ZtmprG ZcommandZ pyversion�filenamerC Zpkg_dirr< r= rK Z content_typeZ expected_urlZresultsr r r � test_uploadq s*
zuploadTestCase.test_uploadc C s) d | _ d | _ | j t | j � d S)Nz Not Foundi� )r' r( Z assertRaisesr rO )r r r r �test_upload_fails� s z uploadTestCase.test_upload_fails)
r r r r! r* r% r? rB rO rP r r )r) r r B s r c C s
t j t � S)N)�unittestZ makeSuiter r r r r �
test_suite� s rR �__main__)�__doc__rF rQ Z test.supportr Zdistutils.commandr r" Zdistutils.command.uploadZdistutils.corer Zdistutils.errorsr Z
distutils.logr Zdistutils.tests.test_configr r rI rA �objectr r rR r r r r r �<module> s Q