관리-도구
편집 파일: test_leaks.cpython-38.pyc
U �g9D � @ s� d Z ddlmZmZmZ ddlZddlZddlZddlZddl Z ddl Z ddl m Z ddl mZ ddl mZ ddl mZ e js�t�G d d � d e�ZG d d � d e �Zed kr�ed��� 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 �I/opt/hc_python/lib64/python3.8/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'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 � z)TestLeaks.test_arg_refs.<locals>.<lambda>�d )r � getrefcountr' �ranger* � assertEqual)r r+ Zrefcount_before�g�_r r r � test_arg_refs+ s � zTestLeaks.test_arg_refsc C sB i }t � dd� �}td�D ]}|jf |� q| �t�|�d� d S )Nc [ s t �� jjf | �S r r& )�kwargsr r r r, 9 r- z+TestLeaks.test_kwarg_refs.<locals>.<lambda>r. � )r' r0 r* r1 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 Z__recycle_threads? s zTestLeaks.__recycle_threadsc s� g � � fdd�}t d�D ]$}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; P s z,TestLeaks.test_threaded_leak.<locals>.workerr6 r< r> � r0 r? r@ rA rB r' r( �_TestLeaks__recycle_threads�gcZcollectZ assertIsNone�r r; r3 rC r2 r rG r �test_threaded_leakN s z TestLeaks.test_threaded_leakc s� g � � fdd�}t d�D ]$}tj|d�}|�� |�d� ~qt�� | �� t�� t� � t�� � D ]}| � |� � qjd S )Nc sN g � t �� _� fdd�} td�D ]}t � | ��� q"��t�t �� �� d S )Nc s � � t�� � d S r )rD r' r( r ��llr r � additionale s zDTestLeaks.test_threaded_adv_leak.<locals>.worker.<locals>.additionalr6 )r' r( rN r0 r* rD rE rF )rO r3 rG rM r r; b s z0TestLeaks.test_threaded_adv_leak.<locals>.workerr6 r< r> rH rK r rG 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_cleanupZassertGreaterEqual�enable_optional_cleanupr1 Z assertGreaterZCLOCKS_PER_SEC)r �usedZused2r r r �assertClocksUsedw s zTestLeaks.assertClocksUsedTFc sb t �g �st�t�� t�� | jtjdd�}t� � �t� � � g ��fdd��g �� �����fdd�}tj |d�}|� � �� d� t�� | jt d d�}t��d 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' rQ 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' rD rE rF r) r* r �waitr( )Zglet)�background_glet_killed�background_glet_runningrX �background_greenlets� bg_main_wrefs�manually_collect_backgroundr r �background_thread� s z4TestLeaks._check_issue251.<locals>.background_threadr< r> Tr r )rJ Z is_tracked�AssertionErrorr r r' r( Z count_objectsr? �Eventr@ rA rY �list�len� assertFalseZdeadr rB �wait_for_pending_cleanups�assertLessEqualrQ rR r1 r rU ) r r^ rW Zgreenlets_beforer_ rC Z lists_beforeZ lists_afterZgreenlets_afterr )rZ r[ rX r\ r] rW r^ r �_check_issue251� sB � zTestLeaks._check_issue251c C s | � � d S r �rg 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' rQ rS rg 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^ rh 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 )NTFrk )Zexpect_greenlet_leakr' rQ rS rg 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^ rW rh r r r r �9test_issue251_issue252_explicit_reference_not_collectable& s �zCTestLeaks.test_issue251_issue252_explicit_reference_not_collectabler. c C s@ t jd dkst�t jd d� dk r.| �d� tr<| �d� d S )Nr � r6 )ro � zOnly observed on 3.11z!Slow and not worth repeating here)r � version_infor` ZskipTestr 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 ]4}t|| �� �}|� | �� }||krH|dkrH q~qH| �||� 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 )r0 r' r* )r3 ��ITERrs r r �run_itF s z?TestLeaks.test_untracked_memory_doesnt_increase.<locals>.run_itro r )rr r0 �get_process_uss�UNTRACK_ATTEMPTS�maxrf )r rw r3 � uss_before�count� uss_afterr ru r �%test_untracked_memory_doesnt_increase; s z/TestLeaks.test_untracked_memory_doesnt_increasec sD | � � dg� � fdd��d���fdd��| �G ���fdd�d�}d }}t| j�D ]�}d� d< |� }tj|d �}|�� |�d � | �|�� � |d kr�|j }t ||j �}�r�| � |j d � | � � d �|j � ~~�s�| � � d d� �r�| �� | �� }||krX|d krX �q qX| �� | �� }| �||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 rs d s z3TestLeaks._check_untracked_memory_thread.<locals>.frt c s2 g } t � �D ] }t���}| �|� |�� q | S r )r0 r' rD r* )�gletsr3 r2 ru r r rw 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|�� q4� rTd| _�� � | _d S )Nzsuspended activer ) rx r{ r0 rv r� �tupleZassertIn�strr} )r r3 r2 ��deallocate_in_threadrw �testr r �__call__ s zETestLeaks._check_untracked_memory_thread.<locals>.ThreadFunc.__call__N)r r r r{ r} r� rv r� r r� r r � ThreadFunc{ s r� r< � r r zafter attempts %d)rr r0 ry r? r@ rA rB rd �is_aliver{ rz r1 r� rv re rx rf )r r� r� r{ r} r| Z thread_funcrC r )r rv r� rs rw r� r � _check_untracked_memory_thread] 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 r4 r7 � staticmethodrI rL rP rU rg ri rj r rl rm rn ry rr r r~ r� r� r� r r r r r" ) s6 � ~ Q r" �__main__Zunittest)�__doc__� __future__r r r r rJ r9 rE r? r' � r Z leakcheckr r r ZGREENLET_USE_GCr` �objectr r" r � __import__�mainr r r r �<module> s&