File: //usr/local/lib/python3.10/dist-packages/greenlet/tests/__pycache__/__init__.cpython-310.pyc
o
���g $ � @ s8 d Z ddlZddlZddlZddlmZ ddlmZ ddlmZ ddl
m Z ddl
m
Z
ddl Z ddl
m
Z dd l
mZ dd
lmZ dd lmZ d d
lmZ ejdd� dkZejdd� dkZej�d�Zej� d�Z ej� d�pse Z ej� d�Z e p}e Z ej� d�Z!G dd� de"�Z#G dd� de#dej$fi ��Z$dS )z
Tests for greenlet.
� N)�collect)� get_objects)� active_count)�sleep)�time)�greenlet)�
getcurrent)�get_pending_cleanup_count)�get_total_main_greenlets� )� leakcheck� )� � )r �
�win�GITHUB_ACTIONS�TRAVIS�APPVEYOR�GREENLET_MANYLINUXc @ s e Zd Zdd� ZdS )�TestCaseMetaClassc C s\ d}t |�� �D ] \}}|�d�r%t|�r%|�|� |r!t�|�}|||< qt�| |||�S )NT�test) �list�items�
startswith�callable�popr �
wrap_refcount�type�__new__)�cls� classname�bases� classDict�check_totalrefcount�key�value� r' �B/usr/local/lib/python3.10/dist-packages/greenlet/tests/__init__.pyr ' s
�zTestCaseMetaClass.__new__N)�__name__�
__module__� __qualname__r r' r' r' r( r $ s r c s� e Zd ZdZdZ ddd�Zedfdd�Zd Zd Z d Z
d
Z d d � Z � fd
d�Z
� fdd�Zdd� Zdd� Zddd�Zddd�Z� ZS ) �TestCaseg����MbP?r Nc C s� |p| j }|p | j}| j}t|� t� | j }t� s't� |ks'| jsUt � |krUt|� t� |krEt
d� t
d|dt� � t
d|dt � � nt� s't� |ks'| jsUt � |ks't � d S )NzTime limit exceeded.zThreads: Waiting for onlyz-->zMGlets : Waiting for only) �threads_before_test�main_greenlets_before_test� cleanup_attempt_sleep_durationr r �cleanup_max_sleep_secondsr �active_thread_count�expect_greenlet_leakr
�printr )�self�initial_active_threads�initial_main_greenlets�
sleep_time�
quit_afterr' r' r( �wait_for_pending_cleanupsB s8
�
�
����
�
z"TestCase.wait_for_pending_cleanupsTc sH t d�D ]}t� q|rt� fdd�t� D ��S t� fdd�t� D ��S )Nr c 3 s � | ] }t |�� u rd V qdS �r N)r ��.0�x��kindr' r( � <genexpr>j s � ��z)TestCase.count_objects.<locals>.<genexpr>c 3 s � | ]
}t |� �rd V qdS r: )�
isinstancer; r> r' r( r@ p s � �
�)�ranger �sumr )r4 r? �
exact_kind�_r' r> r( �
count_objectsd s � �zTestCase.count_objectsr Fc C s | � td�S )zJ
Find all the greenlets and subclasses tracked by the GC.
F)rF � RawGreenlet�r4 r' r' r( �count_greenlets{ s zTestCase.count_greenletsc s> t � �� t� t� | _t� | _| �| j| j� | �� | _ d S �N)
�super�setUpr r1 r- r
r. r9 rI �greenlets_before_testrH �� __class__r' r( rL � s
zTestCase.setUpc s. t | dd�rd S | �| j| j� t� �� d S )N� skipTearDownF)�getattrr9 r- r. rK �tearDownrH rN r' r( rR � s zTestCase.tearDownc C s( dd l }ts|j |j f}|S d}|S )Nr )r l l )�signal�WIN�SIGABRT�SIGSEGV)r4 rS �
expected_exitr' r' r( �,get_expected_returncodes_for_aborted_process� s %�� �� z5TestCase.get_expected_returncodes_for_aborted_processc
C s6 zt �� �� jW S ty } zt�d�|�d}~ww )a�
Return the current process's USS in bytes.
uss is available on Linux, macOS, Windows. Also known as
"Unique Set Size", this is the memory which is unique to a
process and which would be freed if the process was terminated
right now.
If this is not supported by ``psutil``, this raises the
:exc:`unittest.SkipTest` exception.
zuss not supportedN)�psutil�Process�memory_full_info�uss�AttributeError�unittest�SkipTest)r4 �er' r' r( �get_process_uss� s ��zTestCase.get_process_ussc
C s� dd l }tj�tj�t�|�}z
|jtj|gd|j d�W S |j
yB } z|r=t d� t d|� t d� t |j � t d� � d }~ww )Nr zutf-8)�encoding�stderrz-----zFailed to run scriptz~~~~~z------)
�
subprocess�os�path�join�dirname�__file__� check_output�sys�
executable�STDOUT�CalledProcessErrorr3 �output)r4 � script_name� show_outputrd �script�exr' r' r( �
run_script� s(
� �
��zTestCase.run_scriptc C sl dd l }| �|j��}| j|dd�}|}W d � n1 s w Y |d u r+| �� }| �|jj|� |jS )Nr F)rq )rd � assertRaisesrn rt rX �assertIn� exception�
returncode)r4 rp � exitcodesrd �excro �__traceback_info__r' r' r( �assertScriptRaises� s �zTestCase.assertScriptRaises)NN)TrJ )r) r* r+ r/ r0 r9 r rF rM r- r. r2 rI rL rR rX ra rt r| �
__classcell__r' r' rN r( r, : s$
�"
-
r, �NewBase)%�__doc__re rk r^ �gcr r � threadingr r1 r r rY r rG r �greenlet._greenletr r
� r � version_info�PY312�PY313�platformr rT �environ�get�RUNNING_ON_GITHUB_ACTIONS�RUNNING_ON_TRAVIS�RUNNING_ON_APPVEYOR�
RUNNING_ON_CI�RUNNING_ON_MANYLINUXr r r, r' r' r' r( �<module> s:
�