관리-도구
편집 파일: __init__.cpython-38.pyc
U �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�p�e 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 � �winZGITHUB_ACTIONSZTRAVISZAPPVEYORZGREENLET_MANYLINUXc @ s e Zd Zdd� ZdS )�TestCaseMetaClassc C s\ d}t |�� �D ]:\}}|�d�rt|�r|�|� |rBt�|�}|||< qt�| |||�S )NT�test) �list�items� startswith�callable�popr Z wrap_refcount�type�__new__)�cls� classname�basesZ classDictZcheck_totalrefcount�key�value� r �G/opt/hc_python/lib64/python3.8/site-packages/greenlet/tests/__init__.pyr ' s zTestCaseMetaClass.__new__N)�__name__� __module__� __qualname__r r r r r! r $ s r c s� e Zd ZdZdZddd�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� sNt� |ksN| js�t � |kr�t|� t� |kr.t d� t d|dt� � t d|dt � � q�q.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 )�selfZinitial_active_threadsZinitial_main_greenletsZ sleep_timeZ quit_afterr r r! �wait_for_pending_cleanupsB s4 ���� � �z"TestCase.wait_for_pending_cleanupsTc sH t d�D ] }t� q|r0t� fdd�t� D ��S t� fdd�t� D ��S )Nr c 3 s | ]}t |�� krd 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/ )� isinstancer0 r3 r r! r5 p s �)�ranger �sumr )r- r4 Z exact_kind�_r r3 r! � count_objectsd s � �zTestCase.count_objectsr Fc C s | � td�S )zJ Find all the greenlets and subclasses tracked by the GC. F)r: � RawGreenlet�r- r r r! �count_greenlets{ s zTestCase.count_greenletsc s> t � �� t� t� | _t� | _| �| j| j� | �� | _ d S )N) �super�setUpr r* r& r r' r. r= �greenlets_before_testr< �� __class__r r! r? � s zTestCase.setUpc s. t | dd�rd S | �| j| j� t� �� d S )NZ skipTearDownF)�getattrr. r& r' r> �tearDownr< rA r r! rD � s zTestCase.tearDownc C s$ dd l }ts |j |j fnd}|S )Nr )r l l )�signal�WIN�SIGABRT�SIGSEGV)r- rE Z expected_exitr r r! �,get_expected_returncodes_for_aborted_process� s %��� z5TestCase.get_expected_returncodes_for_aborted_processc C sD zt �� �� jW S tk r> } zt�d�|�W 5 d}~X Y nX dS )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�ProcessZmemory_full_infoZuss�AttributeError�unittestZSkipTest)r- �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 k r� } z6|r|t d� t d|� t d� t |j � t d� � W 5 d }~X Y nX d S )Nr zutf-8)�encoding�stderrz-----zFailed to run scriptz~~~~~z------) � subprocess�os�path�join�dirname�__file__� check_output�sys� executable�STDOUT�CalledProcessErrorr, �output)r- � script_name� show_outputrR �script�exr r r! � run_script� s$ � � zTestCase.run_scriptc C sX dd l }| �|j��}| j|dd�}|}W 5 Q R X |d krB| �� }| �|jj|� |jS )Nr F)r_ )rR Z assertRaisesr\ rb rI ZassertIn� exception� returncode)r- r^ Z exitcodesrR �excr] Z__traceback_info__r r r! �assertScriptRaises� s zTestCase.assertScriptRaises)NN)T)N)r" r# r$ r( r) r. r r: r@ r& r' r+ r= r? rD rI rO rb rf � __classcell__r r rA r! r% : s" � " - r% ZNewBase)%�__doc__rS rY rM �gcr r � threadingr r* r r rJ r r; r Zgreenlet._greenletr r � r � version_infoZPY312ZPY313�platformr rF �environ�getZRUNNING_ON_GITHUB_ACTIONSZRUNNING_ON_TRAVISZRUNNING_ON_APPVEYORZ RUNNING_ON_CIZRUNNING_ON_MANYLINUXr r r% r r r r! �<module> s8 �