File: //usr/local/lib/python3.7/distutils/tests/__pycache__/test_archive_util.cpython-37.opt-2.pyc
B
!��g�7 � @ s� d dl Z d dlZd dlZd dlZd dlmZ d dlZd dlmZ d dl m
Z
m Z m Z m
Z
mZ d dlmZmZ d dlmZ d dlmZmZmZmZ yd dlZd dlZdZW n e k
r� d ZY nX yd dl Z dZ W n e k
r� ed
�Z Y nX yd dl Z dZ W n e k
�r d Z Y nX y d dl!Z!W n e k
�rB dZ!Y nX y d dl"Z"W n e k
�rj dZ"Y nX d d � Z#G d
d� dej$ej%e j&�Z'dd� Z(e)dk�r�ee(� � dS )� N)�
splitdrive)� archive_util)�check_archive_formats� make_tarball� make_zipfile� make_archive�ARCHIVE_FORMATS)�find_executable�spawn)�support)�check_warnings� run_unittest�patch�
change_cwdTF�zipc C s8 t jjr dS y| �t�� � W n tk
r2 dS X dS )NTF)�os�path�supports_unicode_filenames�encode�sys�getfilesystemencoding�UnicodeEncodeError)�filename� r �=/usr/local/lib/python3.7/distutils/tests/test_archive_util.py�
can_fs_encode/ s r c @ s� e Zd Ze�ed�dGdd��Ze�ed�dd� �Ze�ed�dd � �Z e�e
d
�d d � �Z e�e d
�d�dd� �Z
e�e d�d�dd� �Zdd� Zdd� Zddddd d gZd d � eD �Zd d!� Ze�ed"�o�ed#�o�ed$�d%d&� �Ze�ed'�d(�d)d*� �Ze�e�oed+�d,d-� �Ze�ed.�d/d0� �Zd1d2� Zd3d4� Zd5d6� Z d7d8� Z e�ed�d9d:� �Z e�ed�d;d<� �Z e�e
d=�d>d?� �Z d@dA� Z!e�edB�e�e"dC�dDdE� ��Z#dFS )H�ArchiveUtilTestCasezNeed zlib support to run�archivec C s, | � � }| �||d� | j||dd d� d S )Nz.tar.gzz.tar)�compress)�
_create_files�
_make_tarball)�self�name�tmpdirr r r �test_make_tarball@ s z%ArchiveUtilTestCase.test_make_tarballc C s | � � }| j|dddd� d S )Nr z.tar.gz�gzip)r )r r )r! r# r r r �test_make_tarball_gzipH s z*ArchiveUtilTestCase.test_make_tarball_gzipzNeed bz2 support to runc C s | � � }| j|dddd� d S )Nr z.tar.bz2Zbzip2)r )r r )r! r# r r r �test_make_tarball_bzip2M s z+ArchiveUtilTestCase.test_make_tarball_bzip2zNeed lzma support to runc C s | � � }| j|dddd� d S )Nr z.tar.xzZxz)r )r r )r! r# r r r �test_make_tarball_xzR s z(ArchiveUtilTestCase.test_make_tarball_xzu årchivz'File system cannot handle this filenamec C s | � d� d S )Nu årchiv)r$ )r! r r r �test_make_tarball_latin1W s z,ArchiveUtilTestCase.test_make_tarball_latin1u のアーカイブc C s | � d� d S )Nu のアーカイブ)r$ )r! r r r �test_make_tarball_extended_ s z.ArchiveUtilTestCase.test_make_tarball_extendedc K s� | � � }t�t|�d t|�d kd� tj�||�}t|�� tt|�d df|� W d Q R X || }| � tj�
|�� | � | � |�| j
� d S )Nr z)source and target should be on same drive� �dist)�mkdtemp�unittest�
skipUnlessr r r �joinr r �
assertTrue�exists� assertEqual�_tarinfo�_created_files)r! r# Z target_name�suffix�kwargs�tmpdir2� base_name�tarballr r r r h s
z!ArchiveUtilTestCase._make_tarballc C s. t �|�}z|�� }|�� |S |�� X d S )N)�tarfile�openZgetnames�sort�close)r! r �tar�namesr r r r4 x s
z ArchiveUtilTestCase._tarinfozdist/z
dist/file1z
dist/file2z dist/sub/zdist/sub/file3z
dist/sub2/c C s g | ]}|� d ��qS )�/)�rstrip)�.0�pr r r �
<listcomp>� s z ArchiveUtilTestCase.<listcomp>c C s~ | � � }tj�|d�}t�|� | �|dgd� | �|dgd� t�tj�|d�� | �|ddgd� t�tj�|d�� |S )Nr, Zfile1�xxxZfile2�subZfile3Zsub2)r- r r r0 �mkdirZ
write_file)r! r# r, r r r r � s
z!ArchiveUtilTestCase._create_filesr? r% z*Need the tar, gzip and zlib command to runc C s� | � � }| �� }tj�|d�}t�� }t�|� zt|d� W d t�|� X |d }| �tj� |�� tj�|d�}ddddg}dd d
dg}t�� }t�|� zt
|� t
|� W d t�|� X | �tj� |�� | � | � |�| j
� | � | � |�| j
� tj�|d�}t�� }t�|� zt|dd d � W d t�|� X |d }| �tj� |�� tj�|d�}t�� }t�|� zt|dd d
d� W d t�|� X |d }| �tj� |�� d S )Nr r, z.tar.gzzarchive2.tar.gzr? z-cfz archive2.tarr% z-fz-9)r z.tarT)r �dry_run)r r- r r r0 �getcwd�chdirr r1 r2 r
r3 r4 r5 ) r! r# r8 r9 �old_dirr: Ztarball2Ztar_cmdZgzip_cmdr r r �test_tarfile_vs_tar� sL
z'ArchiveUtilTestCase.test_tarfile_vs_tarr z The compress program is requiredc C s | � � }tj�| �� d�}t�� }t�|� z.t� � }t� d� t
|ddd� W d Q R X W d t�|� X |d }| � tj� |�� | �
t|j�d� t�|� t�� }t�|� z0t� � }t� d� t
|dddd � W d Q R X W d t�|� X | �tj� |�� | �
t|j�d� d S )
Nr �alwaysr, r )r z.tar.Zr+ T)r rI )r r r r0 r- rJ rK r �warnings� simplefilterr r1 r2 r3 �len�removeZ assertFalse)r! r# r9 rL �wr: r r r �test_compress_deprecated� s.
z,ArchiveUtilTestCase.test_compress_deprecatedz Need zip and zlib support to runc C s� | � � }tj�| �� d�}t|�� t|d� W d Q R X |d }| �tj�|�� t �
|�� }| � t |�
� �| j� W d Q R X d S )Nr r, z.zip)r r r r0 r- r r r1 r2 �zipfile�ZipFiler3 �sorted�namelist�_zip_created_files)r! r# r9 r: �zfr r r �test_make_zipfile� s
z%ArchiveUtilTestCase.test_make_zipfilezNeed zip support to runc s� t | tjdd � g � tj�� �fdd�}t | tjd|� | �� }tj�| �� d�}t |�� t
|d� W d Q R X |d }| � � |dfd tj ifg� | �
tj�|�� t�|�� }| � t|�� �| j� W d Q R X d S )
N�zlibc s* |� dd �tjkr � �| |f� �| |�S )N� compression)�getrU �
ZIP_STORED�append)�a�kw)�called�
zipfile_classr r � fake_zipfile s zCArchiveUtilTestCase.test_make_zipfile_no_zlib.<locals>.fake_zipfilerV r r, z.ziprS r] )r r rU rV r r r r0 r- r r r3 r_ r1 r2 rW rX rY )r! re r# r9 r: rZ r )rc rd r �test_make_zipfile_no_zlib� s
z-ArchiveUtilTestCase.test_make_zipfile_no_zlibc C s4 | � tdddg�d� | �tddddddg�� d S )N�gztarrF r �bztar�xztarZztarr? )r3 r Z assertIsNone)r! r r r �test_check_archive_formats s z.ArchiveUtilTestCase.test_check_archive_formatsc C s* | � � }tj�|d�}| �tt|d� d S )Nr rF )r- r r r0 Z assertRaises�
ValueErrorr )r! r# r9 r r r �test_make_archive s z%ArchiveUtilTestCase.test_make_archivec C sd t �� }dd� }|g dftd< z8ytdd| �� d� W n Y nX | �t �� |� W d td= X d S )Nc _ s
t � �d S )N)� RuntimeError)�argsrb r r r �_breaks! s z:ArchiveUtilTestCase.test_make_archive_cwd.<locals>._breakszxxx filerF )�root_dir)r rJ r r r- r3 )r! Z current_dirro r r r �test_make_archive_cwd s z)ArchiveUtilTestCase.test_make_archive_cwdc C sf | � � }tj�| �� d�}t|d|d�}| �tj�|�� | �tj� |�d� | �| �
|�| j � d S )Nr r? r, z archive.tar) r r r r0 r- r r1 r2 r3 �basenamer4 r5 )r! �base_dirr9 �resr r r �test_make_archive_tar- s z)ArchiveUtilTestCase.test_make_archive_tarc C sf | � � }tj�| �� d�}t|d|d�}| �tj�|�� | �tj� |�d� | �| �
|�| j � d S )Nr rg r, zarchive.tar.gz) r r r r0 r- r r1 r2 r3 rr r4 r5 )r! rs r9 rt r r r �test_make_archive_gztar5 s z+ArchiveUtilTestCase.test_make_archive_gztarc C sf | � � }tj�| �� d�}t|d|d�}| �tj�|�� | �tj� |�d� | �| �
|�| j � d S )Nr rh r, zarchive.tar.bz2) r r r r0 r- r r1 r2 r3 rr r4 r5 )r! rs r9 rt r r r �test_make_archive_bztar> s z+ArchiveUtilTestCase.test_make_archive_bztarzNeed xz support to runc C sf | � � }tj�| �� d�}t|d|d�}| �tj�|�� | �tj� |�d� | �| �
|�| j � d S )Nr ri r, zarchive.tar.xz) r r r r0 r- r r1 r2 r3 rr r4 r5 )r! rs r9 rt r r r �test_make_archive_xztarG s z+ArchiveUtilTestCase.test_make_archive_xztarc C s� t r"t�d�d }t�d�d }nd }}| �� }| �� }tj� | �� d�}t
|d||||d�}| � tj� |�� t
|d||�}| � tj� |�� t
|d||||d�}| � tj� |�� t
|d||ddd�}| � tj� |�� d S ) Nr �rootr r )�owner�groupr? Z kjhkjhkjgZoihohoh)
�UID_GID_SUPPORT�grp�getgrgid�pwd�getpwuidr r- r r r0 r r1 r2 )r! r{ rz rs rp r9 rt r r r � test_make_archive_owner_groupP s$
z1ArchiveUtilTestCase.test_make_archive_owner_groupz
Requires zlibz Requires grp and pwd supportc C s� | � � }tj�| �� d�}t�� }t�|� t�d�d }t �
d�d }zt |dd ||d�}W d t�|� X | � tj�
|�� t�|�}z2x,|�� D ] }| �|jd� | �|jd� q�W W d |�� X d S )Nr r r, )r rz r{ )r r r r0 r- rJ rK r} r~ r r� r r1 r2 r; r< Z
getmembersr3 Zuid�gidr> ) r! r# r9 rL r{ rz Z archive_namer �memberr r r �test_tarfile_root_ownerk s"
z+ArchiveUtilTestCase.test_tarfile_root_ownerN)r )$�__name__�
__module__� __qualname__r. r/ � ZLIB_SUPPORTr$ r& �bz2r' �lzmar( r r) r* r r4 rY r5 r r rM rT � ZIP_SUPPORTr[ rf rj rl rq ru rv rw rx r� r| r� r r r r r < sB
7
r c C s
t �t�S )N)r. Z makeSuiter r r r r �
test_suite� s r� �__main__)*r. r r r; Zos.pathr rO Z distutilsr Zdistutils.archive_utilr r r r r Zdistutils.spawnr r
Zdistutils.testsr Z test.supportr r
r r r} r r| � ImportErrorrU r� r\ r� r� r� r ZTempdirManagerZLoggingSilencerZTestCaser r� r� r r r r �<module> sV
J