File: //usr/local/lib/python3.7/test/__pycache__/test_thread.cpython-37.opt-1.pyc
B
��g�! � @ s� d dl Z d dlZd dlZd dlmZ d dlZd dlZd dlZd dl Z d 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 dQ R X dS )z2Helper function for printing out debugging output.N)r �verbose� _print_mutex�print)�arg� r
�,/usr/local/lib/python3.7/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 d 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 d Q R X td|t|d �f � t�|� td| � | j�2 | jd8 _| jt kr�| jdkr�| j
� � W d Q R X d S )Ng ��@ztask %s will run for %susg ��.Az task %s doner r ) r �randomr �round�time�sleepr r r �NUMTASKSr �release)r �ident�delayr
r
r r 2 s
zThreadRunningTests.taskc C sL t �� �: xtt�D ] }| �� qW td� | j�� td� W d Q R X d S )Nz waiting for tasks to complete...zall 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_size)r r
r
r �test_stack_sizeF s
z"ThreadRunningTests.test_stack_size)�nt�posixztest meant for nt and posixc
C s yt �d� W n: tk
r* td� Y n t jk
rH | �d� Y nX d}x8dD ]0}t �|� | �t �� ||| � td| � qTW xldD ]d}td| � d | _d | _t �
� �: xt t �D ] }| �
� q�W td
� | j�� td � W d Q R X q�W 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)i i r z successfully set stack_size(%d))i i ztrying stack_size = (%d)r z!waiting for all tasks to completezall tasks done)r r/ �
ValueErrorr �errorZskipTestr. r r r r* r+ r& r! r r )r Zfail_msgZtssr, r
r
r � test_nt_and_posix_stack_sizeM s,
z/ThreadRunningTests.test_nt_and_posix_stack_sizec s� t �� }t �� ���� g ���fdd�}t�� �� t �|d� x�sRt�t � qBW | �
t �� |d � �� � g � t �
|� fdd��}~x� s�t�t � q�W | �
t �� |� W d Q R X d S )Nc s �� d � � �� � �� d S )N)�appendr r' r
)�mut�startedr
r r s s
z,ThreadRunningTests.test__count.<locals>.taskr
r c s
� � d �S )N)r6 )�_)�doner
r �<lambda>� � z0ThreadRunningTests.test__count.<locals>.<lambda>)r �_countr r r r* r r$ r% �
POLL_SLEEPr. r' �weakref�ref)r Zorigr Zwrr
)r: r7 r8 r � test__countl s"
z ThreadRunningTests.test__countc
s� �fdd�}� fdd�}t �� �t�d��B}|j� ||_��� t�� � t �|d� ��� W d Q R X W d Q R X | �d|� � � d S )Nc s � � � t�d S )N)r' � SyntaxErrorr
)r8 r
r r � s zCThreadRunningTests.test_save_exception_state_on_error.<locals>.taskc s0 yt �W n t k
r Y nX � | f|�� d S )N)r3 )r �args)�
real_writer
r �mywrite� s
zFThreadRunningTests.test_save_exception_state_on_error.<locals>.mywrite�stderrr
Z Traceback)
r r r Zcaptured_output�writer r* r ZassertIn�getvalue)r r rE rF r
)rD r8 r �"test_save_exception_state_on_error� s
z5ThreadRunningTests.test_save_exception_state_on_errorN)r r r r! r r- r0 �unittestZskipIf�os�namer5 rA rI r
r
r
r r ( s r c @ s e Zd Zdd� Zdd� ZdS )�Barrierc C s. || _ d| _t�� | _t�� | _| j�� d S )Nr )� num_threads�waitingr r �
checkin_mutex�checkout_mutexr )r rN 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 )rP r rO rN rQ r' )r r
r
r �enter� s
z
Barrier.enterN)r r r rR rS r
r
r
r rM � s rM c @ s e Zd Zdd� Zdd� ZdS )� BarrierTestc C sd t �� �R tt�| _t| _x tt�D ]}t�| j |f� q$W t
d� | j � � t
d� W d Q R X d S )Nzwaiting for tasks to endz
tasks done)
r r* rM r& �barr r+ r r �task2r r r )r r, r
r
r � test_barrier� s
zBarrierTest.test_barrierc
C s� x�t t�D ]x}|dkr d}n | j� t�� d }W d Q R X td|t|d �f � t�|� td||f � | j� � td| � q
W | j
� | j d8 _ | j dk}W d Q R X |r�| j �
� d S )Nr g ��@ztask %s will run for %susg ��.Aztask %s entering %sztask %s leaving barrierr )r+ �NUMTRIPSr r" r r# r$ r% rU rS r r r r' )r r( r, r) Zfinishedr
r
r rV � s
zBarrierTest.task2N)r r r rW rV r
r
r
r rT � s
rT c @ s e Zd ZejZdS )� LockTestsN)r r r r r Zlocktyper
r
r
r rY � s rY c @ s< e Zd Zdd� Ze�eed�d�ej dd� ��Z
dd� Z d S )
�TestForkInThreadc C s t �� \| _| _d S )N)rK �pipe�read_fd�write_fd)r r
r
r r � s zTestForkInThread.setUp�forkz need os.forkc sZ d�� �fdd�}t �� �* t�|d� � �t�� jd�dd� W d Q R X � ��d� d S ) Nznot setc s^ t �� } | dkr>z t �� j� t �� jd� W d t �d� X n t �� j� t �| d�\} �d S )Nr s OK)rK r^ �closer\ rG r] �_exit�waitpid)�pid)r �statusr
r �thread1� s z3TestForkInThread.test_forkinthread.<locals>.thread1r
� s OKzUnable to fork() in threadr )r r* r r r. rK �readr\ )r rd r
)r rc r �test_forkinthread� s
z"TestForkInThread.test_forkinthreadc C sP yt �| j� W n tk
r$ Y nX yt �| j� W n tk
rJ Y nX d S )N)rK r_ r\ �OSErrorr] )r r
r
r �tearDown s zTestForkInThread.tearDownN) r r r r rJ Z
skipUnless�hasattrrK r Z reap_threadsrg ri r
r
r
r rZ � s rZ �__main__)rK rJ r" Ztestr �_threadr r$ �sysr? r r& rX r> r r r ZTestCaser
r rM rT rY rZ r �mainr
r
r
r �<module> s* w%,