U
|
¸ý°dE ã @ sô d Z ddlmZmZmZ ddlZddlZddlZddlZddl Z ddl
|
Z
|
ddl Z ddl m Z ddlmZ ddlmZ ddlmZ dd lmZ zdd
|
lmZ W n ek
|
r° Y nX e js¼tG d d d eZG d d de Zedkrðed ¡ dS )z)
|
Testing scenarios that may have leaked.
|
é )Úprint_functionÚabsolute_importÚdivisionNé )ÚTestCase)Úfails_leakcheck)Úignores_leakcheck)ÚRUNNING_ON_GITHUB_ACTIONS)ÚRUNNING_ON_MANYLINUX)Úinternc @ s6 e Zd Ze Zdd Zdd Zdd Zedd Z d S )
|
ÚHasFinalizerTracksInstancesc C s t || _| j t| ¡ d S ©N)r ÚmsgÚEXTANT_INSTANCESÚaddÚid)Úselfr © r úPd:\z\workplace\vscode\pyvenv\venv\Lib\site-packages\greenlet/tests/test_leaks.pyÚ__init__! s
|
z$HasFinalizerTracksInstances.__init__c C s | j t| ¡ d S r )r Úremover ©r r r r Ú__del__$ s z#HasFinalizerTracksInstances.__del__c C s dt | | jf S )Nz(<HasFinalizerTracksInstances at 0x%x %r>)r r r r r r Ú__repr__& s ÿz$HasFinalizerTracksInstances.__repr__c C s | j ¡ d S r )r Úclear)Úclsr r r Úreset* s z!HasFinalizerTracksInstances.resetN)
|
Ú__name__Ú
|
__module__Ú __qualname__Úsetr r r r Ú classmethodr r r r r r s r c @ s¸ e Zd Zdd Zdd Zedd Zdd Zd d
|
Zd d Z d'ddZ
|
dd Z dd Z e dd Ze dd Ze dd ZdZd d Zed d Zd(d d!Zed"d# Zed$d% Zd&S ))Ú TestLeaksc C sJ d}t |¡}t dd ¡}tdD ]}|j| q$| t |¡|¡ d S )N)ÚaÚbÚcc W s t ¡ jj| S r ©ÚgreenletÚ
|
getcurrentÚparentÚswitch)Úargsr r r Ú<lambda>6 ó z)TestLeaks.test_arg_refs.<locals>.<lambda>éd )ÚsysÚ getrefcountr' Úranger* Ú assertEqual)r r+ Zrefcount_beforeÚgÚ_r r r Ú test_arg_refs1 s
|
ÿ zTestLeaks.test_arg_refsc C sB i }t dd ¡}tdD ]}|jf | q| t |¡d¡ d S )Nc [ s t ¡ jjf | S r r& )Úkwargsr r r r, ? r- z+TestLeaks.test_kwarg_refs.<locals>.<lambda>r. é )r' r1 r* r2 r/ r0 )r r6 r3 r4 r r r Útest_kwarg_refs; s ÿ zTestLeaks.test_kwarg_refsc C s4 dd } t j| d}| ¡ t d¡ | d¡ d S )Nc S s t d¡ d S )Nçü©ñÒMbP?)ÚtimeÚsleepr r r r ÚworkerM s z+TestLeaks.__recycle_threads.<locals>.worker©Útargetr9 é
|
)Ú threadingÚThreadÚstartr: r; Újoin)r< Útr r r Z__recycle_threadsE s
|
|
zTestLeaks.__recycle_threadsc s g fdd}t dD ]$}tj|d}| ¡ | d¡ ~qt ¡ | ¡ t ¡ t ¡ t ¡ D ]}|
|
| ¡ qjd S )Nc s t t ¡ ¡¡ d S r )ÚappendÚweakrefÚrefr' r( r ©Zggr r r< V s z,TestLeaks.test_threaded_leak.<locals>.workerr7 r= r? © r1 r@ rA rB rC r' r( Ú_TestLeaks__recycle_threadsÚgcZcollectZ assertIsNone©r r< r4 rD r3 r rH r Útest_threaded_leakT s
|
z TestLeaks.test_threaded_leakc s g fdd}t dD ]$}tj|d}| ¡ | d¡ ~qt ¡ | ¡ t ¡ t ¡ t ¡ D ]}|
|
| ¡ qjd S )Nc sN g t ¡ _ fdd} tdD ]}t | ¡ ¡ q" t t ¡ ¡¡ d S )Nc s t ¡ ¡ d S r )rE r' r( r ©Úllr r Ú
|
additionalk s zDTestLeaks.test_threaded_adv_leak.<locals>.worker.<locals>.additionalr7 )r' r( rO r1 r* rE rF rG )rP r4 rH rN r r< h s
|
z0TestLeaks.test_threaded_adv_leak.<locals>.workerr7 r= r? rI rL r rH r Útest_threaded_adv_leakf s
|
z TestLeaks.test_threaded_adv_leakc C sL t j ¡ }| |d¡ t j d¡ t j ¡ }| ||¡ | t jjd¡ d S )Nr Tr )r' Ú _greenletÚ&get_clocks_used_doing_optional_cleanupZassertGreaterEqualÚenable_optional_cleanupr2 Z assertGreaterZCLOCKS_PER_SEC)r ÚusedZused2r r r ÚassertClocksUsed} s
|
|
zTestLeaks.assertClocksUsedTFc sb t g ¡stt ¡ t ¡ | jtjdd}t ¡ t ¡ g fddg fdd}tj
|
|d}| ¡ d¡ t ¡ | jt d d}td
|
ks¬t| d j¡ d d
= ¡ | d¡ ~| ¡ | jt d d}| jtjdd}| ||¡ sFtj ¡ d k rF| ||¡ rF| tjt ¡ n tj ¡ d k r^| ¡ d S ) NF)Z
|
exact_kindc sL t d} tj dt d¡ r6t ¡ jj}|| g nt ¡ j | g¡ d S )NzDELETING STACK OBJECTZtest_leaks_keyzDELETING THREAD STATE)r r' rR Zset_thread_localr( r) r* )ZjdÚs)Ú explicit_reference_to_switchr r Úbackground_greenlet¥ s þ z6TestLeaks._check_issue251.<locals>.background_greenletc sR t ¡} t | j¡¡ | ¡ | ¡ ~ ¡ d¡ rNt ¡ d S )Nr? ) r' rE rF rG r) r* r Úwaitr( )Zglet)Úbackground_glet_killedÚbackground_glet_runningrY Úbackground_greenletsÚ bg_main_wrefsÚmanually_collect_backgroundr r Úbackground_threadµ s
|
|
|
z4TestLeaks._check_issue251.<locals>.background_threadr= r? Tr r )rK Z
|
is_trackedÚAssertionErrorr r r' r( Z count_objectsr@ ÚEventrA rB rZ ÚlistÚlenÚ assertFalseZdeadr rC Úwait_for_pending_cleanupsÚassertLessEqualrR rS r2 r rV ) r r_ rX Zgreenlets_beforer` rD Z lists_beforeZ lists_afterZgreenlets_afterr )r[ r\ rY r] r^ rX r_ r Ú_check_issue251 s>
|
|
|
zTestLeaks._check_issue251c C s | ¡ d S r ©rh r r r r Ú-test_issue251_killing_cross_thread_leaks_list s z7TestLeaks.test_issue251_killing_cross_thread_leaks_listc C s, t j d¡ z | ¡ W 5 t j d¡ X d S )NFT)r' rR rT rh r r r r Ú#test_issue251_with_cleanup_disabled s z-TestLeaks.test_issue251_with_cleanup_disabledc C s | j dd d S )NF©r_ ri r r r r Ú4test_issue251_issue252_need_to_collect_in_background s z>TestLeaks.test_issue251_issue252_need_to_collect_in_backgroundc C s6 d| _ tj d¡ z| jdd W 5 tj d¡ X d S )NTFrl )Zexpect_greenlet_leakr' rR rT rh r r r r ÚEtest_issue251_issue252_need_to_collect_in_background_cleanup_disabled" s
|
zOTestLeaks.test_issue251_issue252_need_to_collect_in_background_cleanup_disabledc C s | j ddd d S )NFT)r_ rX ri r r r r Ú9test_issue251_issue252_explicit_reference_not_collectable+ s þzCTestLeaks.test_issue251_issue252_explicit_reference_not_collectabler. c C sX t jd dkr*t jd d
dk r*| d¡ t jd dkrFtrF| d¡ trT| d¡ d S )Nr é r7 )rp é zOnly observed on 3.11z!Hard to get a stable pattern herez!Slow and not worth repeating here)r/ Ú version_infoZskipTestr r
|
r r r r Ú_only_test_some_versions3 s
|
|
z"TestLeaks._only_test_some_versionsc s | ¡ dd d fdd}tdD ]
|
}| q*t ¡ ¡ j}t| jD ]@}t|t ¡ ¡ j}| t ¡ ¡ j}||krN|dkrN qqN| ||¡ d S )Nc S s dS )Nr r r r r r ÚfG s z:TestLeaks.test_untracked_memory_doesnt_increase.<locals>.fé' c s t D ]} t ¡ ¡ qd S r )r1 r' r* )r4 ©ÚITERrt r r Úrun_itK s z?TestLeaks.test_untracked_memory_doesnt_increase.<locals>.run_itrp r ) rs r1 ÚpsutilÚProcessÚmemory_full_infoÚussÚUNTRACK_ATTEMPTSÚmaxrg )r rx r4 Ú
|
uss_beforeÚcountÚ uss_afterr rv r Ú%test_untracked_memory_doesnt_increase@ s z/TestLeaks.test_untracked_memory_doesnt_increasec sP | ¡ dg fdddfdd| G fddd}d }}t| jD ]È}d d< | }tj|d }| ¡ | d
|
¡ | | ¡ ¡ |d kr¤|j }t
|
||j }rØ| |j d ¡ | d |j ¡ ~~sð| d d¡ rü| ¡ t ¡ ¡ j}||krX|d krX q"qX| ¡ t ¡ ¡ j}| ||d |f ¡ d S )Nr c s@ zt ¡ j ¡ W n( t jk
|
r: d d7 < Y nX dS )Nr r )r' r( r) r* Z GreenletExitr )Ú
|
EXIT_COUNTr r rt i s z3TestLeaks._check_untracked_memory_thread.<locals>.fru c s2 g } t D ] }t ¡}| |¡ | ¡ q | S r )r1 r' rE r* )Úgletsr4 r3 rv r r rx r s
|
|
|
z8TestLeaks._check_untracked_memory_thread.<locals>.run_itc s, e Zd Zd ZZdZdZ fddZdS )z<TestLeaks._check_untracked_memory_thread.<locals>.ThreadFuncr r r7 c sn t ¡ ¡ j| _t| jD ]}| jt 7 _q| jD ]} dt
|
|¡ q: rZd| _t ¡ ¡ j| _ d S )Nzsuspended activer ) ry rz r{ r| r r1 rw r ÚtupleZassertInÚstrr )r r4 r3 ©Údeallocate_in_threadrx Útestr r Ú__call__ s
|
zETestLeaks._check_untracked_memory_thread.<locals>.ThreadFunc.__call__N)r r r r r r rw r r r r r Ú
|
ThreadFunc s r r= é r r zafter attempts %d)rs r1 r} r@ rA rB rC re Úis_aliver r~ r2 r rw rf ry rz r{ r| rg )r r r r r r Z thread_funcrD r )r rw r rt rx r r Ú _check_untracked_memory_threadb sB
|
z(TestLeaks._check_untracked_memory_threadc C s | j dd d S )NT©r ©r r r r r ÚItest_untracked_memory_doesnt_increase_unfinished_thread_dealloc_in_thread³ s zSTestLeaks.test_untracked_memory_doesnt_increase_unfinished_thread_dealloc_in_threadc C s | j dd d S )NFr r r r r r ÚGtest_untracked_memory_doesnt_increase_unfinished_thread_dealloc_in_main¹ s zQTestLeaks.test_untracked_memory_doesnt_increase_unfinished_thread_dealloc_in_mainN)TF)T)r r r r5 r8 Ú staticmethodrJ rM rQ rV rh rj rk r rm rn ro r} rs r r r r r r r r r r" / s6
|
|
|
|
þ
|
}
|
|
|
|
|
Q
|
r" Ú__main__Zunittest) Ú__doc__Ú
|
__future__r r r r/ rK r: rF r@ ry r' Ú r Z leakcheckr r r r
|
r Ú ImportErrorZGREENLET_USE_GCra Úobjectr r" r Ú
|
__import__Úmainr r r r Ú<module> s2
|
|
