Your IP : 216.73.216.213
�
g f! � @ s� 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 m Z d d l m Z m Z m Z m Z y d d l! Z! d Z" Wn e# k
rZd Z" Yn Xe d Z$ y" d d l% Z% d d l& Z& d Z' Wn e# k
r�d Z' Yn Xy d d l( Z( d Z) Wn! e# k
r�e j* d � Z) Yn Xd
d � Z+ d d
� Z, d d d � Z- d d d � Z. d d � Z/ Gd d � d e j0 � Z1 Gd d � d e j0 � Z2 Gd d � d e j0 � Z3 Gd d � d e j0 � Z4 Gd d � d e j0 � Z5 Gd d � d e j0 � Z6 e7 d k r�e j8 � n 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�
requires_zlibTF�2�zipc O s t t t d d � d � � d S)NZEXDEV� zInvalid cross-device link)�OSError�getattr�errno)�args�kwargs� r �5/opt/alt/python34/lib64/python3.4/test/test_shutil.py� _fake_rename0 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 �wrap5 s
zmock_rename.<locals>.wrap)� functools�wraps)r r! r )r r � mock_rename4 s !r$ c C sY t | t � r$ t j j | � } n t | | r6 d n d � � } | j | � Wd QXd 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_file? s r/ c C sV t | t � r$ t j j | � } n t | | r6 d n d � � } | j � SWd QXd 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_fileK s r3 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 r: W s % r: 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 e9 dj dk � � Z: dl dm � Z; dn do dp � Z< e9 e j
e= j> dq � dr � ds dt � � � Z? e9 e j
e@ du � dv dw � � � ZA e9 e j
e@ du � e j
e= j> dx � dy � dz d{ � � � � ZB e9 e j
e@ du � e j
e= j> d| � d} � d~ d � � � � ZC d� d� � ZD e9 d� d� � � ZE e9 e j
eF d� � d� d� � � � ZG d� d� � ZH d� d� � ZI e9 d� d� � � ZJ d� d� � ZK e9 d� d� � � ZL d� d� � ZM e j
e e= d� � d� � d� d� � � ZN e j
eF d� � e j
e e d� � d� � d� d� � � � ZO d� d� � ZP d� d� � ZQ d� d� � ZR d� d� � ZS � S)��
TestShutilc s t t | � j � g | _ d S)N)�superr? �setUp�tempdirs)�self)� __class__r r rA f s zTestShutil.setUpc sO t t | � j � x5 | j rJ | j j � } t j | t j d k � q Wd S)N�nt�cygwin)zntzcygwin) r@ r? �tearDownrB �pop�shutil�rmtreer r< )rC �d)rD r r rG j 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�mkdtemprB r9 )rC rK r r r rM q 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 QXd S)NZkillmeZsomefile�foorE )rM r r) r* �mkdirr/ �fsencode�assertIsInstance�bytesr< Z assertWarns�DeprecationWarningr rI rJ )rC �tmpZvictim�winr r r �test_rmtree_works_on_bytesz 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)r9 )r )�errorsr r �onerror� s z8TestShutil.test_rmtree_fails_on_symlink.<locals>.onerrorrZ � r � )rM r r) r* rO �symlink� assertRaisesr rI rJ �
assertTrue�exists�lexists� assertEqual�len�assertIsr8 rQ )rC rT Zdir_rX rZ r )rY r � test_rmtree_fails_on_symlink� 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�file1rN �link1�link2�link3) rM r r) r* rO r/ r] rI rJ � assertFalser` r_ )
rC rT rf rg rh rK ri rj rk rl 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 QX| 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)r9 )r )rY r r rZ � s z.TestShutil.test_rmtree_errors.<locals>.onerrorrZ r\ r r[ )rL �mktempr^ �FileNotFoundErrorrI rJ rM r/ r r) r* �NotADirectoryError�assertInZ exception�filenamer_ r` rb rc rd r6 rQ �rmdir)rC ru �tmpdir�cmZ
possible_argsrZ r )rY r �test_rmtree_errors� s2 z TestShutil.test_rmtree_errors�chmodzrequires os.chmod()N� rF 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�brZ � z1Expected call to onerror function did not happen.)�
errorStater rO r �
addCleanuprI rJ r) r* �child_file_path�child_dir_pathr Zcreate_empty_file�stat�st_mode�S_IREAD�S_IEXECrz �check_args_to_onerrorrb )rC 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)Nr\ r r[ r )r� r �unlinkrb r� rv r� rd r6 rt r r_ �
issubclassr )rC 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>.raiserrN )r �lstatrO r r/ rI rJ )rC 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 � n d S)NrN �bar�baz�quuxrE )rM r r) r* r/ r] rz r� �S_IRWXU�S_IRWXG�S_IRWXO�assertNotEqualr� rI �copymoderb r< )rC �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)NrN r� r� r� �follow_symlinksF)rM r r) r* r/ r] rz r� r� r� r� r� rI r� rb r� r� r� )rC r� r� r� r� r� r r r � test_copymode_symlink_to_symlink> 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)NrN r� r� r� r� F)rM r r) r* r/ r] rI r� )rC r� r� r� r� r� r r r �*test_copymode_symlink_to_symlink_wo_lchmod\ 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
� n t t d � rJt t d � rJt j | t j � n t j | � } t t d � r�t j | | d d
�| j | j t j | � j � n t j | | d d �t j | � } t j t j k rx7 d D], } | j t | | � t | | � d � q�Wn t t d � r9| j | j | j � n t t d � rpt | d � rp| j | j | j � n t j | | d d �| j t t j | � j t j | � j � d k � d S)NrN r� r� �quxg E@r� �lchflags� UF_NODUMPr� TF�st_atime�st_mtimer[ �st_flagsg�������?)zst_atimezst_mtime)rM r r) r* r/ r� �utimer� r� r� r] �hasattrr� r� r� r� r� rI �copystatr� �supports_follow_symlinks�assertLessEqualr rb r� r_ �abs)
rC r� r� r� r� r� �src_stat�
src_link_statZ
dst_link_stat�attrr r r �test_copystat_symlinksj 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)Nri �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)
rM r r) r* r/ r� r r� r� rI r� r^ r )rC rw ri 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)NrN 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 s1 | � k r! t t j d � � n � | 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)rM r r) r* r/ rI �
_copyxattr�setxattrrb r5 � listxattr�getxattr�remover r r� rt r� )rC 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)
NrN r� z trusted.foos 42s 43r� Fr� r� )
rM r r) r* r/ r] r� rI r� rb r� r^ r )rC r� r� r� r� r� r r r �test_copyxattr_symlinks� s
"z"TestShutil.test_copyxattr_symlinksc C s{ | 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� n 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 � rw| j t j | � j t j | � j � n d S)NrN r� r� r� r� TF)rM r r) r* r/ r] r� r� r� r� r� rI �copyrm r8 rb r3 r� r_ �readlinkr� r� )rC r� r� r� r� r r r �test_copy_symlinks s$
"z TestShutil.test_copy_symlinksc C sc | 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� n t t d � r� t t d � r� t j | t j � n 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�x7 d D], } | j t | | � t | | � d � q�Wn t t d � r(| j | j | j � | j | j | j � n t t d � r_t | d
� r_| j | j | j � n d S)NrN r� r� r� r� r� r� TFr� r� r[ r� )zst_atimezst_mtime) rM r r) r* r/ r] r� r� r� r� r� r� r� r� rI �copy2rm r8 rb r3 r� r_ r� r� r� r� r r� r� r� ) rC 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)NrN r� zuser.foos 42) rM r r) r* r/ r� rI r� rb r� r� )rC r� r� r� r r r �test_copy2_xattr; 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� rX rN r� F)
rM r r) r* r/ r] rI �copyfiler_ r8 rb r� rm )rC r� r� r� r� rX r r r �test_copyfile_symlinksH 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 �Calledf s r� c s
� � d S)Nr )r r )r� r r �_raiserg s zITestShutil.test_rmtree_uses_safe_fd_version_if_available.<locals>._raiser)r r+ r� r� rv �supports_dir_fdr6 � supports_fdr� r_ rI �_use_fd_functionsrJ Zavoids_symlink_attacksrM r) r* rO Z_rmtree_safe_fd� Exceptionr^ rm )rC r� r� rK Z real_rmtreer� r )r� r �-test_rmtree_uses_safe_fd_version_if_availableY 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) rL Zmkstempr �closer^ rs rI rJ r� )rC Zhandler) r r r � test_rmtree_dont_delete_fileq 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)rL rM r r) r* r� rI rJ �dirnamer/ rO �copytreer_ �isfiler7 r3 rb )rC �src_dir�dst_dir�actualr r r �test_copytree_simplex 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� n t t d � r!t t d
� r!t j | t j
� n 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 � n t t d � r
| j | j | j � n d S)
Nr� r� �subzfile.txtrN rX zsub/linkr� r� r� �symlinksT)rM r r) r* rO r/ r] r� r� r� r� r� r� r� r� rI r� r_ r8 rb r� r� r� ) rC 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
q
W| S)Nr[ r� z.py���r� )r r) r* r7 �splitr9 �splitext)r� �namesr; r< r) r r r �_filter� s
z6TestShutil.test_copytree_with_exclude.<locals>._filter)r r) r* r` rL rM r/ rO rI �ignore_patternsr� r_ rm rJ r� )rC 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� rW r[ )rL rM r r) r* rO r� rI rJ rz r/ r� rb r� r� r� )rC 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 QXd S)Nr� zka-boom)rL rM r r) r* r� rI rJ r� �PermissionErrorZ side_effectr^ r r� )rC Z
mock_patchr� r� r r r �test_copytree_winerror s z!TestShutil.test_copytree_winerrorrE z temporarily disabled on WindowsrX zrequires os.linkc C s� t j t � t j j t d � } t j j t d � } z� t | d � � } | j d � Wd QXt j | | � | j t j
t j | | � t | d � � } | j | j
� d � Wd QXt j | � Wd t j t d d �Xd S)N�cheese�shopr& �cheddarr1 ro T)r rO r r) r* r+ r, rX r^ rI r
r� rb r2 r� rJ )rC 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 QXt j d | � | j t j
t j | | � t | d � � } | j | j
� d � Wd QXt j | � Wd t j t d d �Xd S)Nr� r r& r r1 ro T)r rO r r) r* r+ r, r] r^ rI r
r� rb r2 r� rJ )rC 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 ro T)
r rO r r) r* r] r^ r rI rJ )rC r� r� r r r �test_rmtree_on_symlink6 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^ rI �SpecialFileErrorr� �TESTFN2�__file__r� )rC r r r �test_copyfile_named_pipeE 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 r[ z`%s` is a named pipez$shutil.Error should have been raisedro T)r rO r r) r* r rI r� r r r rb rc �failrJ )rC r� �pipe�erY r� r� Z error_msgr r r �test_copytree_named_pipeP 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)r9 )r� r� )�copiedr r �_copyo s z4TestShutil.test_copytree_special_func.<locals>._copyZ
copy_functionr\ )
rM r r) r* r/ rO rI r� rb rc )rC r� r� r r )r r �test_copytree_special_funcf 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� )rM r r) r* r] rO r/ r^ r rI r� � assertNotInr6 rt )rC r� r� r r r � test_copytree_dangling_symlinksu 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 QXt 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 )rM r r) r* rO r+ r] rI r� rm r8 rt r6 r_ )rC 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� )rM r/ r r) r* )rC �methodr� rw ri �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 rI r� r_ r r) r` rb r� r� )rC ri 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 � � n d S)Nr� r� r[ r� r� )zst_atimezst_mtime)
r rI r� r_ r r) r` r� rb r� r� r r� )rC ri 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 QX| j | | d � | j
t j j | � � | j
t j | � � t j | d � �/ } | j | j � d d d d d
d g � Wd QXt j | � � t | d | d � } Wd QX| j | | d
� | j
t j j | � � | j
t j | � � t j | d � �/ } | j | j � d d d d d
d g � Wd QXd S)Nrp �archive�gztar�.z.tar.gzzr:gzz./subz./sub2z./file1z./file2z ./sub/file3�tarz.tarr1 )�
_create_filesrM r rv r) r� r* �basenamer �
change_cwd�abspathr rb r_ r� �tarfileZ
is_tarfiler+ �assertCountEqual�getnames) rC �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 QXd S)N)r# r+ r% �sortr( )rC 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 � n | | f S)
N�exist_okTri r� r� r� Zfile3Zsub2�outer)rM r r) r* �makedirsr/ rO )rC 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* rM r rb r_ r� �
subprocess�
check_call�DEVNULLr. )rC r& r' r* r+ Ztarball2Ztar_cmdr r r �test_tarfile_vs_tar� s$
" z TestShutil.test_tarfile_vs_tarzNeed zip support to runc C s" | j � \ } } | j � } t j | � t j j | � } t j j t j j | � d � } t j | � �, t j j
| � } t | d | | � } Wd QX| j | | d � | j
t j j | � � | j
t j | � � t j | � �/ } | j | j � d d d d d d g � Wd QXd S)
Nr r z.zipzdist/z dist/sub/z
dist/sub2/z
dist/file1z
dist/file2zdist/sub/file3)r rM r rv r) r� r* r r r! r" r rb r_ r� �zipfileZ
is_zipfile�ZipFiler$ �namelist) rC r& r' r r( r) r* r; �zfr r r �test_make_zipfile s
! z TestShutil.test_make_zipfiler zNeed the zip command to runc
C s6 | 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 QXt j
| � � } | j � } Wd QX| j t | � t | � � d S) Nr r z.zipz archive2.zipz-qz-rr3 r4 )r r r) r* rM r rb r_ r� r5 r6 r7 r9 r: r; r5 )
rC r& r' r* r Zarchive2�zip_cmdr< r� Znames2r r r �test_zipfile_vs_zip3 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 QXd S)
Nr r z.zipr@ z-t�stderrrY �replacez{}
**Unzip Output**
{})r r r) r* rM r rb r_ r� r r! r5 � check_outputZSTDOUT�CalledProcessError�output�decoder �format) rC 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� )rM r r) r* r^ �
ValueErrorr )rC rw r* r r r �test_make_archivee 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 r �owner�groupr Z kjhkjhkjgZoihohoh)
�UID_GID_SUPPORT�grp�getgrgid�pwd�getpwuidr r r) r* rM r r_ r� )rC rN rM r& r' r* r; r r r � test_make_archive_owner_groupj 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 QX| 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/ rM rN )r r r) r* rM rP rQ rR rS r r! r r_ r� r# r+ Z
getmembersrb �uid�gidr� ) rC r& r' r* rN rM 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 rM r� rb r )rC Z current_dirrZ r r r �test_make_archive_cwd� s
z TestShutil.test_make_archive_cwdc C sX | j � } t j | � �7 | j t d d � d � | j t j j d � � Wd QXd S)N�testr ztest.tar) rM r r! rb r r_ r r) r� )rC r& r r r �test_make_tarfile_in_curdir� s z&TestShutil.test_make_tarfile_in_curdirc C sX | j � } t j | � �7 | j t d d � d � | j t j j d � � Wd QXd S)Nr] r ztest.zip) rM r r! rb r r_ r r) r� )rC 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� r[ 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)r` r r r r ra � s r\ r c S s t S)N)r` r r r r ra � 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 )rb r< rc r r r rd � s )r[ r\ )r[ r\ r )r[ r\ )r^ � TypeErrorr r rt r r )rC �formatsr r r � test_register_archive_format� s
z'TestShutil.test_register_archive_formatc
C s( d d d g } t r% | j d � n | j � \ } } t | � } | j d � x� | D]� } t j j | j � d � } t | | | | � } | j � } t
| | � | j t | � | � | j � } t
| | d | �| j t | � | � qW W| j t
j t
t � | j t t
t d d �d S) Nr r r Zbztarr1 r rG r� )�
BZ2_SUPPORTEDr9 r r: r� r r) r* rM r r rb r^ rI Z ReadErrorr rJ )
rC rf r& r' �expectedrG r* ru r Ztmpdir3r r r �test_unpack_archive� s"
z TestShutil.test_unpack_archivec 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)Nr[ z stuff.boo�xx)rb )ru Z extract_dir�extra)rC r r �_boo� s z.TestShutil.test_unpack_registery.<locals>._booZBooz.booz.b2rl r[ z stuff.boork ZBoo2rp )zextrar[ ) r r r r^ r
r rt r rb )rC rf rm r )rC r �test_unpack_registery� s
z TestShutil.test_unpack_registery�
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 ) rI ro r r[ Z
assertGreaterZtotalZused�assertGreaterEqualZfree)rC 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 QX� j t � � t j � d d �Wd QX� j t � � t j � d d �Wd QX� j t � � t j � d � Wd QX� j t � � t j � d � Wd QXt
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)
NrW ztesting chown function�userznon-exising usernamerN znon-exising groupnames spamg ��Q� @c sW t j � � } | d k r1 � j | | j � n | d k rS � j | | j � n d S)N)r r� rb �st_uid�st_gid)r) rU rV �s)ru rC r r � check_chown# s
z*TestShutil.test_chown.<locals>.check_chownrV r )rM rL rq r/ r^ rJ rI rr � LookupErrorre r �getuid�getgidrR rS rP rQ )rC r� rU rV rw rs rN r )ru rC 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)NrN r� ) rI r� r� rM r r) r* r/ rb )rC r� r� r� r� �rvr r r �test_copy_return_valueC 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� rN ) rM r r) r* r/ rI r� r_ r` rb r3 )rC r� r� �dst_file�src_filer| r r r �test_copyfile_return_valueO 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)NrN )
rM r r) r* r/ r^ r
rI r� r )rC r� r r r r �test_copyfile_same_fileZ 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�destTrN ) rM r� rI rJ r r) r* r/ r� rb r6 )rC r� r� r� r| r r r �test_copytree_return_valued s
z%TestShutil.test_copytree_return_value)Tr� r� r� rA rG rM rV r �skip_unless_symlinkre rn ry �unittest�
skipUnlessr� r �skipIf�sys�platformr| 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 r r, r. r rI �whichr8 � ZIP_SUPPORTr= r? rI rK rT rO rW r\ r^ r_ rg rj rn rq r{ r} r� r� r� r r )rD r r? d 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�prefixZTmpTrW �suffixz.Exe)rL rM �temp_dirr� rI rJ ZNamedTemporaryFile� temp_filer rz r< r� �S_IXUSRr� r) r� rW �file)rC r r r rA q s zTestWhich.setUpc C s5 t j | j d | j �} | j | | j j � d S)Nr) )rI r� r� rW rb r� r< )rC r| r r r �
test_basic} s zTestWhich.test_basicc C s8 t j | j j d | j �} | j | | j j � d S)Nr) )rI r� r� r< r� rb )rC 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 QXt j d | j � �' t j | d | �} | j | � Wd QXd S)Nr) )
r r) r� rW r* r� r r! rI r� r� rb � assertIsNone)rC 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 QXd S)Nr) �win32)r r) r� rW r r! rI r� r� r� r� rb r* �curdirr� )rC r' r| r r r �test_cwd� s (zTestWhich.test_cwdr| r znon-root user requiredc C sy t j | j j t j � t j | j j t j � rD | j d � n t j
| j d | j d t j �} | j
| � d S)Nz can't set the file read-onlyr) �mode)r rz r� r< r� r� �access�W_OK�skipTestrI r� r� rW r� )rC r| r r r �test_non_matching_mode� s
$z TestWhich.test_non_matching_modec C sr t j j | j � \ } } t j d | � �? t j | j d | �} | j | t j j
| | j � � Wd QXd S)Nr) ) r r) r� rW r r! rI r� r� rb r* )rC 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) )rI r� rW r� )rC 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� )rI r� r� rW rb r� r< )rC r| r r r �test_pathext_checking� s %z TestWhich.test_pathext_checkingc
C sN t j � �< } | j | d <t j | j � } | j | | j j � Wd QXd S)N�PATH) r �EnvironmentVarGuardrW rI r� r� rb r� r< )rC �envr| r r r �test_environ_path� s
zTestWhich.test_environ_pathc C s| t j j | j � } t j d | j � �L t j � �9 } | j | d <t j | j d d �} | j
| � Wd QXWd QXd S)Nr) r� rp ) r r) r� rW r r! r� rI r� r� r� )rC r' r� r| r r r �test_empty_path� s
zTestWhich.test_empty_pathc
C sH t j � �6 } | j d d � t j | j � } | j | � Wd QXd S)Nr� )r r� rH rI r� r� r� )rC r� r| r r r �test_empty_path_no_PATH� s z!TestWhich.test_empty_path_no_PATHN)r� r� r� rA r� r� r� r� r� r� r� r r| r� r� r� r� r� r� r� r� r� r� r r r r r� o 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" d2 S)3�TestMovec C s� d } t j � | _ t j � | _ t j j | j | � | _ t j j | j | � | _ t | j d � � } | j
d � Wd QXd S)NrN r% s spam) rL rM r� r� r r) r* r r~ r+ r, )rC ru r r r r rA � s zTestMove.setUpc
C sF x? | j | j f D]+ } y | r2 t j | � n Wq Yq Xq Wd S)N)r� r� rI rJ )rC rK r r r rG � s zTestMove.tearDownc C s t | d � � } | j � } Wd QXt j | | � t | d � � } | j | | j � � Wd QX| j t j j | � � d S)Nr0 ) r+ r2 rI �moverb rm r r) r` )rC 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) r5 r r6 rI r� rb rm r) r` )rC 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~ )rC r r r �test_move_file� s zTestMove.test_move_filec C s | j | j | j | j � d S)N)r� r r� r~ )rC r r r �test_move_file_to_dir� s z TestMove.test_move_file_to_dirc C s | j � d S)N)r� )rC r r r �test_move_file_other_fs� s z TestMove.test_move_file_other_fsc C s | j � d S)N)r� )rC 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)rL rq r� r� rI rJ )rC r� r r r �
test_move_dir s zTestMove.test_move_dirc C s | j � d S)N)r� )rC 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 )rC r r r �test_move_dir_to_dir s z TestMove.test_move_dir_to_dirc C s | j � d S)N)r� )rC 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 )rC 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 )rC r r r �test_move_dir_altsep_to_dir$ s z$TestMove.test_move_dir_altsep_to_dirc C s? t | j d � � Wd QX| j t j t j | j | j � d S)Nr% )r+ r~ r^ rI r r� r r� )rC r r r �"test_existing_file_inside_dest_dir) 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^ rI r r� )rC 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/destrH z?_destinsrc() wrongly concluded that dst (%s) is not in src (%s)ro T)zsrcdirr� ) r rO r r) r* r_ rI �
_destinsrcrJ )rC r� r� r r r � test_destinsrc_false_negative4 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.newrH z;_destinsrc() wrongly concluded that dst (%s) is in src (%s)ro T)�srcdirr� )r� r� ) r rO r r) r* rm rI r� rJ )rC r� r� r r r � test_destinsrc_false_positive@ 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 rI r� r~ r_ r8 �samefile)rC r� r r r �test_move_file_symlinkL 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 rI r� r� r_ r8 r� )rC ru r� Z
final_linkr r r � test_move_file_symlink_to_dirU 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� rE )r r) r* r� r] r� rI r� r_ r8 r< rb �realpathr� )rC r� r� r� r r r �test_move_dangling_symlink` 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� rO r] r� rI r� r_ r8 r� )rC r� r� r� r r r �test_move_dir_symlinko 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) rI r� r r� rb r r) r* r )rC 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� )rI r� r r r) r* r� rb )rC r| r r r � test_move_as_rename_return_value� s 'z)TestMove.test_move_as_rename_return_valueN)#r� r� r� rA rG 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� � s8 $
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)rC Z
raise_in_exitZsuppress_at_exitr r r �__init__� s zTestCopyFile.Faux.__init__c G s d S)Nrp r )rC r r r r r2 � s zTestCopyFile.Faux.readc C s
d | _ d S)NT)�_entered)rC r r r � __enter__� s zTestCopyFile.Faux.__enter__c C s: | | | f | _ | j r3 d | _ t d � � n | j S)NTz Cannot close)� _exited_withr� �_raisedr r� )rC �exc_typeZexc_valZexc_tbr r r �__exit__� s
zTestCopyFile.Faux.__exit__)
r� r� r� r� r� r� r� r2 r� r� r r r r �Faux� s r� c C s | j r t ` n d S)N)�_deleterI r+ )rC r r r rG � s zTestCopyFile.tearDownc C s | t _ d | _ d S)NT)rI r+ r� )rC 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)Nr1 c S s+ | d k r t d � � n d s' t � d S)N�srcfilezCannot open "srcfile"r )r �AssertionError)ru r� r r r �_open� s z4TestCopyFile.test_w_source_open_fails.<locals>._openr� �destfile)r� r^ r rI r� )rC 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)
Nr1 c s; | d k r � S| d k r+ t d � � n d s7 t � d S)Nr� r� zCannot open "destfile"r )r r� )ru r� )r� r r r� � s
z2TestCopyFile.test_w_dest_open_fails.<locals>._openr� r� r r[ �Cannot open "destfile")r� )
r� r� rI r� r_ r� r� r rb r )rC 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) NTr1 c s0 | d k r � S| d k r � Sd s, t � d S)Nr� r� r )r� )ru r� )r� r� r r r� � s
z3TestCopyFile.test_w_dest_close_fails.<locals>._openr� r� r r[ � Cannot close)r� ) r� r� rI r� r_ r� r� r� r rb r )rC 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)NTr1 c s0 | d k r � S| d k r � Sd s, t � d S)Nr� r� r )r� )ru r� )r� r� r r r� � s
z5TestCopyFile.test_w_source_close_fails.<locals>._openr� r� r ) r� r� r^ r rI r� r_ r� rm r� r� )rC 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)rL rM r� r� rI rJ r r) r* r� r �upperr� r� r_ r7 rv )rC r� r r r � test_move_dir_caseinsensitive� s z*TestCopyFile.test_move_dir_caseinsensitiveN)
r� r� r� r� �objectr� rG r� r� r� r� r� r� r r r r r� � s
r� c @ sX e Z d Z d d � Z d d � Z e j e j e j
j � � d � 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)rI �get_terminal_sizerp �columns�lines)rC �sizer r r �test_does_not_crash� s z!TermsizeTests.test_does_not_crashc C s� t j � � } d | d <t j � } Wd QX| j | j d � t j � � } d | d <t j � } Wd QX| j | j d � d S)z.Check if environment variables have precedenceZ777�COLUMNSNi Z888�LINESix )r r� rI r� rb r� r� )rC r� r� r r r �test_os_environ_first s
z#TermsizeTests.test_os_environ_firstz
not on ttyc 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 QX| 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 failedr[ r r� r� N)
r5 rC rF r� rr rD r� �intr r� rI r� rb )rC r� ri r� r� r r r �test_stty_match s % z TermsizeTests.test_stty_matchN)
r� r� r� r� r� r� r� r �isattyr� �
__stdout__�filenor� 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 � n | j t t j � t | � � d S) N�__all__Z copyfileobjr� r� r� r� r� r� r� rJ r r Z ExecErrorr r r r r r r r r� rr r� r� r
�statvfsrE ro ) r_ r� rI r r< r9 rb �setr� )rC Z
target_apir r r �test_module_all_attribute. s z(PublicAPITests.test_module_all_attributeN)r� r� r� �__doc__r� r r r r r� + s r� �__main__)9r� Z
unittest.mockrI rL r� r� r Zos.pathr r" r5 �
contextlibr r r r r r r r r
r r r
r# �warningsr] r Z test.supportr r r r �bz2rh � ImportErrorr rP rR rO r9 r� r� r r$ r/ r3 r: ZTestCaser? r� r� r� r� r� r� �mainr r r r �<module> sf L "
� � � � � d�x.