File: //lib/python3/dist-packages/twisted/trial/__pycache__/_asynctest.cpython-38.pyc
U
��W[*9 � @ s� d Z ddlmZmZ ddlZddlZddlmZ ddlm Z m
Z
ddl m Z ddl
mZmZ ddlmZmZmZ g Zeej�G d d
� d
e��ZdS ) zP
Things likely to be used by writers of unit tests.
Maintainer: Jonathan Lange
� )�division�absolute_importN)� implementer)�defer�utils)�failure)�itrial�util)�FailTest�SkipTest�SynchronousTestCasec s� e Zd ZdZd3� fdd� Zdd� ZeZdd� Zd d
� Zd d � Z d
d� Z
dd� Z dd� Z dd� Z
dd� Zdd� Zdd� Zdd � Zd d � Zd d � Zd!d"� Zd#d$� Zd%d&� Zd'd(� Zd)d*� Z� fd+d,�Zd-d.� Zd/d0� Ze fd1d2�Z � Z S )4�TestCasea=
A unit test. The atom of the unit testing universe.
This class extends L{SynchronousTestCase} which extends C{unittest.TestCase}
from the standard library. The main feature is the ability to return
C{Deferred}s from tests and fixture methods and to have the suite wait for
those C{Deferred}s to fire. Also provides new assertions such as
L{assertFailure}.
@ivar timeout: A real number of seconds. If set, the test will
raise an error if it takes longer than C{timeout} seconds.
If not set, util.DEFAULT_TIMEOUT_DURATION is used.
�runTestc s t t| ��|� dS )a�
Construct an asynchronous test case for C{methodName}.
@param methodName: The name of a method on C{self}. This method should
be a unit test. That is, it should be a short method that calls some of
the assert* methods. If C{methodName} is unspecified,
L{SynchronousTestCase.runTest} will be used as the test method. This is
mostly useful for testing Trial.
N)�superr
�__init__)�self�
methodName�� __class__� �:/usr/lib/python3/dist-packages/twisted/trial/_asynctest.pyr . s
zTestCase.__init__c s& �fdd�}� �fdd�}|� ||�S )z�
Fail if C{deferred} does not errback with one of C{expectedFailures}.
Returns the original Deferred with callbacks added. You will need
to return this Deferred from your test case.
c s � � d| f ��d S )Nz&did not catch an error, instead got %r)�failureException)�ignore�r r r �_cbA s �z#TestCase.assertFailure.<locals>._cbc s. | j � � r| jS d� t| �f }��|��d S )Nz
Expected: %r
Got:
%s)�check�value�strr )r �output��expectedFailuresr r r �_ebE s
�z#TestCase.assertFailure.<locals>._eb)� addCallbacks)r Zdeferredr r r! r r r �
assertFailure; s zTestCase.assertFailurec s� ddl m� ��� ������fdd�}t�|tjtd��}t���}t �
|�rht d|f �}t �
|�S t �tj��� |�}���||�� |�� fdd�� |S ) Nr ��reactorc s� t �d�� �f �}t�|�}z| �|� W nZ t jk
r� ��� d�_��� }|d k rv|� |�rv��
�||� n �� �|� Y nX d S )Nz %r (%s) still running at %s secsT) r � TimeoutErrorr �FailureZerrbackZAlreadyCalledError�crash� _timedOut�getTodo�expected�addExpectedFailure�addError)�d�e�f�todo)r r% �resultr �timeoutr r � onTimeoutS s �
z TestCase._run.<locals>.onTimeout��categoryz7%r is a generator function and therefore will never runc s � � � r� �� p| S �N)ZactiveZcancel)�x)�callr r �<lambda>r � z TestCase._run.<locals>.<lambda>)�twisted.internetr% �
getTimeoutr �suppressWarningsr �suppress�DeprecationWarning�getattr�inspectZisgeneratorfunction� TypeErrorr ZfailZ
maybeDeferredZrunWithWarningsSuppressed� _getSuppressZ callLater�addBoth)r r r2 r4 �method�excr. r )r9 r r% r2 r r3 r �_runP s.
�
��
�z
TestCase._runc O s | j ||�S r7 )�run)r �args�kwargsr r r �__call__v s zTestCase.__call__c C s* | � d|�}|j| j| j|f|fd� |S )N�setUp�Z callbackArgsZ errbackArgs)rH r" �deferTestMethod�
_ebDeferSetUp�r �ignoredr2 r. r r r �
deferSetUpz s �zTestCase.deferSetUpc C sN |� t�r$|�| | �| j|j�� n |�| |� |� t�rB|�� | � d |�S r7 )
r r �addSkip�_getSkipReasonrM r r- �KeyboardInterrupt�stop�deferRunCleanups�r r r2 r r r rP � s
zTestCase._ebDeferSetUpc C sH | � | j|�}|j| j| j|f|fd� |�| j|� |�| j|� |S )NrN )rH �_testMethodNamer" �_cbDeferTestMethod�_ebDeferTestMethodrE rX �
deferTearDownrQ r r r rO � s �zTestCase.deferTestMethodc C s( | � � d k r |�| | � � � nd| _|S )NT)r* ZaddUnexpectedSuccess�_passed)r rR r2 r r r r[ � s zTestCase._cbDeferTestMethodc C s� | � � }|d k r*|�|�r*|�| ||� nr|�| jt�rF|�| |� nV|�t�rf|�| |� |� � n6|�t
�r�|� | | � t
| | j�|j�� n |�| |� d S r7 )r* r+ r, r r r
Z
addFailurerV r- rW r rT rU rA rZ r )r r0 r2 r1 r r r r\ � s
�zTestCase._ebDeferTestMethodc C s | � d|�}|�| j|� |S )NZtearDown)rH Z
addErrback�_ebDeferTearDownrQ r r r r] � s zTestCase.deferTearDownc C s( |� | |� |�t�r |�� d| _d S �NF)r- r rV rW r^ rY r r r r_ � s
zTestCase._ebDeferTearDownc C s | � � }|�| j|� |S )zd
Run any scheduled cleanups and report errors (if any to the result
object.
)� _runCleanupsZ addCallback�_cbDeferRunCleanupsrQ r r r rX � s zTestCase.deferRunCleanupsc C s@ |D ]6\}}|t jkr|�| |� |�t�r4|�� d| _qd S r` )r �FAILUREr- r rV rW r^ )r ZcleanupResultsr2 �flagZ testFailurer r r rb � s
z TestCase._cbDeferRunCleanupsc C s� z t �| |��� }|s d| _W n" |�| t�� � d| _Y nX | j�� D ]}|�| |� d| _qL| � � | �
� | jr�|� | � d S r` ) r �_JanitorZpostCaseCleanupr^ r- r r' Z _observerZ getErrorsZflushLoggedErrorsZ_removeObserverZ
addSuccess)r r2 Zclean�errorr r r �_cleanUp� s
zTestCase._cleanUpc C s6 zt �| |��� W n |�| t�� � Y nX d S r7 )r re ZpostClassCleanupr- r r' )r r2 r r r �
_classCleanUp� s zTestCase._classCleanUpc s � �fdd�}|S )z�
Create a method which wraps the reactor method C{name}. The new
method issues a deprecation warning and calls the original.
c s( t jd� � f dtd� �j� | |�S )Nz{reactor.%s cannot be used inside unit tests. In the future, using %s will fail the test and may crash or hang the test run.� )�
stacklevelr6 )�warnings�warnr@ �_reactorMethods)�a�kw��namer r r �_� s � �z&TestCase._makeReactorMethod.<locals>._r )r rq rr r rp r �_makeReactorMethod� s zTestCase._makeReactorMethodc C s6 i | _ dD ]&}t||�| j |< t||| �|�� q
dS )z�
Deprecate C{iterate}, C{crash} and C{stop} on C{reactor}. That is,
each method is wrapped in a function that issues a deprecation
warning, then calls the original.
@param reactor: The Twisted reactor.
)r( ZiteraterW N)rm rA �setattrrs )r r% rq r r r �_deprecateReactor� s zTestCase._deprecateReactorc C s* | j �� D ]\}}t|||� q
i | _ dS )z�
Restore the deprecated reactor methods. Undoes what
L{_deprecateReactor} did.
@param reactor: The Twisted reactor.
N)rm �itemsrt )r r% rq rF r r r �_undeprecateReactor s z TestCase._undeprecateReactorc C sH dd� }g }t | j�dkr>| j�� \}}}|�||||�� q t�|�S )z�
Run the cleanups added with L{addCleanup} in order.
@return: A C{Deferred} that fires when all cleanups are run.
c s � ��fdd�S )Nc s
�� ��S r7 r r �rJ r0 rK r r r: r; z>TestCase._runCleanups.<locals>._makeFunction.<locals>.<lambda>r )r0 rJ rK r rx r �
_makeFunction s z,TestCase._runCleanups.<locals>._makeFunctionr )�lenZ _cleanups�pop�appendr Z_runSequentially)r ry Z callablesr0 rJ rK r r r ra s zTestCase._runCleanupsc C sd ddl m} | �|� d| _z6| �d|�}z| �|� W 5 | �|� | �|� X W 5 | �|� X dS )z�
Really run C{setUp}, the test method, and C{tearDown}. Any of these may
return L{defer.Deferred}s. After they complete, do some reactor cleanup.
@param result: A L{TestResult} object.
r r$ FN) r< r% ru r) rw rS rg rh �_wait)r r2 r% r. r r r �_runFixturesAndTest s
z TestCase._runFixturesAndTestc s t t| �j|f|�|�S )a
Extend the base cleanup feature with support for cleanup functions which
return Deferreds.
If the function C{f} returns a Deferred, C{TestCase} will wait until the
Deferred has fired before proceeding to the next function.
)r r
�
addCleanup)r r0 rJ rK r r r r 4 s zTestCase.addCleanupc C s | � � S r7 )rD r r r r � getSuppress? s zTestCase.getSuppressc C sN t �| jdt j�}z
t|�W S ttfk
rH tjdt d� t j Y S X dS )ae
Returns the timeout value set on this test. Checks on the instance
first, then the class, then the module, then packages. As soon as it
finds something with a C{timeout} attribute, returns that. Returns
L{util.DEFAULT_TIMEOUT_DURATION} if it cannot find anything. See
L{TestCase} docstring for more details.
r3 z)'timeout' attribute needs to be a number.r5 N)
r ZacquireAttributeZ_parentsZDEFAULT_TIMEOUT_DURATION�float�
ValueErrorrC rk rl r@ )r r3 r r r r= C s
�
�zTestCase.getTimeoutc s� |r t d��ddlm� g ��fdd�}� �fdd�}t�|tjdtd ��}� fd
d �}t�|tjdtd ��}|�d � zV|�
|� �r�W �Dd S |�
|� |� _ z � � � W 5 � ` X �s�| j
r�W �d S t� �W 5 d �|� � X d S )
zJTake a Deferred that only ever callbacks. Block until it happens.
z_wait is not reentrantr r$ c s � d k r� � | � d S r7 )r| )�any)�resultsr r r| a s z TestCase._wait.<locals>.appendc s �d k r� � � d S r7 �r( )Zign�r% r� r r r( d s z TestCase._wait.<locals>.crashz reactor\.crash cannot be used.*)�messager6 c s � � � d S r7 r� r r$ r r rW j s z TestCase._wait.<locals>.stopN)� RuntimeErrorr< r% r r> r r? r@ r| r{ rE rW rI r) rV )r r. �runningr| r( rW r r� r r} Y sD �� ��
zTestCase._wait)r ) �__name__�
__module__� __qualname__�__doc__r r# ZfailUnlessFailurerH rL rS rP rO r[ r\ r] r_ rX rb rg rh rs ru rw ra r~ r r� r= �_wait_is_runningr} �
__classcell__r r r r r
s4
&
r
)r� Z
__future__r r rB rk Zzope.interfacer r< r r Ztwisted.pythonr Z
twisted.trialr r Ztwisted.trial._synctestr
r r r� Z ITestCaser
r r r r �<module> s