File: //usr/local/lib/python3.10/dist-packages/greenlet/tests/__pycache__/leakcheck.cpython-310.pyc
o
���g�. � @ s d dl mZ d dlZd dlZd dlZd dlmZ d dlZd dlZej �
d�Z ej �
d�p-e Z ej �
d�Z
e p7e
Zej �
d�ZepEej �
d�Zej �
d �Zej �
d
�Zd d � Zd
d� ZG dd� de�Zeed�rvG dd� de�Ze� e_eZndd� ZG dd� de�ZG dd� de�Z dd� Z dS ) � )�print_functionN)�wraps�GITHUB_ACTIONS�TRAVIS�APPVEYOR�GREENLET_MANYLINUX�GREENLET_SKIP_LEAKCHECKS� GREENLET_SKIP_FAILING_LEAKCHECKS� GREENLET_ONLY_FAILING_LEAKCHECKSc C s
d| _ | S )a�
Ignore the given object during leakchecks.
Can be applied to a method, in which case the method will run, but
will not be subject to leak checks.
If applied to a class, the entire class will be skipped during leakchecks. This
is intended to be used for classes that are very slow and cause problems such as
test timeouts; typically it will be used for classes that are subclasses of a base
class and specify variants of behaviour (such as pool sizes).
T)�ignore_leakcheck��func� r �C/usr/local/lib/python3.10/dist-packages/greenlet/tests/leakcheck.py�ignores_leakcheckL s r c C s d| _ tr t�d�| �} | S )z2
Mark that the function is known to leak.
TzSkipping known failures)�fails_leakcheck�SKIP_FAILING_LEAKCHECKS�unittest�skipr r r r r [ s r c @ s e Zd ZdS )�LeakCheckErrorN)�__name__�
__module__� __qualname__r r r r r d s r �
getobjectsc @ � e Zd Zdd� Zdd� ZdS )�_MockGCc C s
t �d�S )Nr )�sysr ��selfr r r � get_objectsl �
z_MockGC.get_objectsc C s
t t|�S �N)�getattr�gc)r �namer r r � __getattr__n r z_MockGC.__getattr__N)r r r r r% r r r r r k � r c C s | S )a
Decorator for a function that is known to fail when running
strict (``sys.getobjects()``) leakchecks.
This type of leakcheck finds all objects, even those, such as
strings, which are not tracked by the garbage collector.
r r r r r �fails_strict_leakchecks s r' c @ r )�!ignores_types_in_strict_leakcheckc C s
|| _ d S r! )�types)r r) r r r �__init__~ r z*ignores_types_in_strict_leakcheck.__init__c C s | j |_|S r! )r) �leakcheck_ignore_types)r r
r r r �__call__� s z*ignores_types_in_strict_leakcheck.__call__N)r r r r* r, r r r r r( } r& r( c @ sP e Zd ZdZdd� Zdd� Zdd� Zdd � Zd
d � Zd d
� Z dd� Z
dd� Z dS )�_RefCountCheckerr c C s( || _ || _g | _i | _d| _d| _d S )Nr F)�testcase�function�deltas�
peak_stats�
ignored_types� needs_setUp)r r. r/ r r r r* � s
z_RefCountChecker.__init__c C s� || u rdS t |�}|t | j�kr3z| j�|�}W n ty* | j�|�dk}Y nw |tur3|r3dS | j�� D ] }||u rA dS q8|| jv sL|| j v rNdS dS )NFr T)
�type�_include_object_p�__eq__�AttributeError�__cmp__�NotImplemented�__dict__�valuesr2 �
IGNORED_TYPES)r �obj�kind�exact_method_equals�xr r r r5 � s$ � �z"_RefCountChecker._include_object_pc C s t jd | j| jd�S )N)�limitr1 �filter)�objgraph�growthr1 r5 r r r r �_growth� s
�z_RefCountChecker._growthc C sR |sdS g }t dd� |D ��}|D ]\}}}|�d||||f � qd�|�}|S )Nz <Unable to calculate growth>c s s � | ]
\}}}t |�V qd S r! )�len)�.0r$ �_r r r � <genexpr>� s � z0_RefCountChecker._report_diff.<locals>.<genexpr>z %-*s%9d %+9d�
)�max�append�join)r rD �lines�widthr$ �count�delta�diffr r r � _report_diff� s
z _RefCountChecker._report_diffc C s� t �� }t �� | jr| j�� d| j_z)| j| jg|�R i |�� W | j�� | j� � d| j_d| _|r<t �
� d S d S | j�� | j� � d| j_d| _|rVt �
� w w )NFT) r# � isenabled�disabler3 r. �setUp� skipTearDownr/ �tearDown�
doCleanups�enable)r �args�kwargs�
gc_enabledr r r � _run_test� s*
�
�
�z_RefCountChecker._run_testc C s8 dt jv r t jd �� dt jv rt jd �� | �� S )N�urlparsez urllib.parse)r �modules� clear_cacherE r r r r �
_growth_after� s
z _RefCountChecker._growth_afterc C sD | j }|sdS tjrtdtjf ��|dd � ddgkr$t|�dv r$dS |dd � g d�kr0dS t|�d krBt|d
d � �dkrBdS t|�d krk|d dkrk|d |d krk|d |d krk| �|�}td
||f ��t|dd � �dks�t|d
d � �dks�|d
d � �d�dkr�d}nd}t|�|kr�td
|| �|�f ��dS )NTz"Generated uncollectable garbage %r���r )� � F���)r r r � ���re ���zrefcount increased by %r
%srd � � )r0 r# �garbager rF �sumrS rP )r rD r0 rR rA r r r �
_check_deltas� s0 8
> ��z _RefCountChecker._check_deltasc C s� t d�D ]}t�� qt| jdd�}|rd| j_t| jdd�| _| �� }z#| � |�rF| �
||� | � � }| j �
tdd� |D ��� | � |�s*W n
tyT |sQ� Y d S w |r_td | jf ��d S )
Nre r FTr+ r c s s � | ]}|d V qdS )rd Nr )rG �statr r r rI + s � z,_RefCountChecker.__call__.<locals>.<genexpr>z#Expected %s to leak but it did not.)�ranger# �collectr" r/ r. �expect_greenlet_leakr2 rE rn r^ rb r0 rL rm r )r r[ r\ rH �expect_failurerD r r r r, s,
�� ���z_RefCountChecker.__call__N) r r r r< r* r5 rE rS r^ rb rn r, r r r r r- � s %
1r- c s, t � dd�str
� S t� �� fdd��}|S )Nr Fc s@ t | dd�r t�d��trt � dd�st�d��t| � �||�S )Nr Fz$This class ignored during leakchecksr z(Only running tests that fail leakchecks.)r" r �SkipTest�ONLY_FAILING_LEAKCHECKSr- )r r[ r\ ��methodr r �wrapper7 s
z wrap_refcount.<locals>.wrapper)r" �SKIP_LEAKCHECKSr )rw rx r rv r �
wrap_refcount3 s
rz ) �
__future__r �osr r# � functoolsr r rC �environ�get�RUNNING_ON_GITHUB_ACTIONS�RUNNING_ON_TRAVIS�RUNNING_ON_APPVEYOR�
RUNNING_ON_CI�RUNNING_ON_MANYLINUXry r ru r r �AssertionErrorr �hasattr�objectr r' r( r- rz r r r r �<module> s6 !
0