U
|
Z ±d ã @ s¨ d dl Z d dlZd dlmZ d dlZd dlZd dlmZ ddlm Z m
|
Z
|
m Z ddl m Z ddl mZmZ d d
|
Zd d Ze d d Ze dd Zedd Zdd ZdS )é N)ÚQueue)Úcontextmanageré )ÚslowÚgc_collect_harderÚ disable_threading_excepthooké )Ú _thread_cache)Ústart_thread_soonÚ ThreadCachec sR t dd } fdd}t| | ¡ }tjtdd | ¡ W 5 Q R X d S )Nc S s t dd S )NÚhi)Ú RuntimeError© r r úYd:\z\workplace\vscode\pyvenv\venv\Lib\site-packages\trio/_core/tests/test_thread_cache.pyÚfn s z$test_thread_cache_basics.<locals>.fnc s | ¡ d S ©N©Úput©Úoutcome©Úqr r Údeliver s z)test_thread_cache_basics.<locals>.deliverr )Úmatch)r r
|
ÚgetÚpytestZraisesr Úunwrap)r r r r r r Útest_thread_cache_basics s
|
r c sd dgG fddd} t fdd}t| | ¡ }| ¡ dksNtt d s`td S )NFc s e Zd Zdd Z fddZdS )z'test_thread_cache_deref.<locals>.del_mec S s dS )Né* r ©Úselfr r r Ú__call__! s z0test_thread_cache_deref.<locals>.del_me.__call__c s d d< d S )NTr r r ©Úresr r Ú__del__$ s z/test_thread_cache_deref.<locals>.del_me.__del__N)Ú__name__Ú
|
__module__Ú __qualname__r! r$ r r" r r Údel_me s r( c s | ¡ d S r r r r r r r ) s z(test_thread_cache_deref.<locals>.deliverr r )r r
|
r r ÚAssertionErrorr )r( r r r )r r# r Útest_thread_cache_deref s r* c s t d} t| D ]}tdd fdd qt| D ]} ¡ ¡ q6t t ¡ fdd tdd fdd ¡ t d kst
|
d S )
|
Né c S s
|
t d¡S ©Nr )ÚtimeÚsleepr r r r Ú<lambda>A ó zNtest_spawning_new_thread_from_deliver_reuses_starting_thread.<locals>.<lambda>c s
|
| ¡S r r )Úresultr r r r/ A r0 c sD t t ¡ ¡ dkr( ¡ ntdd fdd d S )Nr c S s d S r r r r r r r/ N r0 z_test_spawning_new_thread_from_deliver_reuses_starting_thread.<locals>.deliver.<locals>.<lambda>c s d | S r, r ©Ú_)r Únr r r/ N r0 )ÚprintÚaddÚ threadingÚcurrent_threadÚsetr
|
)r4 r3 )r ÚdoneÚ seen_threads)r4 r r H s
|
|
zMtest_spawning_new_thread_from_deliver_reuses_starting_thread.<locals>.deliverc S s d S r r r r r r r/ P r0 c s
|
d| S )Nr+ r r2 )r r r r/ P r0 r ) r Úranger
|
r r r9 r7 ÚEventÚwaitÚlenr) )ZCOUNTr3 r )r r: r r; r Ú<test_spawning_new_thread_from_deliver_reuses_starting_thread4 s
|
r@ c sL | tdd¡ t tdd fdd ¡ }t d¡ | ¡ rHtd S )NZ IDLE_TIMEOUTg-C ëâ6?c S s d S r r r r r r r/ _ r0 z(test_idle_threads_exit.<locals>.<lambda>c s t ¡ ¡S r )r r7 r8 r2 r r r r/ _ r0 r ) Úsetattrr r r
|
r r- r. Úis_aliver) )Ú monkeypatchZ seen_threadr r r Útest_idle_threads_exitW s
|
rD c c s< t t ¡ } z
|
d V W 5 t ¡ D ]}|| kr | ¡ q X d S r )Ú frozensetr7 Ú enumerateÚjoin)ZbeforeÚthreadr r r Ú_join_started_threadsg s
|
rI c
|
s G dd d}| td|¡ t Z t H t }t ¡ | dd fdd¡ ¡ | dd dd ¡ W 5 Q R X W 5 Q R X d S ) Nc @ s& e Zd Zdd ZdddZdd ZdS ) zAtest_race_between_idle_exit_and_job_assignment.<locals>.JankyLockc S s t ¡ | _d| _d S )Né )r7 ÚLockÚ_lockÚ_counterr r r r Ú__init__
s
|
zJtest_race_between_idle_exit_and_job_assignment.<locals>.JankyLock.__init__Nc S sD | j ¡ |d krdS | jdkr<| jd8 _| j ¡ dS dS d S )NTr r F)rL ÚacquirerM Úrelease)r Útimeoutr r r rO s
|
|
|
zItest_race_between_idle_exit_and_job_assignment.<locals>.JankyLock.acquirec S s | j ¡ d S r )rL rP r r r r rP s zItest_race_between_idle_exit_and_job_assignment.<locals>.JankyLock.release)N)r% r& r' rN rO rP r r r r Ú JankyLock s
|
rR rK c S s d S r r r r r r r/ r0 z@test_race_between_idle_exit_and_job_assignment.<locals>.<lambda>c s ¡ S r )r9 r2 ©r: r r r/ r0 c S s d S r r r r r r r/ ¤ r0 c S s t ¡ S r )ÚsysÚexitr2 r r r r/ ¤ r0 ) rA r r rI r r7 r= r
|
r> )rC rR Ztcr rS r Ú.test_race_between_idle_exit_and_job_assignmentr s rV )r r7 Úqueuer r- rT Ú
|
contextlibr Ztutilr r r Ú r r
|
r r r* r@ rD rI rV r r r r Ú<module> s"
|
"
|
|
