Your IP : 216.73.216.213
�
g f] � @ sK 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 d d � Z e d k rGe � n d S)� N)�support�_thread)�
lock_tests�
� c C s' t j r# t � t | � Wd QXn d S)z2Helper function for printing out debugging output.N)r �verbose� _print_mutex�print)�arg� r �5/opt/alt/python34/lib64/python3.4/test/test_thread.py�
verbose_print s r
c @ s e Z d Z d d � Z d S)�BasicThreadTestc C sY t j � | _ | j j � t j � | _ t j � | _ d | _ d | _ d | _ d S)Nr ) �thread�
allocate_lock�
done_mutex�acquire�
running_mutex� random_mutex�created�running�
next_ident)�selfr 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 sk | j �\ | j d 7_ t d | j � t j | j | j f � | j d 7_ | j d 7_ Wd QXd 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 QXt d | t | d � f � t j | � t d | � | j �B | j d 8_ | j t k r� | j d k r� | j
j � n Wd QXd 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 - 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_threads8 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_size@ 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)r2 r3 r )r2 r3 )r r. �
ValueErrorr
�errorZskipTestr- r r r* r& r! r r )r Zfail_msgZtssr+ r r r � test_nt_and_posix_stack_sizeG 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 k s
z,ThreadRunningTests.test__count.<locals>.taskg{�G�z�?r c s
� j d � S)N)r7 )�_)�doner r �<lambda>y s z0ThreadRunningTests.test__count.<locals>.<lambda>) r �_countr r r r$ r% r- r' �weakref�ref)r Zorigr �wrr )r; r8 r9 r � test__counte 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 QX| j
d | j � � d S)Nc s � j � t � d S)N)r' � SyntaxErrorr )r9 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)r4 )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 rE �crF r )rD r9 r �"test_save_exception_state_on_error s
z5ThreadRunningTests.test_save_exception_state_on_errorN)zntzposix)r r r r! r r, r/ �unittest�skipIf�os�namer6 rA rJ 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 rP 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 )rR r rQ rP rS r' )r r r r �enter� s
z
Barrier.enterN)r r r rT rU r r r r rO � s rO 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) rO 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 QXt d | t | d � f � t j | � t d | | f � | j j � t d | � q
W| j
�# | j d 8_ | j d k } Wd QX| r� | j j
� n d S)Nr g ��@ztask %s will run for %susg ��.Aztask %s entering %sztask %s leaving barrierr )r* �NUMTRIPSr r" r
r# r$ r% rW rU r r r r' )r r( r+ r) Zfinishedr r r rX � s
zBarrierTest.task2N)r r r rY rX r r r r rV � s rV 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)rM �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)rM �fork� RuntimeError�_exit�closer^ rG r_ )�pid)r r r �thread1� s
z3TestForkInThread.test_forkinthread.<locals>.thread1� s OKzUnable to fork() in thread)r r r- rM �readr^ )r rf 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)rM rd r^ �OSErrorr_ )r r r r �tearDown� s
zTestForkInThread.tearDownN)
r r r r rK rL �sys�platform�
startswithr Z reap_threadsri rk r r r r r\ � s
r\ c C s t j t t t t � d S)N)r Z run_unittestr rV r[ r\ r r r r � test_main� s ro �__main__)rM rK r" Ztestr �
import_moduler r$ rl r> r r& rZ r r r
ZTestCaser r rO rV r[ r\ ro r r r r r �<module> s* t$'