File: //lib64/python3.6/test/__pycache__/test_thread.cpython-36.pyc
3
\�! � @ s� d dl Z d dlZd dlZd dlmZ ejd�Zd dlZd dlZd dl Z d dlm
Z
dZ dZ dZ
ej� Zdd � ZG d
d � d ej�ZG d d
� d
e�ZG dd� d�ZG dd� de�ZG dd� de
j�ZG dd� dej�Zedkr�ej� dS )� N)�support�_thread)�
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/lib64/python3.6/test_thread.py�
verbose_print s r
c @ s e Zd Zdd� ZdS )�BasicThreadTestc C sX t j� | _| jj� t j� | _t j� | _d| _d| _d| _t j
� }| 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� Zeje j
dkd �d d
� �Z dd� Z dd� Z
dS )�ThreadRunningTestsc
C s` | j �P | jd7 _td| j � tj| 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 � tj� d }W d Q R X td|t|d �f � tj|� td| � | j�2 | jd8 _| jt kr�| jdkr�| j
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 j� �: xtt�D ] }| j� qW td� | jj� 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 | j tj� dd� tjd� | j tj� 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 jd� W n: tk
r* td� Y n t jk
rH | jd� Y nX d}x8d D ]0}t j|� | jt j� ||| � td| � qTW xld
D ]d}td | � d| _d| _t j
� �: xt t �D ] }| j
� q�W td
� | jj� td � W d Q R X q�W t jd� 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)ztrying stack_size = (%d)z!waiting for all tasks to completezall tasks done)r4 r5 r )r4 r5 )r r0 �
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 j� }t j� ��j� g ���fdd�}tj� �� t j|f � x�sRtjt � qBW | j
t j� |d � �j � g � t j
|� fdd��}~x� s�tjt � q�W | j
t j� |� W d Q R X d S )Nc s �j d � � j� � j� d S )N)�appendr r( r )�mut�startedr r r! s s
z,ThreadRunningTests.test__count.<locals>.taskr c s
� j d �S )N)r9 )�_)�doner r �<lambda>� s z0ThreadRunningTests.test__count.<locals>.<lambda>)r �_countr r r r+ r r% r& �
POLL_SLEEPr/ r( �weakref�ref)r Zorigr! �wrr )r= r: r; r � test__countl s"
z ThreadRunningTests.test__countc s� �fdd�}� fdd�}t j� �tjd��B}|j� ||_�j� tj� � t j|f � �j� W d Q R X W d Q R X | jd|j � � d S )Nc s � j � t�d S )N)r( � SyntaxErrorr )r; 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)r6 )r �args)�
real_writer r �mywrite� s
zFThreadRunningTests.test_save_exception_state_on_error.<locals>.mywrite�stderrZ Traceback)
r r r Zcaptured_output�writer r+ r ZassertIn�getvalue)r r! rH rI r )rG r; r �"test_save_exception_state_on_error� s
z5ThreadRunningTests.test_save_exception_state_on_errorN)r2 r3 )r r r r" r! r. r1 �unittestZskipIf�os�namer8 rD rL r r r r r ( s r c @ s e Zd Zdd� Zdd� ZdS )�Barrierc C s. || _ d| _tj� | _tj� | _| jj� d S )Nr )� num_threads�waitingr r �
checkin_mutex�checkout_mutexr )r rQ r r r �__init__� s
zBarrier.__init__c C s� | j j� | jd | _| j| jkr<| jd | _| jj� d S | j j� | jj� | jd | _| jdkrt| j j� d S | jj� d S )Nr r )rS r rR rQ rT r( )r r r r �enter� s
z
Barrier.enterN)r r r rU rV r r r r rP � s rP c @ s e Zd Zdd� Zdd� ZdS )� BarrierTestc C sd t j� �R tt�| _t| _x tt�D ]}tj| j |f� q$W t
d� | j j � t
d� W d Q R X d S )Nzwaiting for tasks to endz
tasks done)
r r+ rP 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� tj� d }W d Q R X td|t|d �f � tj|� td||f � | jj � td| � q
W | j
� | j d8 _ | j dk}W d Q R X |r�| j 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& rX rV r r r r( )r r) r- r* Zfinishedr r r rY � s
zBarrierTest.task2N)r r r rZ rY r r r r rW � s
rW c @ s e Zd ZejZdS )� LockTestsN)r r r r r Zlocktyper r r r r\ � s r\ c @ s< e Zd Zdd� Zejeed�d�ej dd� ��Z
dd� Z d S )
�TestForkInThreadc C s t j� \| _| _d S )N)rN �pipe�read_fd�write_fd)r r r r r � s zTestForkInThread.setUp�forkz need os.forkc sZ d�� �fdd�}t j� �* tj|f � � jtj� jd�dd� W d Q R X � j�d� d S )Nznot setc s^ t j� } | dkr>z t j� j� t j� jd� W d t jd� X n t j� j� t j| d�\} �d S )Nr s OK)rN ra �closer_ rJ r` �_exit�waitpid)�pid)r �statusr r �thread1� s z3TestForkInThread.test_forkinthread.<locals>.thread1� s OKzUnable to fork() in threadr )r r+ r r r/ rN �readr_ )r rg r )r rf r �test_forkinthread� s
z"TestForkInThread.test_forkinthreadc C sP yt j| j� W n tk
r$ Y nX yt j| j� W n tk
rJ Y nX d S )N)rN rb r_ �OSErrorr` )r r r r �tearDown s zTestForkInThread.tearDownN) r r r r rM Z
skipUnless�hasattrrN r Z reap_threadsrj rl r r r r r] � s r] �__main__)rN rM r# Ztestr �
import_moduler r% �sysrA r r' r[ r@ r r r
ZTestCaser r rP rW r\ r] r �mainr r r r �<module> s*
w%,