File: //opt/alt/python37/lib/python3.7/site-packages/asynctest/__pycache__/_fail_on.cpython-37.pyc
B
�P�e� � @ sT d Z ddlmZ dZddd�ZG dd� d�Zdd � Zd
d � Zdd
d�Zddd�Z d S )a�
:class:`asynctest.TestCase` decorator which controls checks performed after
tests.
This module is separated from :mod:`asynctest.case` to avoid circular imports
in modules registering new checks.
To implement new checks:
* its name must be added in the ``DEFAULTS`` dict,
* a static method of the same name must be added to the :class:`_fail_on`
class,
* an optional static method named ``before_[name of the check]`` can be
added to :class:`_fail_on` to implement some set-up before the test runs.
A check may be only available on some platforms, activated by a conditional
import. In this case, ``DEFAULT`` and :class:`_fail_on` can be updated in the
module. There is an example in the :mod:`asynctest.selector` module.
� )� TimerHandleZ_asynctest_fail_onF)� unused_loop�active_handlesc @ sx e Zd Zddd�Zdd� Zddd�Zd d
� Zd d � Zd
d� Zdd� Z e
dd� �Z e
dd� �Z e
dd� �Ze
dd� �ZdS )�_fail_onNc C s |pi | _ d | _d S )N)�checks�_computed_checks)�selfr � r �C/opt/alt/python37/lib/python3.7/site-packages/asynctest/_fail_on.py�__init__' s
z_fail_on.__init__c C s> t |td �}|r&|�� }|�| j� n| �� }t|t|� |S )N)�getattr�
_FAIL_ON_ATTR�copy�updater �setattr)r �func�checkerr r r
�__call__+ s z_fail_on.__call__Tc C s: |r| j �|� n$x"|�� D ]\}}| j �||� q W d S )N)r r �items�
setdefault)r r �override�check�valuer r r
r 6 s z_fail_on.updatec C s t | j�� �S )N)r r r )r r r r
r = s z
_fail_on.copyc C sX | j d krRt�� }y|�t|td �j� W n tk
r> Y nX |�| j� || _ | j S )N)r �DEFAULTSr r r r
r �AttributeError)r �caser r r r
�
get_checks@ s
z_fail_on.get_checksc
C sT | � |�}xDt|j|�D ]4}yt| d| �|� W q ttfk
rJ Y qX qW d S )NZ before_test_)r �filter�getr r � TypeError)r r r r r r r
� before_testO s
z_fail_on.before_testc C s2 | � |�}x"t|j|�D ]}t| |�|� qW d S )N)r r r r )r r r r r r r
�
check_testW s
z_fail_on.check_testc C s | j js| �d� d S )Nz Loop did not run during the test)�loopZ_asynctest_ran�fail)r r r r
r ^ s z_fail_on.unused_loopc C s t | t�o| j S )N)�
isinstancer �
_cancelled)�handler r r
�_is_live_timer_handlec s z _fail_on._is_live_timer_handlec C s t | j|j�S )N)r r' �
_scheduled)�clsr" r r r
�_live_timer_handlesg s z _fail_on._live_timer_handlesc C s( t | �|j��}|r$|�d�|�� d S )Nz#Loop contained unfinished work {!r})�tupler* r" r# �format)r) r Zhandlesr r r
r k s z_fail_on.active_handles)N)T)�__name__�
__module__� __qualname__r r r r r r r! � staticmethodr r' � classmethodr* r r r r r
r & s
r c K s, x"| D ]}|t krtd�|���qW t| �S )zk
Enable checks on the loop state after a test ran to help testers to
identify common mistakes.
z1fail_on() got an unexpected keyword argument '{}')r r r, r )�kwargs�kwargr r r
�fail_onr s
r4 c s. t t� fdd�tD ���}|d kr&|S ||�S )Nc 3 s | ]}|� fV qd S )Nr )�.0�arg)�flagr r
� <genexpr>� s z _fail_on_all.<locals>.<genexpr>)r �dictr )r7 r r r )r7 r
� _fail_on_all� s r: Nc C s
t d| �S )zM
Activate strict checking of the state of the loop after a test ran.
T)r: )r r r r
�strict� s r; c C s
t d| �S )z1
Deactivate all checks after a test ran.
F)r: )r r r r
�lenient� s r< )N)N)
�__doc__�asyncior r
r r r4 r: r; r< r r r r
�<module> s L