Your IP : 216.73.216.213
3
� f�q � @ sz 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 dZeedd�ZdZd d
� Z d d � Z ej� Z e ee e e��Z d
Z!dZ"dZ#ej$j%d�Z&ej$j%d�Z'ej(j)e!d �Z*dZ+G dd� dej,�Z-G dd� de-�Z.e j/G dd� de.��Z0G dd� dej,�Z1d d � Z2e3d k�rve2� 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 sb t j| �}t|�td�kr<|dk r,t|�}ntdt|� �}tjjtjdt|�|d@ � | }|S )Ng i���l z<iil �� l���� ) �marshal�dumps�type�int� importlib�util� MAGIC_NUMBER�struct�pack)�co�mtime�size�data�pyc� r �8/opt/alt/python36/lib64/python3.6/test/test_zipimport.py�make_pyc s
"r c C s | j 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 jj� tj� | _d S )N) �sysr � meta_path�
path_hooks�path_importer_cache�clearr Z
modules_setup�modules_before)�selfr r r �setUp= s
z ImportHooksBaseTestCase.setUpc C sJ | j tj d d �< | jtjd d �< | jtjd d �< tjj� tj| j� d S )N) r r# r$ r% r& r' r Zmodules_cleanupr( )r) r r r �tearDownD s
z ImportHooksBaseTestCase.tearDownN)�__name__�
__module__� __qualname__r* r+ r r r r r" ; s r" c @ sJ 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(e)j*e+j,dEkdF�dGdH� �Z-dIdJ� Z.dES )K� UncompressedZipImportTestCasec C s t j� tjj� tj| � d S )N)� linecache�
clearcache� zipimport�_zip_directory_cacher' r" r* )r) r r r r* P s
z#UncompressedZipImportTestCase.setUpc
C s� | j tj|� x�|j� D ]�\}\}}tjj||�}|d tjkrXtjj|�s�tj |� qtjj
|�}tjj|�sztj |� t |d��}|j |� 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 �makeTreeW s
z&UncompressedZipImportTestCase.makeTreec K s� | j tj|� t|d��D}x<|j� D ]0\}\}}t|tj|��}| j|_ |j
||� q$W W d Q R X |j dd �} | d k r�t |d��}
|
j
� }W d Q R X t |d��}
|
j| � |
j|� W d Q R X d S )N�w�stuff�rbr5 )r7 r �unlinkr r8 r �time� localtime� compression�
compress_type�writestr�getr= �readr> ) r) r? ZzipName�kw�zr@ r r �zinforD �fr r r �makeZiph s
z%UncompressedZipImportTestCase.makeZipc O st | j |f|� tjjdt� tjdj|��}|jd�}|d k rF||� |rp|j � }| j
|t jjtf|�� | � d S )Nr r �call) rR r# r �insert�TEMP_ZIPr �
import_moduler9 rL Zget_file� assertEqualr )r) Z expected_extr? �modulesrN �modrS �filer r r �doTest} s
z$UncompressedZipImportTestCase.doTestc C s� dt jkr| jd� dt jkr&t jd= dttfi}y| jd|d� W n( tk
rl | jt krh| j
d� Y nX | jt kr�| j
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_namesZskipTestrX �NOW�test_srcr[ � ImportErrorrI r �failr )r) r? r r r �
testAFakeZlib� s
z+UncompressedZipImportTestCase.testAFakeZlibc C s" t d ttfi}| jd|t � d S )Nz.py)�TESTMODr^ r_ r[ )r) r? r r r �testPy� s z$UncompressedZipImportTestCase.testPyc C s" t t ttfi}| jt|t � d S )N)rc �pyc_extr^ �test_pycr[ )r) r? r r r �testPyc� s z%UncompressedZipImportTestCase.testPycc C s. t d ttft t ttfi}| jt|t � d S )Nz.py)rc r^ r_ re rf r[ )r) r? r r r �testBoth� s z&UncompressedZipImportTestCase.testBothc C s" t d tdfi}| jd |t � d S )Nz.py� )rc r^ r[ )r) r? r r r � testEmptyPy� s z)UncompressedZipImportTestCase.testEmptyPyc C sF t t�}|d dN < td ttftt t|fi}| jd|t� d S )Nr � z.py)� bytearrayrf rc r^ r_ re r[ )r) � badmagic_pycr? r r r � testBadMagic� s
z*UncompressedZipImportTestCase.testBadMagicc C s^ t t�}|d dN < tt t|fi}y| jd|t� W n tk
rN Y n X | jd� d S )Nr rk z.pyz)expected ImportError; import from bad pyc)rl rf rc re r^ r[ r` ra )r) rm r? r r r �
testBadMagic2� s z+UncompressedZipImportTestCase.testBadMagic2c C sF t t�}|d dN < td ttftt t|fi}| jd|t� d S )N� � z.py)rl rf rc r^ r_ re r[ )r) Z badtime_pycr? r r r � testBadMTime� s
z*UncompressedZipImportTestCase.testBadMTimec C sB t tj }|d t ttf|t t ttfi}| jt|t t� d S )N�__init__)�TESTPACKr r re r^ rf rc r[ )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}| jt|t tt� d S )Nrs ) rt r r � TESTPACK2re r^ rf rc r[ )r) ru �packdir2r? r r r �testSubPackage� s
z,UncompressedZipImportTestCase.testSubPackagec C sR t tj }|t tj }|tdf|tdf|t t ttfi}| jt|t tt� d S )Nri ) rt r r rw r^ rc re rf r[ )r) ru rx 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}tjj d�}| j
||� t }| j ||� t
jjd|� t
jjd|� tjt �}| jdt|j�� |j\} }
| jtjjt �| jtj�d � | jd|
jtj�d � tj|jtjd�d d
� �}| jdt|j�� t|j�d jdtj �d } | j|d d� | � tjdjt tf��}| jd|jjtj�d � tjdjt td f��}| jtjjt �|jjtj�d � djt tf�} tj| �}| jdt|j�� |j\} }
| jtjjt �| jtj�d � | jd|
jtj�d � tjdj| tf��}| jtjjt �|jjtj�d � tjdj| td f��}| jtjjt �|jjtj�d � tjdj| td f��}| jd|jjtj�d � d S )N�3ri �2z path1.zipr rq r r4 � rk ���r~ r6 r6 ���r r r ���r� r� )rt r r rw r^ rc re rf r �abspathrR �TEMP_DIRrB r# rT r rV rW �len�__path__�basename�splitr �listr9 �__file__)
r) ru rx �packdir3�files1�files2�zip1�zip2rY �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}tjj d�}| j
||� |tdf|t d t ttf|tdf|t d t ttf|t t ttfi}tjj d�}| j
||� t jj d|� t jj d|� t
jt �}| jdt|j�� |j\} }
| jd| jtj�d � | jd|
jtj�d
� t
j|jtjd�d d� �}| jd t|j�� t|j�d jdtj �d } | j|d d� | � t
jdjt tf��}| jd|jjtj�d � t
jdjt td f��}| jd|jjtj�d � djt tf�} t
j| �}| jdt|j�� |j\} }
| jd| jtj�d � | jd|
jtj�d � t
jdj| tf��}| jd|jjtj�d � t
jdj| td f��}| jd|jjtj�d � t
jdj| td f��}| jd|jjtj�d � d S )Nr{ ri z path1.zipr| z path2.zipr rq r r4 r} rk r~ r~ r6 r6 r r r r r� r� r� )rt r r rw r^ rc re rf r r� rR r# rT r rV rW r� r� r� r r� r9 r� )
r) ru rx r� r� r� r� r� rY r� r� r� r� r r r �testNamespacePackageC s\
z2UncompressedZipImportTestCase.testNamespacePackagec C s� t tj }|t tj }|d t ttf|d t ttf|t t ttfdt ttfi}tt d�}�zxB|j
� D ]6\}\}}t |t j
|��}| j|_d|_|j||� qpW |j� tjt �} | j| jt � | j| jt �d� | jd�}
| j|
� | j|
tj� | j|
jd�� |
j d�} | j|
j d�| j � | j t �} | j| j t �| j � t j t �j!d }
tj"j#t t �}| j|
|� | j| j|d �d� | j| j|t �d� | j| j|t �d� |t }t$|�}t j |�} | j%|t&j'k� | j| j(t �d � | j| j(|�d � | j| j |�| j � | j)}| j|j(|�d � | j|j |�| j � tjt tj t �}| j|jt � | j|j*t tj � W d |j� tj+t � X d S )Nrs ZspamrC s spamTr F),rt r r rw re r^ rf rc r rU r8 r rG rH rI rJ �commentrK �closer2 � zipimporterrW �archive�
is_package� find_module�assertIsNotNone�assertIsInstance� assertFalse� load_module� get_filenamer� r rV r� r r9 r! �
assertTruer# rX �
get_source�
__loader__�prefix�remove)r) ru rx r? rO r@ r r rP �ziZfind_mod�load_modrY 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|j
� D ]6\}\}}t |t j
|��}| j|_d|_|j||� qTW |j� tjt tj | �} | j| jt � | j| j|� | j| jt�d� | jt�}
| j| jt�|
j� | j| jttj d �d� | j| jttj t �d� t tj | t } tj| �} | j t�}
| j |
� | j |
tj� | j | jt�� |
jt�}| j|
jt�|j� ttj t }t |�}t!j"|�}
| j#|t$j%k� | j| j&t�d � | j| j&|�d � | j| j|�|
j� |
j'}| j|j&|�d � | j|j|�|
j� W d |j� tj(t � X d S )Nrs rC s eggsTF))rt r r rw re r^ rf rc r rU r8 r rG rH rI rJ r� rK r� r2 r� rW r� r� r� r� r� r� r� r� r� r� r! r rV r� r# rX r� r� r� )r) ru rx r? rO r@ r r rP r� rY 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 ��}|j||� |j� tjt�}| j ||j
|�� | j dt |�� W d |j� t
jt� X d S )NrC z testdata.datc s s | ]
}|V qd S )Nr )�.0�xr r r � <genexpr> s z<UncompressedZipImportTestCase.testGetData.<locals>.<genexpr>� zzipimporter object)r rU rI �bytes�rangerK r� r2 r� rW �get_dataZassertIn�reprr r� )r) rO r@ r r� r r r � testGetData� s
z)UncompressedZipImportTestCase.testGetDatac C st G dd� dt �}ttd�}z@d}d}|j||� |j� tjt�}| j||j||��� W d |j� t j
t� X d S )Nc @ s e Zd Zdd� ZdS )z?UncompressedZipImportTestCase.test_issue31291.<locals>.FunnyStrc S s dS )N�* r )r) �old�newr r r r s zGUncompressedZipImportTestCase.test_issue31291.<locals>.FunnyStr.replaceN)r, r- r. r r r r r �FunnyStr s r� rC z
test31291.dats foo) �strr rU rK r� r2 r� rW r� r r� )r) r� rO r@ r r� r r r �test_issue31291 s
z-UncompressedZipImportTestCase.test_issue31291c C sF d}t t|dd�tt|��}tt t|fdtdfi}| jt|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 �compiler^ r� rc re r[ )r) �srcr r? r r r �testImporterAttr s
z.UncompressedZipImportTestCase.testImporterAttrc C s\ d}t d t|fi}| j|� tjjdt� tjt �}| j |j
d�d� | j t |j
d� d S )Nz[if 1: # indent hack
def test(val):
assert(val)
return val
z.pyr r4 F)
rc r^ rR r# r rT rU r rV rW �test� assertRaises�AssertionError)r) r� r? rY r r r � testDefaultOptimizationLevel( s
z:UncompressedZipImportTestCase.testDefaultOptimizationLevelc C s* t d ttfi}| jd|t dd d� d S )Nz.pys
Some Stuff� )rD )rc r^ r_ r[ )r) r? r r r �testImport_WithStuff5 s
z2UncompressedZipImportTestCase.testImport_WithStuffc C s | j tj|�t� d S )N)rW �inspectZ getsourcer_ )r) �moduler r r �assertModuleSource< s z0UncompressedZipImportTestCase.assertModuleSourcec C s( t d ttfi}| jd|t | jd� d S )Nz.py)rS )rc r^ r_ r[ r� )r) r? r r r �
testGetSource? s z+UncompressedZipImportTestCase.testGetSourcec C sL t ttdd�ttt��}td ttftt t|fi}| jt|t| jd� d S )Nz<???>r
z.py)rS ) r r� r_ r^ r� rc re r[ r� )r) r r? r r r �testGetCompiledSourceC s z3UncompressedZipImportTestCase.testGetCompiledSourcec C s. t d ttfdtdfi}| jd|t |d� d S )Nz.pyzxyz.txtz>>> log.append(True)
)rS )rc r^ r_ r[ )r) �callbackr? r r r �
runDoctestI s z(UncompressedZipImportTestCase.runDoctestc C sF g }t jd }t _zt jd|dt� d� W d |t _X | j|dg� d S )Nzxyz.txtT)�package�module_relative�globs)�doctestZmasterZtestfile�localsrW )r) r� �logZ
old_masterr r r �
doDoctestFileN s z+UncompressedZipImportTestCase.doDoctestFilec C s | j | j� d S )N)r� r� )r) r r r �testDoctestFileZ s z-UncompressedZipImportTestCase.testDoctestFilec C s. g }t jd|dt� d�j� | j|dg� d S )Nzxyz.txtT)r� r� r� )r� Z DocFileTestr� ZrunrW )r) r� r� r r r �doDoctestSuite] s
z,UncompressedZipImportTestCase.doDoctestSuitec C s | j | j� d S )N)r� r� )r) r r r �testDoctestSuitee s z.UncompressedZipImportTestCase.testDoctestSuitec
C s� y |j � W n� tj� d j}t|d�d \}}}}| j|tj� � t|j d�d \}}}}| j|tj� � t
j � }t |d|� | j
|j� jt�� Y n
X td��d S )Nrq r4 r zThis ought to be impossible)Zdo_raiser# �exc_info�tb_nextr rW � raise_src�stripr �tb_frame�io�StringIOr r� �getvalue�endswithr� )r) r� �tbrQ Zlno�n�line�sr r r � doTracebackh s z)UncompressedZipImportTestCase.doTracebackc C s( t d ttfi}| jd |t | jd� d S )Nz.py)rS )rc r^ r� r[ r� )r) r? r r r �
testTracebackz s z+UncompressedZipImportTestCase.testTracebackNz need an unencodable filenamec C sj t jd }t|d�}ttd tjt��}| j|_ |j
|t � |j � zt
j|�jt� W d tj|� X d S )Nz.ziprC z.py)r �TESTFN_UNENCODABLEr r rc rG rH r^ rI rJ rK r_ r� r2 r� r� r r� )r) �filenamerO rP r r r �testUnencodable~ s
z-UncompressedZipImportTestCase.testUnencodablec C s� t jd }| jt j|� t|d��.}ttd tjt ��}| j
|_ |j |t
� W d Q R X tj|� tjtj|�� | jt�� tjttj|��� W d Q R X | jt�� tjttj|��� W d Q R X d S )Nz.ziprC z.py)r ZTESTFNr7 rF r r rc rG rH r^ rI rJ rK r_ r2 r� r �fsencodeZ assertWarns�DeprecationWarningrl �
memoryview)r) r� rO rP r r r �
testBytesPath� s
z+UncompressedZipImportTestCase.testBytesPath)/r, r- r. r rI r* r� rB rU rR r[ rb rd rg rh rj rn ro rr rv ry rz r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� �unittestZskipIfr r� r� r� r r r r r/ L sL
MI<5
r/ c @ s e Zd ZeZdS )�CompressedZipImportTestCaseN)r, r- r. r rI 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 | j tjtj|� d S )N)r� r2 �ZipImportErrorr� )r) r� r r r �assertZipFailure� s z)BadFileZipImportTestCase.assertZipFailurec C s | j d� d S )NZAdfjdkFJKDFJjdklfjs)r� )r) r r r �
testNoFile� s z#BadFileZipImportTestCase.testNoFilec C s | j d� d S )Nri )r� )r) r r r �testEmptyFilename� s z*BadFileZipImportTestCase.testEmptyFilenamec C sB | j ttjd � | j ttjtd d� | j ttjttjt��� d S )N)Zkwd)r� � TypeErrorr2 r� rc r� r r� )r) r r r � testBadArgs� s
z$BadFileZipImportTestCase.testBadArgsc C s | j dd � d S )N�Ai� )r� )r) r r r �testFilenameTooLong� s z,BadFileZipImportTestCase.testFilenameTooLongc C s" t jt� t jt� | jt� d S )N)r rF rc Zcreate_empty_filer� )r) r r r �
testEmptyFile� s
z&BadFileZipImportTestCase.testEmptyFilec C sh t jt� tjttjd�}z0tj|� | jtj ��}tj
t� W d Q R X W d tj td� t jt� X d S )Nr i� ) r rF rc r r= �O_CREATr� r� r2 r� r� �chmod)r) �fd�cmr r r �testFileUnreadable� s
z+BadFileZipImportTestCase.testFileUnreadablec C s8 t jt� ttd�}|jdd � |j� | jt� d S )Nzw+�a� )r rF rc r= r> r� r� )r) rA r r r �testNotZipFile� s
z'BadFileZipImportTestCase.testNotZipFilec
C s t jt� ttd�}|jtjdd�� |jd� |j� tj t�}z�| j
t |j d � | j
t |j
d � | j
t |jd � | j
t |jd � | j
t |jd � | j
t |jd � tj}| j|j d�d � | j
||j
d� | j
||jd� | j
t|jd� | j
||jd� | j
||jd� W d tjj� X d S )Nzw+z=IiPKr� � �abcZaaaaaaaaaaaaaaaaaa)r rF rc r= r> r r r� r2 r� r� r� r� r� r� �get_coder� r� r� rW �OSErrorr3 r' )r) rA rO �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 jttt� W d t jt� X d S )N)r Z run_unittestr/ r� r� rF rc r r r r � test_main� s
r �__main__)4r# r r r �importlib.utilr rG r� r� r Zzipfiler r r r r2 r0 r� r� r� � tracebackr r r r_ r� Ztest_cor� r r! r^ r� rf rc rt rw r r� r� rU r �cache_from_sourceZpyc_filere ZTestCaser" r/ Z
requires_zlibr� r� r r, r r r r �<module> sR TK