Your IP : 216.73.216.60
U
i�fu � @ s� d dl Z d dlZd dlZd dlZd dlZd dlZd dlZd dlZd dlZd dl m
Z
d dl m Z m
Z
mZmZ d dlZd dlZd dlZd dlZd dlZd dlmZmZmZ z d dlZW n ek
r� dZY nX dZe edd�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/�Z0G dd� de0�Z1e
j2G dd� de1��Z3G dd� dej/�Z4d d � Z5e6d k�r�e5� dS ) � N)�support)�ZipFile�ZipInfo�
ZIP_STORED� ZIP_DEFLATED)�
extract_tb�
extract_stack�print_tbzHdef get_name():
return __name__
def get_file():
return __file__
�<???>�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 �8/opt/alt/python38/lib64/python3.8/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.zip�.py�.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 �setUpB s
z ImportHooksBaseTestCase.setUpc C sJ | j tj d d �< | jtjd d �< | jtjd d �< tj�� tj| j� d S r& ) r! r' r( r) r* r+ r Zmodules_cleanupr, r- r r r �tearDownI s
z ImportHooksBaseTestCase.tearDownN)�__name__�
__module__� __qualname__r/ r0 r r r r r% @ s r% c @ sx e Zd ZeZdd� Zefdd�Zefdd�Z dd� Z
d d
� Z d d � Z d
d� Z
dd� Zdd� Zej�dd�dd� �Zdd� Zdd� Zdd � Zd d � Zd d � Zd!d"� Zd#d$� Zd%d&� Zd'd(� Z d)d*� Z d+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+dGdH� Z,e�-e.j/dIkdJ�dKdL� �Z0dMdN� Z1dOdP� Z2dQdR� Z3dSdT� Z4dIS )U� UncompressedZipImportTestCasec C s t �� tj�� t�| � d S r& )� linecache�
clearcache� zipimport�_zip_directory_cacher+ r% r/ r- r r r r/ U s
z#UncompressedZipImportTestCase.setUpc
C s� | � tj|� |�� D ]�\}\}}tj�||�}|d tjkrVtj�|�s�t� |� qtj�
|�}tj�|�sxt� |� t |d��}|� |� W 5 Q R X qd 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��Z}|�� D ]0\}\}}t|t�|��}| j|_ |�
||� q"|� dd �} | d k rn| |_ W 5 Q R X |� dd �}
|
d k r�t
|d��} | �� }W 5 Q R X t
|d��} | �|
� | �|� W 5 Q R X d S )N�w�comment�stuff�rbr: )r; r �unlinkr r< r �time� localtime� compression�
compress_type�writestr�getrH rA �readrB ) r. rC ZzipName�kw�zrD r r �zinforH rI �fr r r �makeZipm s
z%UncompressedZipImportTestCase.makeZipc O st | j |f|� tj�dt� t�d�|��}|�d�}|d k rF||� |rp|� � }| �
|t jjtf|�� | � d S )Nr r �call) rW r' r! �insert�TEMP_ZIPr �
import_moduler= rQ Zget_file� assertEqualr )r. Z expected_extrC �modulesrS �modrX �filer r r �doTest� s
��z$UncompressedZipImportTestCase.doTestc C s� dt jkr| �d� dt jkr&t jd= dttfi}z| �d|d� W n( tk
rl | jt krh| �
d� Y nX | jt kr�| �
d� d S )N�zlibzzlib is a builtin modulezzlib.pyr# z&expected test to not raise ImportErrorz"expected test to raise ImportError) r' �builtin_module_namesZskipTestr] �NOW�test_srcr` � ImportErrorrN r �failr �r. rC r r r �
testAFakeZlib� s
z+UncompressedZipImportTestCase.testAFakeZlibc C s" t d ttfi}| �d|t � d S �Nr# ��TESTMODrc rd r` rg r r r �testPy� s z$UncompressedZipImportTestCase.testPyc C s" t t ttfi}| �t|t � d S r& )rk �pyc_extrc �test_pycr` rg r r r �testPyc� s z%UncompressedZipImportTestCase.testPycc C s. t d ttft t ttfi}| �t|t � d S ri )rk rc rd rm rn r` rg 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 ) N�
state = 'old'�???r Fr# �
state = 'new'r$ � c s � � | jd� d S )N�old�r\ �state�r^ r- r r �check� s zFUncompressedZipImportTestCase.testUncheckedHashBasedPyc.<locals>.check�rX � r r � source_hash�_bootstrap_external�_code_to_hash_pyc�compilerk rc r` �r. �sourcer| �bytecoderC ry r r- r �testUncheckedHashBasedPyc� s
�
� z7UncompressedZipImportTestCase.testUncheckedHashBasedPycz_imp.check_hash_based_pycs�alwaysc sj d}t j�|�}t j�t|dd�|d�}td tdftd td |fi}� fd d
�}� jd |t|d � d S ) Nrq rr r Fr# rs r$ rt c s � � | jd� d S )N�newrv rx r- r r ry � s zOUncompressedZipImportTestCase.test_checked_hash_based_change_pyc.<locals>.checkrz r{ r� r r- r �"test_checked_hash_based_change_pyc� s
�
� z@UncompressedZipImportTestCase.test_checked_hash_based_change_pycc C s" t d tdfi}| �d |t � d S )Nr# � )rk rc r` rg r r r � testEmptyPy� s z)UncompressedZipImportTestCase.testEmptyPyc C sF t t�}|d dN < td ttftt t|fi}| �d|t� d S )Nr � r# �� bytearrayrn rk rc rd rm r` �r. Z badmagic_pycrC r r r � testBadMagic� s �z*UncompressedZipImportTestCase.testBadMagicc C s^ t t�}|d dN < tt t|fi}z| �d|t� W n tk
rN Y n X | �d� d S )Nr r� r# z)expected ImportError; import from bad pyc)r� rn rk rm rc r` re rf r� r r r �
testBadMagic2� s z+UncompressedZipImportTestCase.testBadMagic2c C sF t t�}|d dN < td ttftt t|fi}| �d|t� d S )N� � r# r� )r. Z badtime_pycrC 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 rm rc rn rk r` )r. �packdirrC 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 r� ) r� r r � TESTPACK2rm rc rn rk r` �r. r� �packdir2rC r r r �testSubPackage s
�z,UncompressedZipImportTestCase.testSubPackagec C sR t tj }|t tj }|tdf|tdf|t t ttfi}| �t|t tt� d S �Nr� ) r� r r r� rc rk rm rn 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�3r� �2� path1.zipr r� ���r r9 � ������)r� r r r� rc rk rm rn r! �abspathrW �TEMP_DIRrF r' rY r r[ r\ �len�__path__�basename�splitr �listr= �__file__)
r. r� r� �packdir3�files1�files2�zip1�zip2r^ �p1�p2�mpath�subpkgr r r �testMixedNamespacePackage s�
�
�
�
��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� r� r� r� z path2.zipr r� r� r r9 r� r� r� )r� r r r� rc rk rm rn r! r� rW r' rY 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 �testNamespacePackageh st
�
�
z2UncompressedZipImportTestCase.testNamespacePackagec C s� t tj }|t tj }|d t ttf|d t ttf|t t ttfdt ttfi}| �t j
t � t t d��F}|�
� D ]6\}\}}t|t�|��}| j|_d|_|�||� qzW 5 Q R X 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 � d S )Nr� ZspamrG s spamTr F)-r� r r r� rm rc rn rk r; r rK rZ r r< r rL rM rN rO rH rP r7 � zipimporterr\ �archive�
is_package� find_module�assertIsNotNone�assertIsInstance� assertFalse� load_module� get_filenamer� r r[ r� r! r= r" �
assertTruer' r] �
get_source�
__loader__�prefix)r. r� r� rC rT rD r r rU �ziZfind_mod�load_modr^ Zexisting_pack_pathZexpected_path_path�mod_path�mod_name�loader�zi2r r r �testZipImporterMethods� s^
�
z4UncompressedZipImportTestCase.testZipImporterMethodsc C s< t tj }|t tj }|d t ttf|t t ttfi}| �t j
t � t t d��F}|�
� D ]6\}\}}t|t�|��}| j|_d|_|�||� q^W 5 Q R X 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 � d S )Nr� rG s eggsTF)*r� r r r� rm rc rn rk r; r rK rZ r r< r rL rM rN rO rH rP r7 r� r\ r� r� r� r� r� r� r� r� r� r� r" r r[ r� r' r] r� r� )r. r� r� rC rT rD r r rU r� r^ Zpkg_pathr� Zfind_mod_dottedr� r� r� r� r r r �$testZipImporterMethodsInSubDirectory� s`
�
� �
�
zBUncompressedZipImportTestCase.testZipImporterMethodsInSubDirectoryc C s� | � tjt� ttd��4}| j|_d}tdd� td�D ��}|�||� W 5 Q R X t �
t�}| � ||� |�� | �
dt|�� d S )NrG z testdata.datc s s | ]
}|V qd S r& r )�.0�xr r r � <genexpr> s z<UncompressedZipImportTestCase.testGetData.<locals>.<genexpr>� zzipimporter object)r; r rK rZ r rN �bytes�rangerP r7 r� r\ �get_dataZassertIn�repr)r. rT rD r r� r r r � testGetData s
z)UncompressedZipImportTestCase.testGetDatac 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")
r
r z some.dataz some data)r r rc r� rk rm r` )r. �srcr rC 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
r# r r� F)
rk rc rW r' r! rY rZ r r[ r\ �test� assertRaises�AssertionError)r. r� rC r^ r r r � testDefaultOptimizationLevel1 s
z:UncompressedZipImportTestCase.testDefaultOptimizationLevelc C s& t d ttfi}| jd|t dd� d S )Nr# s6 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)rI rj rg r r r �testImport_WithStuff> s
�z2UncompressedZipImportTestCase.testImport_WithStuffc C s | � t�|�t� d S r& )r\ �inspectZ getsourcerd )r. �moduler r r �assertModuleSourceE s z0UncompressedZipImportTestCase.assertModuleSourcec C s( t d ttfi}| jd|t | jd� d S �Nr# rz )rk rc rd r` r� rg r r r �
testGetSourceH s z+UncompressedZipImportTestCase.testGetSourcec C sL t ttdd�ttt��}td ttftt t|fi}| jt|t| jd� d S )Nr
r r# rz ) r r rd rc r� rk rm r` r� )r. r rC r r r �testGetCompiledSourceL s �z3UncompressedZipImportTestCase.testGetCompiledSourcec C s. t d ttfdtdfi}| jd|t |d� d S )Nr# �xyz.txtz>>> log.append(True)
rz rj )r. �callbackrC r r r �
runDoctestR s
�z(UncompressedZipImportTestCase.runDoctestc C sF g }t jd }t _zt jd|dt� d� W 5 |t _X | �|dg� d S �Nr� T)�packageZmodule_relativeZglobs)�doctestZmasterZtestfile�localsr\ )r. r� �logZ
old_masterr r r �
doDoctestFileW s �
z+UncompressedZipImportTestCase.doDoctestFilec C s | � | j� d S r& )r� r� r- r r r �testDoctestFilec s z-UncompressedZipImportTestCase.testDoctestFilec C s. g }t jd|dt� d��� | �|dg� d S r� )r� Z DocFileTestr� �runr\ )r. r� r� r r r �doDoctestSuitef s �
z,UncompressedZipImportTestCase.doDoctestSuitec C s | � | j� d S r& )r� r� r- r r r �testDoctestSuiten s z.UncompressedZipImportTestCase.testDoctestSuitec C s� z |� � 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 r\ � raise_src�stripr �tb_frame�io�StringIOr r� �getvalue�endswithr� )r. r� �tbrV �lno�n�line�sr r r � doTracebackq s z)UncompressedZipImportTestCase.doTracebackc C s( t d ttfi}| jd |t | jd� d S r� )rk rc r� r` r� rg r r r �
testTraceback� s z+UncompressedZipImportTestCase.testTracebackNz need an unencodable filenamec C sj t jd }| �t j|� t|d��.}ttd t�t ��}| j
|_ |� |t
� W 5 Q R X t�|��t� d S �Nz.ziprG r# )r �TESTFN_UNENCODABLEr; rK r r rk rL rM rc rN rO rP rd r7 r� r� �r. �filenamerT rU r r r �testUnencodable� s
z-UncompressedZipImportTestCase.testUnencodablec C s� t jd }| �t j|� t|d��.}ttd t�t ��}| j
|_ |� |t
� W 5 Q R X t�|� t�t�|�� | �t�� t�tt�|��� W 5 Q R X | �t�� t�tt�|��� W 5 Q R X d S r� )r ZTESTFNr; rK r r rk rL rM rc rN rO rP rd r7 r� r �fsencoder� � TypeErrorr� �
memoryviewr r r r �
testBytesPath� s
z+UncompressedZipImportTestCase.testBytesPathc C s& t d ttfi}| jd|t dd� d S )Nr# s comment�rH rj rg r r r � testComment� s z)UncompressedZipImportTestCase.testCommentc C s( t d ttfi}| jd|t ddd� d S )Nr# s@ cruftcruftcruftcruftcruftcruftcruftcruftcruftcruftcruftcruftcruftcruftcruftcruftcruftcruftcruftcruftcruftcruftcruftcruftcruftcruftcruftcruftcruftcruftcruftcruftcruftcruftcruftcruftcruftcruftcruftcruftcruftcruftcruftcruftcruftcruftcruftcruftcruftcruftcruftcruftcruftcruftcruftcruftcruftcruftcruftcruftcruftcruftcruftcrufts hi)rI rH rj rg r r r � testBeginningCruftAndComment� s z:UncompressedZipImportTestCase.testBeginningCruftAndCommentc C s* t d ttfi}| jd|t dd d� d S )Nr# � ci�� r rj rg r r r �testLargestPossibleComment� s z8UncompressedZipImportTestCase.testLargestPossibleComment)5r1 r2 r3 r rN r/ r� rF rZ rW r` rh rl ro rp r� �unittestZmockZpatchr� 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� ZskipIfr r r r r r
r r r r r r4 Q sZ
MI81
�
r4 c @ s e Zd ZeZdS )�CompressedZipImportTestCaseN)r1 r2 r3 r rN 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 r& )r� r7 �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 r� r r- r r r �testEmptyFilename� s z*BadFileZipImportTestCase.testEmptyFilenamec C sB | � ttjd � | j ttjtd d� | � ttjtt�t��� d S )N)Zkwd)r� r r7 r� rk 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 r& )r rK rk Zcreate_empty_filer r- r r r �
testEmptyFile� s
z&BadFileZipImportTestCase.testEmptyFilec
C sh t �t� t�ttjd�}z0t�|� | �t j
��}t � t� W 5 Q R X W 5 t�td� t �t� X d S )Nr i� ) r rK rk r rA �O_CREAT�chmod�closer� r7 r r� )r. �fd�cmr r r �testFileUnreadable� s
z+BadFileZipImportTestCase.testFileUnreadablec C s4 t �t� ttd�}|�d� |�� | �t� d S )N�w+Zaaaaaaaaaaaaaaaaaaaaaa)r rK rk rA rB r r )r. rE r r r �testNotZipFile� s
z'BadFileZipImportTestCase.testNotZipFilec C s t �t� ttd�}|�t�dd�� |�d� |�� t� t�}z�| � t
|jd � | � t
|jd � | � t
|jd � | � t
|jd � | � t
|jd � | � t
|jd � tj}| �|�d�d � | � ||jd� | � ||jd� | � t|jd� | � ||jd� | � ||jd� W 5 tj
� � X d S )Nr z=IiPKZaaaaaaaaaaaaaaaaaa�abc)r rK rk rA rB r r r r7 r� r8 r+ r� r r� r� r� �get_coder� r� r r\ �OSError)r. rE rT �errorr r r �_testBogusZipFile� s*
z*BadFileZipImportTestCase._testBogusZipFileN) r1 r2 r3 r r r r r r r r r% r r r r r � s r c C s$ zt �ttt� W 5 t �t� X d S r& )r rK rk Z run_unittestr4 r r r r r r � test_main� s �r& �__main__)7r' r r r �importlib.utilr rL r
Z
unittest.mockr� r Zzipfiler r r r r7 r5 r� r� r� � tracebackr r r ra re rd r Ztest_cor� r r" rc r� rn rk r� r� r! r� r� rZ r �cache_from_sourceZpyc_filerm ZTestCaser% r4 Z
requires_zlibr r r& r1 r r r r �<module> s\
aK