Your IP : 216.73.216.213
a
XC?h1u � @ 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 ey� dZY n0 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
�2� G dd� de1��Z3G dd� dej/�Z4d d � Z5e6d k�r�e�7� 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 s6 t �| �}tjjt�ddt|�d@ |d@ � | }|S )Nz<iLLr � �� )�marshal�dumps� importlib�util� MAGIC_NUMBER�struct�pack�int)�co�mtime�size�data�pyc� r �8/opt/alt/python39/lib64/python3.9/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 �setUp= 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 �tearDownD s
z ImportHooksBaseTestCase.tearDownN)�__name__�
__module__� __qualname__r/ r0 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� 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,dIdJ� Z-e�.e/j0dKu dL�dMdN� �Z1dOdP� Z2dQdR� Z3dSdT� Z4dUdV� Z5dKS )W� UncompressedZipImportTestCasec C s t �� tj�� t�| � d S r& )� linecache�
clearcache� zipimport�_zip_directory_cacher+ r% r/ r- r r r r/ P s
z#UncompressedZipImportTestCase.setUpc C s� | � tj|� |�� D ]�\}\}}tj�||�}|d tjkrVtj�|�s�t� |� qtj�
|�}tj�|�sxt� |� t |d��}|� |� W d � q1 s�0 Y qd 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 | � tj|� t|d��d}|�� D ]0\}\}}t|t�|��}| j|_ |�
||� q"|� dd �} | d urn| |_ W d � n1 s�0 Y |� dd �}
|
d u�rt
|d��} | �� }W d � n1 s�0 Y t
|d��$} | �|
� | �|� W d � n1 �s
0 Y 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 �makeZiph s $
&
z%UncompressedZipImportTestCase.makeZipc O sz | j |fi |�� tj�dt� t�d�|��}|�d�}|d urJ||� |rv|� � }| �
|t jjtg|�R � | � 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 jv r| �d� dt jv r&t jd= dttfi}z| �d|d� W n& tyj | jt krf| �
d� Y n0 | 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 tyL Y n 0 | �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 s> t tdtt��}td ttftt t|fi}| �d|t� d S )Nr r# )r �test_co�lenrd rk rc rm r` )r. Ztwenty_thirty_eight_pycrC r r r �
test2038MTime� s
�z+UncompressedZipImportTestCase.test2038MTimec 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\ r� �__path__�basename�splitr �listr= �__file__)
r. r� r� �packdir3�files1�files2�zip1�zip2r^ �p1�p2�mpath�subpkgr r r �testMixedNamespacePackage sl
��
�
��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 �testNamespacePackagek s`
� �
z2UncompressedZipImportTestCase.testNamespacePackagec C s� t tj }|t tj }|d t ttf|d t ttf|t t ttfdt ttfi}| �t j
t � t t d��P}|�
� D ]6\}\}}t|t�|��}| j|_d|_|�||� qzW d � n1 s�0 Y 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)v � | �| �*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� sX
� ,
z4UncompressedZipImportTestCase.testZipImporterMethodsc C sP t tj }|t tj }|d t ttf|t t ttfi}| �t j
t � t t d��P}|�
� D ]6\}\}}t|t�|��}| j|_d|_|�||� q^W d � n1 s�0 Y 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'v � | �| �(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� sX
� ,
��
�
zBUncompressedZipImportTestCase.testZipImporterMethodsInSubDirectoryc C s� | � tjt� ttd��>}| j|_d}tdd� td�D ��}|�||� W d � n1 s\0 Y t �
t�}| � ||� |�� | �
dt|�� d S )NrG z testdata.datc s s | ]
}|V qd S r& r )�.0�xr r r � <genexpr>" � 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 � testDefaultOptimizationLevel4 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_WithStuffA s
�z2UncompressedZipImportTestCase.testImport_WithStuffc C s | � t�|�t� d S r& )r\ �inspectZ getsourcerd )r. �moduler r r �assertModuleSourceH s z0UncompressedZipImportTestCase.assertModuleSourcec C s( t d ttfi}| jd|t | jd� d S �Nr# rz )rk rc rd r` r� rg r r r �
testGetSourceK 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 �testGetCompiledSourceO 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 �
runDoctestU s �z(UncompressedZipImportTestCase.runDoctestc C sL g }t jd }t _z t jd|dt� d� W |t _n|t _0 | �|dg� d S �Nr� T)�packageZmodule_relativeZglobs)�doctestZmasterZtestfile�localsr\ )r. r� �logZ
old_masterr r r �
doDoctestFileZ s �z+UncompressedZipImportTestCase.doDoctestFilec C s | � | j� d S r& )r� r� r- r r r �testDoctestFilef 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 �doDoctestSuitei s �
z,UncompressedZipImportTestCase.doDoctestSuitec C s | � | j� d S r& )r� r� r- r r r �testDoctestSuiteq s z.UncompressedZipImportTestCase.testDoctestSuitec
C s� z |� � W n� ty� } z�|jj}t|d�d \}}}}| �|t�� � t|j d�d \}}}}| �|t�� � t
� � }t |d|� | �
|�� �t�� W Y d }~nd }~0 0 td��d S )Nr� r zThis ought to be impossible)Zdo_raise� Exception�
__traceback__�tb_nextr r\ � raise_src�stripr �tb_frame�io�StringIOr r� �getvalue�endswithr� ) r. r� �e�tbrV �lno�n�line�sr r r � doTracebackt 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 s~ t jd }| �t j|� t|d��8}ttd t�t ��}| j
|_ |� |t
� W d � n1 s`0 Y 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��8}ttd t�t ��}| j
|_ |� |t
� W d � n1 s`0 Y t�|� t�t�|�� | �t��$ t�tt�|��� W d � n1 s�0 Y | �t��$ t�tt�|��� W d � n1 s�0 Y 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
*
2 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)6r1 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 r ZskipIfr r r r r r r r r r r r4 L s\
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 s� t �t� t�ttjd�}zZt�|� | �tj ��}t�
t� W d � n1 sR0 Y W t� td� t �t� nt� td� t �t� 0 d S )Nr i� ) r rK rk r rA �O_CREAT�closer� r7 r r� �chmod)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 |j d � | �
t |j
d � | �
t |jd � | �
t |jd � | �
t |jd � | �
t |jd � tj}| �|� d�d � | �
||j
d� | �
||jd� | �
t|jd� | �
||jd� | �
||jd� W tj�� n tj�� 0 d S )Nr$ z=IiPKZaaaaaaaaaaaaaaaaaa�abc)r rK rk rA rB r r r r7 r� r� r
r� r� r� �get_coder� r� r r\ �OSErrorr8 r+ )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 t �t� d S r& )r rK rk r r r r �tearDownModule s r+ �__main__)8r' 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 r� r� 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 �mainr r r r �<module> s\
iK