Your IP : 216.73.216.85
���]B � @ s? d d l Z d d l Z d d l Z d d l m Z e j d � Z d d l Z d d l Z d d l Z d d l m
Z
d Z d Z e j
� Z d d � Z Gd d
� d
e j � Z Gd d � d e � Z Gd
d � d � Z Gd d � d e � Z Gd d � d e
j � Z Gd d � d e j � Z e d k r;e j � d S)� N)�support�_thread)�
lock_tests�
� c C s% t j r! t � t | � Wd QRXd S)N)r �verbose� _print_mutex�print)�arg� r �0/opt/alt/python35/lib64/python3.5/test_thread.py�
verbose_print s r
c @ s e Z d Z d d � Z d S)�BasicThreadTestc C sx t j � | _ | j j � t j � | _ t j � | _ d | _ d | _ d | _ t j
� } | j t j | � 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 @ s e Z d Z d d � Z d d � Z d d � Z d d � Z e j e j
d k d � d d
� � Z d d � Z d d � Z
d S)�ThreadRunningTestsc
C sl | j �\ | j d 7_ t d | j � t j | j | j f � | j d 7_ | j d 7_ Wd QRXd 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 j � d } Wd QRXt d | t | d � f � t j | � t d | � | j �? | j d 8_ | j t k r� | j d k r� | j
j � Wd QRXd 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! 1 s
zThreadRunningTests.taskc C sF x t t � D] } | j � q
Wt d � | j j � t d � d S)Nz waiting for tasks to complete...zall tasks done)�ranger' r" r
r r )r �ir r r �test_starting_threads< s
z(ThreadRunningTests.test_starting_threadsc C sC | j t j � d d � t j d � | j t j � d d � d S)Nr zinitial stack size is not 0z stack_size not reset to default)� assertEqualr �
stack_size)r r r r �test_stack_sizeD s
z"ThreadRunningTests.test_stack_size�nt�posixztest meant for nt and posixc C s$ y t j d � Wn= t k
r/ t d � Yn" t j k
rP | j d � Yn Xd } xF d D]> } t j | � | j t j � | | | � t d | � q^ Wxp d
D]h } t d | � d | _ d | _ x t t
� D] } | j � q� Wt d
� | j j
� t d � q� Wt j 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)ztrying stack_size = (%d)z!waiting for all tasks to completezall tasks done)r3 r4 r )r3 r4 )r r/ �
ValueErrorr
�errorZskipTestr. r r r+ r' r" r r )r Zfail_msgZtssr, r r r � test_nt_and_posix_stack_sizeK s*
z/ThreadRunningTests.test_nt_and_posix_stack_sizec s� t j � } t j � � � j � g � � � f d d � } t j | f � x � sf t j d � qP W| j t j � | d � � j � g � t j
| � f d d � � } ~ x � s� t j d � q� W| j t j � | � d S)Nc s% � j d � � j � � j � d S)N)�appendr r( r )�mut�startedr r r! o s
z,ThreadRunningTests.test__count.<locals>.taskg{�G�z�?r c s
� j d � S)N)r8 )�_)�doner r �<lambda>} s z0ThreadRunningTests.test__count.<locals>.<lambda>) r �_countr r r r% r& r. r( �weakref�ref)r Zorigr! �wrr )r< r9 r: r � test__counti s
z ThreadRunningTests.test__countc
s� � f d d � } � f d d � } t j � } t j � � t j d � �c } | j � | | _ � j � t j | f � � j � x# t j � | k r� t j d � q� WWd QRX| j
d | 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 y
t � Wn t k
r Yn X� | | � d S)N)r5 )r �args)�
real_writer r �mywrite� s
zFThreadRunningTests.test_save_exception_state_on_error.<locals>.mywrite�stderrg{�G�z�?Z Traceback) r r> r r Zcaptured_output�writer r r% r& ZassertIn�getvalue)r r! rF �crG r )rE r: r �"test_save_exception_state_on_error� s
z5ThreadRunningTests.test_save_exception_state_on_errorN)r1 r2 )r r r r" r! r- r0 �unittest�skipIf�os�namer7 rB rK r r r r r ' s ' r c @ s( e Z d Z d d � Z d d � Z d S)�Barrierc C sA | | _ d | _ t j � | _ t j � | _ | j j � 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 � | j d | _ | j | j k rP | j d | _ | j j � d S| j j � | j j � | j d | _ | j d k r� | j j � d S| j j � 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 Z d Z d d � Z d d � Z d S)� BarrierTestc C sj t t � | _ t | _ x* t t � D] } t j | j | f � q% Wt d � | j j
� t d � d S)Nzwaiting for tasks to endz
tasks done) 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]� } | d k r( d } n! | j � t j � d } Wd QRXt d | t | d � f � t j | � t d | | f � | j j � t d | � q
W| j
�# | j d 8_ | j d k } Wd QRX| 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* �finishedr r r rY � s
zBarrierTest.task2N)r r r rZ rY r r r r rW � s rW c @ s e Z d Z e j Z d S)� LockTestsN)r r r r r Zlocktyper r r r r] � s r] c @ s[ e Z d Z d d � Z e j e j j d � d � e j
d d � � � Z d d � 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�winz5This test is only appropriate for POSIX-like systems.c sH � f d d � } t j | f � � j t j � j d � d d � d S)Nc s� y t j � } Wn t k
r1 t j d � Yn X| d k ry z' t j � j � t j � j d � Wd t j d � Xn t j � j � d S)Nr r s OK)rN �fork� RuntimeError�_exit�closer` rH ra )�pid)r r r �thread1� s
z3TestForkInThread.test_forkinthread.<locals>.thread1� s OKzUnable to fork() in thread)r r r. rN �readr` )r rh r )r r �test_forkinthread� s z"TestForkInThread.test_forkinthreadc C sV y t j | j � Wn t k
r( Yn Xy t j | j � Wn t k
rQ Yn Xd S)N)rN rf r` �OSErrorra )r r r r �tearDown� s
zTestForkInThread.tearDownN)
r r r r rL rM �sys�platform�
startswithr Z reap_threadsrk rm r r r r r^ � s
r^ �__main__)rN rL r# Ztestr �
import_moduler r% rn r? r r' r[ r r r
ZTestCaser r rP rW r] r^ r �mainr r r r �<module> s( t$'