Your IP : 216.73.216.60
3
��2a�- � @ s� d dl Z d dlZ d dlZd dlZd dlZd dlZd dlZd dlZd dlZd dl Z d dl
Z
d dl Z d dl m
Z
d dlmZmZmZmZmZmZmZmZmZmZmZ d dlZd dlZd dlZd dl
Z
d dl m Z d dl m Z m Z e d Z!yd dl"Z"d dl#Z#dZ$W n e%k
�r dZ$Y nX d d
� Z&d d � Z'd d
d�Z(d!dd�Z)dd� Z*G dd� de j+�Z,G dd� de j+�Z-G dd� de j+�Z.G dd� de j+�Z/G dd � d e j+�Z0G d d � d e j+�Z1e2d k�r�e j3� dS )"� 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�FakePath�2TFc O s t ttdd�d��d S )NZEXDEV� zInvalid cross-device link)�OSError�getattr�errno)�args�kwargs� r �0/opt/alt/python36/lib64/python3.6/test_shutil.py� _fake_rename' s r c s t j� �� fdd��}|S )Nc s$ zt j}tt _� | |�S |t _X d 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 sB t | t�rtjj| � } t| |r"dnd��}|j|� W d Q R X d S )N�wb�w)�
isinstance�tupler �path�join�open�write)r& Zcontent�binary�fpr r r �
write_file6 s
r, c C s< t | t�rtjj| � } t| |r"dnd��
}|j� S Q R X d S )N�rb�r)r$ r% r r& r' r( �read)r& r* r+ r r r � read_fileB s
r0 c C s� g }x|t tj| ��D ]j}tjj| |�}tjj|�rttjj|� rt|j|d � x.t|�D ]}|j|d | � qXW q|j|� qW |S )N�/) �sortedr �listdirr& r' �isdir�islink�append�rlistdir)r& �res�name�p�nr r r r7 N s r7 c s� e Zd Z� fdd�Z� fdd�Zdd� Zdd� Zejd d
� �Z ejd d � �Z
d
d� Z e j
eed�d�e jejdd� dkd�e jeed�o�ej� dkd�dd� ���Zdd� Zd d � Ze j
eed�d �ejd d � ��Ze j
eed!�d"�ejd#d$� ��Ze jeed!�d%�ejd&d'� ��Zejd(d)� �Ze j
eed*��oHeed+��oHeed,�d-�d.d/� �Z ej d0d1� �Z ejej e j
eed��o�ej� dkd2�d3d4� ���Z ejd5d6� �Z ejd7d8� �Z!ej d9d:� �Z"ejd;d<� �Z#d=d>� Z$d?d@� Z%dAdB� Z&ejdCdD� �Z'dEdF� Z(dGdH� Z)e j*j+dI�dJdK� �Z,e jej-dLkdM�e j
eedN�dO�dPdQ� ��Z.ejdRdS� �Z/ejdTdU� �Z0e j
eedV�dW�dXdY� �Z1e j
eedV�dW�ejdZd[� ��Z2d\d]� Z3ejd^d_� �Z4ejd`da� �Z5dbdc� Z6e j
eed�d �ddde� �Z7e j
eed�d �e j
eedf�dg�dhdi� ��Z8ej9djdk� �Z:dldm� Z;d�dodp�Z<ej9e j
e=j>dq�dr�dsdt� ��Z?ej9dudv� �Z@ej9e j
e=j>dw�dx�dydz� ��ZAej9e j
e=j>d{�d|�d}d~� ��ZBdd�� ZCej9d�d�� �ZDej9e j
eEd��d�d�� ��ZFd�d�� ZGd�d�� ZHej9d�d�� �ZId�d�� ZJd�d�� ZKd�d�� ZLd�d�� ZMej9d�d�� �ZNejOd�d�� �ZPejQd�d�� �ZRej9d�d�� �ZSd�d�� ZTe j
ee=d��d��d�d�� �ZUe j
eEd��e j
eed��d��d�d�� ��ZVd�d�� ZWd�d�� ZXd�d�� ZYd�d�� ZZ� Z[S )��
TestShutilc s t t| �j� g | _d S )N)�superr< �setUp�tempdirs)�self)� __class__r r r> ] s zTestShutil.setUpc s: t t| �j� x&| jr4| jj� }tj|tjdk� qW d S )N�nt�cygwin)rB rC ) r= r< �tearDownr? �pop�shutil�rmtreer r9 )r@ �d)rA r r rD a s
zTestShutil.tearDownc C s t j� }| jj|� |S )N)�tempfile�mkdtempr? r6 )r@ rH r r r rJ h s zTestShutil.mkdtempc C sX | j � }tjj|d�}tj|� ttjj|d�d� tj|�}| j|t� t j
|� d S )NZkillmeZsomefile�foo) rJ r r& r' �mkdirr, �fsencode�assertIsInstance�bytesrF rG )r@ �tmpZvictimr r r �test_rmtree_works_on_bytesq s
z%TestShutil.test_rmtree_works_on_bytesc s� | j � }tjj|d�}tj|� tjj|d�}tj||� | jttj |� | j
tjj |�� | j
tjj |�� g � � fdd�}tj ||d� | j
t� �d� | j� d d tjj� | j
� d d |� | j� d d d t� d S ) N�dir�linkc s � j | � d S )N)r6 )r )�errorsr r �onerror� s z8TestShutil.test_rmtree_fails_on_symlink.<locals>.onerror)rU � r � )rJ r r& r' rL �symlink� assertRaisesr rF rG �
assertTrue�exists�lexists� assertEqual�len�assertIsr5 rN )r@ rP Zdir_rS rU r )rT r � test_rmtree_fails_on_symlinkz s
z'TestShutil.test_rmtree_fails_on_symlinkc
C s� | j � }tjj|d�}tjj|d�}tjj|d�}x |||fD ]}tj|� q>W tjj|d�}t|d� tjj|d�}tj||� tjj|d�}tj||� tjj|d�} tj|| � tj|� | j tjj
|�� | j tjj
|�� | j tjj
|�� d S ) N�dir1�dir2�dir3�file1rK �link1�link2�link3) rJ r r& r' rL r, rX rF rG � assertFalser[ rZ )
r@ rP ra rb rc rH rd re rf rg r r r � test_rmtree_works_on_symlinks� s$
z(TestShutil.test_rmtree_works_on_symlinksc s� t j� }| jttj|� tj|dd� | j� }t|dfd� tj j
|d�}| jt ��}tj|� W d Q R X |tj j
|d�g}| j |j
j|� | jtj j|�� tj|dd� | jtj j|�� g � � fdd�}tj||d� | jt� �d � | j� d
d
tj� | j� d
d |� | j� d
d d t � | j � d
d d j|� | j� d d
tj� | j� d d |� | j� d d d t � | j � d d d j|� d S ) NT)�
ignore_errorsZtstfile� z*.*c s � j | � d S )N)r6 )r )rT r r rU � s z.TestShutil.test_rmtree_errors.<locals>.onerror)rU rW r rV )rI �mktemprY �FileNotFoundErrorrF rG rJ r, r r& r' �NotADirectoryError�assertInZ exception�filenamerZ r[ r] r^ r_ r3 rN �rmdir)r@ rp �tmpdir�cmZ
possible_argsrU r )rT r �test_rmtree_errors� s2 z TestShutil.test_rmtree_errors�chmodzrequires os.chmod()N� rC z2This test can't be run on Cygwin (issue #1071513).�geteuidr z9This test can't be run reliably as root (issue #1076467).c C s
d| _ tjt� | jtjt� tjjtd�| _ tjjtd�| _
t j | j � tj| j
� tj
t�j}tj
| j �j}tj
| j
�j}t
jt
jB }tj| j |� tj| j
|� tjt|� | jtjt|� | jtj| j |� | jtj| j
|� tjt| jd� | j| j dd� d S )Nr �a�b)rU � z1Expected call to onerror function did not happen.)�
errorStater rL r �
addCleanuprF rG r& r' �child_file_path�child_dir_pathr �create_empty_file�stat�st_mode�S_IREAD�S_IEXECru �check_args_to_onerrorr] )r@ Z old_dir_modeZold_child_file_modeZold_child_dir_modeZnew_moder r r �
test_on_error� s(
zTestShutil.test_on_errorc C s� | j dk r�|tjkr$| j|| j� n:|tjkr>| j|| j� n | j|tj� | j |t
| jg� | j t |d t
�� | j d7 _ n4| j|tj� | j|t
� | j t |d t
�� d| _ d S )NrW r rV rz )r{ r �unlinkr] r} rq r~ r_ r3 ro r rZ �
issubclassr )r@ r �arg�excr r r r� � s
z TestShutil.check_args_to_onerrorc sL z>t j� � fdd�}|t _t jt� ttdfd� tjt� W d � t _X d S )Nc s | t krt� �n� | �S d S )N)r r )�fnr r )�
orig_lstatr r �raiser s zFTestShutil.test_rmtree_does_not_choke_on_failing_lstat.<locals>.raiserrK )r �lstatrL r r, rF rG )r@ 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 � }tjj|d�}tjj|d�}tjj|d�}tjj|d�}t|d� t|d� tj||� tj||� tj|tjtj B � tj|tj
� | j tj|�j tj|�j � t
j||� | jtj|�j tj|�j � tjdk�r�tj|tj
� t
j||� | jtj|�j tj|�j � tj|tj
� t
j||� | jtj|�j tj|�j � tj|tj
� t
j||� | jtj|�j tj|�j � d S )NrK �bar�baz�quuxrB )rJ r r& r' r, rX ru r� �S_IRWXU�S_IRWXG�S_IRWXO�assertNotEqualr� rF �copymoder] r9 )r@ �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 sp | j � }tjj|d�}tjj|d�}tjj|d�}tjj|d�}t|d� t|d� tj||� tj||� tj|tjtj B � tj|tj� tj
|tj tj B � tj
|tj � t j
||dd� | jtj|�jtj|�j� | jtj|�jtj|�j� tj
|tj � t j
||dd� | jtj|�jtj|�j� tj
|tj � t j
||dd� | jtj|�jtj|�j� d S )NrK r� r� r� F)�follow_symlinks)rJ r r& r' r, rX ru r� r� r� r� r� rF r� r] r� r� r� )r@ 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 � }tjj|d�}tjj|d�}tjj|d�}tjj|d�}t|d� t|d� tj||� tj||� tj||dd� d S )NrK r� r� r� F)r� )rJ r r& r' r, rX rF r� )r@ 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 � }tjj|d�}tjj|d�}tjj|d�}tjj|d�}t|d� tj|�}tj||j|jd f� t|d� | j tj|�jtj|�j� tj
||� tj
||� t td�r�tj |tj
� t td�r�t td�r�tj|tj� tj|�}t td��r tj||d d
� | j |jtj|�j� tj||d d
� tj|�}tjtjk�rtx*dD ]"} | jt|| �t|| �d � �qNW t td��r�| j|j|j� t td��r�t |d��r�| j|j|j� tj||d d
� | jttj|�jtj|�j �dk � d S )NrK r� r� �quxg E@r� �lchflags� UF_NODUMPT)r� F�st_atime�st_mtimerV �st_flagsg�������?)r� r� )rJ r r& r' r, r� �utimer� r� r� rX �hasattrr� r� r� r� r� rF �copystatr� �supports_follow_symlinks�assertLessEqualr r] r� rZ �abs)
r@ 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 � }tjj|d�}tjj|d�}t|d� t|d� dd� }tj}zTx*tjtjfD ]}||�t_t j
||� qVW |tjtj �t_| j t t j
||� W d |t_X d S )Nrd �file2�xxxc s t � �dd�� �fdd�
}|S )NT)r� c 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)
rJ r r& r' r, r� r r� r� rF r� rY r )r@ rr rd r� r� Z old_chflagsr� r r r �-test_copystat_handles_harmless_chflags_errors� s
z8TestShutil.test_copystat_handles_harmless_chflags_errorsc s� | j � }tjj|d��t�d� tjj|d�}t|d� tj�|� tj�dd� tj�dd� tj�|� | jt tj
���t tj
|��� | jtj �d�tj |d�� tj |� t|d� t
tjd����fdd �}z.tj�|t_tj�|� | jdtj
|�� W d �t_X d
d �� �fd d
�
}z tj
� |t_
tj�|� W d � t_
X tjj|d��t��� tj�dd� tjj|d�}t||� tj�|� | jtj |d�d� d S )NrK r� zuser.foos 42zuser.bars 43�EPERMc s" |dkr��n� | ||f|� d S )Nzuser.foor )�fnamer� �valr )�
orig_setxattr�os_errorr r �_raise_on_user_foo� s z5TestShutil.test_copyxattr.<locals>._raise_on_user_fooT)r� c s | �krt tjd��� | |d�S )NzOperation not supported)r� )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)rJ r r& r' r, rF �
_copyxattr�setxattrr] r2 � listxattr�getxattr�remover r r� ro r� )r@ r� r� r� r� r )r� r� r� r� r �test_copyxattr� sH
zTestShutil.test_copyxattrzroot privileges requiredc C s� | j � }tjj|d�}tjj|d�}t|d� tj||� tj|dd� tj|dddd� tjj|d�}tjj|d �}t|d� tj||� tj||dd� | j tj
|ddd�d� | j t tj
|d� tj||dd� | j tj
|d�d� d S )
NrK r� z trusted.foos 42s 43F)r� r� r� )
rJ r r& r' r, rX r� rF r� r] r� rY r )r@ r� r� r� r� r� r r r �test_copyxattr_symlinks� s
z"TestShutil.test_copyxattr_symlinksc C s | j � }tjj|d�}tjj|d�}tjj|d�}t|d� tj||� ttd�rftj|tj tj
B � t j ||dd� | j
tjj|�� | jt|�t|�� tj|� t j ||dd� | jtjj|�� | jtj|�tj|�� ttd��r| jtj|�jtj|�j� d S )NrK r� r� r� T)r� F)rJ r r& r' r, rX r� r� r� r� r� rF �copyrh r5 r] r0 r� rZ �readlinkr� r� )r@ r� r� r� r� r r r �test_copy_symlinks� s$
z TestShutil.test_copy_symlinksc C s� | j � }tjj|d�}tjj|d�}tjj|d�}t|d� tj||� ttd�rftj|tj tj
B � ttd�r�ttd�r�tj |tj � tj|�}tj
|�}tj||dd� | jtjj|�� | jt|�t|�� tj|� tj||d d� | jtjj|�� | jtj|�tj|�� tj
|�}tjtjk�rZx*dD ]"}| jt||�t||�d � �q4W ttd��r�| j|j|j� | j |j|j� ttd��r�t|d
��r�| j|j |j � d S )NrK r� r� r� r� r� T)r� Fr� r� rV r� )r� r� ) rJ r r& r' r, rX r� r� r� r� r� r� r� r� rF �copy2rh r5 r] r0 r� rZ r� r� r� r� r r� r� r� ) r@ r� r� r� r� r� r� �dst_statr� r r r �test_copy2_symlinks
s:
z TestShutil.test_copy2_symlinksc C sr | j � }tjj|d�}tjj|d�}t|d� tj|dd� tj||� | jtj |d�tj |d�� tj
|� d S )NrK r� zuser.foos 42) rJ r r& r' r, r� rF r� r] r� r� )r@ r� r� r� r r r �test_copy2_xattr0 s
zTestShutil.test_copy2_xattrc C s� | j � }tjj|d�}tjj|d�}tjj|d�}tjj|d�}t|d� tj||� tj||dd� | jtjj |�� | j
tj |�tj |�� tj||� | j tjj |�� d S )Nr� r� r� rS rK F)r� )
rJ r r& r' r, rX rF �copyfilerZ r5 r] r� rh )r@ r� r� r� r� rS r r r �test_copyfile_symlinks= s
z!TestShutil.test_copyfile_symlinksc s� t jt jt jt jht jko0t jt jko0t jt jk}|r�| j t
j � | j t
j j
� | j� }t jj|d�}t j|� z<t
j}G dd� dt�� � fdd�}|t
_| j� t
j |� W d |t
_X n| jt
j � | jt
j j
� d S )Nrx c @ s e Zd ZdS )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� rq �supports_dir_fdr3 � supports_fdr� rZ rF �_use_fd_functionsrG Zavoids_symlink_attacksrJ r& r' rL Z_rmtree_safe_fd� ExceptionrY rh )r@ r� r� rH 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 s4 t j� \}}tj|� | jttj|� tj|� d S )N) rI Zmkstempr �closerY rn rF rG r� )r@ Zhandler& r r r � test_rmtree_dont_delete_filef s
z'TestShutil.test_rmtree_dont_delete_filec C s t j� }tjjt j� d�}| jtj|� | jtjtjj|�� t |dfd� tj
tjj|d�� t |ddfd� tj ||� | j tjj
tjj|d��� | j tjjtjj|d��� | j tjj
tjj|dd��� t|df�}| j|d� t|ddf�}| j|d� d S )N� destinationztest.txt�123�test_dir�456)rI rJ r r& r' r| rF rG �dirnamer, rL �copytreerZ �isfiler4 r0 r] )r@ �src_dir�dst_dir�actualr r r �test_copytree_simplem s
z TestShutil.test_copytree_simplec C sl | j � }tjj|d�}tjj|d�}tjj|d�}tj|� tj|� t|dfd� tjj|d�}tjj|d�}tjtjj|d�|� ttd�r�tj|t j
t j B � ttd �r�tt d
�r�tj |t j
� tj|�}tj||d d � | jtjjtjj|dd��� | jtjtjj|dd��tjj|d�� tj|�}ttd��rL| j|j|j� ttd ��rh| j|j|j� d S )
Nr� r� �subzfile.txtrK rS zsub/linkr� r� r� T)�symlinks)rJ r r& r' rL r, rX r� r� r� r� r� r� r� r� rF r� rZ r5 r] r� r� r� ) r@ r� r� r� Zsub_dirr� r� r� r� r r r �test_copytree_symlinks� s2
z!TestShutil.test_copytree_symlinksc # C s$ t jj}t jj}tj� }�z�|tj� d�}t|dfd� t|dfd� t j||d�� t|ddfd� t j||d�� t|ddfd� t j||dd�� t j||dd �� t|dddfd� t|dd d
fd� z\tj d d�}tj
|||d � | j |||d��� | j |||d��� | j |||d��� W d tj
|� X z`tj d d
�}tj
|||d � | j |||d��� | j |||dd ��� | j |||dd��� W d tj
|� X zJdd� }tj
|||d � | j |||dd d
��� | j |||dd��� W d tj
|� X W d tj
|� tj
t jj|�� X d S )Nr� ztest.txtr� ztest.tmpr� r� Z test_dir2�subdirZsubdir2ztest.pyz*.tmp)�ignorezsubdir*c S sj g }x`|D ]X}t jj| |�}t jj|�rD|j� d dkrD|j|� q
t jj|�d dkr
|j|� q
W |S )NrV r� z.py���r� )r r& r' r4 �splitr6 �splitext)r� �namesr8 r9 r& r r r �_filter� s
z6TestShutil.test_copytree_with_exclude.<locals>._filter)r r& r' r[ rI rJ r, rL rF �ignore_patternsr� rZ rh rG r� )r@ r' r[ r� r� Zpatternsr� r r r �test_copytree_with_exclude� sL
z%TestShutil.test_copytree_with_excludec C sj t j� }tjj|d�}tj|� tjj|d�}| jtj|� tj |d� t
|dfd� tj tjj|d�d� t
|dfd� tj tjj|d�d� t j|d �}tj |d� tj ||� | j tj
|�jtj
|�j� | j tj
tjj|d��jtj
tjj|d��j� | j tj
tjj|d��jtj
tjj|d��j� tjj|tjj|�d
�}| j tj
|�jtj
|�j� d S ) N�sourcer� i� zpermissive.txtr� zrestrictive.txtr� i� )rR rV )rI rJ r r& r' rL r| rF rG ru r, r� r] r� r� r� )r@ r� r� r� Zrestrictive_subdirZrestrictive_subdir_dstr r r �!test_copytree_retains_permissions� s,
z,TestShutil.test_copytree_retains_permissionszos.chmodc
C sp t j� }tjjt j� d�}| jtj|� | jtjtjj|�� t d�|_
| j tj �� tj
||� W d Q R X d S )Nr� zka-boom)rI rJ r r& r' r| rF rG r� �PermissionErrorZ side_effectrY r r� )r@ Z
mock_patchr� r� r r r �test_copytree_winerror� s
z!TestShutil.test_copytree_winerrorrB z temporarily disabled on WindowsrS zrequires os.linkc ' C s� t jt� t jjtd�}t jjtd�}z�t|d��}|jd� W d Q R X yt j||� W n0 tk
r� } z| j d| � W Y d d }~X nX | j
t j t j
||� t|d��}| j|j� d� W d Q R X t j|� W d t jtdd� X d S ) N�cheese�shopr# �cheddarz
os.link(): %sr. T)rj )r rL r r& r' r( r) rS r� �skipTestrY rF r
r� r] r/ r� rG )r@ r� r� �f�er r r �'test_dont_copy_file_onto_link_to_itself s
z2TestShutil.test_dont_copy_file_onto_link_to_itselfc C s� t jt� t jjtd�}t jjtd�}ztt|d��}|jd� W d Q R X t jd|� | jt j
t j ||� t|d��}| j |j
� d� W d Q R X t j|� W d t jtdd� X d S )Nr� r� r# r� r. T)rj )r rL r r& r' r( r) rX rY rF r
r� r] r/ r� rG )r@ 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 st t jt� zTt jjtd�}t jjtd�}t j|� t j||� | jttj |� tj |dd� W d tj tdd� X d S )Nr� r� T)rj )
r rL r r& r' rX rY r rF rG )r@ r� r� r r r �test_rmtree_on_symlink. s
z!TestShutil.test_rmtree_on_symlink�mkfifozrequires os.mkfifo()c C s~ yt jt� W n0 tk
r> } z| jd| � W Y d d }~X nX z,| jtjtjtt � | jtjtjt
t� W d t j t� X d S )Nzos.mkfifo(): %s) r r r r� r� rY rF �SpecialFileErrorr� �TESTFN2�__file__r� )r@ r r r r �test_copyfile_named_pipe= s z#TestShutil.test_copyfile_named_pipec ' C s t jt� z�t jjtd�}t j|� t jj|d�}yt j|� W n0 tk
rp } z| jd| � W Y d d }~X nX ytj tt
� W n\ tj k
r� } z>|j d }| j
t|�d� |d \}}}| j
d| |� W Y d d }~X n X | jd� W d tjtdd � tjt
dd � X d S )
Nr� Zmypipezos.mkfifo(): %sr rV z`%s` is a named pipez$shutil.Error should have been raisedT)rj )r rL r r& r' r r� r� rF r� r r r r] r^ �failrG )r@ r� �piper rT r� r� Z error_msgr r r �test_copytree_named_pipeK s&
"z#TestShutil.test_copytree_named_pipec s� | j � }tjj| j � d�}t|dfd� tjtjj|d�� t|ddfd� g � � fdd�}tj|||d� | jt � �d � d S )
Nr� ztest.txtr� r� r� c s � j | |f� d S )N)r6 )r� r� )�copiedr r �_copym s z4TestShutil.test_copytree_special_func.<locals>._copy)�
copy_functionrW )
rJ r r& r' r, rL rF r� r] r^ )r@ r� r� r
r )r r �test_copytree_special_funcd s z%TestShutil.test_copytree_special_funcc C s� | j � }tjj| j � d�}tjdtjj|d�� tjtjj|d�� t|ddfd� | jtt j
||� tjj| j � d�}t j
||dd� | j dtj |�� tjj| j � d �}t j
||dd
� | j
dtj |�� d S ) Nr� Z
IDONTEXISTztest.txtr� r� � destination2T)Zignore_dangling_symlinksZ destination3)r� )rJ r r& r' rX rL r, rY r rF r� � assertNotInr3 ro )r@ r� r� r r r � test_copytree_dangling_symlinkss s z*TestShutil.test_copytree_dangling_symlinksc
C s | j � }tjj| j � d�}tjtjj|d�� ttjj|dd�d�� W d Q R X tjtjj|d�tjj|d�dd� tj||dd � | j tjj
tjj|d��� | j dtj tjj|d��� tjj| j � d
�}tj||dd � | j
tjj
tjj|d��� | j dtj tjj|d��� d S ) Nr� Zreal_dirztest.txtr# Z link_to_dirT)Ztarget_is_directoryF)r� r )rJ r r& r' rL r( rX rF r� rh r5 ro r3 rZ )r@ r� r� r r r �test_copytree_symlink_dir� s
z$TestShutil.test_copytree_symlink_dirc C sP d}| j � }t||fd� tjj||�}| j � }|||� tjj||�}||fS )Nztest.txtr� )rJ r, r r& r' )r@ �methodr� rr rd �tmpdir2r� r r r �
_copy_file� s
zTestShutil._copy_filec C sB | j tj�\}}| jtjj|�� | jtj|�j tj|�j � d S )N)
r rF r� rZ r r& r[ r] r� r� )r@ rd r� r r r � test_copy� s zTestShutil.test_copyr� zrequires os.utimec C s� | j tj�\}}| jtjj|�� tj|�}tj|�}| j|j |j � x(dD ] }| j
t ||�t ||�d � qLW t td�r�t |d�r�| jt |d�t |d�� d S )Nr� r� rV r� r� )r� r� )
r rF r� rZ r r& r[ r� r] r� r� r r� )r@ rd r� Z
file1_statZ
file2_statr� r r r �
test_copy2� s
zTestShutil.test_copy2c $ C sf | j d�\}}| j� }tj|� tjj|�}tjjtjj|�d�}tj |�� tjj
|�}t |d|d�}W d Q R X | j ||d � | j
tjj|�� | j
tj|�� tj|d��"}| j|j� dddd d
d g� W d Q R X tj |�� t |d |d�}W d Q R X | j ||d
� | j
tjj|�� | j
tj|�� tj|d��"}| j|j� dddd d
d g� W d Q R X d S )Nrk �archive�gztar�.z.tar.gzzr:gzz./subz./sub2z./file1z./file2z ./sub/file3�tarz.tarr. )�
_create_filesrJ r rq r& r� r' �basenamer �
change_cwd�abspathr r] rZ r� �tarfileZ
is_tarfiler( �assertCountEqual�getnames) r@ �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|�S Q R X d S )N)r! r( r# �sortr% )r@ r& r r� r r r �_tarinfo� s zTestShutil._tarinfo�distc C s� | j � }tjj||�}tj|dd� t|dfd� t|dfd� tjtjj|d�� t|ddfd� tjtjj|d�� |r�t|d fd� ||fS )
NT)�exist_okrd r� r� r� Zfile3Zsub2�outer)rJ r r& r' �makedirsr, rL )r@ r% r$ r- r r r r � s zTestShutil._create_filesr zNeed the tar command to runc C s | j � \}}tjj| j� d�}t|d||�}| j||d � | jtjj|�� tjj|d�}ddd|g}t j
||t j d� | jtjj|�� | j| j |�| j |�� t|d||�}| j||d � | jtjj|�� t|d||d d
�}| j||d � | jtjj|�� d S ) Nr r z.tar.gzz archive2.tarr z-cf)�cwd�stdoutz.tarT)Zdry_run)
r r r& r' rJ r r] rZ r� �
subprocess�
check_call�DEVNULLr, )r@ r$ r% r( r) Ztarball2Ztar_cmdr r r �test_tarfile_vs_tar� s$
z TestShutil.test_tarfile_vs_tarc $ C sl | j � \}}| j� }tj|� tjj|�}tjjtjj|�d�}tj |�� tjj
|�}t |d|�}W d Q R X | j ||d � | j
tjj|�� | j
tj|�� tj|��$}| j|j� dddddd d
g� W d Q R X tj |�� tjj
|�}t |d||�}W d Q R X | j ||d � | j
tjj|�� | j
tj|�� tj|��"}| j|j� dddddd g� W d Q R X d S ) Nr �zipz.zipzdist/z dist/sub/z
dist/sub2/z
dist/file1z
dist/file2zdist/sub/file3r/ )r rJ r rq r& r� r' r r r r r r] rZ r� �zipfileZ
is_zipfile�ZipFiler" �namelist) r@ r$ r% r r& r' r( r8 �zfr r r �test_make_zipfile s4
z TestShutil.test_make_zipfiler7 zNeed the zip command to runc
C s� | j � \}}tjj| j� d�}t|d||�}| j||d � | jtjj|�� tjj|d�}dddd|g}t j
||t j d� | jtjj|�� t j
|��}|j� }W d Q R X t j
|��}|j� } W d Q R X | jt|�t| �� d S )Nr r7 z.zipz archive2.zipz-qz-r)r1 r2 )r r r& r' rJ r r] rZ r� r3 r4 r5 r8 r9 r: r2 )
r@ r$ r% r( r Zarchive2�zip_cmdr; r� Znames2r r r �test_zipfile_vs_zip= s
z TestShutil.test_zipfile_vs_zip�unzipz Need the unzip command to runc C s� | j � \}}tjj| j� d�}t|d||�}| j||d � | jtjj|�� dd|g}t j
|��v yt j |t j
d� W nZ t jk
r� } z<|jjdd�}d |kr�| jd
� d }| j|j||�� W Y d d }~X nX W d Q R X d S ) Nr r7 z.zipr? z-t)�stderr�replace)rT zunrecognized option: tzunzip doesn't support -tz{}
**Unzip Output**
{})r r r& r' rJ r r] rZ r� r r r3 � check_outputZSTDOUT�CalledProcessError�output�decoder� r �format) r@ r$ r% r( r r= r� Zdetails�msgr r r �test_unzip_zipfileW s
z TestShutil.test_unzip_zipfilec C s* | j � }tjj|d�}| jtt|d� d S )Nr r� )rJ r r& r' rY �
ValueErrorr )r@ rr r( r r r �test_make_archiveo s z TestShutil.test_make_archivec C s� t r"tjd�d }tjd�d }nd }}| j� \}}tjj| j � d�}t
|d||||d�}| j tjj |�� t
|d||�}| j tjj |�� t
|d||||d�}| j tjj |�� t
|d||ddd�}| j tjj |�� d S ) Nr �rootr r7 )�owner�groupr Z kjhkjhkjgZoihohoh)
�UID_GID_SUPPORT�grp�getgrgid�pwd�getpwuidr r r& r' rJ r rZ r� )r@ rM rL r$ r% r( r8 r r r � test_make_archive_owner_groupt s"
z(TestShutil.test_make_archive_owner_groupz Requires grp and pwd supportc C s� | j � \}}tjj| j� d�}tjd�d }tjd�d }t j
|�� t |d|d||d�}W d Q R X | j tjj
|�� tj|�}z2x,|j� D ] }| j|jd� | j|jd� q�W W d |j� X d S )Nr r r r- )rL rM )r r r& r' rJ rO rP rQ rR r r r rZ r� r! r( Z
getmembersr] �uid�gidr� ) r@ r$ r% r( rM rL Z archive_namer �memberr r r �test_tarfile_root_owner� s
z"TestShutil.test_tarfile_root_ownerc C sn t j� }dd� }td|g d� z@ytdd| j� d� W n tk
rJ Y nX | jt j� |� W d td� X d S )Nc _ s
t � �d S )N)� RuntimeError)r �kwr r r �_breaks� s z1TestShutil.test_make_archive_cwd.<locals>._breaksr� zxxx file)r$ )r �getcwdr r rJ r� r] r )r@ Z current_dirrZ r r r �test_make_archive_cwd� s z TestShutil.test_make_archive_cwdc C sF | j � }tj|��* | jtdd�d� | jtjjd�� W d Q R X d S )N�testr ztest.tar) rJ r r r] r rZ r r& r� )r@ r$ r r r �test_make_tarfile_in_curdir� s z&TestShutil.test_make_tarfile_in_curdirc C sF | j � }tj|��* | jtdd�d� | jtjjd�� W d Q R X d S )Nr] r7 ztest.zip) rJ r r r] r rZ r r& r� )r@ r$ r r r �test_make_zipfile_in_curdir� s z&TestShutil.test_make_zipfile_in_curdirc C s� | j ttdd� | j ttddd� d� | j ttddd� d
dg� tddd� dgd � d
d � t� D �}| jd|� td� d d � t� D �}| jd|� d S )Nr� rV 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 rW rz c S s t S )N)r` r r r r ra � s zxxx filec S s g | ] \}}|�qS r r )�.0r9 �paramsr r r �
<listcomp>� s z;TestShutil.test_register_archive_format.<locals>.<listcomp>c S s g | ] \}}|�qS r r )rb r9 rc r r r rd � s )rV rW )rV rW rz )rV rW )rY � TypeErrorr r ro r r )r@ �formatsr r r � test_register_archive_format� s
z'TestShutil.test_register_archive_formatc C s. | j |dd� � | j |tj� | j |t� d S )Nc S s | S )Nr )r& r r r ra � s z1TestShutil.check_unpack_archive.<locals>.<lambda>)�#check_unpack_archive_with_converter�pathlibZPathr )r@ rF r r r �check_unpack_archive� s z TestShutil.check_unpack_archivec
C s� | j � \}}t|�}|jd� tjj| j� d�}t||||�}| j� }t||� | j t|�|� | j� } t|| |d� | j t| �|� | j
t j tt
� | j
ttt
dd� d S )Nr/ r )rF r� )r r7 r� r r& r' rJ r r r] rY rF Z ReadErrorr rI )
r@ rF Z converterr$ r% �expectedr( rp r Ztmpdir3r r r rh � s
z.TestShutil.check_unpack_archive_with_converterc C s | j d� d S )Nr )rj )r@ r r r �test_unpack_archive_tar� s z"TestShutil.test_unpack_archive_tarc C s | j d� d S )Nr )rj )r@ r r r �test_unpack_archive_gztar� s z$TestShutil.test_unpack_archive_gztarc C s | j d� d S )NZbztar)rj )r@ r r r �test_unpack_archive_bztar� s z$TestShutil.test_unpack_archive_bztarc C s | j d� d S )NZxztar)rj )r@ r r r �test_unpack_archive_xztar� s z$TestShutil.test_unpack_archive_xztarc C s | j d� d S )Nr7 )rj )r@ r r r �test_unpack_archive_zip� s z"TestShutil.test_unpack_archive_zipc s� t � }� fdd�}tdddg|d g� tdd � � jttd
dg|� td� td
dg|� � jd
dgd ft � � � jddgd ft � � td
� � jt � |� d S )
Nc s( � j |d� � j | d� � j |d� d S )NrV z stuff.boo�xx)r] )rp Z extract_dir�extra)r@ r r �_boo s z-TestShutil.test_unpack_registry.<locals>._booZBooz.booz.b2rr rV z stuff.boorq ZBoo2rk )rr rV ) r r r rY r
r ro r r] )r@ rf rs r )r@ r �test_unpack_registry s
z TestShutil.test_unpack_registry�
disk_usagez)disk_usage not available on this platformc C s` t jtjjt��}| j|jd� | j|jd� | j |j
d� | j |j|j� | j|j|j
� d S )Nr ) rF ru r r& r� r Z
assertGreaterZtotalZused�assertGreaterEqualZfree)r@ �usager r r �test_disk_usage s zTestShutil.test_disk_usage�chownzrequires os.chownc ' s �j � }tj|d�� t� d� �jt�� tj� � W d Q R X �jt�� tj� dd� W d Q R X �jt�� tj� dd� W d Q R X �jt �� tj� d� W d Q R X �jt �� tj� d� W d Q R X t
j � }t
j � }d
� �fd d
� }tj� ||� |� ||� tj� |� |� |� tj� |d� |� |� tj� |d� |� |d � tj|||� ||||� tj||� |||� tj||d� |||� tj||d� |||d � t
j|�d }tj|�d }tj� ||� |� ||� tj|||� ||||� d S )N)rR ztesting chown functionznon-existing username)�userznon-existing groupname)rM s spamg ��Q� @c s: t j� �}|d k r �j||j� |d k r6�j||j� d S )N)r r� r] �st_uid�st_gid)r& rT rU �s)rp r@ r r � check_chownA s
z*TestShutil.test_chown.<locals>.check_chown)rU r )NN)rJ rI rl r, rY rI rF ry � LookupErrorre r �getuid�getgidrQ rR rO rP )r@ r� rT rU r~ rz rM r )rp r@ r �
test_chown& sL
zTestShutil.test_chownc C s� x�t jt jfD ]v}| j� }| j� }tjj|d�}t|d� |||�}| j|tjj|d�� ||tjj|d��}| j|tjj|d�� qW d S )NrK r� ) rF r� r� rJ r r& r' r, r] )r@ r� r� r� r� �rvr r r �test_copy_return_valuea s
z!TestShutil.test_copy_return_valuec C sl | j � }| j � }tjj|d�}tjj|d�}t|d� tj||�}| jtjj|�� | j t
|�t
|�� d S )Nr� rK ) rJ r r& r' r, rF r� rZ r[ r] r0 )r@ r� r� �dst_file�src_filer� r r r �test_copyfile_return_valuem s
z%TestShutil.test_copyfile_return_valuec C sH | j � }tjj|d�}t|d� | jttj||� | jt tj||� d S )NrK )
rJ r r& r' r, rY r
rF r� r )r@ r� r� r r r �test_copyfile_same_filex s
z"TestShutil.test_copyfile_same_filec C s\ | j � }|d }| jtj|d� tjj|d�}t|d� tj||�}| j dgtj
|�� d S )N�destTrK ) rJ r| rF rG r r& r' r, r� r] r3 )r@ r� r� r� r� r r r �test_copytree_return_value� s
z%TestShutil.test_copytree_return_value)r- )\r� r� r� r> rD rJ rQ r �skip_unless_symlinkr` ri rt �unittest�
skipUnlessr� r �skipIf�sys�platformrw r� r� r� r� r� r� r� r r� Zskip_unless_xattrr� r� r� r� r� r� r� r� r� r� r� r� ZmockZpatchr� r9 r r r r r r r r r r r Z
requires_zlibr* r, r rF �whichr6 r< r> rH rJ rS rN rW r\ r^ r_ rg rj rh rl rm Z requires_bz2rn Z
requires_lzmaro rp rt rx r� r� r� r� r� �
__classcell__r r )rA r r< [ s� $
) 8#
A
$
%
:
r< c @ s� e Zd Zdd� Zdd� Zdd� Zdd� Zd d
� Zej e
e d �oHe j � d kd
�dd� �Z
dd� Zdd� Zejejdkd�dd� �Zdd� Zdd� Zd d � Zd S ) � TestWhichc C sp t jdd�| _| jtj| jd� t j| jddd�| _tj | jj
t j � | j| jj
� tjj| jj
�\| _| _d S )NZTmp)�prefixTz.Exe)rR r� �suffix)rI rJ �temp_dirr| rF rG ZNamedTemporaryFile� temp_filer ru r9 r� �S_IXUSRr� r& r� rR �file)r@ r r r r> � s
zTestWhich.setUpc C s& t j| j| jd�}| j|| jj� d S )N)r& )rF r� r� rR r] r� r9 )r@ r� r r r �
test_basic� s zTestWhich.test_basicc C s( t j| jj| jd�}| j|| jj� d S )N)r& )rF r� r� r9 r� r] )r@ r� r r r �test_absolute_cmd� s zTestWhich.test_absolute_cmdc C s� t jj| j�\}}t jj|| j�}tj|d��" tj || j
d�}| j ||� W d Q R X tj| jd�� tj ||d�}| j |� W d Q R X d S )N)r& )
r r& r� rR r' r� r r rF r� r� r] � assertIsNone)r@ r% �tail_dir�relpathr� r r r �test_relative_cmd� s zTestWhich.test_relative_cmdc C sl t jj| j�}tj| jd��F tj| j|d�}t j
dkrT| j |t jj t j
| j�� n
| j|� W d Q R X d S )N)r& �win32)r r& r� rR r r rF r� r� r� r� r] r' �curdirr� )r@ r% r� r r r �test_cwd� s
zTestWhich.test_cwdrw r znon-root user requiredc C sR t j| jjtj� t j| jjt j�r.| jd� t j
| j | j t jd�}| j
|� d S )Nz can't set the file read-only)r& �mode)r ru r� r9 r� r� �access�W_OKr� rF r� r� rR r� )r@ r� r r r �test_non_matching_mode� s
z TestWhich.test_non_matching_modec C sV t jj| j�\}}tj|d��. tj| j|d�}| j |t jj
|| j�� W d Q R X d S )N)r& ) r r& r� rR r r rF r� r� r] r' )r@ r% r� r� r r r �test_relative_path� s z TestWhich.test_relative_pathc C s t jd| jd�}| j|� d S )Nzfoo.exe)r& )rF r� rR r� )r@ r� r r r �test_nonexistent_file� s z TestWhich.test_nonexistent_filer� z pathext check is Windows-onlyc C s: t j| jd d� | jd�}| j|| jjd d� d � d S )N� )r& z.EXE���r� )rF r� r� rR r] r� r9 )r@ r� r r r �test_pathext_checking� s z TestWhich.test_pathext_checkingc
C s> t j� �,}| j|d< tj| j�}| j|| jj� W d Q R X d S )N�PATH) r �EnvironmentVarGuardrR rF r� r� r] r� r9 )r@ �envr� r r r �test_environ_path� s
zTestWhich.test_environ_pathc C sd t jj| j�}tj| jd��> tj� �*}| j|d< tj| j dd�}| j
|� W d Q R X W d Q R X d S )N)r& r� rk ) r r& r� rR r r r� rF r� r� r� )r@ r% r� r� r r r �test_empty_path� s
zTestWhich.test_empty_pathc
C s: t j� �(}|jdd � tj| j�}| j|� W d Q R X d S )Nr� )r r� rE rF r� r� r� )r@ 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 rw r� r� r� r� r� r� r� r� r� r� r r r r r� � s
r� c @ s, e Zd Zdd� Zdd� Zdd� Zdd� Zd d
� Zd d � Ze d
d� �Z
e dd� �Z dd� Z e dd� �Z
dd� Ze dd� �Zdd� Zejejjd�d d � �Zd d � Zd d!� Zd"d#� Zd$d%� Zej e d&d'� ��Z ej e d(d)� ��Z ej e d*d+� ��Z ej e d,d-� ��Z d.d/� Z!d0d1� Z"e d2d3� �Z#e d4d5� �Z$d6S )7�TestMovec C sb d}t j� | _t j� | _tjj| j|�| _tjj| j|�| _t | jd��}|j
d� W d Q R X d S )NrK r" s spam) rI rJ r� r� r r& r' r� r� r( r) )r@ rp r� r r r r> � s
zTestMove.setUpc
C s: x4| j | jfD ]$}y|r"tj|� W q Y qX qW d S )N)r� r� rF rG )r@ rH r r r rD � s zTestMove.tearDownc C sf t |d��}|j� }W d Q R X tj||� t |d��}| j||j� � W d Q R X | jtjj|�� d S )Nr- ) r( r/ rF �mover] rh r r& r[ )r@ r� r� �real_dstr� �contentsr r r �_check_move_file s zTestMove._check_move_filec C sF t tj|��}tj||� | j|t tj|��� | jtjj|�� d S )N) r2 r r3 rF r� r] rh r& r[ )r@ 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� )r@ r r r �test_move_file s zTestMove.test_move_filec C s | j | j| j| j� d S )N)r� r� r� r� )r@ r r r �test_move_file_to_dir s z TestMove.test_move_file_to_dirc C s | j � d S )N)r� )r@ r r r �test_move_file_other_fs s z TestMove.test_move_file_other_fsc C s | j � d S )N)r� )r@ r r r � test_move_file_to_dir_other_fs s z'TestMove.test_move_file_to_dir_other_fsc C s@ t j� }z| j| j||� W d ytj|� W n Y nX X d S )N)rI rl r� r� rF rG )r@ r� r r r �
test_move_dir$ s zTestMove.test_move_dirc C s | j � d S )N)r� )r@ r r r �test_move_dir_other_fs/ s z TestMove.test_move_dir_other_fsc C s, | j | j| jtjj| jtjj| j��� d S )N)r� r� r� r r& r' r )r@ r r r �test_move_dir_to_dir4 s z TestMove.test_move_dir_to_dirc C s | j � d S )N)r� )r@ r r r � test_move_dir_to_dir_other_fs9 s z&TestMove.test_move_dir_to_dir_other_fsc C s4 | j | jtjj | jtjj| jtjj| j��� d S )N)r� r� r r& �sepr� r' r )r@ r r r �test_move_dir_sep_to_dir> s z!TestMove.test_move_dir_sep_to_dirzrequires os.path.altsepc C s4 | j | jtjj | jtjj| jtjj| j��� d S )N)r� r� r r& �altsepr� r' r )r@ r r r �test_move_dir_altsep_to_dirB s z$TestMove.test_move_dir_altsep_to_dirc C s4 t | jd�� W d Q R X | jtjtj| j| j� d S )Nr" )r( r� rY rF r r� r� r� )r@ r r r �"test_existing_file_inside_dest_dirG s
z+TestMove.test_existing_file_inside_dest_dirc C s* t jj| jd�}| jtjtj| j|� d S )Nr� )r r& r' r� rY rF r r� )r@ r� r r r � test_dont_move_dir_in_itselfM s z%TestMove.test_dont_move_dir_in_itselfc
C sp t jt� zPxJdD ]B\}}t jjt|�}t jjt|�}| jtj||�d||f d� qW W d tjtdd� X d S ) N�srcdir� srcdir/destz?_destinsrc() wrongly concluded that dst (%s) is not in src (%s))rG T)rj �r� r� )r� ) r rL r r& r' rZ rF �
_destinsrcrG )r@ r� r� r r r � test_destinsrc_false_negativeR s
z&TestMove.test_destinsrc_false_negativec
C sp t jt� zPxJd
D ]B\}}t jjt|�}t jjt|�}| jtj||�d||f d� qW W d tjtdd� X d S ) Nr� �src/dest�
srcdir.newz;_destinsrc() wrongly concluded that dst (%s) is in src (%s))rG T)rj �r� r� �r� r� )r� r� ) r rL r r& r' rh rF r� rG )r@ r� r� r r r � test_destinsrc_false_positive^ s
z&TestMove.test_destinsrc_false_positivec C s\ t jj| jd�}t j| j|� tj|| j� | j t jj
| j�� | j t jj | j| j�� d S )Nr� ) r r& r' r� rX r� rF r� r� rZ r5 �samefile)r@ r� r r r �test_move_file_symlinkj s
z TestMove.test_move_file_symlinkc C sl d}t jj| j|�}t j| j|� tj|| j� t jj| j|�}| j t jj
|�� | j t jj | j|�� d S )Nr� ) r r& r' r� rX r� rF r� r� rZ r5 r� )r@ rp r� Z
final_linkr r r � test_move_file_symlink_to_dirs s z&TestMove.test_move_file_symlink_to_dirc C s� t jj| jd�}t jj| jd�}t j||� t jj| jd�}tj||� | jt jj |�� t j
dkr�| j t jj |�t j
|�� n | j t jj |�t jj |�� d S )Nr� r� r� rB )r r& r' r� rX r� rF r� rZ r5 r9 r] �realpathr� )r@ r� r� r� r r r �test_move_dangling_symlink~ s
z#TestMove.test_move_dangling_symlinkc C s| t jj| jd�}t jj| jd�}t j|� t j||� t jj| jd�}tj||� | j t jj
|�� | j t jj ||�� d S )Nr� r� r� ) r r& r' r� rL rX r� rF r� rZ r5 r� )r@ r� r� r� r r r �test_move_dir_symlink� s
z TestMove.test_move_dir_symlinkc C s6 t j| j| j�}| j|tjj| jtjj| j��� d S )N) rF r� r� r� r] r r& r' r )r@ r� r r r �test_move_return_value� s z TestMove.test_move_return_valuec C s6 t j| jtjj| jd��}| j|tjj| jd�� d S )Nr� )rF r� r� r r& r' r� r] )r@ r� r r r � test_move_as_rename_return_value� s z)TestMove.test_move_as_rename_return_valuec s8 g � � fdd�}t j| j| j|d� | jt� �d� d S )Nc s � j | |f� d S )N)r6 )r� r� )�movedr r r
� s z7TestMove.test_move_file_special_function.<locals>._copy)r rV )rF r� r� r� r] r^ )r@ r
r )r� r � test_move_file_special_function� s z(TestMove.test_move_file_special_functionc sd g � � fdd�}t jtjj| jd�� t jtjj| jd�� tj| j| j|d� | j t
� �d� d S )Nc s � j | |f� d S )N)r6 )r� r� )r� r r r
� s z6TestMove.test_move_dir_special_function.<locals>._copyZchildZchild1)r rz ) r r r r& r' r� rF r� r� r] r^ )r@ r
r )r� r � test_move_dir_special_function� s z'TestMove.test_move_dir_special_functionN)%r� r� r� r> rD 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 @ sX e Zd ZdZG dd� de�Zdd� Zdd� Zdd � Zd
d � Z d d
� Z
dd� Z dd� Z dS )� TestCopyFileFc @ s: e Zd ZdZdZdZd dd�Zdd� Zdd � Zd
d � Z dS )
zTestCopyFile.FauxFNTc C s || _ || _d S )N)�_raise_in_exit�_suppress_at_exit)r@ Z
raise_in_exitZsuppress_at_exitr r r �__init__� s zTestCopyFile.Faux.__init__c G s dS )Nrk r )r@ r r r r r/ � s zTestCopyFile.Faux.readc C s
d| _ d S )NT)�_entered)r@ r r r � __enter__� s zTestCopyFile.Faux.__enter__c C s& |||f| _ | jr d| _td��| jS )NTz Cannot close)� _exited_withr� �_raisedr r� )r@ �exc_typeZexc_valZexc_tbr r r �__exit__� s
zTestCopyFile.Faux.__exit__)FT)
r� r� r� r� r� r� r� r/ r� r� r r r r �Faux� s
r� c C s | j r
t`d S )N)�_deleterF r( )r@ r r r rD � s zTestCopyFile.tearDownc C s |t _d| _d S )NT)rF r( r� )r@ r r r r �_set_shutil_open� s z TestCopyFile._set_shutil_openc C s* ddd�}| j |� | jttjdd� d S )Nr. c S s | dkrt d��d S )N�srcfilezCannot open "srcfile")r )rp r� r r r �_open� s z4TestCopyFile.test_w_source_open_fails.<locals>._openr� �destfile)r. )r� rY r rF r� )r@ r� r r r �test_w_source_open_fails� s
z%TestCopyFile.test_w_source_open_failsc sd | j � � d � fdd� }| j|� tjdd� | j� j� | j� jd tk� | j� jd j d
� d S ) Nr. c s | dkr � S | dkr t d��d S )Nr� r� zCannot open "destfile")r )rp r� )r� r r r� � s
z2TestCopyFile.test_w_dest_open_fails.<locals>._openr� r� r rV �Cannot open "destfile")r. )r� )
r� r� rF r� rZ r� r� r r] r )r@ r� r )r� r �test_w_dest_open_fails� s
z#TestCopyFile.test_w_dest_open_failsc s� | j � �| j d�� d
� �fdd� }| j|� tjdd� | j�j� | j� j� | j� j� | j�jd tk� | j �jd j
d � d S ) NTr. c s | dkr �S | dkr� S d S )Nr� r� r )rp r� )r� r� r r r� � s
z3TestCopyFile.test_w_dest_close_fails.<locals>._openr� r� r rV � Cannot close)r. )r� ) r� r� rF r� rZ r� r� r� r r] r )r@ r� r )r� r� r �test_w_dest_close_fails� s
z$TestCopyFile.test_w_dest_close_failsc s� | j d��| j � � d� �fdd� }| j|� | jttjdd� | j�j� | j� j� | j� j � | j�j
d d k� | j�j � d S ) NTr. c s | dkr �S | dkr� S d S )Nr� r� r )rp r� )r� r� r r r� s
z5TestCopyFile.test_w_source_close_fails.<locals>._openr� r� r )r. ) r� r� rY r rF r� rZ r� rh r� r� )r@ r� r )r� r� r �test_w_source_close_fails s
z&TestCopyFile.test_w_source_close_failsc C s� t j� | _| jtj| jd� tjjtjj | j�tjj
| j�j � �}| j | j|� z$tj
| j|� | jtjj|�� W d tj|� X d S )NT)rI rJ r� r| rF rG r r& r' r� r �upperr� r� rZ r4 rq )r@ r� r r r � test_move_dir_caseinsensitive s
z*TestCopyFile.test_move_dir_caseinsensitiveN)
r� r� r� r� �objectr� rD r� r� r� r� r� r� r r r r r� � s
r� c @ s^ e Zd Zdd� Zdd� Zdd� Zejej e
j j � �d�eje
ed�d �d
d � ��Zd d
� ZdS )�
TermsizeTestsc C s( t j� }| j|jd� | j|jd� d S )Nr )rF �get_terminal_sizerv �columns�lines)r@ �sizer r r �test_does_not_crash. s z!TermsizeTests.test_does_not_crashc C st t j� � }d|d< |d= tj� }W d Q R X | j|jd� t j� � }|d= d|d< tj� }W d Q R X | j|jd� d S )NZ777�COLUMNS�LINESi Z888ix )r r� rF r� r] r� r� )r@ r� r� r r r �test_os_environ_first8 s
z#TermsizeTests.test_os_environ_firstc
C sL t j� � }d|d< d|d< tj� }W d Q R X | j|jd� | j|jd� d S )Nr� r� Zyyyr� r )r r� rF r� rv r� r� )r@ r� r� r r r �test_bad_environG s
z TermsizeTests.test_bad_environz
not on ttyr� zneed os.get_terminal_size()c C s� yt jddg�j� j� }W n& ttt jfk
r@ | jd� Y nX t|d �t|d �f}t j
� �}|d= |d= t j � }W d Q R X | j
||� d S )NZsttyr� zstty invocation failedrV r r� r� )r3 rB rE r� rm r� rC r� �intr r� rF r� r] )r@ r� rk r� r� r r r �test_stty_matchO s
z TermsizeTests.test_stty_matchc C s� t j� ��}|d= |d= t jtdd �� tjd
d�}W d Q R X | j|jd� | j|jd� t t
j d��,}t jtd|�� tjd d�}W d Q R X W d Q R X | j|jd� | j|jd � W d Q R X d S ) Nr� r� �
__stdout__�
� )�fallbackr# � �( )r r )r r ) r r� Z swap_attrr� rF r� r] r� r� r( r �devnull)r@ r� r� r� r r r �
test_fallbackg 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 Zd Zdd� ZdS )�PublicAPITestsc C s~ | j ttd�� dddddddd d
d d d
dddddddddddddg}ttd�sZtjd krd|jd � | jttj�t|�� d S ) N�__all__Z copyfileobjr� r� r� r� r� r� r� rG r r Z ExecErrorr r r r r r r r r� ry r� r� r
�statvfsrB ru ) rZ r� rF r r9 r6 r] �setr )r@ Z
target_apir r r �test_module_all_attribute~ s
z(PublicAPITests.test_module_all_attributeN)r� r� r� r r r r r r
{ s r
�__main__)F)F)4r� Z
unittest.mockrF rI r� r� r Zos.pathr r ri r3 �
contextlibr r r r r r r r r
r r r
r! r8 �warningsr] r Z test.supportr r r rO rQ rN � ImportErrorr r! r, r0 r7 ZTestCaser< r� r� r� r� r
r� �mainr r r r �<module> sd 4
<d ExN