CasperSecurity
o
�b, � @ s d Z ddlZddlZddlZddlmZ ddlmZ ddl m
Z
m Z ddl m
Z
G dd� dej�Zee
j�G d d
� d
e�e
jd ���Zd d
� Zdd� ZG dd� de�ZG dd� de�ZG dd� de�Ze�eeje
j� e�eeje
j� eedd�Zer�e�eee
j� dd� Z dS )z-
Infrastructure for test running and suites.
� N)� implementer)�
components)�itrial�reporter)� _logObserverc @ � e Zd ZdZdd� ZdS )� TestSuitezh
Extend the standard library's C{TestSuite} with a consistently overrideable
C{run} method.
c C s$ | j D ] }|jr |S ||� q|S )z;
Call C{run} on every member of the suite.
)�_tests�
shouldStop)�self�result�test� r �</usr/lib/python3/dist-packages/twisted/trial/_asyncrunner.py�run s
�z
TestSuite.runN��__name__�
__module__� __qualname__�__doc__r r r r r r � r c @ s e Zd ZdZdd� Zdd� ZdS )�
TestDecoratorz�
Decorator for test cases.
@param _originalTest: The wrapped instance of test.
@type _originalTest: A provider of L{itrial.ITestCase}
c C s
| � |�S �zQ
Run the unit test.
@param result: A TestResult object.
)r �r r r r r �__call__3 s
zTestDecorator.__call__c C s | j �t�|| j��S r )�
_originalTestr r �_AdaptedReporter� __class__r r r r r ; s zTestDecorator.runN)r r r r r r r r r r r & s r r c C s
g | _ dS )z�
Clear all tests from C{suite}.
This messes with the internals of C{suite}. In particular, it assumes that
the suite keeps all of its tests in a list in an instance variable called
C{_tests}.
N)r )�suiter r r � _clearSuiteD s
r c C sN zt | �}W n
ty || � Y S w t| � |D ]
}| �t||�� q| S )a2
Decorate all test cases in C{test} with C{decorator}.
C{test} can be a test case or a test suite. If it is a test suite, then the
structure of the suite is preserved.
L{decorate} tries to preserve the class of the test suites it finds, but
assumes the presence of the C{_tests} attribute on the suite.
@param test: The C{TestCase} or C{TestSuite} to decorate.
@param decorator: A unary callable used to decorate C{TestCase}s.
@return: A decorated C{TestCase} or a C{TestSuite} containing decorated
C{TestCase}s.
)�iter� TypeErrorr �addTest�decorate)r
� decorator�tests�caser r r r# O s �r# c @ s e Zd ZdZdS )�_PyUnitTestCaseAdapterz2
Adapt from pyunit.TestCase to ITestCase.
N)r r r r r r r r r' n s r' c @ r )�_BrokenIDTestCaseAdapterz�
Adapter for pyunit-style C{TestCase} subclasses that have undesirable id()
methods. That is C{unittest.FunctionTestCase} and C{unittest.DocTestCase}.
c C s | j �� }|dur |S | j �� S )zH
Return the fully-qualified Python name of the doctest.
N)r �shortDescription�id)r �testIDr r r r* z s
z_BrokenIDTestCaseAdapter.idN)r r r r r* r r r r r( t r r( c @ r )� _ForceGarbageCollectionDecoratorz�
Forces garbage collection to be run before and after the test. Any errors
logged during the post-test collection are added to the test result as
errors.
c C sR t �� t�| |� t�� t �� t�� D ]}|�| |� qt�� t� � d S )N)
�gc�collectr r r �_add� getErrors�addError� flushErrors�_remove)r r �errorr r r r � s z$_ForceGarbageCollectionDecorator.runNr r r r r r, � s r, � DocTestCasec c sF � zt | �}W n
ty | V Y dS w |D ] }t|�E dH qdS )zF
Iterate through all of the test cases in C{testSuiteOrCase}.
N)r r! �
_iterateTests)�testSuiteOrCaser r
r r r r6 � s � ��r6 ) r �doctestr- �unittest�pyunit�zope.interfacer �twisted.pythonr �
twisted.trialr r �twisted.trial._synctestr r � ITestCase�proxyForInterfacer r r# r' r( r, �registerAdapter�TestCase�FunctionTestCase�getattr� _docTestCaser6 r r r r �<module> s8
��
�