U
e5dX @ s d dl Z d dlZd dlZd dlZd dlZd dlZd dlZd dlmZmZ ddl m
Z
m Z m Z dZ
edejZG dd de
jZd d
Zd d Zd
d Zdd Zdd ZG dd deZe Zd ddZe j dfddZ de j e j fddZ de j e j fd d Z dS ) N)fnmatch fnmatchcase )casesuiteutilTz[_a-z]\w*\.py$c s, e Zd ZdZ fddZ fddZ ZS ) _FailedTestNc s || _ tt| | d S N)
_exceptionsuperr __init__)selfZ method_name exception __class__ '/usr/lib64/python3.8/unittest/loader.pyr s z_FailedTest.__init__c s* | j krtt |S fdd}|S )Nc s
j d S r )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 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 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 r )r )
methodnamer r r! testr r r r 0 s
r c C s< t t|dd }| |i}tdt jf|}||| fS )Nc S s d S r r r r r r testSkipped5 s z'_make_skipped_test..testSkippedZ
ModuleSkipped)r skipstrtypeTestCase)r$ r r r& attrsZ TestClassr r r _make_skipped_test4 s
r, c C s* | dr| d d S tj| d S )Nz $py.classir )lowerendswithospathsplitext)r0 r r r _jython_aware_splitext< s r2 c s e Zd ZdZeejZdZe j
Z dZ fddZ
dd Zdddd 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| g | _t | _d S r )r r3 r errorsset_loading_packagesr r r r r M s zTestLoader.__init__c C sF t |tjrtd| |}|s2t|dr2dg}| t||}|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 loadTestsFromTestCaseT s
z TestLoader.loadTestsFromTestCasepatternc
O s: t |dksd|kr,tdt |dd t |dkrRt |d }td|t |dkrxt|d }td|g }t|D ]4}t ||}t
|t rt |t
jr|| | qt |dd } | |}| d k r6z| | ||W S tk
r4 }
z,t|j|
| j\} } | j| | W Y S d }
~
X Y 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)lenwarningswarnDeprecationWarningpopr9 formatsorteddirgetattr
isinstancer) r7 r r* appendr> r Exceptionr# r r4 )
r
moduler@ argsZkwsZ complainttestsr objrA e
error_case
error_messager r r loadTestsFromModuleb s<
z TestLoader.loadTestsFromModulec C sX | d}d\}}|d kr|d d }|rzd|}t|}W qW q& tk
r | }t|| j\}}|s| j| | Y S Y q&X q&|dd }|} |D ]}
z| t | |
} } W q t
k
rN } zvt | dd d k r|d k r| j| | W Y D S t |
| | jdt
f \}}| j| | W Y
S W 5 d } ~ X Y qX qt| tjrj| | S t| trt| tjr| | S t| tjrt| trt| tjr|d }| |}
tt |
|tjs| |
gS nt| tjr| S t| rH| }t|tjr |S t|tjr6| |gS td| |f n td| d S ) N.)NNr __path__z Failed to access attribute:
%sz"calling %s returned %s, not a testz$don't know how to make test from: %s) splitjoin
__import__r rF r" r r4 rL rJ AttributeErrorr r r rK types
ModuleTyperU r) r7 r r* r> FunctionTyper r8 callabler9 )r
r rN partsrS rT Z
parts_copy module_nameZnext_attributerQ partparentrR instr% r r r loadTestsFromName s
$
z TestLoader.loadTestsFromNamec s fdd|D } |S )Nc s g | ]} | qS r )rf ).0r rN r
r r
s z1TestLoader.loadTestsFromNames..)r )r
namesrN Zsuitesr rh r loadTestsFromNames s z TestLoader.loadTestsFromNamesc s> fdd}t t|t} jr:|jt jd |S )Nc sZ | jsdS t| }t|s&dS djj| f jd kpXt fddjD S )NFz%s.%s.%sc 3 s | ]}t |V qd S r )r )rg r@ ZfullNamer r s zKTestLoader.getTestCaseNames..shouldIncludeMethod..)
startswithtestMethodPrefixrJ r` r r testNamePatternsany)attrnameZtestFuncr
r= rl r shouldIncludeMethod s
z8TestLoader.getTestCaseNames..shouldIncludeMethod)key)listfilterrI sortTestMethodsUsingsort functools
cmp_to_key)r
r= rt Z testFnNamesr rs r r: s
zTestLoader.getTestCaseNamestest*.pyc C sJ d}|d kr | j d k r | j }n|d kr.d}|}tj|}|tjkrRtjd| || _ d}d}g }tjtj|rtj|}||krtjtj|d }npz t | W n t
k
r d}Y nJX tj | }| dd } ztjtj
|j}W n tk
r z
|j}
W n tk
r8 d }
Y nX |
r|
jd kr|
jd k rd}|jD ]P} |s|| |s|qb| |jdtjjd | _ || j| |dd qbn*|jtjkrtdd ntd |d Y nX |r|s | | | _ tj | n tj | |r*t
d | |s@t | ||}| |S )
NFTr __init__.pyrV ) 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/ r0 abspathsysinsertisdirisfilerZ r[ r modulesrY dirname__file__r\ __spec__loadersubmodule_search_locationsrW rn r replacesepextend _find_testsbuiltin_module_namesr9 rG _get_directory_containing_moduleremoverv r ) r
start_dirr@ Z
top_level_dirZset_implicit_topZis_not_importable is_namespacerP Z
the_moduleZtop_partspecr0 r r r discover s
zTestLoader.discoverc C sR t j| }tj|j}tj| drBtj tj |S tj |S d S )Nr} )
r r r/ r0 r r basenamer- rn r )r
rb rN full_pathr r r r ` s
z+TestLoader._get_directory_containing_modulec C sB || j krdS ttj|}tj|| j }|tjjd}|S NrV )r r2 r/ r0 normpathrelpathr r )r
r0 Z_relpathr r r r _get_name_from_pathl s
z TestLoader._get_name_from_pathc C s t | tj| S r )r[ r r )r
r r r r _get_module_from_namex s z TestLoader._get_module_from_namec C s
t ||S r )r )r
r0 r r@ r r r _match_path| s zTestLoader._match_pathFc
c s | |}|dkrD|| jkrD| |||\}}|d k r<|V |sDd S tt|}|D ]t}tj||} | | ||\}}|d k r|V |rV| | }| j| z|
| ||E d H W 5 | j | X qVd S r ) r r6 _find_test_pathrH r/ listdirr0 rZ adddiscardr )
r
r r@ r~ r rP Zshould_recursepathsr0 r r r r r s6
zTestLoader._find_testsc
C st t j|}t j|rVt|s(dS | |||s:dS | |}z| |}W nh t j
k
r } zt ||| j df W Y S d }~X Y n t
|| j \}} | j| |df Y S X t jt|d|}
tt j|
} tt j|} | | kr@t j| }
tt j|}t j|}d}t|||
|f | j||ddfS nt j|rl|st jt j|dsdS d }d }| |}z| |}W nj t j
k
r } zt ||| j df W Y S d }~X Y n t
|| j \}} | j| |df Y S X t|dd }| j| z0| j||d}|d k rP|dfW S |dfW S | j | X ndS d S ) N)NFFr zW%r module incorrectly imported from %r. Expected %r. Is this module globally installed?r? r} rA T) r/ r0 r r VALID_MODULE_NAMEmatchr r r r ZSkipTestr, r r" r4 rL r rJ r2 realpathr- r r rU r rZ r6 r r )r
r r@ r~ r r rN rR rS rT Zmod_filer Zfullpath_noextZ
module_dirZmod_nameZ expected_dirmsgrA rP packager r r r s|
&
&
zTestLoader._find_test_path)N)N)r| N)F)F)r r r ro staticmethodr
three_way_cmprx rp r r8 r r r r> rU rf rk r: r r r r r r r r r r r r r3 B s$
(
N
n
"r3 c C s& t }||_| |_||_|r"||_|S r )r3 rx ro rp r )prefix sortUsingr rp r r r r _makeLoader s r c C s t |||d| S )N)rp )r r: )r= r r rp r r r r: s r: r% c C s t |||| S r )r r> )r= r r r r r r makeSuite s r c C s t |||| S r )r rU )rN r r r r r r
findTestCases s r )NN)!r/ rer r r] rz rC r r r r r Z
__unittestcompile
IGNORECASEr r* r r" r# r r, r2 objectr3 ZdefaultTestLoaderr r r: r8 r r r r r r s: /