3
\V @ s d dl Z d dlZd dlZd dlZd dlZd dlZd dlZd dlmZ ddlm Z m
Z
m Z dZ ej
dejZG dd de jZd d
Zd d Zd
d Zdd Zdd ZG dd deZe Zd ddZe jfddZ de je
j fddZ de je
j fd d Z dS ) N)fnmatch )casesuiteutilTz[_a-z]\w*\.py$c s, e Zd ZdZ fddZ fddZ ZS ) _FailedTestNc s || _ tt| j| d S )N)
_exceptionsuperr __init__)selfZ method_name exception) __class__ '/usr/lib64/python3.6/unittest/loader.pyr
s z_FailedTest.__init__c s* | j krtt j|S fdd}|S )Nc s
j d S )N)r r )r r r testFailure! s z,_FailedTest.__getattr__..testFailure)_testMethodNamer r __getattr__)r namer )r
)r r r s
z_FailedTest.__getattr__)__name__
__module__ __qualname__r r
r
__classcell__r r )r
r r s r c C s" d| t j f }t| t|||S )Nz#Failed to import test module: %s
%s) traceback
format_exc_make_failed_test ImportError)r
suiteClassmessager r r _make_failed_import_test& s r c C s dt j f }t| |||S )Nz Failed to call load_tests:
%s)r r r )r r r r r r r _make_failed_load_tests+ s r c C s t | |}||f|fS )N)r )
methodnamer r r testr r r r 0 s
r c C s< t jt|dd }| |i}tdt jf|}||| fS )Nc S s d S )Nr )r r r r testSkipped5 s z'_make_skipped_test..testSkippedZ
ModuleSkipped)r skipstrtypeTestCase)r r r r" ZattrsZ TestClassr r r _make_skipped_test4 s r' c C s* | j jdr| d d S tjj| d S )Nz $py.class r i)lowerendswithospathsplitext)r, r r r _jython_aware_splitext< s r. c s e Zd ZdZeejZej Z
dZ fddZ dd Z
dddd Zd d
d Zd!d d
Zdd Zd"ddZdd Zdd Zdd Zdd Zd#d d Zd$d d Z ZS )%
TestLoaderr! Nc s t t| j g | _t | _d S )N)r r/ r
errorsset_loading_packages)r )r
r r r
L s zTestLoader.__init__c C sH t |tjrtd| j|}| r4t|dr4dg}| jt||}|S )NzYTest cases should not be derived from TestSuite. Maybe you meant to derive from TestCase?ZrunTest)
issubclassr TestSuite TypeErrorgetTestCaseNameshasattrr map)r
testCaseClassZ
testCaseNamesZ loaded_suiter r r loadTestsFromTestCaseS s
z TestLoader.loadTestsFromTestCase)patternc
O s2 t |dksd|kr,tjdt |jdd t |dkrRt |d }tdj|t |dkrxt|d }tdj|g }x@t|D ]4}t ||}t
|t rt |t
jr|j| j| qW t |dd } | j|}| d k r.y | | ||S tk
r, }
z$t|j|
| j\} } | jj| | S d }
~
X nX |S )Nr Zuse_load_testsz(use_load_tests is deprecated and ignoredr zCloadTestsFromModule() takes 1 positional argument but {} were givenz=loadTestsFromModule() got an unexpected keyword argument '{}'
load_tests)lenwarningswarnDeprecationWarningpopr5 formatsorteddirgetattr
isinstancer% r3 r r& appendr: r Exceptionr r r0 )
r moduler; argsZkwsZ complainttestsr objr< e
error_case
error_messager r r loadTestsFromModulea s4
z TestLoader.loadTestsFromModulec C s> |j d}d\}}|d kr|d d }xb|rydj|}t|}P W q( tk
r |j }t|| j\}}|s| jj| |S Y q(X q(W |dd }|} x|D ]}
y| t | |
} } W q t
k
r2 } z\t | dd d k r|d k r| jj| |S t |
| | jdt j
f \}}| jj| |S W Y d d } ~ X qX qW t| tjrP| j| S t| trtt| tjrt| j| S t| tjrt| trt| tjr|d }| |}
tt |
|tjs| j|
gS nt| tjr| S t| r.| }t|tjr|S t|tjr | j|gS td| |f n td| d S ) N.r __path__z Failed to access attribute:
%sz"calling %s returned %s, not a testz$don't know how to make test from: %s)NN) splitjoin
__import__r rA r r r0 rG rE AttributeErrorr r r rF types
ModuleTyperP r% r3 r r& r: FunctionTyper r4 callabler5 )r r rI partsrN rO Z
parts_copy module_nameZnext_attributerL partparentrM Zinstr! r r r loadTestsFromName sl
z TestLoader.loadTestsFromNamec s fdd|D }j |S )Nc s g | ]}j | qS r )r` ).0r )rI r r r
s z1TestLoader.loadTestsFromNames..)r )r namesrI Zsuitesr )rI r r loadTestsFromNames s z TestLoader.loadTestsFromNamesc C s@ || j fdd}tt|t|}| jr<|jtj| jd |S )Nc S s | j |ott|| S )N)
startswithr[ rE )Zattrnamer9 prefixr r r isTestMethod s
z1TestLoader.getTestCaseNames..isTestMethod)key)testMethodPrefixlistfilterrD sortTestMethodsUsingsort functools
cmp_to_key)r r9 rg Z testFnNamesr r r r6 s zTestLoader.getTestCaseNamestest*.pyc C sR d}|d kr | j d k r | j }n|d kr.d}|}tjj|}|tjkrRtjjd| || _ d}d}g }tjjtjj|rtjj|}||krtjjtjj|d }nxy t | W n t
k
r d}Y nRX tj | }|j dd } ytjjtjj
|j}W n tk
r y
|j}
W n tk
r8 d }
Y nX |
r|
jd kr|
jd k rd}x|jD ]T} | r| j| rqd| j |jjdtjjd | _ |j| j| |dd qdW n*|jtjkrtdd ntdj |d Y nX |r |s| j | | _ tjj | n tjj | |r2t
d | |sHt | j||}| j |S )
NFTr z __init__.pyrQ ) namespacez2Can not use builtin modules as dotted module namesz$don't know how to discover from {!r}z%Start directory is not importable: %r)!_top_level_dirr+ r, abspathsysinsertisdirisfilerU rV r modulesrT dirname__file__rW __spec__loadersubmodule_search_locationsrR re r replacesepextend _find_testsbuiltin_module_namesr5 rB _get_directory_containing_moduleremoverj r ) r start_dirr; Z
top_level_dirZset_implicit_topZis_not_importable is_namespacerK Z
the_moduleZtop_partspecr, r r r discover sv
zTestLoader.discoverc C sR t j| }tjj|j}tjj|j jdrBtjj tjj |S tjj |S d S )Nz __init__.py)
rt rx r+ r, rs rz basenamer) re ry )r r] rI full_pathr r r r X s
z+TestLoader._get_directory_containing_modulec C sB || j krdS ttjj|}tjj|| j }|jtjjd}|S )NrQ )rr r. r+ r, normpathrelpathr~ r )r r, Z_relpathr r r r _get_name_from_pathd s
z TestLoader._get_name_from_pathc C s t | tj| S )N)rV rt rx )r r r r r _get_module_from_namep s z TestLoader._get_module_from_namec C s
t ||S )N)r )r r, r r; r r r _match_patht s zTestLoader._match_pathFc
c s | j |}|dkrD|| jkrD| j|||\}}|d k r<|V |sDd S ttj|}x||D ]t}tjj||} | j| ||\}}|d k r|V |rX| j | }| jj| z| j | ||E d H W d | jj
| X qXW d S )NrQ ) r r2 _find_test_pathrC r+ listdirr, rU addr discard)
r r r; rq r rK Zshould_recursepathsr, r r r r r x s*
zTestLoader._find_testsc C sP t jj|}t jj|rFtj|s(dS | j|||s:d S | j|}y| j|}W nX t j
k
r } zt ||| j dfS d }~X n t
|| j \}} | jj| |dfS t jjt|d|}
tt jj|
} tt jj|} | j | j kr0t jj| }
tt jj|}t jj|}d}t|||
|f | j||ddfS nt jj|rH| rzt jjt jj|d rzd
S d }d }| j|}y| j|}W nZ t j
k
r } zt ||| j dfS d }~X nz t
|| j \}} | jj| |dfS t|dd }| jj| z(| j||d}|d k r0|dfS |dfS | jj | X nd S d S ) NFrz zW%r module incorrectly imported from %r. Expected %r. Is this module globally installed?)r; z __init__.pyr< T)NF)NF)NF)NF) r+ r, r rw VALID_MODULE_NAMEmatchr r r r ZSkipTestr' r r r0 rG rs rE r. realpathr) ry r rP rv rU r2 r r )r r r; rq r r rI rM rN rO Zmod_filer Zfullpath_noextZ
module_dirZmod_nameZ expected_dirmsgr< rK packager r r r sl
zTestLoader._find_test_path)N)N)rp N)F)F)r r r ri staticmethodr
three_way_cmprl r r4 r rr r
r: rP r` rd r6 r r r r r r r r r r )r
r r/ B s"
(
N
n
"r/ c C s t }||_| |_|r ||_|S )N)r/ rl ri r )rf sortUsingr r| r r r _makeLoader s r c C s t ||j| S )N)r r6 )r9 rf r r r r r6 s r6 r! c C s t |||j| S )N)r r: )r9 rf r r r r r makeSuite s r c C s t |||j| S )N)r rP )rI rf r r r r r
findTestCases s r )N) r+ rert r rX rn r> r r r r Z
__unittestcompile
IGNORECASEr r& r r r r r' r. objectr/ ZdefaultTestLoaderr r r6 r4 r r r r r r s6 '