Your IP : 216.73.217.129
U
i�f�# � @ s� d dl Zd dlZd dlZd dlZd dlZd dlZd dlZd dlZd dlm Z d dl
m Z m Z m
Z
mZmZmZmZmZ dd� Zdd� Zdd d
d d �ZG d
d� d�ZG dd� d�ZG dd� dej�Zedd� �Zedkr�e� dS )� N)�mock)�verbose� run_unittest�TESTFN� reap_threads�forget�unlink�rmtree�
start_threadsc
C s� zvz>t|�d r"dd l}dd l}ndd l}dd l}|�dd�}W n2 tk
rr } z|� |� d �� W 5 d }~X Y nX W 5 |� t�� � t|�| k}|r�|�� X d S )N� r � � )
�append� threading� get_ident�len�set� modulefinder�randomZ randrange� Exception�with_traceback) �N�done�
done_tasks�errorsZfinishedr r �x�e� r �>/opt/alt/python38/lib64/python3.8/test/test_threaded_import.py�task s
& r c C s t jddd�| �S )Nzos.register_at_forkT)Zcreate)r Zpatch)�funcr r r �mock_register_at_fork) s r! zaif 1:
import time
time.sleep(%(delay)s)
x = 'a'
import C
zaif 1:
import time
time.sleep(%(delay)s)
x = 'b'
import D
zimport Bzimport A)�A�B�C�Dc @ s e Zd Zdd� Zddd�ZdS )�Finderc C s d| _ d| _t�� | _d S �Nr )�numcallsr r ZLock�lock��selfr r r �__init__G s zFinder.__init__Nc C s> | j � | jd7 _W 5 Q R X | j}t�d� |d | _d S )Nr g{�G�z�?)r) r( r �time�sleep)r+ �name�path�targetr r r r � find_specL s
zFinder.find_spec)NN)�__name__�
__module__� __qualname__r, r2 r r r r r& C s r& c @ s e Zd Zddd�ZdS )�FlushingFinderNc C s t j�� d S �N)�sys�path_importer_cache�clear)r+ r/ r0 r1 r r r r2 [ s zFlushingFinder.find_spec)NN)r3 r4 r5 r2 r r r r r6 W s r6 c @ s\ e Zd Zdd� Zdd� Zedd� �Zdd� Zd d
� Zd d � Z d
d� Z
dd� Z edd� �Z dS )�ThreadedImportTestsc C s t j�dd �| _d S �Nr )r8 �modules�pop�
old_randomr* r r r �setUpa s zThreadedImportTests.setUpc C s | j d k r| j tjd< d S r< )r? r8 r= r* r r r �tearDownd s
z ThreadedImportTests.tearDownc
s t �� rt�d��t�� �dD ]� tr6td� ddd� dD ]&}z tj |= W q: t
k
r^ Y q:X q:g �g ��� � t �
� }t� ���fdd �t� �D ��� W 5 Q R X ��d
�}t �
� | }tr�td |d d
dd� dt��� f }| ��|� | �||� tr td� q d S )Nz"can't run when import lock is held)� �2 rB rC rB rC ZTryingz threads ...� )�end)r r c 3 s$ | ] }t jt� ���fd �V qdS ))r1 �argsN)r �Threadr )�.0�i�r r r r r r � <genexpr> s �
�zAThreadedImportTests.check_parallel_module_init.<locals>.<genexpr>iX z%.1f msg @�@T)�flushrE z done: %s/%szOK.)�imp� lock_held�unittestZSkipTestr ZEventr �printr8 r= �KeyErrorr: r- � monotonicr
�range�waitr � assertFalseZ
assertTrue)r+ �mock_os�modnameZt0Z completedZdtZdbg_infor rJ r �check_parallel_module_initk s8
�
z.ThreadedImportTests.check_parallel_module_initc C s | � � d S r7 )rX r* r r r �test_parallel_module_init� s z-ThreadedImportTests.test_parallel_module_initc C sR t � }tj�d|� z*| �� | �|jd� | �|j |j� W 5 tj�|� X d S r' )
r& r8 � meta_path�insert�removerX �
assertGreaterr( � assertEqualr )r+ �finderr r r �test_parallel_meta_path� s z+ThreadedImportTests.test_parallel_meta_pathc s� t � � t� }� fdd�}tj�d|� tj�|� z4|�d� | � � }| �
� j d� | � � j
� j � W 5 tj�|� tj�|� X d S )Nc s � � d� t�d S )N� )r2 � ImportError)r0 �r_ r r � path_hook� s
z?ThreadedImportTests.test_parallel_path_hooks.<locals>.path_hookr ra )r& r6 r8 �
path_hooksr[ rZ r r\ r2 rX r] r( r^ r )r+ Zflushing_finderrd Znumtestsr rc r �test_parallel_path_hooks� s
z,ThreadedImportTests.test_parallel_path_hooksc C s< z t jd= W n tk
r Y nX dd l}| �|jj� d S )Nz test.threaded_import_hangersr )r8 r= rQ Z test.threaded_import_hangersrU Zthreaded_import_hangersr )r+ Ztestr r r �test_import_hangers� s z'ThreadedImportTests.test_import_hangersc
s d}t �t� | �tjt� tj�dt� | �tjj t� t
� � D ]T\}}|d|i }t t j�
t|d �d��}|�|�d�� W 5 Q R X | �t|� qBt�� g � � fdd�}� fd d
�}tj|d �}tj|d �}|�� |�� |�
� |�
� | �t� �d d
h� d S )Ng �?r �delay�.py�wb�utf-8c s dd l } � �t| dd �� d S �Nr r )r" r �getattr)r" �Zresultsr r � import_ab� s z<ThreadedImportTests.test_circular_imports.<locals>.import_abc s dd l } � �t| dd �� d S rl )r# r rm )r# rn r r � import_ba� s z<ThreadedImportTests.test_circular_imports.<locals>.import_ba)r1 �a�b)�os�mkdirr �
addCleanup�shutilr r8 r0 r[ r\ �circular_imports_modules�items�open�join�write�encoder � importlib�invalidate_cachesr rG �startr^ r ) r+ rh r/ �contents�fro rp �t1�t2r rn r �test_circular_imports� s*
z)ThreadedImportTests.test_circular_importsc C s� d}t j�dtj� | �t jjtj� td }t|d��}|� |�
d�� W 5 Q R X | �t |� | �t t� | �t
d� t�� tt� t jt= d S )Nz�if 1:
import threading
def target():
import random
t = threading.Thread(target=target)
t.start()
t.join()
t = Noner ri rj rk � __pycache__)r8 r0 r[ rs �curdirru r\ r ry r{ r| r r r r} r~ �
__import__r= )r+ rV �code�filenamer� r r r �test_side_effect_import� s z+ThreadedImportTests.test_side_effect_importN)
r3 r4 r5 r@ rA r! rX rY r` rf rg r� r� r r r r r; _ s
!
&r; c C sV d } zt �� } t �d� W n tk
r. Y nX z tt� W 5 | d k rPt �| � X d S )Ng�h㈵��>)r8 �getswitchinterval�setswitchinterval�AttributeErrorr r; )Zold_switchintervalr r r � test_main� s r� �__main__) �_imprM rs r} r8 r- rv r rO r Z test.supportr r r r r r r r
r r! rw r& r6 ZTestCaser; r� r3 r r r r �<module> s0 (
�