B u9aפ@sdZddlZddlZddlZddlZddlZddlZddlZddl m Z ddl m Z m Z ddlZGdddeZGdddeZGd d d eZGd d d eZdZd dZddZGdddejZGdddeZGdddZd5ddZe ddZeddfddZdZedZ d d!Z!d"d#Z"d$d%Z#d&d'Z$Gd(d)d)ej%Z&Gd*d+d+e&Z'Gd,d-d-e&Z(Gd.d/d/e&Z)Gd0d1d1e&Z*d2d3Z+e,d4kre+dS)6a Test the bdb module. A test defines a list of tuples that may be seen as paired tuples, each pair being defined by 'expect_tuple, set_tuple' as follows: ([event, [lineno[, co_name[, eargs]]]]), (set_type, [sargs]) * 'expect_tuple' describes the expected current state of the Bdb instance. It may be the empty tuple and no check is done in that case. * 'set_tuple' defines the set_*() method to be invoked when the Bdb instance reaches this state. Example of an 'expect_tuple, set_tuple' pair: ('line', 2, 'tfunc_main'), ('step', ) Definitions of the members of the 'expect_tuple': event: Name of the trace event. The set methods that do not give back control to the tracer [1] do not trigger a tracer event and in that case the next 'event' may be 'None' by convention, its value is not checked. [1] Methods that trigger a trace event are set_step(), set_next(), set_return(), set_until() and set_continue(). lineno: Line number. Line numbers are relative to the start of the function when tracing a function in the test_bdb module (i.e. this module). co_name: Name of the function being currently traced. eargs: A tuple: * On an 'exception' event the tuple holds a class object, the current exception must be an instance of this class. * On a 'line' event, the tuple holds a dictionary and a list. The dictionary maps each breakpoint number that has been hit on this line to its hits count. The list holds the list of breakpoint number temporaries that are being deleted. Definitions of the members of the 'set_tuple': set_type: The type of the set method to be invoked. This may be the type of one of the Bdb set methods: 'step', 'next', 'until', 'return', 'continue', 'break', 'quit', or the type of one of the set methods added by test_bdb.Bdb: 'ignore', 'enable', 'disable', 'clear', 'up', 'down'. sargs: The arguments of the set method if any, packed in a tuple. N)contextmanager)islicerepeatc@s eZdZdS) BdbExceptionN)__name__ __module__ __qualname__r r /usr/lib/python3.7/test_bdb.pyr>src@seZdZdZdS)BdbErrorz!Error raised by the Bdb instance.N)rrr__doc__r r r r r ?sr c@seZdZdZdS)BdbSyntaxErrorzSyntax error in the test case.N)rrrr r r r r r @sr c@seZdZdZdS)BdbNotExpectedErrorzUnexpected result.N)rrrr r r r r rAsrcCsdtj_itj_dgtj_dS)N)_bdb BreakpointnextZbplist bpbynumberr r r r reset_BreakpointIsrc CsddtjjD}|sdSd}x~|D]v}|s4d}d}|jr>dnd}|jrLd nd }|d |j|||j|jtj |j |j f7}|j r|d |j f7}|d 7}q$W|S)NcSsg|] }|r|qSr r ).0bpr r r Osz$info_breakpoints..Fz!BpNum Temp Enb Hits Ignore Where Tzyes zno Zyeszno z%-5d %s %s %-4d %-6d at %s:%dz stop only if %s )rrr temporaryenabledZnumberhitsignoreospathbasenamefilelinecond)bp_listZ header_addedrinfoZdispZenabr r r info_breakpointsNs"  r&csjeZdZdZfddZdfdd Zfdd Zd d Zd d ZddZ ddZ ddZ ddZ Z S)Bdbz$Extend Bdb to enhance test coverage.csd|_t|||S)N) currentbpsupertrace_dispatch)selfframeeventarg) __class__r r r*fszBdb.trace_dispatchFNc st|trT|tkrt}nt|dd}|j}t||}|j} | j }| j }| j }t j |||||d} t| tr|t| | S)N)rr#funcname) isinstancestr__file__globals importlib import_module__dict__eval__code__ co_filenameco_firstlinenoco_namer) set_breakr ) r+filenamelinenorr#r1globals_modulefunccoderes)r/r r r>js     z Bdb.set_breakcs4t||\|_|_|j|jd|_|j|jfS)Nr)r) get_stackstackindexr,)r+ft)r/r r rF~sz Bdb.get_stackcCs||}|jd7_dS)z8Increment the ignore count of Breakpoint number 'bpnum'.rN)get_bpbynumberr)r+bpnumrr r r set_ignores zBdb.set_ignorecCs||}d|_dS)NT)rKr)r+rLrr r r set_enables zBdb.set_enablecCs||}d|_dS)NF)rKr)r+rLrr r r set_disables zBdb.set_disablecCs|||}|rt|dS)N)Z clear_breakr )r+fnamer@errr r r set_clears z Bdb.set_clearcCs2|jstd|jd8_|j|jd|_dS)zMove up in the frame stack.z Oldest framerrN)rHr rGr,)r+r r r set_upsz Bdb.set_upcCs@|jdt|jkrtd|jd7_|j|jd|_dS)zMove down in the frame stack.rz Newest framerN)rHlenrGr r,)r+r r r set_downsz Bdb.set_down)FNN)rrrr r*r>rFrMrNrOrRrSrU __classcell__r r )r/r r'cs  r'cseZdZdZd,fdd ZddZfdd Zd d Zd d ZddZ ddZ ddZ ddZ ddZ ddZddZddZddZd d!Zd"d#Zd$d%Zd-d&d'Zd(d)Zd*d+ZZS).Tracerz$A tracer for testing the bdb module.NFcs<tj|d||_||_|dk r*d|nd|_|dS)N)skipzDry-run results for %s:)r)__init__ expect_setdry_runheader init_test)r+rZrXr[ test_case)r/r r rYs zTracer.__init__cCsBd|_d|_d|_tt|jddd|_tt|jddd|_dS)Nrr) cur_except expect_set_nobreakpoint_hitslistrrZ expected_listset_list)r+r r r r]s zTracer.init_testc sn|jdk r|j|dkrZyt|||}|Stk rV}z ||_|jSd}~XYqjXnt|||SdS)N exception)r`r)r*r)r+r,r-r.rEe)r/r r r*s zTracer.trace_dispatchcCs(||sdS|d|||dS)Ncall)Z stop_here process_eventnext_set_method)r+r,Z argument_listr r r user_calls zTracer.user_callcCs^|d||jrD|jrDtd}x|dD]}td|q0W|d|_|dS)Nr"rz ) rir[rbr&stripsplitprintdelete_temporariesrj)r+r,r%r"r r r user_lines   zTracer.user_linecCs|d|||dS)Nreturn)rirj)r+r,Z return_valuer r r user_returnszTracer.user_returncCs||_|d||dS)Nrf)exc_inforirj)r+r,rsr r r user_exceptions zTracer.user_exceptioncCs|jg}||f|_dS)N)r(rb)r+r.r$r r r do_clearszTracer.do_clearcCs(|jr$x|jdD]}||qWdS)Nr)rbZclear_bpbynumber)r+nr r r roszTracer.delete_temporariescCsV|jd7_y|jd|_Wn"tk rBtd|jYnX|jd|_dS)Nrrz-expect_set list exhausted, cannot pop item %d)rardpopexpect IndexErrorrre set_tuple)r+r r r pop_nextszTracer.pop_nextcsd}|dkrjd}||jdk rDjsDjg}|gf_|_jrhjdSj r j d|d |dkr dnt j dkr~|dkrFj d\}t}jsdj |jdd  fd d |Dfd d jdDd  t|jddn8|dkr~tjdj ds~djjfdS)Nrfr_rzWrong event type)rhrqr"z(No breakpoints hit at expect_set item %dzBreakpoint numbers do not matchcsg|] }|qSr r )rrv)bpsr r rsz(Tracer.process_event..csg|]}|jqSr )rKr)rrv)r+r r r szWrong breakpoint hit countrzWrong temporary breakpointsz/Wrong exception at expect_set item %d, got '%s')rsrFr(rbr-r{r[ print_stater\rx check_equalcheck_lno_namecheck_expect_max_sizerTsortedkeysraise_not_expectedrar2)r+r-r,argstbr$Z temporariesZbpnumsr )r}r+r risL           zTracer.process_eventcCs&||kr dS|d||j|fdS)Nz"%s at expect_set item %d, got '%s')rra)r+Zexpectedresultmsgr r r r,szTracer.check_equalcCsTt|j}|dkr.|}||jd|d|dkrP||jd|jjjddS)z+Check the line number and function co_name.rzWrong line numberr_zWrong function nameN)rTrx lno_abs2relrr,f_coder=)r+sr@r r r r2s zTracer.check_lno_namecCs&t|j|kr"td|j|jfdS)Nz'Invalid size of the %s expect tuple: %s)rTrxr r-)r+sizer r r r<szTracer.check_expect_max_sizecCs<||jjj}|jj}||tkr8||jjjdS|S)Nr)canonicr,rr;f_linenor4r<)r+rPr@r r r rAszTracer.lno_abs2relcCs.|r*|||tkr*|jjj|dS|S)Nr)rr4r,rr<)r+rPr@r r r lno_rel2absGs*zTracer.lno_rel2abscCs|}|jjj}d|j||f}|jrd}x:|jdD],}|dkrL|d7}|d|||jf7}q8W|d7}d|dt|jdd }|d|7}n|jd kr|d|j dj 7}|d 7}| d t|j d S)Nz('%s', %d, '%s'{rz, z%s: %s}(r)rfz), ,) rr,rr=r-rbrKrr3rsrljustrz)r+r@r=stater}rvr r r get_stateLs   zTracer.get_statecCs:|dk r |jdkr tt|td|j|fdS)Nrz%d: %s)rarnr)r+r\r r r r~^szTracer.print_statecCs6|d7}|dt|j7}|d|7}t|dS)Nrz Expected: %s z Got: )r3rxrr)r+rr r r rdszTracer.raise_not_expectedcCsZ|jd}t|jdkr"|jdnd}t|d|}|dkrF|dS|dkr\||jdS|dkrd}|r||jjj|d}||j|dS|r|dks|d krH|d kr|^}}}|||}||g}||||n"|d kr||n|d kr|d|_| |j r$| n|j r4| |d |ntd |jdS)Nrr_rZset_)stepcontinuequit)rrquntil)breakclearrenabledisable)updown)rr)rrrr|z"%s" is an invalid set_tuple)rzrTgetattrr,rrr;extendr-r{r[r~rxrrrjr )r+set_typerZ set_methodr@rPZremainr r r rjjsL                zTracer.next_set_method)NFN)N)rrrr rYr]r*rkrprrrtruror{rirrrrrrr~rrjrVr r )r/r rWs*    2  rWc@s,eZdZdZd ddZddZd ddZdS) TracerRunzAProvide a context for running a Tracer instance with a test case.NcCs2||_|j|_t|j||j|d|_d|_dS)N)rXr[r^)r^r[rWrZidtracer_original_tracer)r+r^rXr r r rYs zTracerRun.__init__cCstt|_|jS)N)rsysgettracerr)r+r r r __enter__s zTracerRun.__enter__cCstt|jd}|jjr6|d7}|d|jj7}|dk rtt|rt |t r|j r|j d}|rr|d|7}|j rt |dS|j|n|r|j rt |n |j|dS)Nrz+All paired tuples have not been processed, zthe last one was number %drrT)rrsettracerrrera issubclassrr2 BaseExceptionrr[rnr^fail)r+Ztype_value tracebackZ not_emptyZerr_msgr r r __exit__s(     zTracerRun.__exit__)N)NNN)rrrr rYrrr r r r rs rc Csdd}ddg}||t}d|_dd|_t|tdt||_t|(t ||d }| t Wd QRXWd QRXd S) a{Run a test and print the dry-run results. 'modules': A dictionary mapping module names to their source code as a string. The dictionary MUST include one module named 'test_module' with a main() function. 'set_list': A list of set_type tuples to be run on the module. For example, running the following script outputs the following results: ***************************** SCRIPT ******************************** from test.test_bdb import run_test, break_in_func code = ''' def func(): lno = 3 def main(): func() lno = 7 ''' set_list = [ break_in_func('func', 'test_module.py'), ('continue', ), ('step', ), ('step', ), ('step', ), ('quit', ), ] modules = { 'test_module': code } run_test(modules, set_list) **************************** results ******************************** 1: ('line', 2, 'tfunc_import'), ('next',), 2: ('line', 3, 'tfunc_import'), ('step',), 3: ('call', 5, 'main'), ('break', ('test_module.py', None, False, None, 'func')), 4: ('None', 5, 'main'), ('continue',), 5: ('line', 3, 'func', ({1: 1}, [])), ('step',), BpNum Temp Enb Hits Ignore Where 1 no yes 1 0 at test_module.py:2 6: ('return', 3, 'func'), ('step',), 7: ('line', 7, 'main'), ('step',), 8: ('return', 7, 'main'), ('quit',), ************************************************************************* css@y&x t|}t|}|V|VqWWntk r:dSXdS)N)r StopIteration)abxyr r r genszrun_test..gen)r)rTcSsdS)Nr r r r r zrun_test..r )rXN) r BaseTestCaser[rrcriterrZcreate_modulesrruncall tfunc_import)modulesrerXrsltestrr r r run_tests3    rc cstjtjtz\xH|D]@}|d}t|d}| t ||WdQRXt |q$WtdVWdx|D]}tj|qWtjXWdQRXdS)Nz.pyw)rsupportZtemp_cwdrrappendrgetcwdopenwritetextwrapdedent linecache checkcacher6invalidate_cachesZforgetrw)rmrPrIr r r rs     rFcCsd|d|||ffS)Nrr )r1rPrr#r r r break_in_func%srtest_module_for_bdbz.pycCsddl}|dS)Nr)rmain)rr r r r*srcCs d}ttd}d}d}dS)Nr_) tfunc_first tfunc_second)lnor r r tfunc_main.s rcCsd}d}d}dS)Nr_r|r )rr r r r6srcCsd}dS)Nr_r )rr r r r;src@seZdZdZeZdddZdS)rzBase class for all tests.NcCs||ddS)N)ZfailureException)r+rr r r rCszBaseTestCase.fail)N)rrrr r[rr r r r r>src@seZdZdZddZddZddZdd Zd d Zd d Z ddZ ddZ ddZ ddZ ddZddZddZddZdS) StateTestCasez;Test the step, next, return, until and quit 'set_' methods.c Cs8ddddddddg|_t|}|tWdQRXdS)N)r"r_r)r)r"r|r)rhrr)r"r_r)r)rZrrr)r+rr r r test_stepKs   zStateTestCase.test_stepcCsxzdD]r}|j|d\ddddddtdffd d d dd d d igff|fddg |_t|}|tWdQRXWdQRXqWdS)N)rr)r)r"r_r)r)r"r|r)rhrrrr|)Nonerr)rr"rr)r"rr)r)subTestr4rZrrr)r+rrr r r test_step_next_on_last_statementUs    z.StateTestCase.test_step_next_on_last_statementc Cs<ddddddddddg |_t|}|tWdQRXdS) N)r"r_r)r)r"r|r)r)r"rr)rhrr)r"r_r)r)rZrrr)r+rr r r test_nextcs  zStateTestCase.test_nextc CsPd}t|i}t|2ddddg|_t|}|tWdQRXWdQRXdS)Nz9 def main(): lno = 3 )r"r_r)r)r"r|r)r) TEST_MODULErrZrrr)r+rDrrr r r test_next_over_importns   z#StateTestCase.test_next_over_importc Cs8ddddddddg|_t|}|tWdQRXdS)N)r"r_r)r)r"r|r)rhrr)r)r"r_r)r)rZrrr)r+rr r r test_next_on_plain_statement|s   z*StateTestCase.test_next_on_plain_statementc Cs<dddddddddd g |_t|}|tWdQRXdS) N)r"r_r)r)r"r|r)rhrr)r)rr|r)r)r"rr)r)rZrrr)r+rr r r test_next_in_caller_frames  z'StateTestCase.test_next_in_caller_framec Cs@dddddddddddd g |_t|}|tWdQRXdS) N)r"r_r)r)r"r|r)rhrr)r"r_r)rq)rqrr)r"rr)r)rZrrr)r+rr r r test_returns  zStateTestCase.test_returnc Cs<dddddddddd g |_t|}|tWdQRXdS) N)r"r_r)r)r"r|r)rhrr)r)rr|r)rq)rqrr)r)rZrrr)r+rr r r test_return_in_caller_frames  z)StateTestCase.test_return_in_caller_framec Cs<ddddddddddg |_t|}|tWdQRXdS) N)r"r_r)r)r"r|r)rhrr)r"r_r)r)r)r"rr)r)rZrrr)r+rr r r test_untils  zStateTestCase.test_untilc CsLdtddddddddigffdd d g|_t|}|tWdQRXdS) N)r"r_rr)rr_r)rr"r_r)r)i')rqrr)r)rrZrrr)r+rr r r test_until_with_too_large_counts   z-StateTestCase.test_until_with_too_large_countc Cs<dddddddddd g |_t|}|tWdQRXdS) N)r"r_r)r)r"r|r)rhrr)r)rr|r)r)r)r"rr)r)rZrrr)r+rr r r test_until_in_caller_frames  z(StateTestCase.test_until_in_caller_framec Cs\d}t|i}t|>ddddg|_dtf}t||d}|tWdQRXWdQRXdS)Nz9 def main(): lno = 3 )r"r_r)r)r"r|r)rz importlib*)rX)rrrZrrr)r+rDrrXrr r r test_skips  zStateTestCase.test_skipc Cs2ddg|_t|}|t|jtWdQRXdS)N)r"r_r)r)rZr assertRaisesr rr)r+rr r r test_downs  zStateTestCase.test_downc Cs8ddddddddg|_t|}|tWdQRXdS)N)r"r_r)r)r"r|r)rhrr)r)rr|r)r)rZrrr)r+rr r r test_ups   zStateTestCase.test_upN)rrrr rrrrrrrrrrrrrrr r r r rHs         rc@s`eZdZdZddZddZddZdd Zd d Zd d Z ddZ ddZ ddZ ddZ dS)BreakpointTestCasezTest the breakpoint set method.c Cs2ddg|_t|}|t|jtWdQRXdS)N)r"r_r)r)z/non/existent/module.pyr)rZrrr rr)r+rr r r test_bp_on_non_existent_modules  z1BreakpointTestCase.test_bp_on_non_existent_modulec CsZd}t|i}t|<ddtdffg|_t|}|t|jtWdQRXWdQRXdS)Nz9 def main(): lno = 3 )r"r_rrr) rrTEST_MODULE_FNAMErZrrr rr)r+rDrrr r r test_bp_after_last_statements   z/BreakpointTestCase.test_bp_after_last_statementcCsd}t|i}t|rdtdtddtdtddddddd d id gffddddd d id gffd g |_t|}|tWdQRXWdQRXdS) Nz def func(): lno = 3 def main(): for i in range(2): func() )r"r_rrCT)rr_r)rr"r|rr_)r)rrrrrZrrr)r+rDrrr r r test_temporary_bps    z$BreakpointTestCase.test_temporary_bpcCsd}t|i}t|dtdtdtdtddddddd dd d igffd d d d ddd ddd idgffdd ddd dd digffdg|_t|}|tWdQRXWdQRXdS)Nz def func(): lno = 3 def main(): for i in range(3): func() )r"r_rrC)rr_rT)r)r_)rr"r|r)r)r_)rr|rC)r)rr_)r)r)r)rrrrrZrrr)r+rDrrr r r test_disabled_temporary_bps"   z-BreakpointTestCase.test_disabled_temporary_bpc Csnd}t|i}t|Pdtdtdddddd dd d igffd g|_t|}|tWdQRXWdQRXdS) Nz def func(a): lno = 3 def main(): for i in range(3): func(i) )r"r_rrCFza == 2)rr_r)rr"r|r)r)rrrrrZrrr)r+rDrrr r r test_bp_condition9s   z$BreakpointTestCase.test_bp_conditionc Csnd}t|i}t|Pdtdtdddddd dd d igffd g|_t|}|tWdQRXWdQRXdS) Nzk def func(a): lno = 3 def main(): func(0) )r"r_rrCFz1 / 0)rr_r)rr"r|r)r)rrrrrZrrr)r+rDrrr r r )test_bp_exception_on_condition_evaluationMs   z)r)rqr_z)rzexec)rZrruncompilerr)r+rDrr r r test_run_steps   zRunTestCase.test_run_stepc Csld}t|i}t|Nddddddddddg |_d dl}t|}|d ttWdQRXWdQRXdS) Nz9 def main(): lno = 3 )r"rz)r)rhr_r)r"r|r)rqr|r)rqrz)rrztest_module_for_bdb.main())rrrZrrZrunevalr5locals)r+rDrrrr r r test_runeval_steps   zRunTestCase.test_runeval_stepN)rrrr rrr r r r rs rc@s8eZdZdZddZddZddZdd Zd d Zd S) IssuesTestCasezTest fixed bdb issues.c Cszd}d}t|d|i}t|Tdtddddd d dd d igffd d d ddg |_t|}|tWdQRXWdQRXdS)Nz from test_module_for_bdb_2 import func def main(): func() lno = 5 z9 def func(): lno = 3 Ztest_module_for_bdb_2)r"r_rrCztest_module_for_bdb_2.py)rr_r)rr"r|r)r)rqr|rC)r"rr)r)rrrrZrrr)r+Zcode_1Zcode_2rrr r r +test_step_at_return_with_no_trace_in_callers    z:IssuesTestCase.test_step_at_return_with_no_trace_in_callerc Csd}t|i}xdD]}|j|dt|dtdtdddd dd d igff|fg|_|d kr~|jd d dtfdddgn|jddgt|}| t WdQRXWdQRXWdQRXqWdS)Nz def test_gen(): yield 0 lno = 4 return 123 def main(): it = test_gen() next(it) next(it) lno = 11 )rrrq)r)r"r_rtest_gen)rr_r)rr"r|rrqrf r)r)rqrr)r)r"rr) rrrrrrZrrrrr)r+rDrrrr r r #test_next_until_return_in_generators"      z2IssuesTestCase.test_next_until_return_in_generatorcCsd}t|i}t|hdtdtdddddddigffd d d d d d d d ddtfdddg|_t|}|tWdQRXWdQRXdS)Nz def test_gen(): yield 0 lno = 4 yield 1 return 123 def main(): for i in test_gen(): lno = 10 lno = 11 )r"r_rr)rr_r)rr"r|r)r)r"rr)r"rr)r"rrrf r)r)r" r)r) rrrrrrZrrr)r+rDrrr r r 'test_next_command_in_generator_for_loops     z6IssuesTestCase.test_next_command_in_generator_for_loopc Cs~d}t|i}t|`dtdtdddddddigffd d d d d d tfdddg |_t|}|tWdQRXWdQRXdS)Na4 def test_subgen(): yield 0 return 123 def test_gen(): x = yield from test_subgen() return 456 def main(): for i in test_gen(): lno = 12 lno = 13 )r"r_rr)rr_r)rr"rr)r)r"rrfrr)r)r" r)r) rrrrrrZrrr)r+rDrrr r r /test_next_command_in_generator_with_subiterator;s    z>IssuesTestCase.test_next_command_in_generator_with_subiteratorc Csd}t|i}t|hdtdtdddddddigffd d d d tfd d d dtfdddg |_t|}|tWdQRXWdQRXdS)Na4 def test_subgen(): yield 0 return 123 def test_gen(): x = yield from test_subgen() return 456 def main(): for i in test_gen(): lno = 12 lno = 13 )r"r_rZ test_subgen)rr_r)rr"r|r)rqrfrrrr)r)r"r r)r) rrrrrrZrrr)r+rDrrr r r 1test_return_command_in_generator_with_subiteratorZs     z@IssuesTestCase.test_return_command_in_generator_with_subiteratorN) rrrr rrr r r r r r r rs )rcCstjttttdS)N)rrZ run_unittestrrrrr r r r test_mainys r__main__)N)-r ZbdbrrrZunittestrr6r contextlibr itertoolsrrZ test.supportr Exceptionrr r rr[rr&r'rWrrrr4rrrrrrrZTestCaserrrrrrrr r r r 1sR @z. J  (B"(