3
\V @ s d Z ddlZddlZddlZddlZddlZddlZddlZddlmZ ddl m
Z
m Z m Z dZ
ejdejZG dd d e
jZd
d Zd d
Zdd Zdd Zdd ZG dd deZe Zd ddZe j fddZ de j e j fdd Z de j e j fd d Z dS ) zLoading unittests. 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 ZdZdZeejZe j
Z dZ fddZ
dd Zddd d
Zd!d d Zd"d
dZdd Zd#ddZdd Zdd Zdd Zdd Zd$d d Zd%d d Z ZS )&
TestLoaderz
This class is responsible for loading tests according to various criteria
and returning them wrapped in a TestSuite
r! 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 )z;Return a suite of all test cases contained in testCaseClasszYTest 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|}| dk r.y | | ||S tk
r, }
z$t|j|
| j\} } | jj| | S d}
~
X nX |S ) z>Return a suite of all test cases contained in the given moduler Zuse_load_testsz(use_load_tests is deprecated and ignoredNr 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\}}|dkr|dd }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 | dddk r|dk r| jj| |S t |
| | jdt j
f \}}| jj| |S W Y dd} ~ 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| dS )
aS Return a suite of all test cases given a string specifier.
The name may resolve either to a module, a test case class, a
test method within a test case class, or a callable object which
returns a TestCase or TestSuite instance.
The method optionally resolves the names relative to a given module.
.Nr __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 )zReturn a suite of all test cases found using the given sequence
of string specifiers. See 'loadTestsFromName()'.
c 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 )zLReturn a sorted sequence of method names found within testCaseClass
c 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}|dkr | j dk r | j }n|dkr.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|
jdkr|
jdk rd}x|jD ]T} | r| j| rqd| j |jjdtjjd | _ |j| j| |dd qdW n*|jtjkrtddntd j |dY nX |r |s| j | | _ tjj | n tjj | |r2t
d
| |sHt | j||}| j |S ) a% Find and return all test modules from the specified start
directory, recursing into subdirectories to find them and return all
tests found within them. Only test files that match the pattern will
be loaded. (Using shell style pattern matching.)
All test modules must be importable from the top level of the project.
If the start directory is not the top level directory then the top
level directory must be specified separately.
If a test package name (directory with '__init__.py') matches the
pattern then the package will be checked for a 'load_tests' function. If
this exists then it will be called with (loader, tests, pattern) unless
the package has already had load_tests called from the same discovery
invocation, in which case the package module object is not scanned for
tests - this ensures that when a package uses discover to further
discover child tests that infinite recursion does not happen.
If load_tests exists then discovery does *not* recurse into the package,
load_tests is responsible for loading all tests in the package.
The pattern is deliberately not stored as a loader attribute so that
packages can continue discovery themselves. top_level_dir is stored so
load_tests does not need to pass this argument in to loader.discover().
Paths are sorted before being imported to ensure reproducible execution
order even on filesystems with non-alphabetical ordering like ext3/4.
FNTr 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|||\}}|dk r<|V |sDdS ttj|}x||D ]t}tjj||} | j| ||\}}|dk r|V |rX| j | }| jj| z| j | ||E dH W d| jj
| X qXW dS )z/Used by discovery. Yields test suites it loads.rQ N) 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(d S | 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}|dk r0|dfS |dfS | jj | X nd S dS )
zUsed by discovery.
Loads tests from a single file, or a directories' __init__.py when
passed the directory.
Returns a tuple (None_or_tests_from_file, should_recurse).
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 __doc__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 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 s8 '