3
\ï( ã @ sx d dl Z ddlmZ ddlmZ dZdd„ ZG dd „ d eƒZG d
d „ d eƒZG d d
„ d
eƒZ dd„ Z
G dd„ deƒZ dS )é Né )Úcase)ÚutilTc C s t | |dd„ ƒ}|ƒ d S )Nc S s d S )N© r r r ú&/usr/lib64/python3.6/unittest/suite.pyÚ<lambda> s z!_call_if_exists.<locals>.<lambda>)Úgetattr)ÚparentÚattrÚfuncr r r Ú_call_if_exists s r c @ sl e Zd ZdZf fdd„Zdd„ Zdd„ Zdd „ Zd
d „ Zd d
„ Z dd„ Z
dd„ Z dd„ Z dd„ Z
dd„ ZdS )Ú
BaseTestSuiteTc C s g | _ d| _| j|ƒ d S )Nr )Ú_testsÚ_removed_testsÚaddTests)ÚselfÚtestsr r r Ú__init__ s zBaseTestSuite.__init__c C s dt j| jƒt| ƒf S )Nz
<%s tests=%s>)r ÚstrclassÚ __class__Úlist)r r r r Ú__repr__ s zBaseTestSuite.__repr__c C s t || jƒstS t| ƒt|ƒkS )N)Ú
isinstancer ÚNotImplementedr )r Úotherr r r Ú__eq__ s zBaseTestSuite.__eq__c C s
t | jƒS )N)Úiterr )r r r r Ú__iter__" s zBaseTestSuite.__iter__c C s( | j }x | D ]}|r ||jƒ 7 }q W |S )N)r ÚcountTestCases)r ZcasesÚtestr r r r % s
z BaseTestSuite.countTestCasesc C sL t |ƒstdjt|ƒƒƒ‚t|tƒr<t|tjt fƒr<tdƒ‚| j
j |ƒ d S )Nz{} is not callablezNTestCases and TestSuites must be instantiated before passing them to addTest()) ÚcallableÚ TypeErrorÚformatÚreprr ÚtypeÚ
issubclassr ZTestCaseÚ TestSuiter Úappend)r r r r r ÚaddTest, s zBaseTestSuite.addTestc C s. t |tƒrtdƒ‚x|D ]}| j|ƒ qW d S )Nz0tests must be an iterable of tests, not a string)r Ústrr! r( )r r r r r r r 6 s
zBaseTestSuite.addTestsc C s: x4t | ƒD ](\}}|jrP ||ƒ | jr
| j|ƒ q
W |S )N)Ú enumerateÚ
shouldStopÚ_cleanupÚ_removeTestAtIndex)r ÚresultÚindexr r r r Úrun< s zBaseTestSuite.runc C sN y| j | }W n tk
r" Y n(X t|dƒr@| j|jƒ 7 _d | j |< d S )Nr )r r! Úhasattrr r )r r/ r r r r r- E s
z BaseTestSuite._removeTestAtIndexc O s | j ||ŽS )N)r0 )r ÚargsÚkwdsr r r Ú__call__S s zBaseTestSuite.__call__c C s x| D ] }|j ƒ qW d S )N)Údebug)r r r r r r5 V s
zBaseTestSuite.debugN)Ú__name__Ú
__module__Ú __qualname__r, r r r r r r( r r0 r- r4 r5 r r r r r
s
r
c @ sN e Zd Zddd„Zdd„ Zdd„ Zdd „ Zd
d „ Zd d
„ Zdd„ Z dd„ Z
dS )r& Fc C sÖ d}t |ddƒdkr d |_}x’t| ƒD ]†\}}|jr8P t|ƒrˆ| j||ƒ | j||ƒ | j||ƒ |j|_ t |jddƒs(t |ddƒrˆq(|s–||ƒ n|j
ƒ | j r(| j |ƒ q(W |rÒ| jd |ƒ | j
|ƒ d|_|S )NFÚ_testRunEnteredTÚ_classSetupFailedÚ_moduleSetUpFailed)r r9 r* r+ Ú _isnotsuiteÚ_tearDownPreviousClassÚ_handleModuleFixtureÚ_handleClassSetUpr Ú_previousTestClassr5 r, r- Ú_handleModuleTearDown)r r. r5 ZtopLevelr/ r r r r r0 f s0
z
TestSuite.runc C s t ƒ }| j|dƒ d S )NT)Ú _DebugResultr0 )r r5 r r r r5 ‡ s zTestSuite.debugc C sì t |dd ƒ}|j}||kr d S |jr(d S t |ddƒr8d S y
d|_W n tk
rV Y nX t |dd ƒ}|d k rèt|dƒ zdy
|ƒ W nT tk
rÖ } z8t|tƒr ‚ d|_t j
|ƒ}d| }| j |||ƒ W Y d d }~X nX W d t|dƒ X d S ) Nr@ Ú__unittest_skip__FÚ
setUpClassÚ _setupStdoutTzsetUpClass (%s)Ú_restoreStdout) r r r; r: r! r Ú Exceptionr rB r r Ú _addClassOrModuleLevelException) r r r. Ú
previousClassÚ currentClassrD ÚeÚ classNameÚ errorNamer r r r? Ž s2
$zTestSuite._handleClassSetUpc C s" d }t |dd ƒ}|d k r |j}|S )Nr@ )r r7 )r r. ÚpreviousModulerI r r r Ú_get_previous_module® s
z TestSuite._get_previous_modulec C sÜ | j |ƒ}|jj}||kr d S | j|ƒ d|_ytj| }W n tk
rP d S X t|dd ƒ}|d k rØt |dƒ zZy
|ƒ W nJ t
k
rÆ } z.t |t ƒrš‚ d|_d| }| j
|||ƒ W Y d d }~X nX W d t |dƒ X d S )NFÚ setUpModulerE TzsetUpModule (%s)rF )rO r r7 rA r; ÚsysÚmodulesÚKeyErrorr r rG r rB rH ) r r r. rN Z
currentModuleÚmodulerP rK rM r r r r> ¶ s,
$z TestSuite._handleModuleFixturec C sL t |ƒ}t|dd ƒ}|d k r8t|tjƒr8||t|ƒƒ n|j|tjƒ ƒ d S )NÚaddSkip) Ú _ErrorHolderr r r ZSkipTestr) ZaddErrorrQ Úexc_info)r r. Z exceptionrM ÚerrorrU r r r rH Ò s
z)TestSuite._addClassOrModuleLevelExceptionc C sÈ | j |ƒ}|d krd S |jr d S ytj| }W n tk
rB d S X t|dd ƒ}|d k rÄt|dƒ zTy
|ƒ W nD tk
r² } z(t|t ƒrŒ‚ d| }| j
|||ƒ W Y d d }~X nX W d t|dƒ X d S )NÚtearDownModulerE ztearDownModule (%s)rF ) rO r; rQ rR rS r r rG r rB rH )r r. rN rT rY rK rM r r r rA Ú s(
$z TestSuite._handleModuleTearDownc C sÜ t |dd ƒ}|j}||kr d S t |ddƒr.d S t |ddƒr>d S t |ddƒrNd S t |dd ƒ}|d k rØt|dƒ z^y
|ƒ W nN tk
rÆ } z2t|tƒr–‚ tj|ƒ}d| }| j|||ƒ W Y d d }~X nX W d t|d ƒ X d S )
Nr@ r: Fr; rC Ú
tearDownClassrE ztearDownClass (%s)rF ) r r r rG r rB r r rH ) r r r. rI rJ rZ rK rL rM r r r r= ó s,
$z TestSuite._tearDownPreviousClassN)F) r6 r7 r8 r0 r5 r? rO r> rH rA r= r r r r r& \ s
! r& c @ sP e Zd ZdZdd„ Zdd„ Zdd„ Zdd „ Zd
d „ Zd d
„ Z dd„ Z
dd„ Z dS )rV Nc C s
|| _ d S )N)Ú description)r r[ r r r r s z_ErrorHolder.__init__c C s | j S )N)r[ )r r r r Úid s z_ErrorHolder.idc C s d S )Nr )r r r r ÚshortDescription s z _ErrorHolder.shortDescriptionc C s d| j f S )Nz <ErrorHolder description=%r>)r[ )r r r r r # s z_ErrorHolder.__repr__c C s | j ƒ S )N)r\ )r r r r Ú__str__& s z_ErrorHolder.__str__c C s d S )Nr )r r. r r r r0 ) s z_ErrorHolder.runc C s
| j |ƒS )N)r0 )r r. r r r r4 . s z_ErrorHolder.__call__c C s dS )Nr r )r r r r r 1 s z_ErrorHolder.countTestCases) r6 r7 r8 ZfailureExceptionr r\ r] r r^ r0 r4 r r r r r rV s
rV c C s& y t | ƒ W n tk
r dS X dS )NTF)r r! )r r r r r< 4 s
r< c @ s e Zd ZdZdZdZdS )rB NF)r6 r7 r8 r@ r; r+ r r r r rB = s rB ) rQ Ú r r Z
__unittestr Úobjectr
r& rV r< rB r r r r Ú<module> s L 3&
All system for education purposes only. For more tools: Telegram @jackleet