Your IP : 216.73.216.213
��Yf(+ � @ si d d l Z d d l Z d d l Z d d l Z d d l Z d d l Z d d l Z d d l Z d d l Z d d l Z d d l
Z
d d l m Z d d l m
Z
m Z m Z m Z m Z m Z m Z m Z m Z m Z m Z d d l Z d d l Z d d l Z d d l m Z d d l m Z m Z m Z e d Z! y" d d l" Z" d d l# Z# d Z$ Wn e% k
rvd Z$ Yn Xd d
� Z& d d � Z' d d
d � Z( d d d � Z) d d � Z* Gd d � d e j+ � Z, Gd d � d e j+ � Z- Gd d � d e j+ � Z. Gd d � d e j+ � Z/ Gd d � d e j+ � Z0 Gd d � d e j+ � Z1 e2 d k ree j3 � d S) � N)� ExitStack) � make_archive�register_archive_format�unregister_archive_format�get_archive_formats�Error�unpack_archive�register_unpack_format�
RegistryError�unregister_unpack_format�get_unpack_formats�
SameFileError)�support)�TESTFN�check_warnings�captured_stdout�2TFc O s t t t d d � d � � d S)NZEXDEV� zInvalid cross-device link)�OSError�getattr�errno)�args�kwargs� r �5/opt/alt/python35/lib64/python3.5/test/test_shutil.py� _fake_rename% s r c s% t j � � � f d d � � } | S)Nc s4 z# t j } t t _ � | | � SWd | t _ Xd S)N)�os�renamer )r r Zbuiltin_rename)�funcr r �wrap* s
zmock_rename.<locals>.wrap)� functools�wraps)r r r )r r � mock_rename) s !r" c C sW t | t � r! t j j | � } t | | r3 d n d � � } | j | � Wd QRXd S)z�Write *content* to a file located at *path*.
If *path* is a tuple instead of a string, os.path.join will be used to
make a path. If *binary* is true, the file will be opened in binary
mode.
�wb�wN)�
isinstance�tupler �path�join�open�write)r' Zcontent�binary�fpr r r �
write_file4 s r- c C sT t | t � r! t j j | � } t | | r3 d n d � � } | j � SWd QRXd S)z�Return contents from a file located at *path*.
If *path* is a tuple instead of a string, os.path.join will be used to
make a path. If *binary* is true, the file will be opened in binary
mode.
�rb�rN)r% r& r r' r( r) �read)r' r+ r, r r r � read_file@ s r1 c C s� g } x� t t j | � � D]� } t j j | | � } t j j | � r� t j j | � r� | j | d � x9 t | � D] } | j | d | � qz Wq | j | � q W| S)N�/) �sortedr �listdirr' r( �isdir�islink�append�rlistdir)r' �res�name�p�nr r r r8 L s % r8 c s� e Z d Z � f d d � Z � f d d � Z d d � Z d d � Z e j d d
� � Z e j d d � � Z
d
d � Z e j
e e d � d � e j e j d d � d k d � e j e e d � o� e j � d k d � d d � � � � Z d d � Z d d � Z e j
e e d � d � e j d d � � � Z e j
e e d! � d" � e j d# d$ � � � Z e j e e d! � d% � e j d& d' � � � Z e j d( d) � � Z e j
e e d* � o�e e d+ � o�e e d, � d- � d. d/ � � Z e j d0 d1 � � Z e j e j e j
e e d � o>e j � d k d2 � d3 d4 � � � � Z e j d5 d6 � � Z e j d7 d8 � � Z! e j d9 d: � � Z" e j d; d< � � Z# d= d> � Z$ d? d@ � Z% dA dB � Z& e j dC dD � � Z' dE dF � Z( dG dH � Z) e j* j+ dI � dJ dK � � Z, e j e j- dL k dM � e j
e e dN � dO � dP dQ � � � Z. e j dR dS � � Z/ e j dT dU � � Z0 e j
e e dV � dW � dX dY � � Z1 e j
e e dV � dW � e j dZ d[ � � � Z2 d\ d] � Z3 e j d^ d_ � � Z4 e j d` da � � Z5 db dc � Z6 e j
e e d � d � dd de � � Z7 e j
e e d � d � e j
e e df � dg � dh di � � � Z8 e j9 dj dk � � Z: dl dm � Z; dn do dp � Z< e j9 e j
e= j> dq � dr � ds dt � � � Z? e j9 du dv � � Z@ e j9 e j
e= j> dw � dx � dy dz � � � ZA e j9 e j
e= j> d{ � d| � d} d~ � � � ZB d d� � ZC e j9 d� d� � � ZD e j9 e j
eE d� � d� d� � � � ZF d� d� � ZG d� d� � ZH e j9 d� d� � � ZI d� d� � ZJ d� d� � ZK d� d� � ZL e j9 d� d� � � ZM e jN d� d� � � ZO e jP d� d� � � ZQ e j9 d� d� � � ZR d� d� � ZS e j
e e= d� � d� � d� d� � � ZT e j
eE d� � e j
e e d� � d� � d� d� � � � ZU d� d� � ZV d� d� � ZW d� d� � ZX d� d� � ZY � S)��
TestShutilc s t t | � j � g | _ d S)N)�superr= �setUp�tempdirs)�self)� __class__r r r? [ s zTestShutil.setUpc sO t t | � j � x5 | j rJ | j j � } t j | t j d k � q Wd S)N�nt�cygwin)rC rD ) r> r= �tearDownr@ �pop�shutil�rmtreer r: )rA �d)rB r r rE _ s zTestShutil.tearDownc C s t j � } | j j | � | S)zjCreate a temporary directory that will be cleaned up.
Returns the path of the directory.
)�tempfile�mkdtempr@ r7 )rA rI r r r rK f s zTestShutil.mkdtempc C s� | j � } t j j | d � } t j | � t t j j | d � d � t j | � } | j | t � t j d k } | r� | j
t � n t � � t
j | � Wd QRXd S)NZkillmeZsomefile�foorC )rK r r' r( �mkdirr- �fsencode�assertIsInstance�bytesr: Z assertWarns�DeprecationWarningr rG rH )rA �tmpZvictim�winr r r �test_rmtree_works_on_byteso s
z%TestShutil.test_rmtree_works_on_bytesc s2 | j � } t j j | d � } t j | � t j j | d � } t j | | � | j t t j | � | j
t j j | � � | j
t j j | � � g � � f d d � } t j | d | �| j
t � � d � | j � d d t j j � | j
� d d | � | j � d d d t � d S) N�dir�linkc s � j | � d S)N)r7 )r )�errorsr r �onerror� s z8TestShutil.test_rmtree_fails_on_symlink.<locals>.onerrorrX � r � )rK r r' r( rM �symlink� assertRaisesr rG rH �
assertTrue�exists�lexists� assertEqual�len�assertIsr6 rO )rA rR Zdir_rV rX r )rW r � test_rmtree_fails_on_symlinkz s
z'TestShutil.test_rmtree_fails_on_symlinkc
C s_ | j � } t j j | d � } t j j | d � } t j j | d � } x$ | | | f D] } t j | � q[ Wt j j | d � } t | d � t j j | d � } t j | | � t j j | d � } t j | | � t j j | d � } t j | | � t j | � | j t j j
| � � | j t j j
| � � | j t j j
| � � d S) N�dir1�dir2�dir3�file1rL �link1�link2�link3) rK r r' r( rM r- r[ rG rH � assertFalser^ r] )
rA rR rd re rf rI rg rh ri rj r r r � test_rmtree_works_on_symlinks� s$
z(TestShutil.test_rmtree_works_on_symlinksc s& t j � } | j t t j | � t j | d d �| j � } t | d f d � t j j
| d � } | j t � � } t j | � Wd QRX| t j j
| d � g } | j | j
j | � | j t j j | � � t j | d d �| j t j j | � � g � � f d d � } t j | d | �| j t � � d � | j � d
d
t j � | j � d
d | � | j � d
d d t � | j � d
d d j | � | j � d d
t j � | j � d d | � | j � d d d t � | j � d d d j | � d S) N�
ignore_errorsTZtstfile� z*.*c s � j | � d S)N)r7 )r )rW r r rX � s z.TestShutil.test_rmtree_errors.<locals>.onerrorrX rZ r rY )rJ �mktempr\ �FileNotFoundErrorrG rH rK r- r r' r( �NotADirectoryError�assertInZ exception�filenamer] r^ r` ra rb r4 rO �rmdir)rA rs �tmpdir�cmZ
possible_argsrX r )rW r �test_rmtree_errors� s2 z TestShutil.test_rmtree_errors�chmodzrequires os.chmod()N� rD z2This test can't be run on Cygwin (issue #1071513).�geteuidr z9This test can't be run reliably as root (issue #1076467).c C ss d | _ t j t � | j t j t � t j j t d � | _ t j j t d � | _
t j | j � t j | j
� t j
t � j } t j
| j � j } t j
| j
� j } t
j t
j B} t j | j | � t j | j
| � t j t | � | j t j t | � | j t j | j | � | j t j | j
| � t j t d | j �| j | j d d � d S)Nr �a�brX � z1Expected call to onerror function did not happen.)�
errorStater rM r �
addCleanuprG rH r' r( �child_file_path�child_dir_pathr �create_empty_file�stat�st_mode�S_IREAD�S_IEXECrx �check_args_to_onerrorr` )rA Z old_dir_modeZold_child_file_modeZold_child_dir_modeZnew_moder r r �
test_on_error� s(
zTestShutil.test_on_errorc C s� | j d k r� | t j k r4 | j | | j � nQ | t j k rY | j | | j � n, | j | t j � | j | t
| j g � | j t | d t
� � | j d 7_ nF | j | t j � | j | t
� | j t | d t
� � d | _ d S)NrZ r rY r} )r~ r �unlinkr` r� rt r� rb r4 rr r r] �
issubclassr )rA r �arg�excr r r r� � s z TestShutil.check_args_to_onerrorc sf zU t j � � f d d � } | t _ t j t � t t d f d � t j t � Wd � t _ Xd S)Nc s& | t k r t � � n
� | � Sd S)N)r r )�fnr r )�
orig_lstatr r �raiser s zFTestShutil.test_rmtree_does_not_choke_on_failing_lstat.<locals>.raiserrL )r �lstatrM r r- rG rH )rA r� r )r� r �+test_rmtree_does_not_choke_on_failing_lstat s
z6TestShutil.test_rmtree_does_not_choke_on_failing_lstatzrequires os.chmodc C s | j � } t j j | d � } t j j | d � } t j j | d � } t j j | d � } t | d � t | d � t j | | � t j | | � t j | t j t j B� t j | t j
� | j t j | � j t j | � j � t
j | | � | j t j | � j t j | � j � t j d k rt j | t j
� t
j | | � | j t j | � j t j | � j � t j | t j
� t
j | | � | j t j | � j t j | � j � t j | t j
� t
j | | � | j t j | � j t j | � j � d S)NrL �bar�baz�quuxrC )rK r r' r( r- r[ rx r� �S_IRWXU�S_IRWXG�S_IRWXO�assertNotEqualr� rG �copymoder` r: )rA �tmp_dir�src�dst�src_link�dst_linkr r r � test_copymode_follow_symlinks s0
((((z(TestShutil.test_copymode_follow_symlinks�lchmodzrequires os.lchmodc C s | j � } t j j | d � } t j j | d � } t j j | d � } t j j | d � } t | d � t | d � t j | | � t j | | � t j | t j t j B� t j | t j � t j
| t j t j B� t j
| t j � t j
| | d d �| j t j | � j t j | � j � | j t j | � j t j | � j � t j
| t j � t j
| | d d �| j t j | � j t j | � j � t j
| t j � t j
| | d d �| j t j | � j t j | � j � d S)NrL r� r� r� �follow_symlinksF)rK r r' r( r- r[ rx r� r� r� r� r� rG r� r` r� r� r� )rA r� r� r� r� r� r r r � test_copymode_symlink_to_symlink3 s.
((z+TestShutil.test_copymode_symlink_to_symlinkz requires os.lchmod to be missingc C s� | j � } t j j | d � } t j j | d � } t j j | d � } t j j | d � } t | d � t | d � t j | | � t j | | � t j | | d d �d S)NrL r� r� r� r� F)rK r r' r( r- r[ rG r� )rA r� r� r� r� r� r r r �*test_copymode_symlink_to_symlink_wo_lchmodQ s
z5TestShutil.test_copymode_symlink_to_symlink_wo_lchmodc
C s� | j � } t j j | d � } t j j | d � } t j j | d � } t j j | d � } t | d � t j | � } t j | | j | j d f � t | d � | j t j | � j t j | � j � t j
| | � t j
| | � t t d � rt j | t j
� t t d � rDt t d � rDt j | t j � t j | � } t t d � r�t j | | d d
�| j | j t j | � j � t j | | d d �t j | � } t j t j k rx4 d D], } | j t | | � t | | � d � q�Wt t d � r*| j | j | j � t t d � r^t | d � r^| j | j | j � t j | | d d �| j t t j | � j t j | � j � d k � d S)NrL r� r� �quxg E@r� �lchflags� UF_NODUMPr� TF�st_atime�st_mtimerY �st_flagsg�������?)r� r� )rK r r' r( r- r� �utimer� r� r� r[ �hasattrr� r� r� r� r� rG �copystatr� �supports_follow_symlinks�assertLessEqualr r` r� r] �abs)
rA r� r� r� r� r� �src_stat�
src_link_statZ
dst_link_stat�attrr r r �test_copystat_symlinks_ sD
(
+z!TestShutil.test_copystat_symlinks�chflags�
EOPNOTSUPP�ENOTSUPz)requires os.chflags, EOPNOTSUPP & ENOTSUPc C s� | j � } t j j | d � } t j j | d � } t | d � t | d � d d � } t j } zr x9 t j t j f D]% } | | � t _ t j
| | � q{ W| t j t j � t _ | j t t j
| | � Wd | t _ Xd S)Nrg �file2�xxxc s( t � � d d � � f d d � } | S)Nr� Tc s � � _ � � d S)N)r )r' �flagsr� )�err�exr r �_chflags_raiser� s znTestShutil.test_copystat_handles_harmless_chflags_errors.<locals>.make_chflags_raiser.<locals>._chflags_raiser)r )r� r� r )r� r� r �make_chflags_raiser� s zUTestShutil.test_copystat_handles_harmless_chflags_errors.<locals>.make_chflags_raiser)
rK r r' r( r- r� r r� r� rG r� r\ r )rA ru rg r� r� Z old_chflagsr� r r r �-test_copystat_handles_harmless_chflags_errors� s
z8TestShutil.test_copystat_handles_harmless_chflags_errorsc sN | j � } t j j | d � � t � d � t j j | d � } t | d � t j � | � t j � d d � t j � d d � t j � | � | j t t j
� � � t t j
| � � � | j t j � d � t j | d � � t j | � t | d � t
t j d � � � � f d d � } z? t j � | t _ t j � | � | j d t j
| � � Wd � t _ Xd
d � � f d d
� } z& t j
� | t _
t j � | � Wd � t _
Xt j j | d � � t � � � t j � d d � t j j | d � } t | | � t j � | � | j t j | d � d � d S)NrL r� zuser.foos 42zuser.bars 43�EPERMc s, | d k r � � n � | | | | � d S)Nzuser.foor )�fnamer� �valr )�
orig_setxattr�os_errorr r �_raise_on_user_foo� s z5TestShutil.test_copyxattr.<locals>._raise_on_user_foor� Tc s. | � k r t t j d � � � | d | �S)NzOperation not supportedr� )r r r� )r� r� )�orig_listxattrr� r r �
_raise_on_src� s z0TestShutil.test_copyxattr.<locals>._raise_on_srcZ the_originalzuser.the_values fiddlyZthe_copy)rK r r' r( r- rG �
_copyxattr�setxattrr` r3 � listxattr�getxattr�remover r r� rr r� )rA r� r� r� r� r )r� r� r� r� r �test_copyxattr� sH
.
zTestShutil.test_copyxattrzroot privileges requiredc C sM | j � } t j j | d � } t j j | d � } t | d � t j | | � t j | d d � t j | d d d d �t j j | d � } t j j | d � } t | d � t j | | � t j | | d d �| j t j
| d d d �d � | j t t j
| d � t j | | d d �| j t j
| d � d � d S)
NrL r� z trusted.foos 42s 43r� Fr� r� )
rK r r' r( r- r[ r� rG r� r` r� r\ r )rA r� r� r� r� r� r r r �test_copyxattr_symlinks� s
"z"TestShutil.test_copyxattr_symlinksc C su | j � } t j j | d � } t j j | d � } t j j | d � } t | d � t j | | � t t d � r� t j | t j t j
B� t j | | d d �| j
t j j | � � | j t | � t | � � t j | � t j | | d d �| j t j j | � � | j t j | � t j | � � t t d � rq| j t j | � j t j | � j � d S)NrL r� r� r� r� TF)rK r r' r( r- r[ r� r� r� r� r� rG �copyrk r6 r` r1 r� r] �readlinkr� r� )rA r� r� r� r� r r r �test_copy_symlinks� s$
"z TestShutil.test_copy_symlinksc C sT | j � } t j j | d � } t j j | d � } t j j | d � } t | d � t j | | � t t d � r� t j | t j t j
B� t t d � r� t t d � r� t j | t j � t j | � } t j
| � } t j | | d d �| j t j j | � � | j t | � t | � � t j | � t j | | d d �| j t j j | � � | j t j | � t j | � � t j
| � } t j t j k r�x4 d D], } | j t | | � t | | � d � q�Wt t d � r | j | j | j � | j | j | j � t t d � rPt | d
� rP| j | j | j � d S)NrL r� r� r� r� r� r� TFr� r� rY r� )r� r� ) rK r r' r( r- r[ r� r� r� r� r� r� r� r� rG �copy2rk r6 r` r1 r� r] r� r� r� r� r r� r� r� ) rA r� r� r� r� r� r� �dst_statr� r r r �test_copy2_symlinks
s:
"
z TestShutil.test_copy2_symlinksc C s� | j � } t j j | d � } t j j | d � } t | d � t j | d d � t j | | � | j t j | d � t j | d � � t j
| � d S)NrL r� zuser.foos 42) rK r r' r( r- r� rG r� r` r� r� )rA r� r� r� r r r �test_copy2_xattr0 s
zTestShutil.test_copy2_xattrc C s� | j � } t j j | d � } t j j | d � } t j j | d � } t j j | d � } t | d � t j | | � t j | | d d �| j t j j | � � | j
t j | � t j | � � t j | | � | j t j j | � � d S)Nr� r� r� rV rL r� F)
rK r r' r( r- r[ rG �copyfiler] r6 r` r� rk )rA r� r� r� r� rV r r r �test_copyfile_symlinks= s
"z!TestShutil.test_copyfile_symlinksc s- t j t j t j t j h t j k oH t j t j k oH t j t j k } | r| j t
j � | j t
j j
� | j � } t j j | d � } t j | � zT t
j } Gd d � d t � � � f d d � } | t
_ | j � t
j | � Wd | t
_ Xn# | j t
j � | j t
j j
� d S)Nr{ c @ s e Z d Z d S)zHTestShutil.test_rmtree_uses_safe_fd_version_if_available.<locals>.CalledN)�__name__�
__module__� __qualname__r r r r �Called[ s r� c s
� � d S)Nr )r r )r� r r �_raiser\ s zITestShutil.test_rmtree_uses_safe_fd_version_if_available.<locals>._raiser)r r) r� r� rt �supports_dir_fdr4 � supports_fdr� r] rG �_use_fd_functionsrH Zavoids_symlink_attacksrK r' r( rM Z_rmtree_safe_fd� Exceptionr\ rk )rA r� r� rI Z real_rmtreer� r )r� r �-test_rmtree_uses_safe_fd_version_if_availableN s&
z8TestShutil.test_rmtree_uses_safe_fd_version_if_availablec C sF t j � \ } } t j | � | j t t j | � t j | � d S)N) rJ Zmkstempr �closer\ rq rG rH r� )rA Zhandler' r r r � test_rmtree_dont_delete_filef s
z'TestShutil.test_rmtree_dont_delete_filec C st t j � } t j j t j � d � } | j t j | � | j t j t j j | � � t | d f d � t j
t j j | d � � t | d d f d � t j | | � | j t j j
t j j | d � � � | j t j j t j j | d � � � | j t j j
t j j | d d � � � t | d f � } | j | d � t | d d f � } | j | d � d S)N� destinationztest.txt�123�test_dir�456)rJ rK r r' r( r rG rH �dirnamer- rM �copytreer] �isfiler5 r1 r` )rA �src_dir�dst_dir�actualr r r �test_copytree_simplem s ((
z TestShutil.test_copytree_simplec C s | j � } t j j | d � } t j j | d � } t j j | d � } t j | � t j | � t | d f d � t j j | d � } t j j | d � } t j t j j | d � | � t t d � r� t j | t j
t j B� t t d � rt t d
� rt j | t j
� t j | � } t j | | d d �| j t j j t j j | d d � � � | j t j t j j | d d � � t j j | d � � t j | � } t t d � r�| j | j | j � t t d � r�| j | j | j � d S)
Nr� r� �subzfile.txtrL rV zsub/linkr� r� r� �symlinksT)rK r r' r( rM r- r[ r� r� r� r� r� r� r� r� rG r� r] r6 r` r� r� r� ) rA r� r� r� Zsub_dirr� r� r� r� r r r �test_copytree_symlinks� s2
+$z!TestShutil.test_copytree_symlinksc # C s� t j j } t j j } t j � } z�| t j � d � } t | d f d � t | d f d � t j | | d � � t | d d f d � t j | | d � � t | d d f d � t j | | d d � � t j | | d d � � t | d d d f d � t | d d d
f d � z� t j d d � } t j
| | d | �| j | | | d � � � | j | | | d � � � | j | | | d � � � Wd t j
| � Xz� t j d d
� } t j
| | d | �| j | | | d � � � | j | | | d d � � � | j | | | d d � � � Wd t j
| � Xzg d d � } t j
| | d | �| j | | | d d d
� � � | j | | | d d � � � Wd t j
| � XWd t j
| � t j
t j j | � � Xd S)Nr� ztest.txtr� ztest.tmpr� r� Z test_dir2�subdirZsubdir2ztest.pyz*.tmp�ignorezsubdir*c S s� g } x� | D]| } t j j | | � } t j j | � r` | j � d d k r` | j | � q
t j j | � d d k r
| j | � q
W| S)NrY r� z.py���r� )r r' r( r5 �splitr7 �splitext)r� �namesr9 r: r' r r r �_filter� s
z6TestShutil.test_copytree_with_exclude.<locals>._filter)r r' r( r^ rJ rK r- rM rG �ignore_patternsr� r] rk rH r� )rA r( r^ r� r� Zpatternsr� r r r �test_copytree_with_exclude� sL #
#
z%TestShutil.test_copytree_with_excludec C s t j � } t j j | d � } t j | � t j j | d � } | j t j | � t j | d � t
| d f d � t j t j j | d � d � t
| d f d � t j t j j | d � d � t j d | � } t j | d � t j | | � | j t j
| � j t j
| � j � | j t j
t j j | d � � j t j
t j j | d � � j � | j t j
t j j | d � � j t j
t j j | d � � j � t j j | t j j | � d
� } | j t j
| � j t j
| � j � d S) N�sourcer� i� zpermissive.txtr� zrestrictive.txtr� i� rU rY )rJ rK r r' r( rM r rG rH rx r- r� r` r� r� r� )rA r� r� r� Zrestrictive_subdirZrestrictive_subdir_dstr r r �!test_copytree_retains_permissions� s,
($"$" z,TestShutil.test_copytree_retains_permissionszos.chmodc
C s� t j � } t j j t j � d � } | j t j | � | j t j t j j | � � t d � | _
| j t j � � t j
| | � Wd QRXd S)Nr� zka-boom)rJ rK r r' r( r rG rH r� �PermissionErrorZ side_effectr\ r r� )rA Z
mock_patchr� r� r r r �test_copytree_winerror� s z!TestShutil.test_copytree_winerrorrC z temporarily disabled on WindowsrV zrequires os.linkc C s� t j t � t j j t d � } t j j t d � } z� t | d � � } | j d � Wd QRXt j | | � | j t j
t j | | � t | d � � } | j | j
� d � Wd QRXt j | � Wd t j t d d �Xd S)N�cheese�shopr$ �cheddarr/ rm T)r rM r r' r( r) r* rV r\ rG r
r� r` r0 r� rH )rA r� r� �fr r r �'test_dont_copy_file_onto_link_to_itself s
z2TestShutil.test_dont_copy_file_onto_link_to_itselfc C s� t j t � t j j t d � } t j j t d � } z� t | d � � } | j d � Wd QRXt j d | � | j t j
t j | | � t | d � � } | j | j
� d � Wd QRXt j | � Wd t j t d d �Xd S)Nr� r� r$ r r/ rm T)r rM r r' r( r) r* r[ r\ rG r
r� r` r0 r� rH )rA r� r� r r r r �*test_dont_copy_file_onto_symlink_to_itself s
z5TestShutil.test_dont_copy_file_onto_symlink_to_itselfc
C s� t j t � zt t j j t d � } t j j t d � } t j | � t j | | � | j t t j | � t j | d d �Wd t j t d d �Xd S)Nr� r� rm T)
r rM r r' r( r[ r\ r rG rH )rA r� r� r r r �test_rmtree_on_symlink+ s
z!TestShutil.test_rmtree_on_symlink�mkfifozrequires os.mkfifo()c C s^ t j t � z< | j t j t j t t � | j t j t j t t � Wd t j t � Xd S)N)
r r r r\ rG �SpecialFileErrorr� �TESTFN2�__file__r� )rA r r r �test_copyfile_named_pipe: s
z#TestShutil.test_copyfile_named_pipec C s t j t � z� t j j t d � } t j | � t j j | d � } t j | � y t j t t � Wnr t j k
r� } zO | j
d } | j t | � d � | d \ } } } | j d | | � WYd d } ~ Xn X| j
d � Wd t j t d d �t j t d d �Xd S) Nr� Zmypiper rY z`%s` is a named pipez$shutil.Error should have been raisedrm T)r rM r r' r( r rG r� r r r r` ra �failrH )rA r� �pipe�erW r� r� Z error_msgr r r �test_copytree_named_pipeE s
'z#TestShutil.test_copytree_named_pipec s� | j � } t j j | j � d � } t | d f d � t j t j j | d � � t | d d f d � g � � f d d � } t j | | d | �| j t � � d � d S)
Nr� ztest.txtr� r� r� c s � j | | f � d S)N)r7 )r� r� )�copiedr r �_copyd s z4TestShutil.test_copytree_special_func.<locals>._copy�
copy_functionrZ )
rK r r' r( r- rM rG r� r` ra )rA r� r� r r )r r �test_copytree_special_func[ s z%TestShutil.test_copytree_special_funcc C s) | j � } t j j | j � d � } t j d t j j | d � � t j t j j | d � � t | d d f d � | j t t j
| | � t j j | j � d � } t j
| | d d �| j d t j | � � t j j | j � d � } t j
| | d
d �| j
d t j | � � d S) Nr� Z
IDONTEXISTztest.txtr� r� � destination2Zignore_dangling_symlinksTZ destination3r� )rK r r' r( r[ rM r- r\ r rG r� � assertNotInr4 rr )rA r� r� r r r � test_copytree_dangling_symlinksj s z*TestShutil.test_copytree_dangling_symlinksc
C s� | j � } t j j | j � d � } t j t j j | d � � t t j j | d d � d � � Wd QRXt j t j j | d � t j j | d � d d �t j | | d d �| j t j j
t j j | d � � � | j d t j t j j | d � � � t j j | j � d
� } t j | | d d �| j
t j j
t j j | d � � � | j d t j t j j | d � � � d S) Nr� Zreal_dirztest.txtr$ Z link_to_dirZtarget_is_directoryTr� Fr )rK r r' r( rM r) r[ rG r� rk r6 rr r4 r] )rA r� r� r r r �test_copytree_symlink_dir s "(((z$TestShutil.test_copytree_symlink_dirc C sr d } | j � } t | | f d � t j j | | � } | j � } | | | � t j j | | � } | | f S)Nztest.txtr� )rK r- r r' r( )rA �methodr� ru rg �tmpdir2r� r r r �
_copy_file� s
zTestShutil._copy_filec C s] | j t j � \ } } | j t j j | � � | j t j | � j t j | � j � d S)N)
r rG r� r] r r' r^ r` r� r� )rA rg r� r r r � test_copy� s zTestShutil.test_copyr� zrequires os.utimec C s� | j t j � \ } } | j t j j | � � t j | � } t j | � } | j | j | j � x4 d D], } | j
t | | � t | | � d � ql Wt t d � r� t | d � r� | j t | d � t | d � � d S)Nr� r� rY r� r� )r� r� )
r rG r� r] r r' r^ r� r` r� r� r r� )rA rg r� Z
file1_statZ
file2_statr� r r r �
test_copy2� s
zTestShutil.test_copy2c $ C s� | j d � \ } } | j � } t j | � t j j | � } t j j t j j | � d � } t j | � �, t j j
| � } t | d | d � } Wd QRX| j | | d � | j
t j j | � � | j
t j | � � t j | d � �/ } | j | j � d d d d d
d g � Wd QRXt j | � � t | d | d � } Wd QRX| j | | d
� | j
t j j | � � | j
t j | � � t j | d � �/ } | j | j � d d d d d
d g � Wd QRXd S)Nrn �archive�gztar�.z.tar.gzzr:gzz./subz./sub2z./file1z./file2z ./sub/file3�tarz.tarr/ )�
_create_filesrK r rt r' r� r( �basenamer �
change_cwd�abspathr r` r] r� �tarfileZ
is_tarfiler) �assertCountEqual�getnames) rA �root_dir�base_dirr �work_dir�
rel_base_name� base_name�tarballZtfr r r �test_make_tarball� s0
! z TestShutil.test_make_tarballc C s= t j | � �' } | j � } | j � t | � SWd QRXd S)N)r# r) r% �sortr& )rA r' r r� r r r �_tarinfo� s
zTestShutil._tarinfo�distc C s� | j � } t j j | | � } t j | d d �t | d f d � t | d f d � t j t j j | d � � t | d d f d � t j t j j | d � � | r� t | d f d � | | f S)
N�exist_okTrg r� r� r� Zfile3Zsub2�outer)rK r r' r( �makedirsr- rM )rA r' r& r/ r r r r � s zTestShutil._create_filesr zNeed the tar command to runc C s{ | j � \ } } t j j | j � d � } t | d | | � } | j | | d � | j t j j | � � t j j | d � } d d d | g } t j
| d | d t j �| j t j j | � � | j | j | � | j | � � t | d | | � } | j | | d � | j t j j | � � t | d | | d
d �} | j | | d � | j t j j | � � d S) Nr r z.tar.gzz archive2.tarr z-cf�cwd�stdoutz.tarZdry_runT)
r r r' r( rK r r` r] r� �
subprocess�
check_call�DEVNULLr. )rA r& r' r* r+ Ztarball2Ztar_cmdr r r �test_tarfile_vs_tar� s$
" z TestShutil.test_tarfile_vs_tarc $ C s� | j � \ } } | j � } t j | � t j j | � } t j j t j j | � d � } t j | � �) t j j
| � } t | d | � } Wd QRX| j | | d � | j
t j j | � � | j
t j | � � t j | � �2 } | j | j � d d d d d d d
g � Wd QRXt j | � �, t j j
| � } t | d | | � } Wd QRX| j | | d � | j
t j j | � � | j
t j | � � t j | � �/ } | j | j � d d d d d d g � Wd QRXd S) Nr �zipz.zipzdist/z dist/sub/z
dist/sub2/z
dist/file1z
dist/file2zdist/sub/file3r1 )r rK r rt r' r� r( r r r! r" r r` r] r� �zipfileZ
is_zipfile�ZipFiler$ �namelist) rA r& r' r r( r) r* r9 �zfr r r �test_make_zipfile s4
! z TestShutil.test_make_zipfiler9 zNeed the zip command to runc
C s8 | j � \ } } t j j | j � d � } t | d | | � } | j | | d � | j t j j | � � t j j | d � } d d d d | g } t j
| d | d t j �| j t j j | � � t j
| � � } | j � } Wd QRXt j
| � � } | j � } Wd QRX| j t | � t | � � d S) Nr r9 z.zipz archive2.zipz-qz-rr3 r4 )r r r' r( rK r r` r] r� r5 r6 r7 r: r; r< r3 )
rA r& r' r* r Zarchive2�zip_cmdr= r� Znames2r r r �test_zipfile_vs_zip4 s
z TestShutil.test_zipfile_vs_zip�unzipz Need the unzip command to runc C s | j � \ } } t j j | j � d � } t | d | | � } | j | | d � | j t j j | � � d d | g } t j
| � �~ y t j | d t j
�Wn\ t j k
r} z9 | j j d d � } d } | j | j | | � � WYd d } ~ Xn XWd QRXd S)
Nr r9 z.ziprA z-t�stderrrW �replacez{}
**Unzip Output**
{})r r r' r( rK r r` r] r� r r! r5 � check_outputZSTDOUT�CalledProcessError�output�decoder
�format) rA r& r' r* r r? r� Zdetails�msgr r r �test_unzip_zipfileN s z TestShutil.test_unzip_zipfilec C s; | j � } t j j | d � } | j t t | d � d S)Nr r� )rK r r' r( r\ �
ValueErrorr )rA ru r* r r r �test_make_archived s z TestShutil.test_make_archivec C sF t r/ t j d � d } t j d � d } n
d } } | j � \ } } t j j | j � d � } t
| d | | d | d | �} | j t j j | � � t
| d | | � } | j t j j | � � t
| d | | d | d | �} | j t j j | � � t
| d | | d d d d �} | j t j j | � � d S)
Nr �rootr r9 �owner�groupr Z kjhkjhkjgZoihohoh)
�UID_GID_SUPPORT�grp�getgrgid�pwd�getpwuidr r r' r( rK r r] r� )rA rO rN r& r' r* r9 r r r � test_make_archive_owner_groupi s"
z(TestShutil.test_make_archive_owner_groupz Requires grp and pwd supportc C s | j � \ } } t j j | j � d � } t j d � d } t j d � d } t j
| � �& t | d | d d | d | �} Wd QRX| j t j j
| � � t j | � } zA x: | j � D], } | j | j d � | j | j d � q� WWd | j � Xd S)Nr r r r/ rN rO )r r r' r( rK rQ rR rS rT r r! r r] r� r# r) Z
getmembersr` �uid�gidr� ) rA r& r' r* rO rN Z archive_namer Zmemberr r r �test_tarfile_root_owner� s z"TestShutil.test_tarfile_root_ownerc C s� t j � } d d � } t d | g d � zL y t d d d | j � �Wn t k
r_ Yn X| j t j � | � Wd t d � Xd S)Nc _ s
t � � d S)N)� RuntimeError)r �kwr r r �_breaks� s z1TestShutil.test_make_archive_cwd.<locals>._breaksr� zxxx filer& )r �getcwdr r rK r� r` r )rA Z current_dirr[ r r r �test_make_archive_cwd� s
z TestShutil.test_make_archive_cwdc C sY | j � } t j | � �7 | j t d d � d � | j t j j d � � Wd QRXd S)N�testr ztest.tar) rK r r! r` r r] r r' r� )rA r& r r r �test_make_tarfile_in_curdir� s z&TestShutil.test_make_tarfile_in_curdirc C sY | j � } t j | � �7 | j t d d � d � | j t j j d � � Wd QRXd S)Nr^ r9 ztest.zip) rK r r! r` r r] r r' r� )rA r& r r r �test_make_zipfile_in_curdir� s z&TestShutil.test_make_zipfile_in_curdirc C s� | j t t d d � | j t t d d d � d � | j t t d d d � d
d g � t d d d � d g d � d
d � t � D� } | j d | � t d � d d � t � D� } | j d | � d S)Nr� rY c S s t S)N)�xr r r r �<lambda>� s z9TestShutil.test_register_archive_format.<locals>.<lambda>c S s t S)N)ra r r r r rb � s rZ r} c S s t S)N)ra r r r r rb � s zxxx filec S s g | ] \ } } | � q Sr r )�.0r: �paramsr r r �
<listcomp>� s z;TestShutil.test_register_archive_format.<locals>.<listcomp>c S s g | ] \ } } | � q Sr r )rc r: rd r r r re � s )rY rZ )rY rZ r} )rY rZ )r\ � TypeErrorr r rr r r )rA �formatsr r r � test_register_archive_format� s
z'TestShutil.test_register_archive_formatc C s� | j � \ } } t | � } | j d � t j j | j � d � } t | | | | � } | j � } t | | � | j t | � | � | j � } t | | d | �| j t | � | � | j
t j t t
� | j
t t t
d d �d S)Nr1 r rH r� )r r8 r� r r' r( rK r r r` r\ rG Z ReadErrorr rK ) rA rH r& r' �expectedr* rs r Ztmpdir3r r r �check_unpack_archive� s
z TestShutil.check_unpack_archivec C s | j d � d S)Nr )rj )rA r r r �test_unpack_archive_tar� s z"TestShutil.test_unpack_archive_tarc C s | j d � d S)Nr )rj )rA r r r �test_unpack_archive_gztar� s z$TestShutil.test_unpack_archive_gztarc C s | j d � d S)NZbztar)rj )rA r r r �test_unpack_archive_bztar� s z$TestShutil.test_unpack_archive_bztarc C s | j d � d S)NZxztar)rj )rA r r r �test_unpack_archive_xztar� s z$TestShutil.test_unpack_archive_xztarc C s | j d � d S)Nr9 )rj )rA r r r �test_unpack_archive_zip� s z"TestShutil.test_unpack_archive_zipc s� t � } � f d d � } t d d d g | d g � t d d � � j t t d
d g | � t d � t d
d g | � � j d
d g d f t � � � j d d g d f t � � t d
� � j t � | � d S)
Nc s4 � j | d � � j | d � � j | d � d S)NrY z stuff.boo�xx)r` )rs Z extract_dir�extra)rA r r �_boo� s z-TestShutil.test_unpack_registry.<locals>._booZBooz.booz.b2rq rY z stuff.boorp ZBoo2rn )rq rY ) r r r r\ r
r rr r r` )rA rg rr r )rA r �test_unpack_registry� s
z TestShutil.test_unpack_registry�
disk_usagez)disk_usage not available on this platformc C s~ t j t j � � } | j | j d � | j | j d � | j | j d � | j | j | j � | j | j | j � d S)Nr ) rG rt r r\ Z
assertGreaterZtotalZused�assertGreaterEqualZfree)rA Zusager r r �test_disk_usage s zTestShutil.test_disk_usage�chownzrequires os.chownc ( s� � j � } t j d | � � t � d � � j t � � t j � � Wd QRX� j t � � t j � d d �Wd QRX� j t � � t j � d d �Wd QRX� j t � � t j � d � Wd QRX� j t � � t j � d � Wd QRXt
j � } t
j � } d d � � f d d
� } t j � | | � | � | | � t j � | � | � | � t j � d | �| � | � t j � d | �| � d | �t j | | | � | | | | � t j | | � | | | � t j | d | �| | | � t j | d | �| | d | �t
j | � d } t j | � d } t j � | | � | � | | � t j | | | � | | | | � d S)
NrU ztesting chown function�userznon-exising usernamerO znon-exising groupnames spamg ��Q� @c sQ t j � � } | d k r. � j | | j � | d k rM � j | | j � d S)N)r r� r` �st_uid�st_gid)r' rV rW �s)rs rA r r � check_chown1 s
z*TestShutil.test_chown.<locals>.check_chownrW r )rK rJ ro r- r\ rK rG rw � LookupErrorrf r �getuid�getgidrS rT rQ rR )rA r� rV rW r| rx rO r )rs rA r �
test_chown sL
zTestShutil.test_chownc C s� x� t j t j f D]� } | j � } | j � } t j j | d � } t | d � | | | � } | j | t j j | d � � | | t j j | d � � } | j | t j j | d � � q Wd S)NrL r� ) rG r� r� rK r r' r( r- r` )rA r� r� r� r� �rvr r r �test_copy_return_valueQ s
z!TestShutil.test_copy_return_valuec C s� | j � } | j � } t j j | d � } t j j | d � } t | d � t j | | � } | j t j j | � � | j t
| � t
| � � d S)Nr� rL ) rK r r' r( r- rG r� r] r^ r` r1 )rA r� r� �dst_file�src_filer� r r r �test_copyfile_return_value] s
z%TestShutil.test_copyfile_return_valuec C sd | j � } t j j | d � } t | d � | j t t j | | � | j t t j | | � d S)NrL )
rK r r' r( r- r\ r
rG r� r )rA r� r� r r r �test_copyfile_same_fileh s
z"TestShutil.test_copyfile_same_filec C s� | j � } | d } | j t j | d � t j j | d � } t | d � t j | | � } | j d g t j
| � � d S)N�destTrL ) rK r rG rH r r' r( r- r� r` r4 )rA r� r� r� r� r r r �test_copytree_return_valuer s
z%TestShutil.test_copytree_return_value)Zr� r� r� r? rE rK rT r �skip_unless_symlinkrc rl rw �unittest�
skipUnlessr� r �skipIf�sys�platformrz r� r� r� r� r� r� r� r r� Zskip_unless_xattrr� r� r� r� r� r� r� r� r� r� r� r� ZmockZpatchr� r: r r r r r
r r r r r r Z
requires_zlibr, r. r rG �whichr8 r> r@ rJ rL rU rP rX r] r_ r` rh rj rk rl Z requires_bz2rm Z
requires_lzmarn ro rs rv r� r� r� r� r� r r )rB r r= Y s� $ $
) 8$#
A *'
'*$ % ! *:
r= c @ s� e Z d Z d d � Z d d � Z d d � Z d d � Z d d
� Z e j e
e d � ol e j � d k d
� d d � � Z
d d � Z d d � Z e j e j d k d � d d � � Z d d � Z d d � Z d d � Z d S) � TestWhichc C s� t j d d � | _ | j t j | j d � t j d | j d d d d � | _ t j | j j
t j � | j | j j
� t j j | j j
� \ | _ | _ d S)N�prefixZTmpTrU �suffixz.Exe)rJ rK �temp_dirr rG rH ZNamedTemporaryFile� temp_filer rx r: r� �S_IXUSRr� r' r� rU �file)rA r r r r? s zTestWhich.setUpc C s5 t j | j d | j �} | j | | j j � d S)Nr' )rG r� r� rU r` r� r: )rA r� r r r �
test_basic� s zTestWhich.test_basicc C s8 t j | j j d | j �} | j | | j j � d S)Nr' )rG r� r� r: r� r` )rA r� r r r �test_absolute_cmd� s zTestWhich.test_absolute_cmdc C s� t j j | j � \ } } t j j | | j � } t j d | � �- t j | d | j
�} | j | | � Wd QRXt j d | j � �' t j | d | �} | j | � Wd QRXd S)Nr' )
r r' r� rU r( r� r r! rG r� r� r` � assertIsNone)rA r' �tail_dir�relpathr� r r r �test_relative_cmd� s zTestWhich.test_relative_cmdc C s� t j j | j � } t j d | j � �a t j | j d | �} t j
d k rz | j | t j j t j
| j � � n
| j | � Wd QRXd S)Nr' �win32)r r' r� rU r r! rG r� r� r� r� r` r( �curdirr� )rA r' r� r r r �test_cwd� s (zTestWhich.test_cwdrz r znon-root user requiredc C sv t j | j j t j � t j | j j t j � rA | j d � t j
| j d | j d t j �} | j
| � d S)Nz can't set the file read-onlyr' �mode)r rx r� r: r� r� �access�W_OK�skipTestrG r� r� rU r� )rA r� r r r �test_non_matching_mode� s
$z TestWhich.test_non_matching_modec C ss t j j | j � \ } } t j d | � �? t j | j d | �} | j | t j j
| | j � � Wd QRXd S)Nr' ) r r' r� rU r r! rG r� r� r` r( )rA r' r� r� r r r �test_relative_path� s z TestWhich.test_relative_pathc C s) t j d d | j �} | j | � d S)Nzfoo.exer' )rG r� rU r� )rA r� r r r �test_nonexistent_file� s z TestWhich.test_nonexistent_filer� z pathext check is Windows-onlyc C sM t j | j d d � d | j �} | j | | j j d d � d � d S)N� r' z.EXE���r� )rG r� r� rU r` r� r: )rA r� r r r �test_pathext_checking� s %z TestWhich.test_pathext_checkingc
C sO t j � �< } | j | d <t j | j � } | j | | j j � Wd QRXd S)N�PATH) r �EnvironmentVarGuardrU rG r� r� r` r� r: )rA �envr� r r r �test_environ_path� s
zTestWhich.test_environ_pathc C s~ t j j | j � } t j d | j � �M t j � �9 } | j | d <t j | j d d �} | j
| � Wd QRXWd QRXd S)Nr' r� rn ) r r' r� rU r r! r� rG r� r� r� )rA r' r� r� r r r �test_empty_path� s
zTestWhich.test_empty_pathc
C sI t j � �6 } | j d d � t j | j � } | j | � Wd QRXd S)Nr� )r r� rF rG r� r� r� )rA r� r� r r r �test_empty_path_no_PATH� s z!TestWhich.test_empty_path_no_PATHN)r� r� r� r? r� r� r� r� r� r� r� r rz r� r� r� r� r� r� r� r� r� r� r r r r r� } s
$ r� c @ s� e Z d Z d d � Z d d � Z d d � Z d d � Z d d
� Z d d � Z e d
d � � Z
e d d � � Z d d � Z e d d � � Z
d d � Z e d d � � Z d d � Z e j e j j d � d d � � Z d d � Z d d! � Z d" d# � Z d$ d% � Z e j e d& d' � � � Z e j e d( d) � � � Z e j e d* d+ � � � Z e j e d, d- � � � Z d. d/ � Z! d0 d1 � Z" e d2 d3 � � Z# e d4 d5 � � Z$ d6 S)7�TestMovec C s� d } t j � | _ t j � | _ t j j | j | � | _ t j j | j | � | _ t | j d � � } | j
d � Wd QRXd S)NrL r# s spam) rJ rK r� r� r r' r( r� r� r) r* )rA rs r r r r r? � s zTestMove.setUpc
C sC x< | j | j f D]( } y | r/ t j | � Wq Yq Xq Wd S)N)r� r� rG rH )rA rI r r r rE � s zTestMove.tearDownc C s� t | d � � } | j � } Wd QRXt j | | � t | d � � } | j | | j � � Wd QRX| j t j j | � � d S)Nr. ) r) r0 rG �mover` rk r r' r^ )rA r� r� �real_dstr �contentsr r r �_check_move_file� s zTestMove._check_move_filec C sa t t j | � � } t j | | � | j | t t j | � � � | j t j j | � � d S)N) r3 r r4 rG r� r` rk r' r^ )rA r� r� r� r� r r r �_check_move_dir� s zTestMove._check_move_dirc C s | j | j | j | j � d S)N)r� r� r� )rA r r r �test_move_file s zTestMove.test_move_filec C s | j | j | j | j � d S)N)r� r� r� r� )rA r r r �test_move_file_to_dir s z TestMove.test_move_file_to_dirc C s | j � d S)N)r� )rA r r r �test_move_file_other_fs
s z TestMove.test_move_file_other_fsc C s | j � d S)N)r� )rA r r r � test_move_file_to_dir_other_fs s z'TestMove.test_move_file_to_dir_other_fsc C sJ t j � } z | j | j | | � Wd y t j | � Wn Yn XXd S)N)rJ ro r� r� rG rH )rA r� r r r �
test_move_dir s zTestMove.test_move_dirc C s | j � d S)N)r� )rA r r r �test_move_dir_other_fs s z TestMove.test_move_dir_other_fsc C s> | j | j | j t j j | j t j j | j � � � d S)N)r� r� r� r r' r( r )rA r r r �test_move_dir_to_dir$ s z TestMove.test_move_dir_to_dirc C s | j � d S)N)r� )rA r r r � test_move_dir_to_dir_other_fs) s z&TestMove.test_move_dir_to_dir_other_fsc C sH | j | j t j j | j t j j | j t j j | j � � � d S)N)r� r� r r' �sepr� r( r )rA r r r �test_move_dir_sep_to_dir. s z!TestMove.test_move_dir_sep_to_dirzrequires os.path.altsepc C sH | j | j t j j | j t j j | j t j j | j � � � d S)N)r� r� r r' �altsepr� r( r )rA r r r �test_move_dir_altsep_to_dir2 s z$TestMove.test_move_dir_altsep_to_dirc C s@ t | j d � � Wd QRX| j t j t j | j | j � d S)Nr# )r) r� r\ rG r r� r� r� )rA r r r �"test_existing_file_inside_dest_dir7 s z+TestMove.test_existing_file_inside_dest_dirc C s; t j j | j d � } | j t j t j | j | � d S)Nr� )r r' r( r� r\ rG r r� )rA r� r r r � test_dont_move_dir_in_itself= s z%TestMove.test_dont_move_dir_in_itselfc
C s� t j t � zq xj d g D]_ \ } } t j j t | � } t j j t | � } | j t j | | � d d | | f �q WWd t j t d d �Xd S)N�srcdir� srcdir/destrI z?_destinsrc() wrongly concluded that dst (%s) is not in src (%s)rm T)r� r� ) r rM r r' r( r] rG �
_destinsrcrH )rA r� r� r r r � test_destinsrc_false_negativeB s
z&TestMove.test_destinsrc_false_negativec
C s� t j t � zt xm d d g D]_ \ } } t j j t | � } t j j t | � } | j t j | | � d d | | f �q WWd t j t d d �Xd S)
Nr� �src/dest�
srcdir.newrI z;_destinsrc() wrongly concluded that dst (%s) is in src (%s)rm T)r� r� )r� r� ) r rM r r' r( rk rG r� rH )rA r� r� r r r � test_destinsrc_false_positiveN s
z&TestMove.test_destinsrc_false_positivec C s� t j j | j d � } t j | j | � t j | | j � | j t j j
| j � � | j t j j | j | j � � d S)Nr� ) r r' r( r� r[ r� rG r� r� r] r6 �samefile)rA r� r r r �test_move_file_symlinkZ s
z TestMove.test_move_file_symlinkc C s� d } t j j | j | � } t j | j | � t j | | j � t j j | j | � } | j t j j
| � � | j t j j | j | � � d S)Nr� ) r r' r( r� r[ r� rG r� r� r] r6 r� )rA rs r� Z
final_linkr r r � test_move_file_symlink_to_dirc s z&TestMove.test_move_file_symlink_to_dirc C s� t j j | j d � } t j j | j d � } t j | | � t j j | j d � } t j | | � | j t j j | � � t j
d k r� | j t j j | � t j
| � � n( | j t j j | � t j j | � � d S)Nr� r� r� rC )r r' r( r� r[ r� rG r� r] r6 r: r` �realpathr� )rA r� r� r� r r r �test_move_dangling_symlinkn s (z#TestMove.test_move_dangling_symlinkc C s� t j j | j d � } t j j | j d � } t j | � t j | | � t j j | j d � } t j | | � | j t j j
| � � | j t j j | | � � d S)Nr� r� r� ) r r' r( r� rM r[ r� rG r� r] r6 r� )rA r� r� r� r r r �test_move_dir_symlink} s
z TestMove.test_move_dir_symlinkc C sM t j | j | j � } | j | t j j | j t j j | j � � � d S)N) rG r� r� r� r` r r' r( r )rA r� r r r �test_move_return_value� s z TestMove.test_move_return_valuec C sM t j | j t j j | j d � � } | j | t j j | j d � � d S)Nr� )rG r� r� r r' r( r� r` )rA r� r r r � test_move_as_rename_return_value� s 'z)TestMove.test_move_as_rename_return_valuec sN g � � f d d � } t j | j | j d | �| j t � � d � d S)Nc s � j | | f � d S)N)r7 )r� r� )�movedr r r � s z7TestMove.test_move_file_special_function.<locals>._copyr rY )rG r� r� r� r` ra )rA r r )r� r � test_move_file_special_function� s z(TestMove.test_move_file_special_functionc s� g � � f d d � } t j t j j | j d � � t j t j j | j d � � t j | j | j d | �| j t
� � d � d S)Nc s � j | | f � d S)N)r7 )r� r� )r� r r r � s z6TestMove.test_move_dir_special_function.<locals>._copyZchildZchild1r r} ) r r� r r' r( r� rG r� r� r` ra )rA r r )r� r � test_move_dir_special_function� s z'TestMove.test_move_dir_special_functionN)%r� r� r� r? rE r� r� r� r� r" r� r� r� r� r� r� r� r� r� r r' r� r� r� r� r� r� r r� r� r� r� r� r� r� r� r� r r r r r� � s< $
r� c @ s� e Z d Z d Z Gd d � d e � Z d d � Z d d � Z d d � Z d
d � Z d d
� Z
d d � Z d d � Z d S)� TestCopyFileFc @ sX e Z d Z d Z d Z d Z d d d d � Z d d � Z d d � Z d
d � Z d S) zTestCopyFile.FauxFNTc C s | | _ | | _ d S)N)�_raise_in_exit�_suppress_at_exit)rA Z
raise_in_exitZsuppress_at_exitr r r �__init__� s zTestCopyFile.Faux.__init__c G s d S)Nrn r )rA r r r r r0 � s zTestCopyFile.Faux.readc C s
d | _ d S)NT)�_entered)rA r r r � __enter__� s zTestCopyFile.Faux.__enter__c C s7 | | | f | _ | j r0 d | _ t d � � | j S)NTz Cannot close)� _exited_withr� �_raisedr r� )rA �exc_typeZexc_valZexc_tbr r r �__exit__� s
zTestCopyFile.Faux.__exit__)
r� r� r� r� r� r� r� r0 r� r� r r r r �Faux� s r� c C s | j r t ` d S)N)�_deleterG r) )rA r r r rE � s zTestCopyFile.tearDownc C s | t _ d | _ d S)NT)rG r) r� )rA r r r r �_set_shutil_open� s z TestCopyFile._set_shutil_openc C s9 d d d � } | j | � | j t t j d d � d S)Nr/ c S s( | d k r t d � � d s$ t � d S)N�srcfilezCannot open "srcfile"r )r �AssertionError)rs r� r r r �_open� s z4TestCopyFile.test_w_source_open_fails.<locals>._openr� �destfile)r� r\ r rG r� )rA r� r r r �test_w_source_open_fails� s
z%TestCopyFile.test_w_source_open_failsc s� | j � � d � f d d � } | j | � t j d d � | j � j � | j � j d t k � | j � j d j d � d S)
Nr/ c s8 | d k r � S| d k r( t d � � d s4 t � d S)Nr� r� zCannot open "destfile"r )r r� )rs r� )r� r r r� � s
z2TestCopyFile.test_w_dest_open_fails.<locals>._openr� r� r rY �Cannot open "destfile")r� )
r� r� rG r� r] r� r� r r` r )rA r� r )r� r �test_w_dest_open_fails� s
z#TestCopyFile.test_w_dest_open_failsc s� | j � � | j d � � d � � f d d � } | j | � t j d d � | j � j � | j � j � | j � j � | j � j d t k � | j � j d j
d
� d S) NTr/ c s0 | d k r � S| d k r � Sd s, t � d S)Nr� r� r )r� )rs r� )r� r� r r r� � s
z3TestCopyFile.test_w_dest_close_fails.<locals>._openr� r� r rY � Cannot close)r� ) r� r� rG r� r] r� r� r� r r` r )rA r� r )r� r� r �test_w_dest_close_fails� s
z$TestCopyFile.test_w_dest_close_failsc s� | j d � � | j � � d � � f d d � } | j | � | j t t j d d � | j � j � | j � j � | j � j � | j � j
d d k � | j � j � d S)NTr/ c s0 | d k r � S| d k r � Sd s, t � d S)Nr� r� r )r� )rs r� )r� r� r r r� � s
z5TestCopyFile.test_w_source_close_fails.<locals>._openr� r� r ) r� r� r\ r rG r� r] r� rk r� r� )rA r� r )r� r� r �test_w_source_close_fails� s
z&TestCopyFile.test_w_source_close_failsc C s� t j � | _ | j t j | j d � t j j t j j | j � t j j
| j � j � � } | j | j | � z0 t j
| j | � | j t j j | � � Wd t j | � Xd S)NT)rJ rK r� r rG rH r r' r( r� r �upperr� r� r] r5 rt )rA r� r r r � test_move_dir_caseinsensitive s z*TestCopyFile.test_move_dir_caseinsensitiveN)
r� r� r� r� �objectr� rE r� r� r� r� r� r� r r r r r� � s
r� c @ s� e Z d Z d d � Z d d � Z d d � Z e j e j e
j j � � d � e j e
e d � d � d
d � � � Z d d
� Z d S)�
TermsizeTestsc C s6 t j � } | j | j d � | j | j d � d S)z�Check if get_terminal_size() returns a meaningful value.
There's no easy portable way to actually check the size of the
terminal, so let's check if it returns something sensible instead.
r N)rG �get_terminal_sizeru �columns�lines)rA �sizer r r �test_does_not_crash s z!TermsizeTests.test_does_not_crashc C s� t j � �$ } d | d <| d =t j � } Wd QRX| j | j d � t j � �$ } | d =d | d <t j � } Wd QRX| j | j d � d S)z.Check if environment variables have precedenceZ777�COLUMNS�LINESNi Z888ix )r r� rG r� r` r� r� )rA r� r� r r r �test_os_environ_first( s
z#TermsizeTests.test_os_environ_firstc
C s` t j � �' } d | d <d | d <t j � } Wd QRX| j | j d � | j | j d � d S)Nr� r� Zyyyr� r )r r� rG r� ru r� r� )rA r� r� r r r �test_bad_environ7 s
z TermsizeTests.test_bad_environz
not on ttyr� zneed os.get_terminal_size()c C s� y% t j d d g � j � j � } Wn( t t j f k
rO | j d � Yn Xt | d � t | d � f } t j � �! } | d =| d =t
j � } Wd QRX| j | | � d S) z�Check if stty returns the same results ignoring env
This test will fail if stdin and stdout are connected to
different terminals with different sizes. Nevertheless, such
situations should be pretty rare.
Zsttyr� zstty invocation failedrY r r� r� N)
r5 rD rG r� rp rE r� �intr r� rG r� r` )rA r� ri r� r� r r r �test_stty_match? s
% z TermsizeTests.test_stty_matchc C s� t j � �� } | d =| d =t j t d d � � t j d d
� } Wd QRX| j | j d � | j | j d � t t
j d � �6 } t j t d | � � t j d d � } Wd QRXWd QRX| j | j d � | j | j d � Wd QRXd S) Nr� r� �
__stdout__Zfallback�
� r$ � �( )r� r� )r� r ) r r� Z swap_attrr� rG r� r` r� r� r) r �devnull)rA r� r� r r r r �
test_fallbackV s zTermsizeTests.test_fallbackN)r� r� r� r� r� r� r� r� r �isattyr� r� �filenor� r� r r r r r r� s
!r� c @ s" e Z d Z d Z d d � Z d S)�PublicAPITestsz>Ensures that the correct values are exposed in the public API.c C s� | j t t d � � d d d d d d d d d
d d d
d d d d d d d d d d d d d g } t t d � s� t j d k r� | j d � | j t t j � t | � � d S) N�__all__Z copyfileobjr� r� r� r� r� r� r� rH r r Z ExecErrorr r r r r r r r r� rw r� r� r
�statvfsrC rt ) r] r� rG r r: r7 r` �setr )rA Z
target_apir r r �test_module_all_attributem s
z(PublicAPITests.test_module_all_attributeN)r� r� r� �__doc__r r r r r r j s r �__main__)4r� Z
unittest.mockrG rJ r� r� r Zos.pathr r r5 �
contextlibr r r r r r r r r
r r r
r# r: �warningsr^ r Z test.supportr r r r rQ rS rP � ImportErrorr r" r- r1 r8 ZTestCaser= r� r� r� r� r r� �mainr r r r �<module> sT L
� � � � � )d�xM