File: //proc/thread-self/root/usr/local/lib/python3.7/test/__pycache__/test_zipimport.cpython-37.pyc
B
��g^y � @ s� d dl Z d dlZd dlZd dlZd dlZd dlZd dlZd dlZd dlm Z d dl
m Z m Z m
Z
mZ d dlZd dlZd dlZd dlZd dlZd dlmZmZmZ y d dlZW n ek
r� dZY nX dZeedd�Z dZ d d
� Z d d � Z e�� Z e e e e!e��Z"d
Z#dZ$dZ%ej&�'d�Z(ej&�'d�Z)ej*�+e#d �Z,dZ-G dd� dej.�Z/G dd� de/�Z0e j1G dd� de0��Z2G dd� dej.�Z3d d � Z4e5d k�r�e4� dS ) � N)�support)�ZipFile�ZipInfo�
ZIP_STORED� ZIP_DEFLATED)�
extract_tb�
extract_stack�print_tbzHdef get_name():
return __name__
def get_file():
return __file__
z<???>�execz def do_raise(): raise TypeError
c C sd t �| �}t|�td�kr<|dk r,t|�}ntdt|� �}tjjt�ddt|�|d@ � | }|S )Ng i���l���� z<iiir l �� ) �marshal�dumps�type�int� importlib�util� MAGIC_NUMBER�struct�pack)�co�mtime�size�data�pyc� r �//usr/local/lib/python3.7/test/test_zipimport.py�make_pyc" s
$r c C s | � tjd�S )N�.)�replace�os�sep)�pathr r r �module_path_to_dotted_name. s r! Z
ziptestmoduleZziptestpackageZziptestpackage2Z junk95142z
junk95142.zipz.pyz.pycc @ s e Zd Zdd� Zdd� ZdS )�ImportHooksBaseTestCasec C sH t jd d � | _t jd d � | _t jd d � | _t j�� t�� | _d S )N) �sysr � meta_path�
path_hooks�path_importer_cache�clearr Z
modules_setup�modules_before)�selfr r r �setUpA s
z ImportHooksBaseTestCase.setUpc C sJ | j tj d d �< | jtjd d �< | jtjd d �< tj�� tj| j� d S )N) r r# r$ r% r& r' r Zmodules_cleanupr( )r) r r r �tearDownH s
z ImportHooksBaseTestCase.tearDownN)�__name__�
__module__� __qualname__r* r+ r r r r r" ? s r" c @ s` e Zd ZeZdd� Zefdd�Zefdd�Z dd� Z
d d
� Z d d � Z d
d� Z
dd� Zdd� Zdd� Zdd� Zdd� Zdd� Zdd � Zd d � Zd d � Zd!d"� Zd#d$� Zd%d&� Zd'd(� Zd)d*� Zd+d,� Z d-d.� Z d/d0� Z d1d2� Z d3d4� Z d5d6� Z!d7d8� Z"d9d:� Z#d;d<� Z$d=d>� Z%d?d@� Z&dAdB� Z'dCdD� Z(dEdF� Z)e*�+e,j-dGkdH�dIdJ� �Z.dKdL� Z/e,j0dMdN� �Z1dGS )O� UncompressedZipImportTestCasec C s t �� tj�� t�| � d S )N)� linecache�
clearcache� zipimport�_zip_directory_cacher' r" r* )r) r r r r* T s
z#UncompressedZipImportTestCase.setUpc
C s� | � tj|� x�|�� D ]�\}\}}tj�||�}|d tjkrXtj�|�s�t� |� qtj�
|�}tj�|�szt� |� t |d��}|� |� W d Q R X qW d S )N����wb)
�
addCleanupr Zrmtree�itemsr r �joinr �isdir�makedirs�dirname�open�write) r) �filesZdirName�namer r r Zdname�fpr r r �makeTree[ s
z&UncompressedZipImportTestCase.makeTreec K s� | � tj|� t|d��D}x<|�� D ]0\}\}}t|t�|��}| j|_ |�
||� q$W W d Q R X |� dd �} | d k r�t |d��}
|
�
� }W d Q R X t |d��}
|
�| � |
�|� W d Q R X d S )N�w�stuff�rbr5 )r6 r �unlinkr r7 r �time� localtime� compression�
compress_type�writestr�getr< �readr= ) r) r> ZzipName�kw�zr? r r �zinforC �fr r r �makeZipl s
z%UncompressedZipImportTestCase.makeZipc O st | j |f|� tj�dt� t�d�|��}|�d�}|d k rF||� |rp|� � }| �
|t jjtf|�� | � d S )Nr r �call) rQ r# r �insert�TEMP_ZIPr �
import_moduler8 rK Zget_file� assertEqualr )r) Z expected_extr> �modulesrM �modrR �filer r r �doTest� s
z$UncompressedZipImportTestCase.doTestc C s� dt jkr| �d� dt jkr&t jd= dttfi}y| �d|d� W n( tk
rl | jt krh| �
d� Y nX | jt kr�| �
d� d S )N�zlibzzlib is a builtin modulezzlib.pyz.pyz&expected test to not raise ImportErrorz"expected test to raise ImportError) r# �builtin_module_namesZskipTestrW �NOW�test_srcrZ � ImportErrorrH r �failr )r) r> r r r �
testAFakeZlib� s
z+UncompressedZipImportTestCase.testAFakeZlibc C s" t d ttfi}| �d|t � d S )Nz.py)�TESTMODr] r^ rZ )r) r> r r r �testPy� s z$UncompressedZipImportTestCase.testPyc C s" t t ttfi}| �t|t � d S )N)rb �pyc_extr] �test_pycrZ )r) r> r r r �testPyc� s z%UncompressedZipImportTestCase.testPycc C s. t d ttft t ttfi}| �t|t � d S )Nz.py)rb r] r^ rd re rZ )r) r> r r r �testBoth� s z&UncompressedZipImportTestCase.testBothc sj d}t j�|�}t j�t|dd�|d�}td tdftd td |fi}� fd d
�}� jd |t|d � d S ) Ns
state = 'old'z???r
Fz.pyz
state = 'new'z.pyc� c s � � | jd� d S )N�old)rV �state)rX )r) r r �check� s zFUncompressedZipImportTestCase.testUncheckedHashBasedPyc.<locals>.check)rR ) r r � source_hash�_bootstrap_external�_code_to_hash_pyc�compilerb r] rZ )r) �sourcerl �bytecoder> rk r )r) r �testUncheckedHashBasedPyc� s
z7UncompressedZipImportTestCase.testUncheckedHashBasedPycc C s" t d tdfi}| �d |t � d S )Nz.py� )rb r] rZ )r) r> r r r � testEmptyPy� s z)UncompressedZipImportTestCase.testEmptyPyc C sF t t�}|d dN < td ttftt t|fi}| �d|t� d S )Nr � z.py)� bytearrayre rb r] r^ rd rZ )r) � badmagic_pycr> r r r � testBadMagic� s
z*UncompressedZipImportTestCase.testBadMagicc C s^ t t�}|d dN < tt t|fi}y| �d|t� W n tk
rN Y n X | �d� d S )Nr ru z.pyz)expected ImportError; import from bad pyc)rv re rb rd r] rZ r_ r` )r) rw r> r r r �
testBadMagic2� s z+UncompressedZipImportTestCase.testBadMagic2c C sF t t�}|d dN < td ttftt t|fi}| �d|t� d S )N� � z.py)rv re rb r] r^ rd rZ )r) Z badtime_pycr> r r r � testBadMTime� s
z*UncompressedZipImportTestCase.testBadMTimec C sB t tj }|d t ttf|t t ttfi}| �t|t t� d S )N�__init__)�TESTPACKr r rd r] re rb rZ )r) �packdirr> r r r � testPackage� s
z)UncompressedZipImportTestCase.testPackagec C sb t tj }|t tj }|d t ttf|d t ttf|t t ttfi}| �t|t tt� d S )Nr} ) r~ r r � TESTPACK2rd r] re rb rZ )r) r �packdir2r> r r r �testSubPackage� s
z,UncompressedZipImportTestCase.testSubPackagec C sR t tj }|t tj }|tdf|tdf|t t ttfi}| �t|t tt� d S )Nrs ) r~ r r r� r] rb rd re rZ )r) r r� r> r r r �testSubNamespacePackage� s
z5UncompressedZipImportTestCase.testSubNamespacePackagec
C sR t tj }|t tj }|t d tj }|tdf|t t ttf|tdf|tdf|t t ttf|t d t ttf|t t ttfi}|tdf|t d t ttf|tdf|t d t ttf|t t ttfi}tj� d�}| �
||� t }| � ||� t
j�d|� t
j�d|� t�t �}| �dt|j�� |j\} }
| �tj�t �| �tj�d � | �d|
�tj�d � t�|�tjd�d d � �}| �d
t|j�� t|j�d �dtj �d
} | �|d d � | � t�d�t tf��}| �d|j�tj�d � t�d�t td f��}| �tj�t �|j�tj�d � d�t tf�} t�| �}| �dt|j�� |j\} }
| �tj�t �| �tj�d � | �d|
�tj�d � t�d�| tf��}| �tj�t �|j�tj�d � t�d�| td f��}| �tj�t �|j�tj�d � t�d�| td f��}| �d|j�tj�d � d S )
N�3rs �2z path1.zipr r{ ���r r4 � ������)r~ r r r� r] rb rd re r �abspathrQ �TEMP_DIRrA r# rS r rU rV �len�__path__�basename�splitr �listr8 �__file__)
r) r r� �packdir3�files1�files2�zip1�zip2rX �p1�p2�mpath�subpkgr r r �testMixedNamespacePackage sb
z7UncompressedZipImportTestCase.testMixedNamespacePackagec
C s2 t tj }|t tj }|t d tj }|tdf|t t ttf|tdf|tdf|t t ttf|t d t ttf|t t ttfi}tj� d�}| �
||� |tdf|t d t ttf|tdf|t d t ttf|t t ttfi}tj� d�}| �
||� t j� d|� t j� d|� t
�t �}| �dt|j�� |j\} }
| �d| �tj�d � | �d|
�tj�d � t
�|�tjd �d d
� �}| �d t|j�� t|j�d �dtj �d } | �|d d
� | � t
�d �t tf��}| �d|j�tj�d � t
�d �t td f��}| �d|j�tj�d � d �t tf�} t
�| �}| �dt|j�� |j\} }
| �d| �tj�d � | �d|
�tj�d � t
�d �| tf��}| �d|j�tj�d
� t
�d �| td f��}| �d|j�tj�d
� t
�d �| td f��}| �d|j�tj�d
� d S )Nr� rs z path1.zipr� z path2.zipr r{ r� r r4 r� r� r� )r~ r r r� r] rb rd re r r� rQ r# rS r rU rV r� r� r� r r� r8 r� )
r) r r� r� r� r� r� r� rX r� r� r� r� r r r �testNamespacePackageU s\
z2UncompressedZipImportTestCase.testNamespacePackagec C s� t tj }|t tj }|d t ttf|d t ttf|t t ttfdt ttfi}tt d�}�zxB|�
� D ]6\}\}}t |t �
|��}| j|_d|_|�||� qpW |�� t�t �} | �| jt � | �| �t �d� | �d�}
| �|
� | �|
tj� | �|
�d�� |
� d�} | �|
� d�| j � | � t �} | �| � t �| j � t � t �j!d }
tj"�#t t �}| �|
|� | �| �|d �d� | �| �|t �d� | �| �|t �d� |t }t$|�}t � |�} | �%|t&j'k� | �| �(t �d � | �| �(|�d � | �| � |�| j � | j)}| �|�(|�d � | �|� |�| j � t�t tj t �}| �|jt � | �|j*t tj � W d |�� t�+t � X d S )Nr} ZspamrB s spamTr F),r~ r r r� rd r] re rb r rT r7 r rF rG rH rI �commentrJ �closer2 � zipimporterrV �archive�
is_package� find_module�assertIsNotNone�assertIsInstance� assertFalse� load_module� get_filenamer� r rU r� r r8 r! �
assertTruer# rW �
get_source�
__loader__�prefix�remove)r) r r� r> rN r? r r rO �ziZfind_mod�load_modrX Zexisting_pack_pathZexpected_path_path�mod_path�mod_name�loader�zi2r r r �testZipImporterMethods� s\
z4UncompressedZipImportTestCase.testZipImporterMethodsc C sJ t tj }|t tj }|d t ttf|t t ttfi}tt d�}�z�xB|�
� D ]6\}\}}t |t �
|��}| j|_d|_|�||� qTW |�� t�t tj | �} | �| jt � | �| j|� | �| �t�d� | �t�}
| �| �t�|
j� | �| �ttj d �d� | �| �ttj t �d� t tj | t } t�| �} | � t�}
| � |
� | � |
tj� | � | �t�� |
�t�}| �|
�t�|j� ttj t }t |�}t!�"|�}
| �#|t$j%k� | �| �&t�d � | �| �&|�d � | �| �|�|
j� |
j'}| �|�&|�d � | �|�|�|
j� W d |�� t�(t � X d S )Nr} rB s eggsTF))r~ r r r� rd r] re rb r rT r7 r rF rG rH rI r� rJ r� r2 r� rV r� r� r� r� r� r� r� r� r� r� r! r rU r� r# rW r� r� r� )r) r r� r> rN r? r r rO r� rX Zpkg_pathr� Zfind_mod_dottedr� r� r� r� r r r �$testZipImporterMethodsInSubDirectory� sV
zBUncompressedZipImportTestCase.testZipImporterMethodsInSubDirectoryc C s� t td�}| j|_z^d}tdd� td�D ��}|�||� |�� t�t�}| � ||�
|�� | � dt |�� W d |�� t
�t� X d S )NrB z testdata.datc s s | ]
}|V qd S )Nr )�.0�xr r r � <genexpr> s z<UncompressedZipImportTestCase.testGetData.<locals>.<genexpr>� zzipimporter object)r rT rH �bytes�rangerJ r� r2 r� rV �get_dataZassertIn�reprr r� )r) rN r? r r� r r r � testGetData s
z)UncompressedZipImportTestCase.testGetDatac C st G dd� dt �}ttd�}z@d}d}|�||� |�� t�t�}| �||�||��� W d |�� t �
t� X d S )Nc @ s e Zd Zdd� ZdS )z?UncompressedZipImportTestCase.test_issue31291.<locals>.FunnyStrc S s dS )N�* r )r) ri �newr r r r ! s zGUncompressedZipImportTestCase.test_issue31291.<locals>.FunnyStr.replaceN)r, r- r. r r r r r �FunnyStr s r� rB z
test31291.dats foo) �strr rT rJ r� r2 r� rV r� r r� )r) r� rN r? r r� r r r �test_issue31291 s
z-UncompressedZipImportTestCase.test_issue31291c C sF d}t t|dd�tt|��}tt t|fdtdfi}| �t|t� d S )Nz�if 1: # indent hack
def get_file():
return __file__
if __loader__.get_data("some.data") != b"some data":
raise AssertionError("bad data")
z<???>r
z some.dataz some data)r ro r] r� rb rd rZ )r) �srcr r> r r r �testImporterAttr/ s
z.UncompressedZipImportTestCase.testImporterAttrc C s\ d}t d t|fi}| �|� tj�dt� t�t �}| � |�
d�d� | � t |j
d� d S )Nz[if 1: # indent hack
def test(val):
assert(val)
return val
z.pyr r� F)
rb r] rQ r# r rS rT r rU rV �test� assertRaises�AssertionError)r) r� r> rX r r r � testDefaultOptimizationLevel: s
z:UncompressedZipImportTestCase.testDefaultOptimizationLevelc C s& t d ttfi}| jd|t dd� d S )Nz.pys6 Some StuffSome StuffSome StuffSome StuffSome StuffSome StuffSome StuffSome StuffSome StuffSome StuffSome StuffSome StuffSome StuffSome StuffSome StuffSome StuffSome StuffSome StuffSome StuffSome StuffSome StuffSome StuffSome StuffSome StuffSome StuffSome StuffSome StuffSome StuffSome StuffSome StuffSome Stuff)rC )rb r] r^ rZ )r) r> r r r �testImport_WithStuffG s
z2UncompressedZipImportTestCase.testImport_WithStuffc C s | � t�|�t� d S )N)rV �inspectZ getsourcer^ )r) �moduler r r �assertModuleSourceN s z0UncompressedZipImportTestCase.assertModuleSourcec C s( t d ttfi}| jd|t | jd� d S )Nz.py)rR )rb r] r^ rZ r� )r) r> r r r �
testGetSourceQ s z+UncompressedZipImportTestCase.testGetSourcec C sL t ttdd�ttt��}td ttftt t|fi}| jt|t| jd� d S )Nz<???>r
z.py)rR ) r ro r^ r] r� rb rd rZ r� )r) r r> r r r �testGetCompiledSourceU s z3UncompressedZipImportTestCase.testGetCompiledSourcec C s. t d ttfdtdfi}| jd|t |d� d S )Nz.pyzxyz.txtz>>> log.append(True)
)rR )rb r] r^ rZ )r) �callbackr> r r r �
runDoctest[ s z(UncompressedZipImportTestCase.runDoctestc C sF g }t jd }t _zt jd|dt� d� W d |t _X | �|dg� d S )Nzxyz.txtT)�package�module_relative�globs)�doctestZmasterZtestfile�localsrV )r) r� �logZ
old_masterr r r �
doDoctestFile` s z+UncompressedZipImportTestCase.doDoctestFilec C s | � | j� d S )N)r� r� )r) r r r �testDoctestFilel s z-UncompressedZipImportTestCase.testDoctestFilec C s. g }t jd|dt� d��� | �|dg� d S )Nzxyz.txtT)r� r� r� )r� Z DocFileTestr� �runrV )r) r� r� r r r �doDoctestSuiteo s
z,UncompressedZipImportTestCase.doDoctestSuitec C s | � | j� d S )N)r� r� )r) r r r �testDoctestSuitew s z.UncompressedZipImportTestCase.testDoctestSuitec C s� y |� � W n� t�� d j}t|d�d \}}}}| �|t�� � t|j d�d \}}}}| �|t�� � t
� � }t |d|� | �
|�� �t�� Y n
X td��d S )Nr{ r� r zThis ought to be impossible)Zdo_raiser# �exc_info�tb_nextr rV � raise_src�stripr �tb_frame�io�StringIOr r� �getvalue�endswithr� )r) r� �tbrP Zlno�n�line�sr r r � doTracebackz s z)UncompressedZipImportTestCase.doTracebackc C s( t d ttfi}| jd |t | jd� d S )Nz.py)rR )rb r] r� rZ r� )r) r> r r r �
testTraceback� s z+UncompressedZipImportTestCase.testTracebackNz need an unencodable filenamec C sj t jd }t|d�}ttd t�t��}| j|_ |�
|t � |� � zt
�|��t� W d t�|� X d S )Nz.ziprB z.py)r �TESTFN_UNENCODABLEr r rb rF rG r] rH rI rJ r^ r� r2 r� r� r r� )r) �filenamerN rO r r r �testUnencodable� s
z-UncompressedZipImportTestCase.testUnencodablec C s� t jd }| �t j|� t|d��.}ttd t�t ��}| j
|_ |� |t
� W d Q R X t�|� t�t�|�� | �t�� t�tt�|��� W d Q R X | �t�� t�tt�|��� W d Q R X d S )Nz.ziprB z.py)r ZTESTFNr6 rE r r rb rF rG r] rH rI rJ r^ r2 r� r �fsencodeZ assertWarns�DeprecationWarningrv �
memoryview)r) r� rN rO r r r �
testBytesPath� s
z+UncompressedZipImportTestCase.testBytesPathc C s� t j�t j�}| �t|jd� | �t|jd� | �t|jd� | �t|jd� | �t|j d� | �t|j
d� | �t|j d� | �t|j d� d S )NZfoo)
r2 r� �__new__r� �
ValueErrorr� � find_loaderr� r� r� r� �get_coder� )r) r� r r r � testUninitializedZipimporter� s z:UncompressedZipImportTestCase.testUninitializedZipimporter)2r, r- r. r rH r* r� rA rT rQ rZ ra rc rf rg rr rt rx ry r| r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� �unittestZskipIfr r� r� r� � cpython_onlyr� r r r r r/ P sP
MI<5
r/ c @ s e Zd ZeZejdd� �ZdS )�CompressedZipImportTestCasec C sr dd� }t td��"}| �tjt� |�ddt� W d Q R X t�t�}t� t
d|�� | � t |j
d� W d Q R X d S )Nc W s d S )Nr )�argsr r r �bad_decompress� s zCCompressedZipImportTestCase.test_issue31602.<locals>.bad_decompressrB zbar.pys print("hello world")Z
decompressZbar)r rT r6 r rE rJ r r2 r� Z swap_attrr[ r� � TypeErrorr� )r) r� Zzip_filer� r r r �test_issue31602� s
z+CompressedZipImportTestCase.test_issue31602N)r, r- r. r rH r r� r� r r r r r� � s r� c @ sT e Zd Zdd� Zdd� Zdd� Zdd� Zd d
� Zd d � Zd
d� Z dd� Z
dd� Z dS )�BadFileZipImportTestCasec C s | � tjtj|� d S )N)r� r2 �ZipImportErrorr� )r) r� r r r �assertZipFailure� s z)BadFileZipImportTestCase.assertZipFailurec C s | � d� d S )NZAdfjdkFJKDFJjdklfjs)r )r) r r r �
testNoFile� s z#BadFileZipImportTestCase.testNoFilec C s | � d� d S )Nrs )r )r) r r r �testEmptyFilename� s z*BadFileZipImportTestCase.testEmptyFilenamec C sB | � ttjd � | j ttjtd d� | � ttjtt�t��� d S )N)Zkwd)r� r� r2 r� rb r� r r� )r) r r r � testBadArgs� s
z$BadFileZipImportTestCase.testBadArgsc C s | � dd � d S )N�Ai� )r )r) r r r �testFilenameTooLong� s z,BadFileZipImportTestCase.testFilenameTooLongc C s" t �t� t �t� | �t� d S )N)r rE rb Zcreate_empty_filer )r) r r r �
testEmptyFile� s
z&BadFileZipImportTestCase.testEmptyFilec
C sh t �t� t�ttjd�}z0t�|� | �tj ��}t�
t� W d Q R X W d t� td� t �t� X d S )Nr i� ) r rE rb r r< �O_CREATr� r� r2 r r� �chmod)r) �fd�cmr r r �testFileUnreadable� s
z+BadFileZipImportTestCase.testFileUnreadablec C s4 t �t� ttd�}|�d� |�� | �t� d S )Nzw+Zaaaaaaaaaaaaaaaaaaaaaa)r rE rb r< r= r� r )r) r@ r r r �testNotZipFile� s
z'BadFileZipImportTestCase.testNotZipFilec C s t �t� ttd�}|�t�dd�� |�d� |�� t� t�}z�| �
t |j d � | �
t |j
d � | �
t |jd � | �
t |jd � | �
t |jd � | �
t |jd � tj}| �|� d�d � | �
||j
d� | �
||jd� | �
t|jd� | �
||jd� | �
||jd� W d tj�� X d S )Nzw+z=IiPKZaaaaaaaaaaaaaaaaaa�abc)r rE rb r< r= r r r� r2 r� r� r� r� r� r� r� r� r� r rV �OSErrorr3 r' )r) r@ rN �errorr r r �_testBogusZipFile� s*
z*BadFileZipImportTestCase._testBogusZipFileN) r, r- r. r r r r r r r r
r r r r r r� � s r� c C s$ zt �ttt� W d t �t� X d S )N)r Z run_unittestr/ r� r� rE rb r r r r � test_main s
r �__main__)6r# r r r �importlib.utilr rF r� r� r Zzipfiler r r r r2 r0 r� r� r� � tracebackr r r r[ r_ r^ ro Ztest_cor� r r! r] r� re rb r~ r� r r� r� rT r �cache_from_sourceZpyc_filerd ZTestCaser" r/ Z
requires_zlibr� r� r r, r r r r �<module> sZ
pK