0x1949 Team - FAZEMRX - MANAGER
Edit File: _asynctest.cpython-310.pyc
o �b@7 � @ s� U d Z ddlZddlZddlmZ ddlmZ ddlmZm Z ddl m Z ddl m Z mZ ddlmZmZmZ g Zed ed < ee j�G d d � d e��ZdS ) zP Things likely to be used by writers of unit tests. Maintainer: Jonathan Lange � N)�List)� implementer)�defer�utils)�failure)�itrial�util)�FailTest�SkipTest�SynchronousTestCase�_wait_is_runningc s� e Zd ZdZd/� fdd� Zdd� ZeZdd� Zd d � Zd d � Z d d� Z dd� Z dd� Z dd� Z dd� Zdd� Zejdd� �Zdd � Zd d � Zd d � Zd!d"� Zd#d$� Zd%d&� Z� fd'd(�Zd)d*� Zd+d,� Ze fd-d.�Z � Z S )0�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 � �|� 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)�super�__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| ����)Nz$did not catch an error, instead got )�failureException)�ignore�r r r �_cb@ s �z#TestCase.assertFailure.<locals>._cbc s* | j � � r| jS d�� t| ��}��|��)Nz Expected: {!r} Got: {})�check�value�format�strr )r �output��expectedFailuresr r r �_ebE s � z#TestCase.assertFailure.<locals>._eb)� addCallbacks)r �deferredr! r r" r r r � assertFailure9 s zTestCase.assertFailurec s� ddl m� ��� ������fdd�}t�|tjtd��}t���}t � |�r4t d� |��}t �|�S t �tj��� |�}���||�� |�� fdd�� |S ) Nr ��reactorc s� t ���d� � d�� d��}t�|�}z| �|� W d S t jyJ ��� d�_��� }|d urA|� |�rA�� �||� Y d S �� �|� Y d S w )Nz (z) still running at z secsT) r � TimeoutErrorr �Failure�errback�AlreadyCalledError�crash� _timedOut�getTodo�expected�addExpectedFailure�addError)�d�e�f�todo)r r' �resultr �timeoutr r � onTimeoutW s � �z TestCase._run.<locals>.onTimeout��categoryz9{!r} is a generator function and therefore will never runc s � � � r� �� p | S �N)�active�cancel)�x)�callr r �<lambda>| s z TestCase._run.<locals>.<lambda>)�twisted.internetr' � getTimeoutr �suppressWarningsr �suppress�DeprecationWarning�getattr�inspect�isgeneratorfunction� TypeErrorr r �fail� maybeDeferred�runWithWarningsSuppressed� _getSuppress� callLater�addBoth)r r r6 r8 �method�excr2 r )r? r r' r6 r r7 r �_runR s( � �� �z TestCase._runc O s | j |i |��S r; )�run)r �args�kwargsr r r �__call__ s zTestCase.__call__c C s* | � d|�}|j| j| j|f|fd� |S )N�setUp�� callbackArgs� errbackArgs)rR r# �deferTestMethod� _ebDeferSetUp�r �ignoredr6 r2 r r r � deferSetUp� s �zTestCase.deferSetUpc C sN |� t�r|�| | �| j|j�� n|�| |� |� t�r!|�� | � d |�S r; ) r r �addSkip�_getSkipReasonrW r r1 �KeyboardInterrupt�stop�deferRunCleanups�r r r6 r r r r\ � s zTestCase._ebDeferSetUpc C sH | � | j|�}|j| j| j|f|fd� |�| j|� |�| j|� |S )NrX )rR �_testMethodNamer# �_cbDeferTestMethod�_ebDeferTestMethodrO rd � deferTearDownr] r r r r[ � s �zTestCase.deferTestMethodc C s* | � � d ur|�| | � � � |S d| _|S )NT)r. �addUnexpectedSuccess�_passed)r r^ r6 r r r rg � s �zTestCase._cbDeferTestMethodc C s� | � � }|d ur|�|�r|�| ||� d S |�| jt�r%|�| |� d S |�t�r6|�| |� |� � d S |�t �rL|� | | � t | | j�|j�� d S |�| |� d S r; )r. r/ r0 r r r � addFailurerb r1 rc r r` ra rF rf r )r r4 r6 r5 r r r rh � s �zTestCase._ebDeferTestMethodc C s | � d|�}|�| j|� |S )N�tearDown)rR � addErrback�_ebDeferTearDownr] r r r ri � s zTestCase.deferTearDownc C s( |� | |� |�t�r|�� d| _d S �NF)r1 r rb rc rk re r r r ro � s zTestCase._ebDeferTearDownc c sv � g }| j ddd� D ] \}}}z ||i |��V W q ty* |�t�� � Y q w |D ] }|�| |� d| _q-dS )zf Run any scheduled cleanups and report errors (if any) to the result. object. N���F)� _cleanups� Exception�appendr r) r1 rk )r r^ r6 �failures�funcrT rU r4 r r r rd � s � � �zTestCase.deferRunCleanupsc C s� zt �| |��� }|sd| _W n ty# |�| t�� � d| _Y nw | j� � D ] }|�| |� d| _q)| � � | � � | jrG|� | � d S d S rp ) r �_Janitor�postCaseCleanuprk � BaseExceptionr1 r r) � _observer� getErrors�flushLoggedErrors�_removeObserver� addSuccess)r r6 �clean�errorr r r �_cleanUp� s" � � �zTestCase._cleanUpc C s< z t �| |��� W d S ty |�| t�� � Y d S w r; )r rw �postClassCleanupry r1 r r) )r r6 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� | i |��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.� )� stacklevelr: )�warnings�warnrE �_reactorMethods)�a�kw��namer r r �_� s ��z&TestCase._makeReactorMethod.<locals>._r )r r� r� r r� r �_makeReactorMethod� s zTestCase._makeReactorMethodc C s6 i | _ dD ]}t||�| j |< t||| �|�� qdS )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, �iteraterc N)r� rF �setattrr� )r r' r� r r r �_deprecateReactor� s �zTestCase._deprecateReactorc C s* | j �� D ] \}}t|||� qi | _ dS )z� Restore the deprecated reactor methods. Undoes what L{_deprecateReactor} did. @param reactor: The Twisted reactor. N)r� �itemsr� )r r' r� rP r r r �_undeprecateReactor s z TestCase._undeprecateReactorc C s� ddl m} | �|� d| _z+| �d|�}z| �|� W | �|� | �|� n | �|� | �|� w W | �|� dS | �|� w )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) rA r' r� r- r_ �_waitr� r� r� )r r6 r' r2 r r r �_runFixturesAndTest s �z TestCase._runFixturesAndTestc s t � j|g|�R i |��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 � addCleanup)r r4 rT rU r r r r� ) s zTestCase.addCleanupc C s | � � S r; )rM r r r r � getSuppress3 s zTestCase.getSuppressc C sH t �| jdt j�}zt|�W S ttfy# tjdt d� t j Y S w )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. r7 z)'timeout' attribute needs to be a number.r9 ) r �acquireAttribute�_parents�DEFAULT_TIMEOUT_DURATION�float� ValueErrorrI r� r� rE )r r7 r r r rB 6 s � � �zTestCase.getTimeoutc s� |rt d��ddlm� g ��fdd�}� �fdd�}t�|tjdtd ��}� fd d �}t�|tjdtd ��}|�d � z5|� |� �rMW d �|� � d S |� |� |� _ z� � � W � ` n� ` w �sf| j roW d �|� � d S t� �d �|� � w ) zATake a Deferred that only ever callbacks. Block until it happens.z_wait is not reentrantr r& c s � d ur � � | � d S d S r; )rt )�any)�resultsr r rt V s �z TestCase._wait.<locals>.appendc s �d ur � � � d S d S r; �r, )�ign�r' r� r r r, Z s �z TestCase._wait.<locals>.crashz reactor\.crash cannot be used.*)�messager: c s � � � d S r; r� r r& r r rc e s z TestCase._wait.<locals>.stopN)� RuntimeErrorrA r' r rC r rD rE rt rO �poprc rS r- rb )r r2 �runningrt r, rc r r� r r� M sL �� �� � � zTestCase._wait)r ) �__name__� __module__� __qualname__�__doc__r r% �failUnlessFailurerR rV r_ r\ r[ rg rh ri ro r �inlineCallbacksrd r� r� r� r� r� r� r� r� rB r r� � __classcell__r r r r r s4 - r )r� rG r� �typingr �zope.interfacer rA r r �twisted.pythonr � twisted.trialr r �twisted.trial._synctestr r r r �__annotations__� ITestCaser r r r r �<module> s