cb2dZdZddlmZddlZddlZddlmZddl Z ddl Z ddl m Z ddl mZddlmZmZmZmZmZmZmZmZddlmZddlZddlZdd lmZmZddlZ dd l!m"Z"dd l#m$Z$m%Z%e&e d ds ddl'Zdd l'm(Z(n ddlZdd lm(Z(gdZ+dZ,edeZ-dZ.eddedefdZ/dZ0dZ1e,dddfdZ2dZ3edLdeedefdZ4e4Z5d edefd!Z6Gd"d#e7Z8ejrZ9Gd$d%e:Z;Gd&d'e<Z=ej|Z?Gd(d)e(Z@d*eeAd+fdeeeeBeeAfeffd,ZCGd-d.e<ZDGd/d0eEZFd1ZGGd2d3ejZHed4eHjeH_IGd5d6ejrZJGd7d8ejZKGd9d:eHe;ZLGd<d=ZMd*edd>fd?ZNd@eeOdAeOddfdBZPGdCdDZQdEeOdFedefdGZRdHeOdefdIZSdJeOdefdKZTy#e)$r e)de*zwxYw)MaRun tests. This will find all modules whose name match a given prefix in the test directory, and run them. Various command line options provide additional facilities. Command line options: -v verbose -- run tests in verbose mode with output to stdout -q quiet -- don't print anything except if a test fails -t testdir -- directory where the tests will be found -x exclude -- add a test to exclude -p profile -- profiled execution -d dbc -- enable design-by-contract -m match -- only run test matching the tag pattern which follow If no non-option arguments are present, prefixes used are 'test', 'regrtest', 'smoketest' and 'unittest'. zrestructuredtext en)contextmanagerN)rmtree)isgeneratorfunction)AnyIteratorUnionOptionalCallableDictListTuple)NoReturn)class_deprecatedcallable_deprecatedwraps)cached classproperty __package__)SkipTestz.You have to install python-unittest2 to use %s) unittest_main find_tests nocoverage pause_trace)testregrtest smoketestunittestfunc validationz*[lgc 0.63] use inspect.isgeneratorfunctionz.proxybs!,,.J!))=J)H  2!4262AG2:T:'1H$zH$2:T:'1H$zH$sF6B$A6 A& B$& A33B$6B!8 B B! BB!!B$kargsc.tj}tjd}|t_ |i| t |d|t_S#|t_wxYw# t |d|t_w#|t_wxYwxYwr(r+)r%r5r0r1r"s r3r4zwith_tempdir..proxyss{!,,.J!))=J)H  2//2:T:'1H$zH$2:T:'1H$zH$s/A) A A&)B+ B8 B BB)rrrrstrrr"r4s` r3 with_tempdirr9\sv8$ 2 2 2xRU?V9W0X 2  x 2 2s 2s 2  2 c.tfd}|S)zEA decorator moving the enclosed function inside the tempfile.tempfdirctj}tjtj |i|tj|S#tj|wxYwN)osgetcwdchdirr,r/)r%r5old_cwdr"s r3r4zin_tempdir..proxysK))+ !!" T+U+ HHW BHHW s AA.rr8s` r3 in_tempdirrBs" 8_ Lr:cPtt|}|j|_|S)zMA decorator run the enclosed function inside a tmpdir removed after execution)r9rB__name__r8s r3within_tempdirrEs# H- .E&&EN Lr:z.pyTc(g}tj|D]g}|r|j|s|D]L}|j|s|r |j|r|dt | }||vs<|j |Ni|j |S)z7 Return a list of all applicable test modules. N)r>listdirendswith startswithlenappendsort)testdirprefixessuffixexcludes remove_suffixtestsnamer)s r3rrs E 7#+v." +??6*$v)>#Ns6{l38+ T*  ++ JJL Lr:c |j}|j|jz}t|dk(r|dj y d}t dt dj t|Dcgc]\}\}}d|d|c}}}t dt  td}|jjd k(r t y t|}|||d}|j cc}}}w#ttf$rt d |Y'wxYw#ttf$r t Yy wxYw) z?starts an interactive shell so that the user can inspect errorsr!rzChoose a test to debug:  z : zType 'exit' (or ^D) to quitzEnter a test name: exitzERROR: invalid test number N) debuggers error_descrs fail_descrsrKstartprintjoin enumerateinputstriplowerint ValueError IndexErrorEOFErrorKeyboardInterrupt) resultrYdescrs testindexi_descrtodebugdebuggers r3start_interactive_moderps9  I  6#5#5 5F 9~! I + , $))9U[K\]]-!ZaAu5]^ _ / 0 G  56==?((*f4G)$'L #,VI->q-A#B!('^' 3LJKL/0  s<5D -6D6%D;D6D30D62D33D66EEtracec#ZKtj}tj| d|#t|st |dr |j }tj|y#|#t|st |dr |j }tj|wxYww)z.not_covereds* ] )(( ) ) )s')rwr{r)rr}s` r3rrsEt[! DN)3)#)#) !K r:ceZdZdZdZdZdZy) _DebugResultzCUsed by the TestSuite to hold previous class when running in debug.NF)rD __module__ __qualname____doc___previousTestClass_moduleSetUpFailed shouldStoprFr:r3rrsIJr:rceZdZdZy)keywordsz5Keyword args (**kwargs) support for generative tests.NrDrrrrFr:r3rrs?r:rceZdZdZdZy)starargsz0Variable arguments (*args) for generative tests.c.tj||Sr=)tuple__new__)clsr%s r3rzstarargs.__new__ s}}S$''r:N)rDrrrrrFr:r3rrs :(r:rceZdZdZy)InnerTestSkippedzraised when a test is skippedNrrFr:r3rrs'r:rparams.c*g}d}i}d}|D]}t|tr#|}|r td|dz}|t|z }6t|tr|}|dzr td|dz}^|dzs|dzr td|j |||fS)NrFrzfound starargs after keywords !z got multiple keywords parametersz%found parameters after kwargs or args) isinstancer TypeErrorlistrrL)rr%varargsr&flagsparams r3parse_generative_argsrs DGF E eX &G ABB QJE DM !D x (Fqy BCC QJE QY%!)CD D KK " <r:ceZdZdZy) InnerTestc@tj||}||_|Sr=)rrrT)rrTdatainstances r3rzInnerTest.__new__/s==d+ r:N)rDrrrrFr:r3rr.sr:rc\eZdZdZdededdffd ZdedefdZd edefd Z d fd Z xZ S) Tagsz(A set of tag able validate an expressiontagsr&r#Nc|jdd|_|rtd|jzt |dk(rt |dt s|d}tt|'|y)NinheritTz4%s are an invalid keyword argument for this functionr!r) poprrkeysrKrr7superr__init__)selfrr& __class__s r3rz Tags.__init__8sczz)T2 RU[U`U`Ubbc c t9>*T!Wc":7D dD"4(r:keyc ||vSr=rF)rrs r3 __getitem__zTags.__getitem__As d{r:expct|i|Sr=)eval)rrs r3matchz Tags.matchDsCT""r:c6ttt| |Sr=)rr__or__)rotherrs r3rz Tags.__or__MsU4-e455r:)rrr#r) rDrrrr7rrboolrrr __classcell__rs@r3rr5sP2)c)S)T)st###66r:rcfd}|S)Ncdtjdjztd|i|S)NzPlease use %s instead.r)warningswarnrDDeprecationWarning)r%r& original_funcs r3deprecated_funcz#_deprecate..deprecated_funcSs1 /-2H2HHK]_`ad-f--r:rF)rrs` r3 _deprecaterRs. r:c eZdZdZdZeZddeddffd Ze e defdZ e dedefdZ d Zdeeffd Zd ed ed ededef dZdefdZddZddZ dd edededefdZ dd eded eeeeeedfdeedef dZddedefdZ e!jDdk\re#jHjJZ&xZ'Se#jHjLZ%xZ'S)TestCasez;A unittest.TestCase extension with some additional methods.N methodNamer#ctt| |tj|_|j |_d|_d|_ yr=) rrrrtexc_info_TestCase__exc_info_testMethodName_TestCase__testMethodName_current_test_descr _options_)rrrs r3rzTestCase.__init__`s: h&z2,, $ 4 4#' r:ctj|j}tjtj tj |jdS)zphelper attribute holding the standard test's data directory NOTE: this is a logilab's standard r)rtmodulesrospr^dirnameabspath__file__)rmods r3datadirzTestCase.datadirgs=kk#..)xx CKK $=>GGr:fnamecBtj|jg|S)z&joins the object's datadir and `fname`)rr^r)rrs r3datapathzTestCase.datapathtsxx ,e,,r:c||_y)zsets the current test's description. This can be useful for generative tests because it allows to specify a description per yield N)r)rrms r3set_descriptionzTestCase.set_descriptionys $) r:cX|j |jStt|S)z`override default unittest shortDescription to handle correctly generative tests )rrrshortDescription)rrs r3rzTestCase.shortDescriptions-  # # /++ +Xt577r:rhrr%r&cp ||i|y#ttf$rtj$r^}t |dr|j |t |n,tjdtd|j|Yd}~yd}~wt$r#|j||jYywxYw)NaddSkip4TestResult has no addSkip method, skips not reportedrFT)rg SystemExitrrrwrr7rrRuntimeWarning addSuccess ExceptionaddErrorr)rrhrr%r&es r3 quiet_runzTestCase.quiet_runs  $ !& !":.     vy)tSV, JN\]!!$'  OOD$//"3 4 s "B5AB,B54B5c.t||jS)zreturn the test method)getattrrrs r3_get_test_methodzTestCase._get_test_methodstT1122r:c0t|j||S)z>return the option value or default if the option is not define)rr)roptiondefaults r3optvalzTestCase.optvalst~~vw77r:c||j}||_|j}t|jdds t|ddr t|jddxs t|dd}t |dr|j ||n,tjdtd|j||j|y|r ||sy|j| |j||js |j|yt|}|r|j!|||n|j#||}|d k(}|j||j$s |j|y|sr|j||j|y#|j|wxYw#|j|wxYw) zrewrite TestCase.__call__ to support generative tests This is mostly a copy/paste from unittest.py (i.e same variable names, same logic, except for the generative tests part) N__unittest_skip__F__unittest_skip_why__rrrr)defaultTestResultrrrrrwrrrrrstopTest startTestrsetUpr_proceed_generative_proceedtearDown) rrh runconditionoptions testMethodskip_why generativestatussuccesss r3__call__zTestCase.__call__s >++-F **, 4>>#6 >' +UC  &"4>>3JBOSZ 7T69-NN42MMNP^`a%%d+%   Z 8  ">>&$**5 OOD !-Z8J((\Jvz: A+>>&$--8 OOD ! '!!$' OOD !-%, OOD !s&A1F,/GAGG,F?Gtestfuncrc|xjdzc_d} |D]}|r#||dst|tr||s(t|ttfs|f}|d}t |dd\}}|xjdz c_|j ||||} | dk(r|j|d}nd}|js|S |S#|j$r&|j||jd}Y|St$r} |j|| Yd} ~ |Sd} ~ wt$r&|j||jd}Y|SwxYw)Nr!TF) skipgeneratorr)testsRunrrrrrrrrfailureException addFailurerrrrr) rrhrrrrrr%r&rrs r3rzTestCase._proceed_generativesb 1 "*  XU(S&vy9l6>R !&5$-8$YFay4VABZ@ f1$vtT6BQ;%%d+"G#G$$= <$$    dDOO$5 6G  $ NN4 # #    OOD$//"3 4G  s0B.CC C1EE D##.EErFc~|xsi} ||i|y#|j$r#|j||jYyt$rt$r}|j ||Yd}~yd}~wt $r}|j ||Yd}~yd}~wt$r#|j||jYywxYw)zproceed the actual test returns 0 on success, 1 on failure, 2 on error Note: addSuccess can't be called here because we have to wait for tearDown to be successfully executed to declare the test as successful r!Nrr) rrrrgrrrrr)rrhrr%r&rs r3rzTestCase._proceeds2  d %f %$$    dDOO$5 6     NN4 #  NN4 #  OOD$//"3 4 s-/B<B<A** B<6B  ,B<;B<msgc$|xsd}t|)z6mark a generative test as skipped for the reasonztest was skipped)r)rrs r3 innerSkipzTestCase.innerSkips''s##r:)r)runTestr=NNN)rFN)(rDrrrmaxDiffrrr7rrrr classmethodrrr rrr rrrrrrrr rcr r rrrrt version_inforrassertCountEqualassertItemsEqualrrs@r3rrZsEG 6D3t HH H-c-c--)8(3-88C3SW&3(381"hIM%%%-%=E% %V9;!% DJS 5945    B$S$H$  6!#,,==#,,==r:rz4assertItemsEqual is deprecated, use assertCountEqualceZdZdZy) SkippedSuitec&|jdy)$just there to trigger test executionz(doctest module has no DocTestSuite classN) skipped_testrs r3rzSkippedSuite.test0s DEr:N)rDrrrrFr:r3rr/sFr:rceZdZdZdZy) DocTestFindercx|jdd|_tjj|g|i|y)NskippedrF)rrdoctestrr)rr%r&s r3rzDocTestFinder.__init__6s2zz)R0 &&t=d=f=r:c~t|dd|jvrytjj ||||||S)zroverride default _get_test method to be able to skip tests according to skipped attribute's value rDrN)rrrr _get_test)robjrTmoduleglobs source_liness r3rzDocTestFinder._get_test:s= 3 B '4<< 7$$..tS$|\\r:N)rDrrrrrFr:r3rr5s >]r:rc*eZdZdZdZdZddZeZdZy)DocTestztrigger module doctest I don't know how to make unittest.main consider the DocTestSuite instance without this hack z4use stdlib doctest module with unittest API directlyrFNcH t|j}tj|j|}t j tj_tjj} |j|tjjtjj!|S#t$r t}YwxYw#tjjtjj!|wxYw)N)r) test_finder)rrr DocTestSuiterrtr DocTestCaserAttributeErrorrbuiltins__dict__copyrunclearupdate)rrhrrfindersuite old_builtinss r3rzDocTest.__call__Ns #"4<<8F((&IE69llG   3  ((--/  399V$    # # %    $ $\ 2 # NE #    # # %    $ $\ 2sAC ;C" CC"?D!cy)rNrFrs r3rz DocTest.testar:r ) rDrrr__deprecation_warning__rrr+rrFr:r3r"r"Es& UG3" C3r:r") metaclasscBeZdZdZdZdZd dZdZdZdZ d Z d Z y) MockConnectionz@fake DB-API 2.0 connexion AND cursor (i.e. cursor() return self)c.g|_g|_||_yr=)receivedstatesresults)rr:s r3rzMockConnection.__init__hs   r:c|S)zMock cursor methodrFrs r3cursorzMockConnection.cursorms r:Nc>|jj||fy)zMock execute methodN)r8rL)rqueryr%s r3executezMockConnection.executeqs eT]+r:c |jdS)zMock fetchone methodrr:rs r3fetchonezMockConnection.fetchoneus||Ar:c|jS)zMock fetchall methodrArs r3fetchallzMockConnection.fetchallys ||r:cd|jjdt|jfy)zMock commiy methodcommitNr9rLrKr8rs r3rFzMockConnection.commit}s" Hc$--&89:r:cd|jjdt|jfy)zMock rollback methodrollbackNrGrs r3rIzMockConnection.rollbacks" JDMM(:;>> option = mock_object(verbose=False, index=range(5)) >>> option.verbose False >>> option.index [0, 1, 2, 3, 4] rLrF)type)rs r3 mock_objectrOs $4F # %%r:pathschrootctt}}|D]y}tj||}tj|}|dk(r|j |E|j tj ||j |{|D]-}tj |rtj|/|D]}t|djy)aCreates directories and files found in . :param paths: list of relative paths to files or directories :param chroot: the root directory in which paths will be created >>> from os.path import isdir, isfile >>> isdir('/tmp/a') False >>> create_files(['a/b/foo.py', 'a/b/c/', 'a/b/c/d/e.py'], '/tmp') >>> isdir('/tmp/a') True >>> isdir('/tmp/a/b/c') True >>> isfile('/tmp/a/b/c/d/e.py') True >>> isfile('/tmp/a/b/foo.py') True rwN) setrr^basenameaddrisdirr>makedirsopenrK)rPrQdirsfilespathfilenamedirpathfilepaths r3 create_filesr`s&%%D xx%<<% r> HHTN HHS[[& ' IIdO !yy! KK !$ Xs!!#$r:ceZdZdZy) AttrObjectc :|jj|yr=)r)r-)rr&s r3rzAttrObject.__init__s V$r:N)rDrrrrFr:r3rbrbs%r:rbr%r&c0dtdtffd }|S)z#descriptor adding tag to a functionrr#cDt|drJti|_|S)Nr)rwrr)rr%r&s r3descztag..descs+4((($)&)  r:r )r%r&rfs`` r3tagrhs8 Kr:versionc,dtdtffd }|S)zXCompare version of python interpreter to the given one. Skip the test if older. fr#cjd} t|Dcgc] }t|c}}tj dd|krfd}|j |_|S|Scc}w#t$rtdzwxYw)N.z0%s is not a correct version : should be X.Y[.Z].r c |jdddjDcgc] }t|c}dycc}w)NzNeed at least z' version of python. Current version is rm)skipTestr^r7)rr%r&elementcurrentris r3new_fz=require_version..check_require_version..new_fs5 g)N7#g,)N OQ)NsA)splitrrcrdrtrrD)rkversion_elementsvcomparerrrqris @r3check_require_versionz.require_version..check_require_versions"==- [-=>SV>?G""2A& W   ZZENLH? [ORYYZ Z [s A/A*A/*A//Brg)rirws` r3require_versionrxs h( ! r:rc,dtdtffd }|S)z:Check if the given module is loaded. Skip the test if not.rkr#cn t|S#t$rfd}|j|_|cYSwxYw)Nc.|jdzy)Nz%s can not be imported.)ro)rr%r&rs r3rrz;require_module..check_require_module..new_fs 7&@Ar:) __import__ ImportErrorrD)rkrrrs r3check_require_modulez,require_module..check_require_modules<  v H  BZZENL  s !44rg)rr~s` r3require_modulers  X  r:r=)Ur __docformat__ contextlibrrtr>os.pathr\rr,rshutilrinspectrtypingrrrr r r r r mypy_extensionsrr(rlogilab.common.deprecationrrrunittest_legacy functoolsrlogilab.common.decoratorsrrr unittest2rr}rD__all__DEFAULT_PREFIXES is_generator __unittestr9rBrErrpryrrobjectr TestSuitedictrrrmainrrrcrrrrTrrrrrrr"r6rOr7r`rbrhrxrrFr:r3rs$,& & 'NNN$L"; t4W$&  ! GVP"#OP  ST#8##U#L "2%"\`$H (+ x   X(.6    @t@(u( (x(%S/eE$t*dSViBWx  M>`/:  F8%%F ]G)) ] 3h"23@! ! L&#&&&"$S "$3"$4"$J%% s c h !S!X!8 3 8 E WJXUVVWs 4 HH