Your IP : 216.73.216.213
U
i�f�! � @ s� d dl Z d dlZd dlZd dlmZ d dlZd dlZd dlZd dlm Z dZ
dZ dZ e�
� Zdd� ZG d d
� d
ej�ZG d d � d e�ZG d
d� d�ZG dd� de�ZG dd� de j�ZG dd� dej�Zedkr�e�� dS )� N)�support)�
lock_tests�
� g{�G�z�?c C s" t jr t� t| � W 5 Q R X dS )z2Helper function for printing out debugging output.N)r �verbose� _print_mutex�print)�arg� r
�5/opt/alt/python38/lib64/python3.8/test/test_thread.py�
verbose_print s r c @ s e Zd Zdd� ZdS )�BasicThreadTestc C sX t �� | _| j�� t �� | _t �� | _d| _d| _d| _t �
� }| j t j f|�� d S �Nr )
�thread�
allocate_lock�
done_mutex�acquire�
running_mutex� random_mutex�created�running�
next_identr Zthreading_setupZ
addCleanupZthreading_cleanup)�self�keyr
r
r �setUp s
zBasicThreadTest.setUpN)�__name__�
__module__� __qualname__r r
r
r
r r
s r
c @ sV e Zd Zdd� Zdd� Zdd� Zdd� Ze�e j
d kd
�d d � �Z d
d� Z dd� Z
dS )�ThreadRunningTestsc C s` | j �P | jd7 _td| j � t�| j| jf� | jd7 _| jd7 _W 5 Q R X d S )N� zcreating task %s)r r r r �start_new_thread�taskr r �r r
r
r �newtask) s zThreadRunningTests.newtaskc C s� | j � t�� d }W 5 Q R X td|t|d �f � t�|� td| � | j�2 | jd8 _| jt kr�| jdkr�| j
� � W 5 Q R X d S )N� ��@�task %s will run for %sus� ��.Az task %s doner r ) r �randomr �round�time�sleepr r r �NUMTASKSr �release)r �ident�delayr
r
r r! 1 s
zThreadRunningTests.taskc C sH t �� �6 tt�D ] }| �� qtd� | j�� td� W 5 Q R X d S )Nz waiting for tasks to complete...�all tasks done)r �wait_threads_exit�ranger+ r# r r r �r �ir
r
r �test_starting_threads< s
z(ThreadRunningTests.test_starting_threadsc C s2 | � t�� dd� t�d� | � t�� dd� d S )Nr zinitial stack size is not 0z stack_size not reset to default)� assertEqualr �
stack_sizer" r
r
r �test_stack_sizeE s
z"ThreadRunningTests.test_stack_size)�nt�posixztest meant for nt and posixc
C s� zt �d� W n: tk
r* td� Y n t jk
rH | �d� Y nX d}dD ]0}t �|� | �t �� ||| � td| � qRdD ]`}td| � d | _d | _t �
� �6 t t �D ] }| �
� q�td
� | j�� td � W 5 Q R X q�t �d � d S ) Ni z3caught expected ValueError setting stack_size(4096)z4platform does not support changing thread stack sizez&stack_size(%d) failed - should succeed)� � r z successfully set stack_size(%d))r: r; ztrying stack_size = (%d)r z!waiting for all tasks to completer/ )r r6 �
ValueErrorr �errorZskipTestr5 r r r r0 r1 r+ r# r r )r Zfail_msgZtssr3 r
r
r � test_nt_and_posix_stack_sizeL s,
z/ThreadRunningTests.test_nt_and_posix_stack_sizec s� t �� }t �� ���� g ���fdd�}t�� �x t �|d� �sPt�t � q@| �
t �� |d � �� � g � t �
|� fdd��}~� s�t�t � q�| �
t �� |� W 5 Q R X d S )Nc s �� d � � �� � �� d S �N)�appendr r, r
)�mut�startedr
r r! r s
z,ThreadRunningTests.test__count.<locals>.taskr
r c s
� � d �S r? )r@ )�_)�doner
r �<lambda>� � z0ThreadRunningTests.test__count.<locals>.<lambda>)r �_countr r r r0 r r) r* �
POLL_SLEEPr5 r, �weakref�ref)r Zorigr! Zwrr
)rD rA rB r � test__countk s"
z ThreadRunningTests.test__countc
s� � fdd�}t �� � t�� �x}t�� �" � �� t �|d� � �� W 5 Q R X | �t|j j
�d� | � |j j |� | �|j j
d� | �|j j� W 5 Q R X d S )Nc s � � � td��d S )N� task failed)r, r< r
�rB r
r r! � s z:ThreadRunningTests.test_unraisable_exception.<locals>.taskr
rL z&Exception ignored in thread started by)r r r Zcatch_unraisable_exceptionr0 r r r5 �strZ
unraisable� exc_valueZassertIs�object�err_msgZassertIsNotNone�
exc_traceback)r r! �cmr
rM r �test_unraisable_exception� s
�z,ThreadRunningTests.test_unraisable_exceptionN)r r r r# r! r4 r7 �unittestZskipIf�os�namer> rK rT r
r
r
r r ' s
r c @ s e Zd Zdd� Zdd� ZdS )�Barrierc C s. || _ d| _t�� | _t�� | _| j�� d S r )� num_threads�waitingr r �
checkin_mutex�checkout_mutexr )r rY r
r
r �__init__� s
zBarrier.__init__c C s� | j �� | jd | _| j| jkr<| jd | _| j�� d S | j �� | j�� | jd | _| jdkrt| j �� d S | j�� d S )Nr r )r[ r rZ rY r\ r, r" r
r
r �enter� s
z
Barrier.enterN)r r r r] r^ r
r
r
r rX � s rX c @ s e Zd Zdd� Zdd� ZdS )� BarrierTestc C s` t �� �N tt�| _t| _tt�D ]}t�| j |f� q"t
d� | j � � t
d� W 5 Q R X d S )Nzwaiting for tasks to endz
tasks done)
r r0 rX r+ �barr r1 r r �task2r r r r2 r
r
r � test_barrier� s
zBarrierTest.test_barrierc
C s� t t�D ]x}|dkrd}n | j� t�� d }W 5 Q R X td|t|d �f � t�|� td||f � | j� � td| � q| j
� | j d8 _ | j dk}W 5 Q R X |r�| j �
� d S )Nr r$ r% r& ztask %s entering %sztask %s leaving barrierr )r1 �NUMTRIPSr r' r r( r) r* r` r^ r r r r, )r r- r3 r. Zfinishedr
r
r ra � s" �
zBarrierTest.task2N)r r r rb ra r
r
r
r r_ � s
r_ c @ s e Zd ZejZdS )� LockTestsN)r r r r r Zlocktyper
r
r
r rd � s rd c @ s< e Zd Zdd� Ze�eed�d�ej dd� ��Z
dd� Z d S )
�TestForkInThreadc C s t �� \| _| _d S r? )rV �pipe�read_fd�write_fdr" r
r
r r � s zTestForkInThread.setUp�forkz need os.forkc sZ d�� �fdd�}t �� �* t�|d� � �t�� jd�dd� W 5 Q R X � ��d� d S ) Nznot setc s^ t �� } | dkr>z t �� j� t �� jd� W 5 t �d� X n t �� j� t �| d�\} �d S )Nr � OK)rV ri �_exit�closerg �writerh �waitpid)�pid�r Zstatusr
r �thread1� s z3TestForkInThread.test_forkinthread.<locals>.thread1r
� rj zUnable to fork() in threadr )r r0 r r r5 rV �readrg )r rq r
rp r �test_forkinthread� s
�z"TestForkInThread.test_forkinthreadc C sP zt �| j� W n tk
r$ Y nX zt �| j� W n tk
rJ Y nX d S r? )rV rl rg �OSErrorrh r" r
r
r �tearDown� s zTestForkInThread.tearDownN) r r r r rU Z
skipUnless�hasattrrV r Z reap_threadsrt rv r
r
r
r re � s
re �__main__)rV rU r' Ztestr �_threadr r) rI r r+ rc rH r r r ZTestCaser
r rX r_ rd re r �mainr
r
r
r �<module> s( t%,