File: //usr/local/lib/python3.10/dist-packages/greenlet/tests/__pycache__/test_leaks.cpython-310.pyc
o
���g9D � @ s� d Z ddlmZmZmZ ddlZddlZddlZddlZddl Z ddl
Z
ddl m Z ddl
mZ ddl
mZ ddl
mZ e
jsAJ �G d d
� d
e�ZG d d � d e �Zed
kr]ed��� dS dS )z)
Testing scenarios that may have leaked.
� )�print_function�absolute_import�divisionN� )�TestCase)�fails_leakcheck)�ignores_leakcheck)�RUNNING_ON_MANYLINUXc @ s6 e Zd Ze� Zdd� Zdd� Zdd� Zedd� �Z d S )
�HasFinalizerTracksInstancesc C s t �|�| _| j�t| �� d S �N)�sys�intern�msg�EXTANT_INSTANCES�add�id)�selfr � r �D/usr/local/lib/python3.10/dist-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'dd�Z
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� �}td�D ]}|j|� q| �t �|�|� d S )N)�a�b�cc W s t �� jj| � S r ��greenlet�
getcurrent�parent�switch)�argsr r r �<lambda>0 s z)TestLeaks.test_arg_refs.<locals>.<lambda>�d )r � getrefcountr' �ranger* � assertEqual)r r+ �refcount_before�g�_r r r �
test_arg_refs+ s
� zTestLeaks.test_arg_refsc C sF i }t � dd� �}td�D ]
}|jdi |�� q
| �t�|�d� d S )Nc [ s t �� jjdi | ��S )Nr r& )�kwargsr r r r, 9 s z+TestLeaks.test_kwarg_refs.<locals>.<lambda>r- � r )r' r/ r* r0 r r. )r r5 r2 r3 r r r �test_kwarg_refs5 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 �workerG s z+TestLeaks.__recycle_threads.<locals>.worker��targetr8 �
)� threading�Thread�startr9 r: �join)r; �tr r r �__recycle_threads? s
zTestLeaks.__recycle_threadsc � g � � fdd�}t d�D ]}tj|d�}|�� |�d� ~q t�� | �� t�� t� � t�� � D ]}| �
|� � q5d S )Nc s � � t�t�� �� d S r )�append�weakref�refr' r( r ��ggr r r; P s z,TestLeaks.test_threaded_leak.<locals>.workerr6 r< r> � r/ r? r@ rA rB r' r( �_TestLeaks__recycle_threads�gc�collect� assertIsNone�r r; r3 rC r2 r rI r �test_threaded_leakN s
�z TestLeaks.test_threaded_leakc rE )Nc sN g � t �� _� fdd�} td�D ] }t � | ��� q��t�t �� �� d S )Nc s � � t�� � d S r )rF r' r( r ��llr r �
additionale s zDTestLeaks.test_threaded_adv_leak.<locals>.worker.<locals>.additionalr6 )r' r( rS r/ r* rF rG rH )rT r3 rI rR r r; b s
z0TestLeaks.test_threaded_adv_leak.<locals>.workerr6 r< r> rK rP r rI r �test_threaded_adv_leak` 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_cleanup�assertGreaterEqual�enable_optional_cleanupr0 �
assertGreater�CLOCKS_PER_SEC)r �used�used2r r r �assertClocksUsedw s
zTestLeaks.assertClocksUsedTFc s` t �g �sJ �t�� t�� | jtjdd�}t�� �t�� � g ��fdd��g �� �����fdd�}tj |d�}|�
� �� d� t�� | jt d d�}t
��d
ksVJ �| ��d j� �d d �= � �� |�d� ~| �� | jt d d�}| jtjdd�}| �||� �s�tj�� d ur�| �||� �r�| �tjt� � n tj�� d ur�| �� d S d S ) NF)�
exact_kindc sN t d�} tj�dt d�� � r t�� jj}|| g� d S t�� j�| g� d S )NzDELETING STACK OBJECT�test_leaks_keyzDELETING THREAD STATE)r
r' rV �set_thread_localr( r) r* )�jd�s)� explicit_reference_to_switchr r �background_greenlet� s � z6TestLeaks._check_issue251.<locals>.background_greenletc sV t � ��} ��t�| j�� ��| � | �� ~ ��� � �d� �r)t �� d S d S )Nr> ) r' rF rG rH r) r* r �waitr( )�glet)�background_glet_killed�background_glet_runningre �background_greenlets�
bg_main_wrefs�manually_collect_backgroundr r �background_thread� s
�z4TestLeaks._check_issue251.<locals>.background_threadr< r> Tr r )rM �
is_trackedr
r r' r( �
count_objectsr? �Eventr@ rA rf �list�len� assertFalse�deadr rB �wait_for_pending_cleanups�assertLessEqualrV rW r0 r r^ ) r rl rd �greenlets_beforerm rC � lists_before� lists_after�greenlets_afterr )rh ri re rj rk rd rl r �_check_issue251� sF
�
�zTestLeaks._check_issue251c C s | � � d S r �r{ r r r r �-test_issue251_killing_cross_thread_leaks_list� s z7TestLeaks.test_issue251_killing_cross_thread_leaks_listc C s6 t j�d� z
| �� W t j�d� d S t j�d� w )NFT)r' rV rY r{ r r r r �#test_issue251_with_cleanup_disabled s
z-TestLeaks.test_issue251_with_cleanup_disabledc C � | j dd� d S )NF�rl r| r r r r �4test_issue251_issue252_need_to_collect_in_background s z>TestLeaks.test_issue251_issue252_need_to_collect_in_backgroundc C s@ d| _ tj�d� z| jdd� W tj�d� d S tj�d� w )NTFr� )�expect_greenlet_leakr' rV rY r{ 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)rl rd r| r r r r �9test_issue251_issue252_explicit_reference_not_collectable& s
�zCTestLeaks.test_issue251_issue252_explicit_reference_not_collectabler- c C sD t jd dks J �t jd d� dk r| �d� tr | �d� d S d S )Nr � r6 )r� � zOnly observed on 3.11z!Slow and not worth repeating here)r � version_info�skipTestr r r r r �_only_test_some_versions. s
�z"TestLeaks._only_test_some_versionsc s� | � � dd� �d� � �fdd�}td�D ]}|� q| �� }t| j�D ]}t|| �� �}|� | �� }||kr>|dkr> nq$| �||� d S )Nc S s dS )Nr r r r r r �fB s z:TestLeaks.test_untracked_memory_doesnt_increase.<locals>.f�' c s t � �D ] } t����� qd S r )r/ r' r* )r3 ��ITERr� r r �run_itF s �z?TestLeaks.test_untracked_memory_doesnt_increase.<locals>.run_itr� r )r� r/ �get_process_uss�UNTRACK_ATTEMPTS�maxrv )r r� r3 �
uss_before�count� uss_afterr r� r �%test_untracked_memory_doesnt_increase; s �z/TestLeaks.test_untracked_memory_doesnt_increasec sB | � � dg� � fdd��d���fdd��| �G ���fdd�d�}d }}t| j�D ]`}d� d< |� }tj|d �}|�� |�d
� | �|�� � |d u rR|j }t
||j �}�rl| � |j d � | � � d �|j
� ~~�sx| � � d d� �r~| �� | �� }||kr�|d kr� nq,| �� | �� }| �||d
|f � d S )Nr c s8 z
t �� j�� W dS t jy � d d7 < � w )Nr r )r' r( r) r* � GreenletExitr )�
EXIT_COUNTr r r� d s ��z3TestLeaks._check_untracked_memory_thread.<locals>.fr� c s2 g } t � �D ]}t���}| �|� |�� q| S r )r/ r' rF r* )�gletsr3 r2 r� r r r� m s
z8TestLeaks._check_untracked_memory_thread.<locals>.run_itc s, e Zd Zd ZZdZdZ� ��fdd�ZdS )z<TestLeaks._check_untracked_memory_thread.<locals>.ThreadFuncr r r6 c sb �� � | _t| j�D ] }| jt�� �7 _q
| jD ]
}��dt|�� q� r*d| _�� � | _d S )Nzsuspended activer ) r� r� r/ r� r� �tuple�assertIn�strr� )r r3 r2 ��deallocate_in_threadr� �testr r �__call__ s
zETestLeaks._check_untracked_memory_thread.<locals>.ThreadFunc.__call__N)r r r r� r� r� r� r� r r� r r �
ThreadFunc{ s
r� r< � r r zafter attempts %d)r� r/ r� r? r@ rA rB rs �is_aliver� r� r0 r� r� ru r� rv )r r� r� r� r� r� � thread_funcrC r )r� r� r� r� r� r� r � _check_untracked_memory_thread] sD
�z(TestLeaks._check_untracked_memory_threadc C r )NT�r� �r� r r r r �Itest_untracked_memory_doesnt_increase_unfinished_thread_dealloc_in_thread� � zSTestLeaks.test_untracked_memory_doesnt_increase_unfinished_thread_dealloc_in_threadc C r )NFr� r� r r r r �Gtest_untracked_memory_doesnt_increase_unfinished_thread_dealloc_in_main� r� zQTestLeaks.test_untracked_memory_doesnt_increase_unfinished_thread_dealloc_in_mainN)TF)T)r r r r4 r7 � staticmethodrL rQ rU r^ r{ r} r~ r r� r� r� r� r� r r� r� r� r� r r r r r" ) s8
�~
Q
r" �__main__�unittest)�__doc__�
__future__r r r r rM r9 rG r? r' � r � leakcheckr r r �GREENLET_USE_GC�objectr
r" r �
__import__�mainr r r r �<module> s*
�