Your IP : 216.73.216.213
�
f f%f � @ s� d Z d d l Z d d l m Z d d l Z d d l Z d d l m Z d Z e j e � Z Gd d � d e j � Z
Gd d � d e j � Z Gd
d � d e j � Z Gd d
� d
e j � Z
e d k r� e j � n d S)zTests for lock.py� N)�mock)�
test_utilszt^<(?P<class>.*?) object at (?P<address>.*?)\[(?P<extras>(set|unset|locked|unlocked)(,value:\d)?(,waiters:\d+)?)\]>\Zc @ s� e Z d Z d d � Z d d � Z d d � Z d d � Z d d
� Z d d � Z d
d � Z d d � Z
d d � Z d d � Z d d � Z
d d � Z d d � Z d S) � LockTestsc C s | j � | _ d S)N)�
new_test_loop�loop)�self� r �A/opt/alt/python34/lib64/python3.4/test/test_asyncio/test_locks.py�setUp s zLockTests.setUpc C s` t j � } t j d | � } | j | j | � t j d | j � } | j | j | j � d S)Nr )r �Mock�asyncio�Lock�assertIs�_loopr )r r �lockr r r �test_ctor_loop s
zLockTests.test_ctor_loopc C s6 t j | j � t j � } | j | j | j � d S)N)r �set_event_loopr r
r r )r r r r r �test_ctor_noloop s zLockTests.test_ctor_noloopc s� t j d | j � � | j t � � j d � � | j t j t � � � � t j � f d d � � } | j j | � � | j t � � j d � � | j t j t � � � � d S)Nr z [unlocked]>c 3 s
� Dd Hd S)Nr r )r r r � acquire_lock* s z)LockTests.test_repr.<locals>.acquire_lockz [locked]>)
r r
r �
assertTrue�repr�endswith�RGX_REPR�match� coroutine�run_until_complete)r r r )r r � test_repr% s zLockTests.test_reprc s� t j d | j � � t j � f d d � � } | j j | � � } | j | � | j � j � � � j � | j � j � � d S)Nr c 3 s � Dd HS)Nr r )r r r r 5 s z)LockTests.test_lock.<locals>.acquire_lock) r r
r r r r �locked�release� assertFalse)r r �resr )r r � test_lock2 s
zLockTests.test_lockc s. t j d | j � � g } | j | j j � j � � � t j � f d d � � } t j � f d d � � } t j � f d d � � } t j | | � d | j �} t j | | � d | j �} t j | j � | j
g | � � j � t j | j � | j
d g | � t j | j � | j
d g | � t j | | � d | j �} � j � t j | j � | j
d d g | � � j � t j | j � | j
d d d
g | � | j | j � � | j | j
� � | j | j � � | j | j
� � | j | j � � | j | j
� � d S) Nr c 3 s% � j � Dd Hr! | j d � n d S)N� T)�acquire�append)�result)r r r �c1G s z"LockTests.test_acquire.<locals>.c1c 3 s% � j � Dd Hr! | j d � n d S)N� T)r# r$ )r% )r r r �c2M s z"LockTests.test_acquire.<locals>.c2c 3 s% � j � Dd Hr! | j d � n d S)N� T)r# r$ )r% )r r r �c3S s z"LockTests.test_acquire.<locals>.c3r" r' r) )r r
r r r r# r �Taskr � run_briefly� assertEqualr �doner% )r r% r& r( r* �t1�t2�t3r )r r � test_acquireA s8
zLockTests.test_acquirec C s� t j d | j � } | j | j j | j � � � t j | j � d | j �} | j j | j � | j t j
| j j | � | j | j � d S)Nr )
r r
r r r r# r+ � call_soon�cancel� assertRaises�CancelledErrorr �_waiters)r r Ztaskr r r �test_acquire_cancelw s z LockTests.test_acquire_cancelc s� t j d | j � � t j � f d d � � } t j d | j � } t j | d | � d | j �} t j | j � | j � j � � t j | d d � d | j �} t j | j � | j
t � j � d � t j | d d � d | j �} t j | j � | j
t � j � d � | j
d � | j � | j � j d j � � t j | j � | j � j � � | j | j � � | j | j � � | j | j � � d S)
Nr c
3 s= � j � Dd Hz | d k r* | Dd Hn Wd � j � Xd S)N)r# r )�nameZblocker)r r r �lockit� s
z*LockTests.test_cancel_race.<locals>.lockit�A�Br" �Cr' r )r r
r r ZFuturer+ r r, r r r- �lenr7 Z
set_resultr4 Z cancelledr r. )r r: ZfaZta�tbZtcr )r r �test_cancel_race� s( !!!
zLockTests.test_cancel_racec C s, t j d | j � } | j t | j � d S)Nr )r r
r r5 � RuntimeErrorr )r r r r r �test_release_not_acquired� s z#LockTests.test_release_not_acquiredc C s_ t j d | j � } | j j | j � � | j | j � � | j � | j | j � � d S)Nr ) r r
r r r# r r r r )r r r r r �test_release_no_waiters� s
z!LockTests.test_release_no_waitersc sv t j d | j � � t j � f d d � � } | j j | � � � | j � j � � Wd QX| j � j � � d S)Nr c 3 s � Dd HS)Nr r )r r r r � s z4LockTests.test_context_manager.<locals>.acquire_lock)r r
r r r r r r )r r r )r r �test_context_manager� s
z LockTests.test_context_managerc s� t j d | j � � t j � f d d � � } | j j | � � } | � | j � j � � Wd QX| j � j � � | j t � � | � Wd QXWd QXd S)Nr c 3 s � Dd HS)Nr r )r r r r � s z?LockTests.test_context_manager_cant_reuse.<locals>.acquire_lock)
r r
r r r r r r r5 �AttributeError)r r �cmr )r r � test_context_manager_cant_reuse� s z)LockTests.test_context_manager_cant_reusec C s� t j d | j � } y | � | j d � Wd QXWn; t k
rp } z | j t | � d � WYd d } ~ Xn X| j | j � � d S)Nr z-RuntimeError is not raised in with expressionz9"yield from" should be used as context manager expression) r r
r �failrA r- �strr r )r r �errr r r � test_context_manager_no_yield� s z'LockTests.test_context_manager_no_yieldN)�__name__�
__module__� __qualname__r
r r r r! r2 r8 r@ rB rC rD rG rK r r r r r s
6 0 r c @ s| e Z d Z d d � Z d d � Z d d � Z d d � Z d d
� Z d d � Z d
d � Z d d � Z
d d � Z d S)�
EventTestsc C s | j � | _ d S)N)r r )r r r r r
� s zEventTests.setUpc C s` t j � } t j d | � } | j | j | � t j d | j � } | j | j | j � d S)Nr )r r r �Eventr r r )r r �evr r r r � s
zEventTests.test_ctor_loopc C s6 t j | j � t j � } | j | j | j � d S)N)r r r rP r r )r rQ r r r r � s zEventTests.test_ctor_noloopc C s� t j d | j � } | j t | � j d � � t j t | � � } | j | j d � d � | j
� | j t | � j d � � | j t j t | � � � | j j t
j � � | j d t | � k � | j t j t | � � � d S)Nr z[unset]>ZextrasZunsetz[set]>z waiters:1)r rP r r r r r r r- �group�setr7 r$ r r )r rQ r r r r r � s
zEventTests.test_reprc s� t j d | j � � | j � j � � g } t j � f d d � � } t j � f d d � � } t j � f d d � � } t j | | � d | j �} t j | | � d | j �} t j | j � | j g | � t j | | � d | j �} � j
� t j | j � | j d d d
g | � | j | j � � | j
| j � � | j | j � � | j
| j � � | j | j � � | j
| j � � d S) Nr c 3 s% � j � Dd Hr! | j d � n d S)Nr" )�waitr$ )r% )rQ r r r& s z EventTests.test_wait.<locals>.c1c 3 s% � j � Dd Hr! | j d � n d S)Nr' )rT r$ )r% )rQ r r r( s z EventTests.test_wait.<locals>.c2c 3 s% � j � Dd Hr! | j d � n d S)Nr) )rT r$ )r% )rQ r r r* s z EventTests.test_wait.<locals>.c3r) r" r' )r rP r r �is_setr r+ r r, r- rS r r. Z assertIsNoner% )r r% r& r( r* r/ r0 r1 r )rQ r � test_wait s(
zEventTests.test_waitc C sH t j d | j � } | j � | j j | j � � } | j | � d S)Nr )r rP r rS r rT r )r rQ r r r r �test_wait_on_set3 s
zEventTests.test_wait_on_setc C sv t j d | j � } t j | j � d | j �} | j j | j � | j t j | j j | � | j
| j � d S)Nr ) r rP r r+ rT r3 r4 r5 r6 r r r7 )r rQ rT r r r �test_wait_cancel: s zEventTests.test_wait_cancelc C sf t j d | j � } | j | j � � | j � | j | j � � | j � | j | j � � d S)Nr )r rP r r rU rS r �clear)r rQ r r r �
test_clearD s
zEventTests.test_clearc s. t j d | j � � g } t j � f d d � � } t j | | � d | j �} t j | j � | j g | � � j � � j � | j
� j � � � j � � j � | j d t � j
� � t j | j � | j d g | � | j d t � j
� � | j | j � � | j | j � � d S)Nr c 3 s% � j � Dd Hr! | j d � n d S)Nr" T)rT r$ )r% )rQ r r r& R s z.EventTests.test_clear_with_waiters.<locals>.c1r" r )r rP r r r+ r r, r- rS rY r rU r> r7 r r. r% )r r% r& �tr )rQ r �test_clear_with_waitersN s"
z"EventTests.test_clear_with_waitersN) rL rM rN r
r r r rV rW rX rZ r\ r r r r rO � s (
rO c @ s� e Z d Z d d � Z d d � Z d d � Z d d � Z d d
� Z d d � Z d
d � Z d d � Z
d d � Z d d � Z d d � Z
d d � Z d d � Z d d � Z d d � Z d d � Z d! d" � Z d# S)$�ConditionTestsc C s | j � | _ d S)N)r r )r r r r r
n s zConditionTests.setUpc C s` t j � } t j d | � } | j | j | � t j d | j � } | j | j | j � d S)Nr )r r r � Conditionr r r )r r �condr r r r q s
z ConditionTests.test_ctor_loopc C s6 t j | j � t j � } | j | j | j � d S)N)r r r r^ r r )r r_ r r r r y s z ConditionTests.test_ctor_noloopc s� t j d | j � � g } t j � f d d � � } t j � f d d � � } t j � f d d � � } t j | | � d | j �} t j | | � d | j �} t j | | � d | j �} t j | j � | j g | � | j � j � � | j
| j j � j � � � � j
� t j | j � | j g | � | j
� j � � � j � t j | j � | j d g | � | j
� j � � � j
d � t j | j � | j d g | � | j
� j � � � j � t j | j � | j d d g | � | j
� j � � � j � t j | j � | j d d d
g | � | j
� j � � | j
| j � � | j
| j � � | j
| j � � | j
| j � � | j
| j � � | j
| j � � d S) Nr c 3 s4 � j � Dd H� j � Dd Hr0 | j d � n d S)Nr" T)r# rT r$ )r% )r_ r r r& � s z$ConditionTests.test_wait.<locals>.c1c 3 s4 � j � Dd H� j � Dd Hr0 | j d � n d S)Nr' T)r# rT r$ )r% )r_ r r r( � s z$ConditionTests.test_wait.<locals>.c2c 3 s4 � j � Dd H� j � Dd Hr0 | j d � n d S)Nr) T)r# rT r$ )r% )r_ r r r* � s z$ConditionTests.test_wait.<locals>.c3r" r' r) )r r^ r r r+ r r, r- r r r r r# �notifyr r. r% )r r% r& r( r* r/ r0 r1 r )r_ r rV ~ sL
zConditionTests.test_waitc C s� t j d | j � } | j j | j � � t j | j � d | j �} | j j | j � | j t j
| j j | � | j | j � | j
| j � � d S)Nr )r r^ r r r# r+ rT r3 r4 r5 r6 r r7 r r )r r_ rT r r r rX � s z ConditionTests.test_wait_cancelc C s8 t j d | j � } | j t | j j | j � � d S)Nr )r r^ r r5 rA r rT )r r_ r r r �test_wait_unacquired� s z#ConditionTests.test_wait_unacquiredc sV t j d | j � � d � � f d d � � g } t j � � f d d � � } t j | | � d | j �} t j | j � | j g | � | j j � j � � � j
� � j � t j | j � | j g | � d � | j j � j � � � j
� � j � t j | j � | j d g | � | j | j
� � | j | j � � d S) Nr Fc s � S)Nr r )�presultr r � predicate� s z/ConditionTests.test_wait_for.<locals>.predicatec 3 sA � j � Dd H� j � � Dd Hr= | j d � � j � n d S)Nr" T)r# �wait_forr$ r )r% )r_ rc r r r& � s
z(ConditionTests.test_wait_for.<locals>.c1Tr" )r r^ r r r+ r r, r- r r# r` r r r. r% )r r% r& r[ r )r_ rc rb r �
test_wait_for� s*
z ConditionTests.test_wait_forc C s{ t j d | j � } | j j | j d d � � � } | j d d d g | � | j t | j j | j d d � � � d S)Nr c S s
d d d g S)Nr" r' r) r r r r r �<lambda>� s z9ConditionTests.test_wait_for_unacquired.<locals>.<lambda>r" r' r) c S s d S)NFr r r r r rf s )r r^ r r rd r- r5 rA )r r_ r r r r �test_wait_for_unacquired� s ! z'ConditionTests.test_wait_for_unacquiredc s t j d | j � � g } t j � f d d � � } t j � f d d � � } t j � f d d � � } t j | | � d | j �} t j | | � d | j �} t j | | � d | j �} t j | j � | j g | � | j j � j � � � j
d � � j � t j | j � | j d g | � | j j � j � � � j
d � � j
d � � j � t j | j � | j d d
d g | � | j | j
� � | j | j � � | j | j
� � | j | j � � | j | j
� � | j | j � � d S) Nr c 3 s> � j � Dd H� j � Dd Hr: | j d � � j � n d S)Nr" T)r# rT r$ r )r% )r_ r r r& s
z&ConditionTests.test_notify.<locals>.c1c 3 s> � j � Dd H� j � Dd Hr: | j d � � j � n d S)Nr' T)r# rT r$ r )r% )r_ r r r( s
z&ConditionTests.test_notify.<locals>.c2c 3 s> � j � Dd H� j � Dd Hr: | j d � � j � n d S)Nr) T)r# rT r$ r )r% )r_ r r r* s
z&ConditionTests.test_notify.<locals>.c3r" i r' r) )r r^ r r r+ r r, r- r r# r` r r r. r% )r r% r& r( r* r/ r0 r1 r )r_ r � test_notify s6
zConditionTests.test_notifyc sM t j d | j � � g } t j � f d d � � } t j � f d d � � } t j | | � d | j �} t j | | � d | j �} t j | j � | j g | � | j j � j � � � j
� � j � t j | j � | j d d g | � | j | j
� � | j | j � � | j | j
� � | j | j � � d S)Nr c 3 s> � j � Dd H� j � Dd Hr: | j d � � j � n d S)Nr" T)r# rT r$ r )r% )r_ r r r& @ s
z*ConditionTests.test_notify_all.<locals>.c1c 3 s> � j � Dd H� j � Dd Hr: | j d � � j � n d S)Nr' T)r# rT r$ r )r% )r_ r r r( H s
z*ConditionTests.test_notify_all.<locals>.c2r" r' )r r^ r r r+ r r, r- r r# �
notify_allr r r. r% )r r% r& r( r/ r0 r )r_ r �test_notify_all; s"
z ConditionTests.test_notify_allc C s, t j d | j � } | j t | j � d S)Nr )r r^ r r5 rA r` )r r_ r r r �test_notify_unacquireda s z%ConditionTests.test_notify_unacquiredc C s, t j d | j � } | j t | j � d S)Nr )r r^ r r5 rA ri )r r_ r r r �test_notify_all_unacquirede s z)ConditionTests.test_notify_all_unacquiredc C s t j d | j � } | j d t | � k � | j t j t | � � � | j j | j � � | j d t | � k � | j j
t j � � | j d t | � k � | j t j t | � � � | j j
t j � � | j d t | � k � | j t j t | � � � d S)Nr Zunlockedr z waiters:1z waiters:2)
r r^ r r r r r r r# r7 r$ r r )r r_ r r r r i s zConditionTests.test_reprc sv t j d | j � � t j � f d d � � } | j j | � � � | j � j � � Wd QX| j � j � � d S)Nr c 3 s � Dd HS)Nr r )r_ r r � acquire_cond| s z9ConditionTests.test_context_manager.<locals>.acquire_cond)r r^ r r r r r r )r rm r )r_ r rD y s
z#ConditionTests.test_context_managerc C s� t j d | j � } y | � | j d � Wd QXWn; t k
rp } z | j t | � d � WYd d } ~ Xn X| j | j � � d S)Nr z-RuntimeError is not raised in with expressionz9"yield from" should be used as context manager expression) r r^ r rH rA r- rI r r )r r_ rJ r r r rK � s z,ConditionTests.test_context_manager_no_yieldc C sZ t j d | j � } t j | d | j �} | j | j | � | j | j | j � d S)Nr )r r
r r^ r �_lockr )r r r_ r r r �test_explicit_lock� s z!ConditionTests.test_explicit_lockc C s^ | j � } | j | j � t j d | j � } | j t � � t j | d | �Wd QXd S)Nr ) r Z
addCleanup�closer r
r r5 �
ValueErrorr^ )r r r r r r �test_ambiguous_loops� s
z#ConditionTests.test_ambiguous_loopsN)rL rM rN r
r r rV rX ra re rg rh rj rk rl r rD rK ro rr r r r r r] l s" B & 7 &
r] c @ s� e Z d Z d d � Z d d � Z d d � Z d d � Z d d
� Z d d � Z d
d � Z d d � Z
d d � Z d d � Z d d � Z
d d � Z d d � Z d d � Z d d � Z d S) �SemaphoreTestsc C s | j � | _ d S)N)r r )r r r r r
� s zSemaphoreTests.setUpc C s` t j � } t j d | � } | j | j | � t j d | j � } | j | j | j � d S)Nr )r r r � Semaphorer r r )r r �semr r r r � s
z SemaphoreTests.test_ctor_loopc C s6 t j | j � t j � } | j | j | j � d S)N)r r r rt r r )r ru r r r r � s z SemaphoreTests.test_ctor_noloopc C s/ t j d d | j �} | j | j � � d S)Nr r )r rt r r r )r ru r r r �test_initial_value_zero� s z&SemaphoreTests.test_initial_value_zeroc C sN t j d | j � } | j t | � j d � � | j t j t | � � � | j j | j � � | j t | � j d � � | j d t | � k � | j t j t | � � � | j
j t j
� � | j d t | � k � | j t j t | � � � | j
j t j
� � | j d t | � k � | j t j t | � � � d S)Nr z[unlocked,value:1]>z [locked]>�waitersz waiters:1z waiters:2)r rt r r r r r r r r# r7 r$ r r )r ru r r r r � s zSemaphoreTests.test_reprc s� t j d | j � � | j d � j � t j � f d d � � } | j j | � � } | j | � | j � j � � | j d � j � � j � | j
� j � � | j d � j � d S)Nr r" c 3 s � Dd HS)Nr r )ru r r r � s z3SemaphoreTests.test_semaphore.<locals>.acquire_lockr ) r rt r r- �_valuer r r r r r )r r r r )ru r �test_semaphore� s
z SemaphoreTests.test_semaphorec C s | j t t j d � d S)Nr" ���)r5 rq r rt )r r r r �test_semaphore_value� s z#SemaphoreTests.test_semaphore_valuec s� t j d d | j �� g } | j | j j � j � � � | j | j j � j � � � | j � j � � t j � f d d � � } t j � f d d � � } t j � f d d � � } t j � f d d
� � } t j | | � d | j �} t j | | � d | j �} t j | | � d | j �} t
j | j � | j d g | � | j � j � � | j d t
� j � � | j d
� j � t j | | � d | j �} � j � � j � | j d � j � t
j | j � | j d
� j � | j d t
| � � | j � j � � | j d t
� j � � | j d
� j � | j | j � � | j | j � � | | | g }
d d � |
D� } | j d t
| � � � j � | j j t j |
� � d S)Nr) r c 3 s � j � Dd H| j d � d S)Nr" T)r# r$ )r% )ru r r r& � s
z'SemaphoreTests.test_acquire.<locals>.c1c 3 s � j � Dd H| j d � d S)Nr' T)r# r$ )r% )ru r r r( � s
z'SemaphoreTests.test_acquire.<locals>.c2c 3 s � j � Dd H| j d � d S)Nr) T)r# r$ )r% )ru r r r* � s
z'SemaphoreTests.test_acquire.<locals>.c3c 3 s � j � Dd H| j d � d S)N� T)r# r$ )r% )ru r r �c4� s
z'SemaphoreTests.test_acquire.<locals>.c4r" r' r c S s. g | ]$ } | j � r | j � r | � q Sr )r. r% )�.0r[ r r r �
<listcomp> s z/SemaphoreTests.test_acquire.<locals>.<listcomp>)r rt r r r r# r r r r+ r r, r- r> r7 rx r r. r% Zgather) r r% r& r( r* r} r/ r0 r1 �t4Z
race_tasksZ
done_tasksr )ru r r2 � sD
zSemaphoreTests.test_acquirec C s� t j d | j � } | j j | j � � t j | j � d | j �} | j j | j � | j t j | j j | � | j
| j p� t d d � | j D� � � d S)Nr c s s | ] } | j � Vq d S)N)r. )r~ Zwaiterr r r � <genexpr>* s z5SemaphoreTests.test_acquire_cancel.<locals>.<genexpr>)
r rt r r r# r+ r3 r4 r5 r6 r r7 �all)r ru r# r r r r8 s z"SemaphoreTests.test_acquire_cancelc C s( t j d d d | j � } t j | j � d | j �} t j | j � d | j �} t j | j � d | j �} t j | j � d | j �} t j | j � | j � | j � | j � t j | j � t d d � | | g D� � } | j
| d � | j � | j � t j | j � d S)N�valuer r c s s | ] } | j � Vq d S)N)r. )r~ r[ r r r r� ; s zCSemaphoreTests.test_acquire_cancel_before_awoken.<locals>.<genexpr>r" ) r rt r r+ r# r r, r r4 �sumr- )r ru r/ r0 r1 r� Znum_doner r r �!test_acquire_cancel_before_awoken, s
z0SemaphoreTests.test_acquire_cancel_before_awokenc C s� t j d d d | j � } t j | j � d | j �} t j | j � d | j �} t j | j � | j � | j � t j | j � | j | j
� � d S)Nr� r r ) r rt r r+ r# r r, r r4 r r )r ru r/ r0 r r r �test_acquire_hangB s
z SemaphoreTests.test_acquire_hangc C s, t j d | j � } | j t | j � d S)Nr )r ZBoundedSemaphorer r5 rq r )r ru r r r rB P s z(SemaphoreTests.test_release_not_acquiredc C s_ t j d | j � } | j j | j � � | j | j � � | j � | j | j � � d S)Nr ) r rt r r r# r r r r )r ru r r r rC U s
z&SemaphoreTests.test_release_no_waitersc s� t j d d | j �� t j � f d d � � } | j j | � � �Z | j � j � � | j d � j � | j j | � � � | j � j � � Wd QXWd QX| j d � j � d S)Nr' r c 3 s � Dd HS)Nr r )ru r r r ` s z9SemaphoreTests.test_context_manager.<locals>.acquire_lockr" )
r rt r r r r r r- rx r )r r r )ru r rD ] s z#SemaphoreTests.test_context_managerc C s� t j d d | j �} y | � | j d � Wd QXWn; t k
rs } z | j t | � d � WYd d } ~ Xn X| j d | j � d S)Nr' r z-RuntimeError is not raised in with expressionz9"yield from" should be used as context manager expression)r rt r rH rA r- rI rx )r ru rJ r r r rK m s z,SemaphoreTests.test_context_manager_no_yieldN)rL rM rN r
r r rv r ry r{ r2 r8 r� r� rB rC rD rK r r r r rs � s A rs �__main__)�__doc__Zunittestr �rer r Z STR_RGX_REPR�compiler ZTestCaser rO r] rs rL �mainr r r r �<module> s ��� 7�