B u9as@sddlZddlZddlZddlZddlZddlZddlZddlZddlm Z ddl m Z ddl m Z ddlmZddlmZGdddeZd d Zd d Zd dZddZddZddZddZddZedZedejddZ ddZ!dd Z"d!d"Z#d#d$Z$d%d&Z%d'd(Z&d)d*Z'd+d,Z(d-d.Z)d/d0Z*d1d2Z+d3d4Z,d5d6Z-d7d8Z.d9d:Z/Gd;d<dZ2e3d?kre4dS)@N) ExitStack)StringIO)support) _FakeInput)patchc@s(eZdZdZddZddZddZdS) PdbTestInputz:Context manager that makes testing Pdb in doctests easier.cCs ||_dS)N)input)selfrr /usr/lib/python3.7/test_pdb.py__init__szPdbTestInput.__init__cCs0tj|_t|jt_ttdr&tnd|_dS)Ngettrace)sysstdin real_stdinrrhasattrr orig_trace)r r r r __enter__s zPdbTestInput.__enter__cGs|jt_|jrt|jdS)N)rrrrsettrace)r excr r r __exit__szPdbTestInput.__exit__N)__name__ __module__ __qualname____doc__r rrr r r r rsrcCsdS)a1This tests the custom displayhook for pdb. >>> def test_function(foo, bar): ... import pdb; pdb.Pdb(nosigint=True, readrc=False).set_trace() ... pass >>> with PdbTestInput([ ... 'foo', ... 'bar', ... 'for i in range(5): print(i)', ... 'continue', ... ]): ... test_function(1, None) > (3)test_function() -> pass (Pdb) foo 1 (Pdb) bar (Pdb) for i in range(5): print(i) 0 1 2 3 4 (Pdb) continue Nr r r r r test_pdb_displayhook%srcCsdS)aTest the basic commands of pdb. >>> def test_function_2(foo, bar='default'): ... print(foo) ... for i in range(5): ... print(i) ... print(bar) ... for i in range(10): ... never_executed ... print('after for') ... print('...') ... return foo.upper() >>> def test_function3(arg=None, *, kwonly=None): ... pass >>> def test_function(): ... import pdb; pdb.Pdb(nosigint=True, readrc=False).set_trace() ... ret = test_function_2('baz') ... test_function3(kwonly=True) ... print(ret) >>> with PdbTestInput([ # doctest: +ELLIPSIS, +NORMALIZE_WHITESPACE ... 'step', # entering the function call ... 'args', # display function args ... 'list', # list function source ... 'bt', # display backtrace ... 'up', # step up to test_function() ... 'down', # step down to test_function_2() again ... 'next', # stepping to print(foo) ... 'next', # stepping to the for loop ... 'step', # stepping into the for loop ... 'until', # continuing until out of the for loop ... 'next', # executing the print(bar) ... 'jump 8', # jump over second for loop ... 'return', # return out of function ... 'retval', # display return value ... 'next', # step to test_function3() ... 'step', # stepping into test_function3() ... 'args', # display function args ... 'continue', ... ]): ... test_function() > (3)test_function() -> ret = test_function_2('baz') (Pdb) step --Call-- > (1)test_function_2() -> def test_function_2(foo, bar='default'): (Pdb) args foo = 'baz' bar = 'default' (Pdb) list 1 -> def test_function_2(foo, bar='default'): 2 print(foo) 3 for i in range(5): 4 print(i) 5 print(bar) 6 for i in range(10): 7 never_executed 8 print('after for') 9 print('...') 10 return foo.upper() [EOF] (Pdb) bt ... (21)() -> test_function() (3)test_function() -> ret = test_function_2('baz') > (1)test_function_2() -> def test_function_2(foo, bar='default'): (Pdb) up > (3)test_function() -> ret = test_function_2('baz') (Pdb) down > (1)test_function_2() -> def test_function_2(foo, bar='default'): (Pdb) next > (2)test_function_2() -> print(foo) (Pdb) next baz > (3)test_function_2() -> for i in range(5): (Pdb) step > (4)test_function_2() -> print(i) (Pdb) until 0 1 2 3 4 > (5)test_function_2() -> print(bar) (Pdb) next default > (6)test_function_2() -> for i in range(10): (Pdb) jump 8 > (8)test_function_2() -> print('after for') (Pdb) return after for ... --Return-- > (10)test_function_2()->'BAZ' -> return foo.upper() (Pdb) retval 'BAZ' (Pdb) next > (4)test_function() -> test_function3(kwonly=True) (Pdb) step --Call-- > (1)test_function3() -> def test_function3(arg=None, *, kwonly=None): (Pdb) args arg = None kwonly = True (Pdb) continue BAZ Nr r r r r test_pdb_basic_commandsBs|rcCsdS)ahTest basic commands related to breakpoints. >>> def test_function(): ... import pdb; pdb.Pdb(nosigint=True, readrc=False).set_trace() ... print(1) ... print(2) ... print(3) ... print(4) First, need to clear bdb state that might be left over from previous tests. Otherwise, the new breakpoints might get assigned different numbers. >>> from bdb import Breakpoint >>> Breakpoint.next = 1 >>> Breakpoint.bplist = {} >>> Breakpoint.bpbynumber = [None] Now test the breakpoint commands. NORMALIZE_WHITESPACE is needed because the breakpoint list outputs a tab for the "stop only" and "ignore next" lines, which we don't want to put in here. >>> with PdbTestInput([ # doctest: +NORMALIZE_WHITESPACE ... 'break 3', ... 'disable 1', ... 'ignore 1 10', ... 'condition 1 1 < 2', ... 'break 4', ... 'break 4', ... 'break', ... 'clear 3', ... 'break', ... 'condition 1', ... 'enable 1', ... 'clear 1', ... 'commands 2', ... 'p "42"', ... 'print("42", 7*6)', # Issue 18764 (not about breakpoints) ... 'end', ... 'continue', # will stop at breakpoint 2 (line 4) ... 'clear', # clear all! ... 'y', ... 'tbreak 5', ... 'continue', # will stop at temporary breakpoint ... 'break', # make sure breakpoint is gone ... 'continue', ... ]): ... test_function() > (3)test_function() -> print(1) (Pdb) break 3 Breakpoint 1 at :3 (Pdb) disable 1 Disabled breakpoint 1 at :3 (Pdb) ignore 1 10 Will ignore next 10 crossings of breakpoint 1. (Pdb) condition 1 1 < 2 New condition set for breakpoint 1. (Pdb) break 4 Breakpoint 2 at :4 (Pdb) break 4 Breakpoint 3 at :4 (Pdb) break Num Type Disp Enb Where 1 breakpoint keep no at :3 stop only if 1 < 2 ignore next 10 hits 2 breakpoint keep yes at :4 3 breakpoint keep yes at :4 (Pdb) clear 3 Deleted breakpoint 3 at :4 (Pdb) break Num Type Disp Enb Where 1 breakpoint keep no at :3 stop only if 1 < 2 ignore next 10 hits 2 breakpoint keep yes at :4 (Pdb) condition 1 Breakpoint 1 is now unconditional. (Pdb) enable 1 Enabled breakpoint 1 at :3 (Pdb) clear 1 Deleted breakpoint 1 at :3 (Pdb) commands 2 (com) p "42" (com) print("42", 7*6) (com) end (Pdb) continue 1 '42' 42 42 > (4)test_function() -> print(2) (Pdb) clear Clear all breaks? y Deleted breakpoint 2 at :4 (Pdb) tbreak 5 Breakpoint 4 at :5 (Pdb) continue 2 Deleted breakpoint 4 at :5 > (5)test_function() -> print(3) (Pdb) break (Pdb) continue 3 4 Nr r r r r test_pdb_breakpoint_commandsskrcCsdS)Nr r r r r do_nothing/srcCs tddS)N*)printr r r r do_something2sr!cCsdS)a Test the list and source commands of pdb. >>> def test_function_2(foo): ... import test.test_pdb ... test.test_pdb.do_nothing() ... 'some...' ... 'more...' ... 'code...' ... 'to...' ... 'make...' ... 'a...' ... 'long...' ... 'listing...' ... 'useful...' ... '...' ... '...' ... return foo >>> def test_function(): ... import pdb; pdb.Pdb(nosigint=True, readrc=False).set_trace() ... ret = test_function_2('baz') >>> with PdbTestInput([ # doctest: +ELLIPSIS, +NORMALIZE_WHITESPACE ... 'list', # list first function ... 'step', # step into second function ... 'list', # list second function ... 'list', # continue listing to EOF ... 'list 1,3', # list specific lines ... 'list x', # invalid argument ... 'next', # step to import ... 'next', # step over import ... 'step', # step into do_nothing ... 'longlist', # list all lines ... 'source do_something', # list all lines of function ... 'source fooxxx', # something that doesn't exit ... 'continue', ... ]): ... test_function() > (3)test_function() -> ret = test_function_2('baz') (Pdb) list 1 def test_function(): 2 import pdb; pdb.Pdb(nosigint=True, readrc=False).set_trace() 3 -> ret = test_function_2('baz') [EOF] (Pdb) step --Call-- > (1)test_function_2() -> def test_function_2(foo): (Pdb) list 1 -> def test_function_2(foo): 2 import test.test_pdb 3 test.test_pdb.do_nothing() 4 'some...' 5 'more...' 6 'code...' 7 'to...' 8 'make...' 9 'a...' 10 'long...' 11 'listing...' (Pdb) list 12 'useful...' 13 '...' 14 '...' 15 return foo [EOF] (Pdb) list 1,3 1 -> def test_function_2(foo): 2 import test.test_pdb 3 test.test_pdb.do_nothing() (Pdb) list x *** ... (Pdb) next > (2)test_function_2() -> import test.test_pdb (Pdb) next > (3)test_function_2() -> test.test_pdb.do_nothing() (Pdb) step --Call-- > ...test_pdb.py(...)do_nothing() -> def do_nothing(): (Pdb) longlist ... -> def do_nothing(): ... pass (Pdb) source do_something ... def do_something(): ... print(42) (Pdb) source fooxxx *** ... (Pdb) continue Nr r r r r test_list_commands5s]r"cCsdS)aTest post mortem traceback debugging. >>> def test_function_2(): ... try: ... 1/0 ... finally: ... print('Exception!') >>> def test_function(): ... import pdb; pdb.Pdb(nosigint=True, readrc=False).set_trace() ... test_function_2() ... print('Not reached.') >>> with PdbTestInput([ # doctest: +ELLIPSIS, +NORMALIZE_WHITESPACE ... 'next', # step over exception-raising call ... 'bt', # get a backtrace ... 'list', # list code of test_function() ... 'down', # step into test_function_2() ... 'list', # list code of test_function_2() ... 'continue', ... ]): ... try: ... test_function() ... except ZeroDivisionError: ... print('Correctly reraised.') > (3)test_function() -> test_function_2() (Pdb) next Exception! ZeroDivisionError: division by zero > (3)test_function() -> test_function_2() (Pdb) bt ... (10)() -> test_function() > (3)test_function() -> test_function_2() (3)test_function_2() -> 1/0 (Pdb) list 1 def test_function(): 2 import pdb; pdb.Pdb(nosigint=True, readrc=False).set_trace() 3 -> test_function_2() 4 print('Not reached.') [EOF] (Pdb) down > (3)test_function_2() -> 1/0 (Pdb) list 1 def test_function_2(): 2 try: 3 >> 1/0 4 finally: 5 -> print('Exception!') [EOF] (Pdb) continue Correctly reraised. Nr r r r r test_post_mortems;r#cCsdS)a[This illustrates the simple case of module skipping. >>> def skip_module(): ... import string ... import pdb; pdb.Pdb(skip=['stri*'], nosigint=True, readrc=False).set_trace() ... string.capwords('FOO') >>> with PdbTestInput([ ... 'step', ... 'continue', ... ]): ... skip_module() > (4)skip_module() -> string.capwords('FOO') (Pdb) step --Return-- > (4)skip_module()->None -> string.capwords('FOO') (Pdb) continue Nr r r r r test_pdb_skip_modulessr$Zmodule_to_skipz6def foo_pony(callback): x = 1; callback(); return NonecCsdS)aAThis illustrates skipping of modules that call into other code. >>> def skip_module(): ... def callback(): ... return None ... import pdb; pdb.Pdb(skip=['module_to_skip*'], nosigint=True, readrc=False).set_trace() ... mod.foo_pony(callback) >>> with PdbTestInput([ ... 'step', ... 'step', ... 'step', ... 'step', ... 'step', ... 'continue', ... ]): ... skip_module() ... pass # provides something to "step" to > (5)skip_module() -> mod.foo_pony(callback) (Pdb) step --Call-- > (2)callback() -> def callback(): (Pdb) step > (3)callback() -> return None (Pdb) step --Return-- > (3)callback()->None -> return None (Pdb) step --Return-- > (5)skip_module()->None -> mod.foo_pony(callback) (Pdb) step > (10)() -> pass # provides something to "step" to (Pdb) continue Nr r r r r #test_pdb_skip_modules_with_callbacks(r%cCsdS)aTest that "continue" and "next" work properly in bottom frame (issue #5294). >>> def test_function(): ... import pdb, sys; inst = pdb.Pdb(nosigint=True, readrc=False) ... inst.set_trace() ... inst.botframe = sys._getframe() # hackery to get the right botframe ... print(1) ... print(2) ... print(3) ... print(4) >>> with PdbTestInput([ # doctest: +ELLIPSIS ... 'next', ... 'break 7', ... 'continue', ... 'next', ... 'continue', ... 'continue', ... ]): ... test_function() > (4)test_function() -> inst.botframe = sys._getframe() # hackery to get the right botframe (Pdb) next > (5)test_function() -> print(1) (Pdb) break 7 Breakpoint ... at :7 (Pdb) continue 1 2 > (7)test_function() -> print(3) (Pdb) next 3 > (8)test_function() -> print(4) (Pdb) continue 4 Nr r r r r test_pdb_continue_in_bottomframes'r&cCsttjddd||dS)zRun pdb.method(arg).TF)ZnosigintZreadrcN)getattrpdbPdb)methodargr r r pdb_invokeDsr,cCsdS)aTesting run and runeval with incorrect first argument. >>> pti = PdbTestInput(['continue',]) >>> with pti: ... pdb_invoke('run', lambda x: x) Traceback (most recent call last): TypeError: exec() arg 1 must be a string, bytes or code object >>> with pti: ... pdb_invoke('runeval', lambda x: x) Traceback (most recent call last): TypeError: eval() arg 1 must be a string, bytes or code object Nr r r r r $test_pdb_run_with_incorrect_argumentIs r-cCsdS)a Testing run and runeval with code object as a first argument. >>> with PdbTestInput(['step','x', 'continue']): # doctest: +ELLIPSIS ... pdb_invoke('run', compile('x=1', '', 'exec')) > (1)()... (Pdb) step --Return-- > (1)()->None (Pdb) x 1 (Pdb) continue >>> with PdbTestInput(['x', 'continue']): ... x=0 ... pdb_invoke('runeval', compile('x+1', '', 'eval')) > (1)()->None (Pdb) x 1 (Pdb) continue Nr r r r r test_pdb_run_with_code_objectYsr.cCsdS)a Test that pdb stops after a next/until/return issued at a return debug event. >>> def test_function_2(): ... x = 1 ... x = 2 >>> def test_function(): ... import pdb; pdb.Pdb(nosigint=True, readrc=False).set_trace() ... test_function_2() ... test_function_2() ... test_function_2() ... end = 1 >>> from bdb import Breakpoint >>> Breakpoint.next = 1 >>> with PdbTestInput(['break test_function_2', ... 'continue', ... 'return', ... 'next', ... 'continue', ... 'return', ... 'until', ... 'continue', ... 'return', ... 'return', ... 'continue']): ... test_function() > (3)test_function() -> test_function_2() (Pdb) break test_function_2 Breakpoint 1 at :1 (Pdb) continue > (2)test_function_2() -> x = 1 (Pdb) return --Return-- > (3)test_function_2()->None -> x = 2 (Pdb) next > (4)test_function() -> test_function_2() (Pdb) continue > (2)test_function_2() -> x = 1 (Pdb) return --Return-- > (3)test_function_2()->None -> x = 2 (Pdb) until > (5)test_function() -> test_function_2() (Pdb) continue > (2)test_function_2() -> x = 1 (Pdb) return --Return-- > (3)test_function_2()->None -> x = 2 (Pdb) return > (6)test_function() -> end = 1 (Pdb) continue Nr r r r r &test_next_until_return_at_return_eventos?r/cCsdS)aXTesting skip unwindng stack on yield for generators for "next" command >>> def test_gen(): ... yield 0 ... return 1 ... yield 2 >>> def test_function(): ... import pdb; pdb.Pdb(nosigint=True, readrc=False).set_trace() ... it = test_gen() ... try: ... if next(it) != 0: ... raise AssertionError ... next(it) ... except StopIteration as ex: ... if ex.value != 1: ... raise AssertionError ... print("finished") >>> with PdbTestInput(['step', ... 'step', ... 'step', ... 'next', ... 'next', ... 'step', ... 'step', ... 'continue']): ... test_function() > (3)test_function() -> it = test_gen() (Pdb) step > (4)test_function() -> try: (Pdb) step > (5)test_function() -> if next(it) != 0: (Pdb) step --Call-- > (1)test_gen() -> def test_gen(): (Pdb) next > (2)test_gen() -> yield 0 (Pdb) next > (3)test_gen() -> return 1 (Pdb) step --Return-- > (3)test_gen()->1 -> return 1 (Pdb) step StopIteration: 1 > (7)test_function() -> next(it) (Pdb) continue finished Nr r r r r #test_pdb_next_command_for_generators9r0cCsdS)aTesting skip unwindng stack on yield for coroutines for "next" command >>> import asyncio >>> async def test_coro(): ... await asyncio.sleep(0) ... await asyncio.sleep(0) ... await asyncio.sleep(0) >>> async def test_main(): ... import pdb; pdb.Pdb(nosigint=True, readrc=False).set_trace() ... await test_coro() >>> def test_function(): ... loop = asyncio.new_event_loop() ... loop.run_until_complete(test_main()) ... loop.close() ... print("finished") >>> with PdbTestInput(['step', ... 'step', ... 'next', ... 'next', ... 'next', ... 'step', ... 'continue']): ... test_function() > (3)test_main() -> await test_coro() (Pdb) step --Call-- > (1)test_coro() -> async def test_coro(): (Pdb) step > (2)test_coro() -> await asyncio.sleep(0) (Pdb) next > (3)test_coro() -> await asyncio.sleep(0) (Pdb) next > (4)test_coro() -> await asyncio.sleep(0) (Pdb) next Internal StopIteration > (3)test_main() -> await test_coro() (Pdb) step --Return-- > (3)test_main()->None -> await test_coro() (Pdb) continue finished Nr r r r r #test_pdb_next_command_for_coroutines5r1cCsdS)aTesting skip unwindng stack on yield for coroutines for "next" command >>> import asyncio >>> async def agen(): ... yield 1 ... await asyncio.sleep(0) ... yield 2 >>> async def test_coro(): ... async for x in agen(): ... print(x) >>> async def test_main(): ... import pdb; pdb.Pdb(nosigint=True, readrc=False).set_trace() ... await test_coro() >>> def test_function(): ... loop = asyncio.new_event_loop() ... loop.run_until_complete(test_main()) ... loop.close() ... print("finished") >>> with PdbTestInput(['step', ... 'step', ... 'next', ... 'next', ... 'step', ... 'next', ... 'continue']): ... test_function() > (3)test_main() -> await test_coro() (Pdb) step --Call-- > (1)test_coro() -> async def test_coro(): (Pdb) step > (2)test_coro() -> async for x in agen(): (Pdb) next > (3)test_coro() -> print(x) (Pdb) next 1 > (2)test_coro() -> async for x in agen(): (Pdb) step --Call-- > (2)agen() -> yield 1 (Pdb) next > (3)agen() -> await asyncio.sleep(0) (Pdb) continue 2 finished Nr r r r r "test_pdb_next_command_for_asyncgen"s:r2cCsdS)aTesting no unwindng stack on yield for generators for "return" command >>> def test_gen(): ... yield 0 ... return 1 ... yield 2 >>> def test_function(): ... import pdb; pdb.Pdb(nosigint=True, readrc=False).set_trace() ... it = test_gen() ... try: ... if next(it) != 0: ... raise AssertionError ... next(it) ... except StopIteration as ex: ... if ex.value != 1: ... raise AssertionError ... print("finished") >>> with PdbTestInput(['step', ... 'step', ... 'step', ... 'return', ... 'step', ... 'step', ... 'continue']): ... test_function() > (3)test_function() -> it = test_gen() (Pdb) step > (4)test_function() -> try: (Pdb) step > (5)test_function() -> if next(it) != 0: (Pdb) step --Call-- > (1)test_gen() -> def test_gen(): (Pdb) return StopIteration: 1 > (7)test_function() -> next(it) (Pdb) step > (8)test_function() -> except StopIteration as ex: (Pdb) step > (9)test_function() -> if ex.value != 1: (Pdb) continue finished Nr r r r r %test_pdb_return_command_for_generator^s5r3cCsdS)aTesting no unwindng stack on yield for coroutines for "return" command >>> import asyncio >>> async def test_coro(): ... await asyncio.sleep(0) ... await asyncio.sleep(0) ... await asyncio.sleep(0) >>> async def test_main(): ... import pdb; pdb.Pdb(nosigint=True, readrc=False).set_trace() ... await test_coro() >>> def test_function(): ... loop = asyncio.new_event_loop() ... loop.run_until_complete(test_main()) ... loop.close() ... print("finished") >>> with PdbTestInput(['step', ... 'step', ... 'next', ... 'continue']): ... test_function() > (3)test_main() -> await test_coro() (Pdb) step --Call-- > (1)test_coro() -> async def test_coro(): (Pdb) step > (2)test_coro() -> await asyncio.sleep(0) (Pdb) next > (3)test_coro() -> await asyncio.sleep(0) (Pdb) continue finished Nr r r r r %test_pdb_return_command_for_coroutines'r4cCsdS)aTesting no unwindng stack on yield for generators for "until" command if target breakpoing is not reached >>> def test_gen(): ... yield 0 ... yield 1 ... yield 2 >>> def test_function(): ... import pdb; pdb.Pdb(nosigint=True, readrc=False).set_trace() ... for i in test_gen(): ... print(i) ... print("finished") >>> with PdbTestInput(['step', ... 'until 4', ... 'step', ... 'step', ... 'continue']): ... test_function() > (3)test_function() -> for i in test_gen(): (Pdb) step --Call-- > (1)test_gen() -> def test_gen(): (Pdb) until 4 0 1 > (4)test_gen() -> yield 2 (Pdb) step --Return-- > (4)test_gen()->2 -> yield 2 (Pdb) step > (4)test_function() -> print(i) (Pdb) continue 2 finished Nr r r r r $test_pdb_until_command_for_generators*r5cCsdS)aTesting no unwindng stack for coroutines for "until" command if target breakpoing is not reached >>> import asyncio >>> async def test_coro(): ... print(0) ... await asyncio.sleep(0) ... print(1) ... await asyncio.sleep(0) ... print(2) ... await asyncio.sleep(0) ... print(3) >>> async def test_main(): ... import pdb; pdb.Pdb(nosigint=True, readrc=False).set_trace() ... await test_coro() >>> def test_function(): ... loop = asyncio.new_event_loop() ... loop.run_until_complete(test_main()) ... loop.close() ... print("finished") >>> with PdbTestInput(['step', ... 'until 8', ... 'continue']): ... test_function() > (3)test_main() -> await test_coro() (Pdb) step --Call-- > (1)test_coro() -> async def test_coro(): (Pdb) until 8 0 1 2 > (8)test_coro() -> print(3) (Pdb) continue 3 finished Nr r r r r $test_pdb_until_command_for_coroutines,r6cCsdS)azThe next command on returning from a generator controlled by a for loop. >>> def test_gen(): ... yield 0 ... return 1 >>> def test_function(): ... import pdb; pdb.Pdb(nosigint=True, readrc=False).set_trace() ... for i in test_gen(): ... print('value', i) ... x = 123 >>> with PdbTestInput(['break test_gen', ... 'continue', ... 'next', ... 'next', ... 'next', ... 'continue']): ... test_function() > (3)test_function() -> for i in test_gen(): (Pdb) break test_gen Breakpoint 6 at :1 (Pdb) continue > (2)test_gen() -> yield 0 (Pdb) next value 0 > (3)test_gen() -> return 1 (Pdb) next Internal StopIteration: 1 > (3)test_function() -> for i in test_gen(): (Pdb) next > (5)test_function() -> x = 123 (Pdb) continue Nr r r r r +test_pdb_next_command_in_generator_for_loops'r7cCsdS)aThe next command in a generator with a subiterator. >>> def test_subgenerator(): ... yield 0 ... return 1 >>> def test_gen(): ... x = yield from test_subgenerator() ... return x >>> def test_function(): ... import pdb; pdb.Pdb(nosigint=True, readrc=False).set_trace() ... for i in test_gen(): ... print('value', i) ... x = 123 >>> with PdbTestInput(['step', ... 'step', ... 'next', ... 'next', ... 'next', ... 'continue']): ... test_function() > (3)test_function() -> for i in test_gen(): (Pdb) step --Call-- > (1)test_gen() -> def test_gen(): (Pdb) step > (2)test_gen() -> x = yield from test_subgenerator() (Pdb) next value 0 > (3)test_gen() -> return x (Pdb) next Internal StopIteration: 1 > (3)test_function() -> for i in test_gen(): (Pdb) next > (5)test_function() -> x = 123 (Pdb) continue Nr r r r r !test_pdb_next_command_subiteratorAs-r8cCsdS)a(Test for reference leaks when the SIGINT handler is set. >>> def test_function(): ... i = 1 ... while i <= 2: ... sess = pdb.Pdb() ... sess.set_trace(sys._getframe()) ... print('pdb %d: %s' % (i, sess._previous_sigint_handler)) ... i += 1 >>> with PdbTestInput(['continue', ... 'continue']): ... test_function() > (6)test_function() -> print('pdb %d: %s' % (i, sess._previous_sigint_handler)) (Pdb) continue pdb 1: > (5)test_function() -> sess.set_trace(sys._getframe()) (Pdb) continue pdb 2: Nr r r r r test_pdb_issue_20766psr9c@seZdZddZddZddZddZd d Zd d Zd dZ ddZ ddZ ddZ ddZ ddZddZddZddZdd Zd!d"Zd#d$Zd%d&Zd'd(Zd)d*Zd+d,Zd-S). PdbTestCasecCsttjdS)N)runlinkTESTFN)r r r r tearDownszPdbTestCase.tearDownc Csz|tjdtjddg|}tj|tjtjtjd}| t |\}}WdQRX|obt |}|opt |}||fS)N __pycache__z-mr()stdoutrstderr) addCleanuprrmtreer executable subprocessPopenPIPESTDOUT communicatestrencodebytesdecode)r Zpdb_argscommandscmdprocr?r@r r r _run_pdbs zPdbTestCase._run_pdbc CsFd}t|d}|t|WdQRX|tj|||g|S)z3Run 'script' lines with pdb and the pdb 'commands'.zmain.pywN)openwritetextwrapdedentrArr;rP)r scriptrMfilenamefr r r run_pdb_scripts  zPdbTestCase.run_pdb_scriptc Csd|_t|j|jd}|jd}t|jt|d}WdQRXt|d}|t|WdQRX| tj|j| d|jg|S)z(Runs the script code as part of a modulet_mainz /__main__.pyz /__init__.pyrQNz-m) module_namerrBosmkdirrRrSrTrUrArP)r rVrM main_file init_filerXr r r run_pdb_modules       zPdbTestCase.run_pdb_modulec Csbt|}ttjd}||WdQRX|s4dn|dtj|df}||t|tjdS)NrQr) rTrUrRrr<rS assertEqualr(Z find_function)r Z file_contentZ func_nameexpectedrXr r r _assert_find_functions z!PdbTestCase._assert_find_functioncCs|ddddS)NZfoo)rd)r r r r test_find_function_empty_filesz)PdbTestCase.test_find_function_empty_filecCs|ddddS)Nz def foo(): pass def bar(): pass def quux(): pass bar)rg)rd)r r r r test_find_function_founds z$PdbTestCase.test_find_function_foundc Csxttjd}|dWdQRXtjddtjg}tj|tjtjtj d}| |j j | d\}}|d|ddS) Nwbsprint("testing my pdb") z-mr()r?rr@squit s SyntaxErrorz0Got a syntax error running test script under PDB)rRrr<rSrrCrDrErFrGrAr?closerH assertNotIn)r rXrNrOr?r@r r r test_issue7964s zPdbTestCase.test_issue7964c Csrd}d}d}tdd}|t|WdQRX|tjd|||\}}|t dd| DddS) Nz from bar import bar def foo(): bar() def nope(): pass def foobar(): foo() nope() foobar() z from bar import bar break bar continue step step quit z5 def bar(): pass zbar.pyrQcss|]}d|kVqdS)zmain.py(5)foo()->NoneNr ).0lr r r sz.PdbTestCase.test_issue13183..z+Fail to step into the caller after a return) rRrSrTrUrArr;rY assertTrueany splitlines)r rVrMrgrXr?r@r r r test_issue13183s zPdbTestCase.test_issue13183c Csttjd}|tddWdQRXtjdtjg}t j |t j t j t j d}| |jj|d\}}|d|ddS) Nrja import threading import pdb def start_pdb(): pdb.Pdb(readrc=False).set_trace() x = 1 y = 1 t = threading.Thread(target=start_pdb) t.start()asciiz-u)r?rr@scont Errorz*Got an error running test script under PDB)rRrr<rSrTrUrJrrCrDrErFrGrAr?rkrHrlrL)r rXrNrOr?r@r r r test_issue13120s   zPdbTestCase.test_issue13120cCs6d}d}d}|||\}}|||d||dS)Nz def f: pass rez SyntaxError:zE Expected: {} Got: {} Fail to handle a syntax error in the debuggee.)rYassertInformat)r rVrMrcr?r@r r r test_issue16180szPdbTestCase.test_issue16180c Cstd}tjdd}zttdd}|dWdQRXtdd}||WdQRXt j dg}t j |t j t j t j d}|$|d\}}|d |WdQRXWdQRXWd|dk r|tjd<XdS) Nz_ import pdb; pdb.Pdb(readrc=False).set_trace() print('hello') HOMEz.pdbrcrQzinvalid zmain.py)r?rr@sq z(NameError: name 'invalid' is not defined)rTrUr\environpoprZtemp_cwdrRrSrrCrDrErFrHrlrL)r rVZ save_homerXrNrOr?r@r r r test_readrc_kwarg)s*     "zPdbTestCase.test_readrc_kwargc Csdt}d}t6}|td||ttjdtj|dWdQRX|| |ddS)Nz"Nobody expects... blah, blah, blahz sys.stdout set_trace)header ) rr enter_contextrobjectr(r)rrbgetvalue)r r?rZ resourcesr r r test_headerIszPdbTestCase.test_headercCs:d}d}|||\}}|tdd|D|dS)Nzprint("SUCCESS")z/ continue quit css|]}d|kVqdS)SUCCESSNr )rnror r r rpYsz.PdbTestCase.test_run_module..)r`rqrrrs)r rVrMr?r@r r r test_run_moduleRszPdbTestCase.test_run_modulecCs:d}d}|||\}}|tdd|D|dS)NzQ if __name__ == '__main__': print("SUCCESS") z/ continue quit css|]}d|kVqdS)rNr )rnror r r rpesz9PdbTestCase.test_module_is_run_as_main..)r`rqrrrs)r rVrMr?r@r r r test_module_is_run_as_main[sz&PdbTestCase.test_module_is_run_as_maincCsXd}d}|||\}}|tdd|D||tdd|D|dS)Nz{ if __name__ == '__main__': pass print("SUCCESS") pass z* b 3 quit css|]}d|kVqdS)zBreakpoint 1 atNr )rnror r r rpssz.PdbTestCase.test_breakpoint..css|]}d|kVqdS)rNr )rnror r r rpts)r`rqrrrsall)r rVrMr?r@r r r test_breakpointgs zPdbTestCase.test_breakpointcCs2d}|ddg|\}}|tj|dddS)Nz( c quit z-mr( re)rPrxr(Z_usagereplace)r rMr?r@r r r test_run_pdb_with_pdbvs z!PdbTestCase.test_run_pdb_with_pdbc Csld}t||d}t|t|d}WdQRX|tj||d|gd\}}|d|dS)NrZz /__init__.pyrQz-mrez,ImportError: No module named t_main.__main__) rrBr\r]rRrArPrxrs)r r[r_rXr?r@r r r test_module_without_a_mains    z&PdbTestCase.test_module_without_a_maincCs:d}d}|||\}}|tdd|D|dS)NzY #This is a comment, on line 2 print("SUCCESS") z quit css|]}d|kVqdS)z__main__.py(4)()Nr )rnror r r rpsz=PdbTestCase.test_blocks_at_first_code_line..)r`rqrrrs)r rVrMr?r@r r r test_blocks_at_first_code_lines  z*PdbTestCase.test_blocks_at_first_code_linec Cs2d|_t|j|jd}|jd}|jd}|tj|jt|jt|d}|t dWdQRXt|d}|t dWdQRXt|d}|t dWdQRXd }| d |jg|\}}| t d d | D|| t d d | D| t dd | DdS)NrZz /__main__.pyz /__init__.pyz /module.pyrQz6 top_var = "VAR from top" z from . import top_var from .module import var from . import module pass # We'll stop here and print the vars zY var = "VAR from module" var2 = "second var" zz b 5 c p top_var p var p module.var2 quit z-mcss|]}d|kVqdS)zVAR from moduleNr )rnror r r rpsz4PdbTestCase.test_relative_imports..css|]}d|kVqdS)z VAR from topNr )rnror r r rpscss|]}d|kVqdS)z second varNr )rnror r r rps)r[rrBrAr\r]rRrSrTrUrPrqrrrs)r r^r_ module_filerXrMr?_r r r test_relative_importss*        z!PdbTestCase.test_relative_importsc Csd|_t|j|jd}|jd}|jd}|tj|jt|jt|d}|t dWdQRXt|d}|t dWdQRXt|d}|t dWdQRXd }| d |jd g|\}}| t d d | D|dS)NrZz /runme.pyz /__init__.pyz /module.pyrQz6 top_var = "VAR from top" zl from . import module pass # We'll stop here and print the vars z5 var = "VAR from module" zQ b 3 c p module.var quit z-mz.runmecss|]}d|kVqdS)zVAR from moduleNr )rnror r r rpszDPdbTestCase.test_relative_imports_on_plain_module..)r[rrBrAr\r]rRrSrTrUrPrqrrrs)r r^r_rrXrMr?rr r r %test_relative_imports_on_plain_modules&        z1PdbTestCase.test_relative_imports_on_plain_modulec CsTdddddg}|d|d\}}||dddd d d d d d d dg dS)Nrzprint(z debug print(zdebug doesnotexistcreraz3(Pdb) *** SyntaxError: unexpected EOF while parsingz!(Pdb) ENTERING RECURSIVE DEBUGGERz-*** SyntaxError: unexpected EOF while parsingzLEAVING RECURSIVE DEBUGGERz> (1)()z9((Pdb)) *** NameError: name 'doesnotexist' is not definedz(Pdb) )joinrYrbrs)r rMr?rr r r test_errors_in_commands z"PdbTestCase.test_errors_in_commandN)rrrr=rPrYr`rdrfrirmrtrwrzr~rrrrrrrrrrr r r r r:s, $        %r:cGs*ddlm}ttt|g}t|S)Nr)test_pdb)testrunittestZ makeSuiter:doctestZ DocTestSuiteZ TestSuite)argsrZsuitesr r r load_testss  r__main__)5rr\r(rtypesrrDrT contextlibriorrrZtest.test_doctestrZ unittest.mockrrrrrrrr!r"r#r$ ModuleTypemodexec__dict__r%r&r,r-r.r/r0r1r2r3r4r5r6r7r8r9ZTestCaser:rrmainr r r r sZ     n`>  +*A;7<7),.)/n