YfV @ sp d Z d d l Z d d l Z d d l Z d d l Z d d l Z d d l Z d d l Z d d l m Z d d l m
Z
m Z m Z d Z
e j d e j Z Gd d d e
j Z d
d Z d d
Z d d Z d d Z d d Z Gd d d e Z e Z d d d Z e j d d Z d e j e j d d Z d e j e j d d Z d S) zLoading unittests. N)fnmatch )casesuiteutilTz[_a-z]\w*\.py$c s: e Z d Z d Z f d d Z f d d Z S) _FailedTestNc s# | | _ t t | j | d S)N)
_exceptionsuperr __init__)selfZ method_name exception) __class__ 4/opt/alt/python35/lib64/python3.5/unittest/loader.pyr
s z_FailedTest.__init__c s; | j k r% t t j | S f d d } | 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 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& d t 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 | f S)N)r )
methodnamer r r testr r r r 0 s r c C sX t j t | d d } | | i } t d t j f | } | | | f S)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 s7 | j j d r# | d d St j j | d S)Nz $py.class r i)lowerendswithospathsplitext)r+ r r r _jython_aware_splitext< s r- c s e Z d Z d Z d Z e e j Z e j
Z d Z f d d Z
d d Z d d d d
Z d d d Z d d
d Z d d Z d d d d Z d d Z d d Z d d Z d d Z d d d Z d d d Z S)!
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 | t j r t d | j | } | rL t | d rL d g } | j t | | } | 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.loadTestsFromTestCasepatternc
O s t | d k s d | k r> t j d t | j d d t | d k ru t | d } t d j | t | d k r t | d } t d j | g } xZ t | D]L } t | | } t
| t r t | t
j r | j | j | q Wt | d d } | j | } | d k ry | | | | SWnW t k
r}
z7 t | j |
| j \ } } | j j | | SWYd d }
~
Xn X| 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)lenwarningswarnDeprecationWarningpopr4 formatsorteddirgetattr
isinstancer$ r2 r r% appendr9 r Exceptionr r r/ )
r moduler: argsZkwsZ complainttestsr objr; e
error_case
error_messager r r loadTestsFromModulea s4 ! z TestLoader.loadTestsFromModulec C s | j d } d \ } } | d k r | d d } x} | r y d j | } t | } PWq: t k
r | j } t | | j \ } } | s | j j | | SYq: Xq: W| d d } | } x | D] }
y | t | |
} } Wq t
k
r} z| t | d d d k rA| d k rA| j j | | St |
| | j d t j
f \ } } | j j | | SWYd d } ~ Xq Xq Wt | t j r| j | St | t rt | t j r| j | St | t j r[t | t r[t | t j r[| d } | | }
t t |
| t j sq| j |
g Sn t | t j rq| St | r| } t | t j r| St | t j r| j | g St d | | f n t d | d S)
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 r@ r r r/ rF rD AttributeErrorr r r rE types
ModuleTyperO r$ r2 r r% r9 FunctionTyper r3 callabler4 )r r rH partsrM rN Z
parts_copy module_nameZnext_attributerK partparentrL Zinstr r r r loadTestsFromName sl
!
z TestLoader.loadTestsFromNamec s) f d d | D } j | S)zReturn a suite of all test cases found using the given sequence
of string specifiers. See 'loadTestsFromName()'.
c s" g | ] } j | q Sr )r_ ).0r )rH r r r
s z1TestLoader.loadTestsFromNames..)r )r namesrH Zsuitesr )rH r r loadTestsFromNames s z TestLoader.loadTestsFromNamesc C sY | | j d d } t t | t | } | j rU | j d t j | j | S)zLReturn a sorted sequence of method names found within testCaseClass
c S s" | j | o! t t | | S)N)
startswithrZ rD )Zattrnamer8 prefixr r r isTestMethod s z1TestLoader.getTestCaseNames..isTestMethodkey)testMethodPrefixlistfilterrC sortTestMethodsUsingsort functools
cmp_to_key)r r8 rf Z testFnNamesr r r r5 s zTestLoader.getTestCaseNamesztest*.pyc C s d } | d k r- | j d k r- | j } n | d k rE d } | } t j j | } | t j k ry t j j d | | | _ d } d } g } t j j t j j | r t j j | } | | k rt j j t j j | d } ny t | Wn t
k
r d } YnXt j | } | j d d } y% t j j t j j
| j } Wnt k
ry
| j }
Wn t k
rd }
Yn X|
r@|
j d k r@|
j d k r|d } x | j D]j } | r| j | rq| j | j j d t j j d | _ | j | j | | d d qWn< | j t j k rdt d d n t d j | d Yn X| r| s| j | | _ t j j | n t j j | | rt
d
| | st | 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__.pyrP 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+ abspathsysinsertisdirisfilerT rU r modulesrS dirname__file__rV __spec__loadersubmodule_search_locationsrQ rd r replacesepextend _find_testsbuiltin_module_namesr4 rA _get_directory_containing_moduleremoveri r ) r start_dirr: Z
top_level_dirZset_implicit_topZis_not_importable is_namespacerJ Z
the_moduleZtop_partspecr+ r r r discover sv %
zTestLoader.discoverc C ss t j | } t j j | j } t j j | j j d r_ t j j t j j | St j j | Sd S)Nz __init__.py)
rr rv r* r+ rq rx basenamer( rd rw )r r\ rH full_pathr r r r X s
! z+TestLoader._get_directory_containing_modulec C s | | j k r d St t j j | } t j j | | j } t j j | sb t d | j d s~ t d | j t j j
d } | S)NrP z Path must be within the projectz..) rp r- r* r+ normpathrelpathisabsAssertionErrorrd r| r} )r r+ Z_relpathr r r r _get_name_from_pathd s z TestLoader._get_name_from_pathc C s t | t j | S)N)rU rr rv )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 | } | d k r` | | j k r` | j | | | \ } } | d k rV | V| s` d St t j | } x | D] } t j j | | } | j | | | \ } } | d k r | V| r| | j | } | j j | z | j | | | Ed HWd | j j
| Xq| Wd S)z/Used by discovery. Yields test suites it loads.rP N) r r1 _find_test_pathrB r* listdirr+ rT addr discard)
r r r: ro r rJ Zshould_recursepathsr+ r r r r r x s*
zTestLoader._find_testsc C s0 t j j | } t j j | rt j | s7 d S| j | | | sP d
S| j | } y | j | } Wnz t j
k
r } z t | | | j d f SWYd d } ~ Xq,t
| | j \ } } | j j | | d f SYq,Xt j j t | d | }
t t j j |
} t t j j | } | j | j k rt j j | }
t t j j | } t j j | } d } t | | |
| f | j | d | d f Sn`t j j | r(| r t j j t j j | d r d Sd } d } | j | } y | j | } Wnz t j
k
r|} z t | | | j d f SWYd d } ~ Xq,t
| | j \ } } | j j | | d f SYq,Xt | d d } | j j | z9 | j | d | } | d k r| d f S| d f SWd | j j | Xn d Sd S)
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).
NFrx 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 ru VALID_MODULE_NAMEmatchr r r r ZSkipTestr& r r r/ rF rq rD r- realpathr( rw r rO rt rT r1 r r )r r r: ro r r rH rL rM rN Zmod_filer Zfullpath_noextZ
module_dirZmod_nameZ expected_dirmsgr; rJ packager r r r sl + "+
zTestLoader._find_test_path)r r r __doc__rh staticmethodr
three_way_cmprk r r3 r rp r
r9 rO r_ rc r5 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. rk rh r )re sortUsingr rz r r r _makeLoader s r c C s t | | j | S)N)r r5 )r8 re r r r r r5 s r5 r c C s t | | | j | S)N)r r9 )r8 re r r r r r makeSuite s r c C s t | | | j | S)N)r rO )rH re r r r r r
findTestCases s r )!r r* rerr r rW rm r= r r r r Z
__unittestcompile
IGNORECASEr r% r r r r r& r- objectr. ZdefaultTestLoaderr r r5 r3 r r r r r r s4