a
    ze7                     @   s  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Zd dlZd dl	Z	d dl
mZmZ d dlmZ d dlmZ d dlmZ d dlmZ d dlmZ dd	 ZG d
d deZdd Zdd Zdd Zdd Zdd Zdd Zdd Z dd Z!dd Z"dd Z#e$d Z%e&d!e%j' d"d# Z(d$d% Z)d&d' Z*d(d) Z+d*d+ Z,d,d- Z-d.d/ Z.d0d1 Z/d2d3 Z0d4d5 Z1d6d7 Z2d8d9 Z3d:d; Z4d<d= Z5d>d? Z6d@dA Z7dBdC Z8G dDdE dEej9Z:G dFdG dGej9Z;dHdI Z<e=dJkre>  dS )K    N)	ExitStackredirect_stdout)StringIO)support)
_FakeInput)patch)
Breakpointc                   C   s   dt _i t _d gt _d S )N   )r   nextZbplistZ
bpbynumber r   r   #/usr/lib/python3.9/test/test_pdb.pyreset_Breakpoint   s    r   c                   @   s(   e Zd ZdZdd Zdd Zdd ZdS )	PdbTestInputz:Context manager that makes testing Pdb in doctests easier.c                 C   s
   || _ d S N)input)selfr   r   r   r   __init__    s    zPdbTestInput.__init__c                 C   s0   t j| _t| jt _tt dr&t  nd | _d S )Ngettrace)sysstdin
real_stdinr   r   hasattrr   
orig_tracer   r   r   r   	__enter__#   s    zPdbTestInput.__enter__c                 G   s   | j t_| jrt| j d S r   )r   r   r   r   settrace)r   excr   r   r   __exit__(   s    zPdbTestInput.__exit__N)__name__
__module____qualname____doc__r   r   r   r   r   r   r   r      s   r   c                   C   s   dS )a1  This 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)
    > <doctest test.test_pdb.test_pdb_displayhook[0]>(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.   s    r"   c                   C   s   dS )a  Test 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_function4(a, b, c, /):
    ...     pass

    >>> def test_function():
    ...     import pdb; pdb.Pdb(nosigint=True, readrc=False).set_trace()
    ...     ret = test_function_2('baz')
    ...     test_function3(kwonly=True)
    ...     test_function4(1, 2, 3)
    ...     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
    ...     'return',     # return out of function
    ...     'next',       # step to test_function4()
    ...     'step',       # stepping to test_function4()
    ...     'args',       # display function args
    ...     'continue',
    ... ]):
    ...    test_function()
    > <doctest test.test_pdb.test_pdb_basic_commands[3]>(3)test_function()
    -> ret = test_function_2('baz')
    (Pdb) step
    --Call--
    > <doctest test.test_pdb.test_pdb_basic_commands[0]>(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
    ...
      <doctest test.test_pdb.test_pdb_basic_commands[4]>(25)<module>()
    -> test_function()
      <doctest test.test_pdb.test_pdb_basic_commands[3]>(3)test_function()
    -> ret = test_function_2('baz')
    > <doctest test.test_pdb.test_pdb_basic_commands[0]>(1)test_function_2()
    -> def test_function_2(foo, bar='default'):
    (Pdb) up
    > <doctest test.test_pdb.test_pdb_basic_commands[3]>(3)test_function()
    -> ret = test_function_2('baz')
    (Pdb) down
    > <doctest test.test_pdb.test_pdb_basic_commands[0]>(1)test_function_2()
    -> def test_function_2(foo, bar='default'):
    (Pdb) next
    > <doctest test.test_pdb.test_pdb_basic_commands[0]>(2)test_function_2()
    -> print(foo)
    (Pdb) next
    baz
    > <doctest test.test_pdb.test_pdb_basic_commands[0]>(3)test_function_2()
    -> for i in range(5):
    (Pdb) step
    > <doctest test.test_pdb.test_pdb_basic_commands[0]>(4)test_function_2()
    -> print(i)
    (Pdb) until
    0
    1
    2
    3
    4
    > <doctest test.test_pdb.test_pdb_basic_commands[0]>(5)test_function_2()
    -> print(bar)
    (Pdb) next
    default
    > <doctest test.test_pdb.test_pdb_basic_commands[0]>(6)test_function_2()
    -> for i in range(10):
    (Pdb) jump 8
    > <doctest test.test_pdb.test_pdb_basic_commands[0]>(8)test_function_2()
    -> print('after for')
    (Pdb) return
    after for
    ...
    --Return--
    > <doctest test.test_pdb.test_pdb_basic_commands[0]>(10)test_function_2()->'BAZ'
    -> return foo.upper()
    (Pdb) retval
    'BAZ'
    (Pdb) next
    > <doctest test.test_pdb.test_pdb_basic_commands[3]>(4)test_function()
    -> test_function3(kwonly=True)
    (Pdb) step
    --Call--
    > <doctest test.test_pdb.test_pdb_basic_commands[1]>(1)test_function3()
    -> def test_function3(arg=None, *, kwonly=None):
    (Pdb) args
    arg = None
    kwonly = True
    (Pdb) return
    --Return--
    > <doctest test.test_pdb.test_pdb_basic_commands[1]>(2)test_function3()->None
    -> pass
    (Pdb) next
    > <doctest test.test_pdb.test_pdb_basic_commands[3]>(5)test_function()
    -> test_function4(1, 2, 3)
    (Pdb) step
    --Call--
    > <doctest test.test_pdb.test_pdb_basic_commands[2]>(1)test_function4()
    -> def test_function4(a, b, c, /):
    (Pdb) args
    a = 1
    b = 2
    c = 3
    (Pdb) continue
    BAZ
    Nr   r   r   r   r   test_pdb_basic_commandsK   s    r#   c                   C   s   dS )a  Test 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.

    >>> reset_Breakpoint()

    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()
    > <doctest test.test_pdb.test_pdb_breakpoint_commands[0]>(3)test_function()
    -> print(1)
    (Pdb) break 3
    Breakpoint 1 at <doctest test.test_pdb.test_pdb_breakpoint_commands[0]>:3
    (Pdb) disable 1
    Disabled breakpoint 1 at <doctest test.test_pdb.test_pdb_breakpoint_commands[0]>: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 <doctest test.test_pdb.test_pdb_breakpoint_commands[0]>:4
    (Pdb) break 4
    Breakpoint 3 at <doctest test.test_pdb.test_pdb_breakpoint_commands[0]>:4
    (Pdb) break
    Num Type         Disp Enb   Where
    1   breakpoint   keep no    at <doctest test.test_pdb.test_pdb_breakpoint_commands[0]>:3
            stop only if 1 < 2
            ignore next 10 hits
    2   breakpoint   keep yes   at <doctest test.test_pdb.test_pdb_breakpoint_commands[0]>:4
    3   breakpoint   keep yes   at <doctest test.test_pdb.test_pdb_breakpoint_commands[0]>:4
    (Pdb) clear 3
    Deleted breakpoint 3 at <doctest test.test_pdb.test_pdb_breakpoint_commands[0]>:4
    (Pdb) break
    Num Type         Disp Enb   Where
    1   breakpoint   keep no    at <doctest test.test_pdb.test_pdb_breakpoint_commands[0]>:3
            stop only if 1 < 2
            ignore next 10 hits
    2   breakpoint   keep yes   at <doctest test.test_pdb.test_pdb_breakpoint_commands[0]>:4
    (Pdb) condition 1
    Breakpoint 1 is now unconditional.
    (Pdb) enable 1
    Enabled breakpoint 1 at <doctest test.test_pdb.test_pdb_breakpoint_commands[0]>:3
    (Pdb) clear 1
    Deleted breakpoint 1 at <doctest test.test_pdb.test_pdb_breakpoint_commands[0]>:3
    (Pdb) commands 2
    (com) p "42"
    (com) print("42", 7*6)
    (com) end
    (Pdb) continue
    1
    '42'
    42 42
    > <doctest test.test_pdb.test_pdb_breakpoint_commands[0]>(4)test_function()
    -> print(2)
    (Pdb) clear
    Clear all breaks? y
    Deleted breakpoint 2 at <doctest test.test_pdb.test_pdb_breakpoint_commands[0]>:4
    (Pdb) tbreak 5
    Breakpoint 4 at <doctest test.test_pdb.test_pdb_breakpoint_commands[0]>:5
    (Pdb) continue
    2
    Deleted breakpoint 4 at <doctest test.test_pdb.test_pdb_breakpoint_commands[0]>:5
    > <doctest test.test_pdb.test_pdb_breakpoint_commands[0]>(5)test_function()
    -> print(3)
    (Pdb) break
    (Pdb) continue
    3
    4
    Nr   r   r   r   r   test_pdb_breakpoint_commands   s    r$   c                   C   s   dS )a  Test that do_p/do_pp do not swallow exceptions.

    >>> class BadRepr:
    ...     def __repr__(self):
    ...         raise Exception('repr_exc')
    >>> obj = BadRepr()

    >>> def test_function():
    ...     import pdb; pdb.Pdb(nosigint=True, readrc=False).set_trace()

    >>> with PdbTestInput([  # doctest: +NORMALIZE_WHITESPACE
    ...     'p obj',
    ...     'pp obj',
    ...     'continue',
    ... ]):
    ...    test_function()
    --Return--
    > <doctest test.test_pdb.test_pdb_pp_repr_exc[2]>(2)test_function()->None
    -> import pdb; pdb.Pdb(nosigint=True, readrc=False).set_trace()
    (Pdb) p obj
    *** Exception: repr_exc
    (Pdb) pp obj
    *** Exception: repr_exc
    (Pdb) continue
    Nr   r   r   r   r   test_pdb_pp_repr_excL  s    r%   c                   C   s   d S r   r   r   r   r   r   
do_nothingh  s    r&   c                   C   s   t d d S )N*   )printr   r   r   r   do_somethingk  s    r)   c                   C   s   dS )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()
    > <doctest test.test_pdb.test_list_commands[1]>(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--
    > <doctest test.test_pdb.test_list_commands[0]>(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
    > <doctest test.test_pdb.test_list_commands[0]>(2)test_function_2()
    -> import test.test_pdb
    (Pdb) next
    > <doctest test.test_pdb.test_list_commands[0]>(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_commandsn  s    r*   c                   C   s   dS )aI  Test the whatis command

    >>> myvar = (1,2)
    >>> def myfunc():
    ...     pass

    >>> class MyClass:
    ...    def mymethod(self):
    ...        pass

    >>> def test_function():
    ...   import pdb; pdb.Pdb(nosigint=True, readrc=False).set_trace()

    >>> with PdbTestInput([  # doctest: +ELLIPSIS, +NORMALIZE_WHITESPACE
    ...    'whatis myvar',
    ...    'whatis myfunc',
    ...    'whatis MyClass',
    ...    'whatis MyClass()',
    ...    'whatis MyClass.mymethod',
    ...    'whatis MyClass().mymethod',
    ...    'continue',
    ... ]):
    ...    test_function()
    --Return--
    > <doctest test.test_pdb.test_pdb_whatis_command[3]>(2)test_function()->None
    -> import pdb; pdb.Pdb(nosigint=True, readrc=False).set_trace()
    (Pdb) whatis myvar
    <class 'tuple'>
    (Pdb) whatis myfunc
    Function myfunc
    (Pdb) whatis MyClass
    Class test.test_pdb.MyClass
    (Pdb) whatis MyClass()
    <class 'test.test_pdb.MyClass'>
    (Pdb) whatis MyClass.mymethod
    Function mymethod
    (Pdb) whatis MyClass().mymethod
    Method mymethod
    (Pdb) continue
    Nr   r   r   r   r   test_pdb_whatis_command  s    r+   c                   C   s   dS )a  Test 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.')
    > <doctest test.test_pdb.test_post_mortem[1]>(3)test_function()
    -> test_function_2()
    (Pdb) next
    Exception!
    ZeroDivisionError: division by zero
    > <doctest test.test_pdb.test_post_mortem[1]>(3)test_function()
    -> test_function_2()
    (Pdb) bt
    ...
      <doctest test.test_pdb.test_post_mortem[2]>(10)<module>()
    -> test_function()
    > <doctest test.test_pdb.test_post_mortem[1]>(3)test_function()
    -> test_function_2()
      <doctest test.test_pdb.test_post_mortem[0]>(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
    > <doctest test.test_pdb.test_post_mortem[0]>(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_mortem  s    r,   c                   C   s   dS )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()
    > <doctest test.test_pdb.test_pdb_skip_modules[0]>(4)skip_module()
    -> string.capwords('FOO')
    (Pdb) step
    --Return--
    > <doctest test.test_pdb.test_pdb_skip_modules[0]>(4)skip_module()->None
    -> string.capwords('FOO')
    (Pdb) continue
    Nr   r   r   r   r   test_pdb_skip_modules5  s    r-   Zmodule_to_skipz6def foo_pony(callback): x = 1; callback(); return Nonec                   C   s   dS )aA  This 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
    > <doctest test.test_pdb.test_pdb_skip_modules_with_callback[0]>(5)skip_module()
    -> mod.foo_pony(callback)
    (Pdb) step
    --Call--
    > <doctest test.test_pdb.test_pdb_skip_modules_with_callback[0]>(2)callback()
    -> def callback():
    (Pdb) step
    > <doctest test.test_pdb.test_pdb_skip_modules_with_callback[0]>(3)callback()
    -> return None
    (Pdb) step
    --Return--
    > <doctest test.test_pdb.test_pdb_skip_modules_with_callback[0]>(3)callback()->None
    -> return None
    (Pdb) step
    --Return--
    > <doctest test.test_pdb.test_pdb_skip_modules_with_callback[0]>(5)skip_module()->None
    -> mod.foo_pony(callback)
    (Pdb) step
    > <doctest test.test_pdb.test_pdb_skip_modules_with_callback[1]>(10)<module>()
    -> pass  # provides something to "step" to
    (Pdb) continue
    Nr   r   r   r   r   #test_pdb_skip_modules_with_callbackQ  s    r.   c                   C   s   dS )a  Test 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()
    > <doctest test.test_pdb.test_pdb_continue_in_bottomframe[0]>(4)test_function()
    -> inst.botframe = sys._getframe()  # hackery to get the right botframe
    (Pdb) next
    > <doctest test.test_pdb.test_pdb_continue_in_bottomframe[0]>(5)test_function()
    -> print(1)
    (Pdb) break 7
    Breakpoint ... at <doctest test.test_pdb.test_pdb_continue_in_bottomframe[0]>:7
    (Pdb) continue
    1
    2
    > <doctest test.test_pdb.test_pdb_continue_in_bottomframe[0]>(7)test_function()
    -> print(3)
    (Pdb) next
    3
    > <doctest test.test_pdb.test_pdb_continue_in_bottomframe[0]>(8)test_function()
    -> print(4)
    (Pdb) continue
    4
    Nr   r   r   r   r    test_pdb_continue_in_bottomframe|  s    r/   c                 C   s   t tjddd| | dS )zRun pdb.method(arg).TF)ZnosigintZreadrcN)getattrpdbPdb)methodargr   r   r   
pdb_invoke  s    r5   c                   C   s   dS )a  Testing 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_argument  s    r6   c                   C   s   dS )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', '<string>', 'exec'))
    > <string>(1)<module>()...
    (Pdb) step
    --Return--
    > <string>(1)<module>()->None
    (Pdb) x
    1
    (Pdb) continue

    >>> with PdbTestInput(['x', 'continue']):
    ...     x=0
    ...     pdb_invoke('runeval', compile('x+1', '<string>', 'eval'))
    > <string>(1)<module>()->None
    (Pdb) x
    1
    (Pdb) continue
    Nr   r   r   r   r   test_pdb_run_with_code_object  s    r7   c                   C   s   dS )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

    >>> reset_Breakpoint()
    >>> with PdbTestInput(['break test_function_2',
    ...                    'continue',
    ...                    'return',
    ...                    'next',
    ...                    'continue',
    ...                    'return',
    ...                    'until',
    ...                    'continue',
    ...                    'return',
    ...                    'return',
    ...                    'continue']):
    ...     test_function()
    > <doctest test.test_pdb.test_next_until_return_at_return_event[1]>(3)test_function()
    -> test_function_2()
    (Pdb) break test_function_2
    Breakpoint 1 at <doctest test.test_pdb.test_next_until_return_at_return_event[0]>:1
    (Pdb) continue
    > <doctest test.test_pdb.test_next_until_return_at_return_event[0]>(2)test_function_2()
    -> x = 1
    (Pdb) return
    --Return--
    > <doctest test.test_pdb.test_next_until_return_at_return_event[0]>(3)test_function_2()->None
    -> x = 2
    (Pdb) next
    > <doctest test.test_pdb.test_next_until_return_at_return_event[1]>(4)test_function()
    -> test_function_2()
    (Pdb) continue
    > <doctest test.test_pdb.test_next_until_return_at_return_event[0]>(2)test_function_2()
    -> x = 1
    (Pdb) return
    --Return--
    > <doctest test.test_pdb.test_next_until_return_at_return_event[0]>(3)test_function_2()->None
    -> x = 2
    (Pdb) until
    > <doctest test.test_pdb.test_next_until_return_at_return_event[1]>(5)test_function()
    -> test_function_2()
    (Pdb) continue
    > <doctest test.test_pdb.test_next_until_return_at_return_event[0]>(2)test_function_2()
    -> x = 1
    (Pdb) return
    --Return--
    > <doctest test.test_pdb.test_next_until_return_at_return_event[0]>(3)test_function_2()->None
    -> x = 2
    (Pdb) return
    > <doctest test.test_pdb.test_next_until_return_at_return_event[1]>(6)test_function()
    -> end = 1
    (Pdb) continue
    Nr   r   r   r   r   &test_next_until_return_at_return_event  s    r8   c                   C   s   dS )aX  Testing 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()
    > <doctest test.test_pdb.test_pdb_next_command_for_generator[1]>(3)test_function()
    -> it = test_gen()
    (Pdb) step
    > <doctest test.test_pdb.test_pdb_next_command_for_generator[1]>(4)test_function()
    -> try:
    (Pdb) step
    > <doctest test.test_pdb.test_pdb_next_command_for_generator[1]>(5)test_function()
    -> if next(it) != 0:
    (Pdb) step
    --Call--
    > <doctest test.test_pdb.test_pdb_next_command_for_generator[0]>(1)test_gen()
    -> def test_gen():
    (Pdb) next
    > <doctest test.test_pdb.test_pdb_next_command_for_generator[0]>(2)test_gen()
    -> yield 0
    (Pdb) next
    > <doctest test.test_pdb.test_pdb_next_command_for_generator[0]>(3)test_gen()
    -> return 1
    (Pdb) step
    --Return--
    > <doctest test.test_pdb.test_pdb_next_command_for_generator[0]>(3)test_gen()->1
    -> return 1
    (Pdb) step
    StopIteration: 1
    > <doctest test.test_pdb.test_pdb_next_command_for_generator[1]>(7)test_function()
    -> next(it)
    (Pdb) continue
    finished
    Nr   r   r   r   r   #test_pdb_next_command_for_generator  s    r9   c                   C   s   dS )a,  Testing 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()
    ...     asyncio.set_event_loop_policy(None)
    ...     print("finished")

    >>> with PdbTestInput(['step',
    ...                    'step',
    ...                    'next',
    ...                    'next',
    ...                    'next',
    ...                    'step',
    ...                    'continue']):
    ...     test_function()
    > <doctest test.test_pdb.test_pdb_next_command_for_coroutine[2]>(3)test_main()
    -> await test_coro()
    (Pdb) step
    --Call--
    > <doctest test.test_pdb.test_pdb_next_command_for_coroutine[1]>(1)test_coro()
    -> async def test_coro():
    (Pdb) step
    > <doctest test.test_pdb.test_pdb_next_command_for_coroutine[1]>(2)test_coro()
    -> await asyncio.sleep(0)
    (Pdb) next
    > <doctest test.test_pdb.test_pdb_next_command_for_coroutine[1]>(3)test_coro()
    -> await asyncio.sleep(0)
    (Pdb) next
    > <doctest test.test_pdb.test_pdb_next_command_for_coroutine[1]>(4)test_coro()
    -> await asyncio.sleep(0)
    (Pdb) next
    Internal StopIteration
    > <doctest test.test_pdb.test_pdb_next_command_for_coroutine[2]>(3)test_main()
    -> await test_coro()
    (Pdb) step
    --Return--
    > <doctest test.test_pdb.test_pdb_next_command_for_coroutine[2]>(3)test_main()->None
    -> await test_coro()
    (Pdb) continue
    finished
    Nr   r   r   r   r   #test_pdb_next_command_for_coroutineL  s    r:   c                   C   s   dS )a*  Testing 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()
    ...     asyncio.set_event_loop_policy(None)
    ...     print("finished")

    >>> with PdbTestInput(['step',
    ...                    'step',
    ...                    'next',
    ...                    'next',
    ...                    'step',
    ...                    'next',
    ...                    'continue']):
    ...     test_function()
    > <doctest test.test_pdb.test_pdb_next_command_for_asyncgen[3]>(3)test_main()
    -> await test_coro()
    (Pdb) step
    --Call--
    > <doctest test.test_pdb.test_pdb_next_command_for_asyncgen[2]>(1)test_coro()
    -> async def test_coro():
    (Pdb) step
    > <doctest test.test_pdb.test_pdb_next_command_for_asyncgen[2]>(2)test_coro()
    -> async for x in agen():
    (Pdb) next
    > <doctest test.test_pdb.test_pdb_next_command_for_asyncgen[2]>(3)test_coro()
    -> print(x)
    (Pdb) next
    1
    > <doctest test.test_pdb.test_pdb_next_command_for_asyncgen[2]>(2)test_coro()
    -> async for x in agen():
    (Pdb) step
    --Call--
    > <doctest test.test_pdb.test_pdb_next_command_for_asyncgen[1]>(2)agen()
    -> yield 1
    (Pdb) next
    > <doctest test.test_pdb.test_pdb_next_command_for_asyncgen[1]>(3)agen()
    -> await asyncio.sleep(0)
    (Pdb) continue
    2
    finished
    Nr   r   r   r   r   "test_pdb_next_command_for_asyncgen  s    r;   c                   C   s   dS )a  Testing 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()
    > <doctest test.test_pdb.test_pdb_return_command_for_generator[1]>(3)test_function()
    -> it = test_gen()
    (Pdb) step
    > <doctest test.test_pdb.test_pdb_return_command_for_generator[1]>(4)test_function()
    -> try:
    (Pdb) step
    > <doctest test.test_pdb.test_pdb_return_command_for_generator[1]>(5)test_function()
    -> if next(it) != 0:
    (Pdb) step
    --Call--
    > <doctest test.test_pdb.test_pdb_return_command_for_generator[0]>(1)test_gen()
    -> def test_gen():
    (Pdb) return
    StopIteration: 1
    > <doctest test.test_pdb.test_pdb_return_command_for_generator[1]>(7)test_function()
    -> next(it)
    (Pdb) step
    > <doctest test.test_pdb.test_pdb_return_command_for_generator[1]>(8)test_function()
    -> except StopIteration as ex:
    (Pdb) step
    > <doctest test.test_pdb.test_pdb_return_command_for_generator[1]>(9)test_function()
    -> if ex.value != 1:
    (Pdb) continue
    finished
    Nr   r   r   r   r   %test_pdb_return_command_for_generator  s    r<   c                   C   s   dS )a%  Testing 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()
    ...     asyncio.set_event_loop_policy(None)
    ...     print("finished")

    >>> with PdbTestInput(['step',
    ...                    'step',
    ...                    'next',
    ...                    'continue']):
    ...     test_function()
    > <doctest test.test_pdb.test_pdb_return_command_for_coroutine[2]>(3)test_main()
    -> await test_coro()
    (Pdb) step
    --Call--
    > <doctest test.test_pdb.test_pdb_return_command_for_coroutine[1]>(1)test_coro()
    -> async def test_coro():
    (Pdb) step
    > <doctest test.test_pdb.test_pdb_return_command_for_coroutine[1]>(2)test_coro()
    -> await asyncio.sleep(0)
    (Pdb) next
    > <doctest test.test_pdb.test_pdb_return_command_for_coroutine[1]>(3)test_coro()
    -> await asyncio.sleep(0)
    (Pdb) continue
    finished
    Nr   r   r   r   r   %test_pdb_return_command_for_coroutine  s    r=   c                   C   s   dS )a  Testing no unwindng stack on yield for generators
       for "until" command if target breakpoint 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()
    > <doctest test.test_pdb.test_pdb_until_command_for_generator[1]>(3)test_function()
    -> for i in test_gen():
    (Pdb) step
    --Call--
    > <doctest test.test_pdb.test_pdb_until_command_for_generator[0]>(1)test_gen()
    -> def test_gen():
    (Pdb) until 4
    0
    1
    > <doctest test.test_pdb.test_pdb_until_command_for_generator[0]>(4)test_gen()
    -> yield 2
    (Pdb) step
    --Return--
    > <doctest test.test_pdb.test_pdb_until_command_for_generator[0]>(4)test_gen()->2
    -> yield 2
    (Pdb) step
    > <doctest test.test_pdb.test_pdb_until_command_for_generator[1]>(4)test_function()
    -> print(i)
    (Pdb) continue
    2
    finished
    Nr   r   r   r   r   $test_pdb_until_command_for_generator"  s    r>   c                   C   s   dS )a  Testing no unwindng stack for coroutines
       for "until" command if target breakpoint 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()
    ...     asyncio.set_event_loop_policy(None)
    ...     print("finished")

    >>> with PdbTestInput(['step',
    ...                    'until 8',
    ...                    'continue']):
    ...     test_function()
    > <doctest test.test_pdb.test_pdb_until_command_for_coroutine[2]>(3)test_main()
    -> await test_coro()
    (Pdb) step
    --Call--
    > <doctest test.test_pdb.test_pdb_until_command_for_coroutine[1]>(1)test_coro()
    -> async def test_coro():
    (Pdb) until 8
    0
    1
    2
    > <doctest test.test_pdb.test_pdb_until_command_for_coroutine[1]>(8)test_coro()
    -> print(3)
    (Pdb) continue
    3
    finished
    Nr   r   r   r   r   $test_pdb_until_command_for_coroutineN  s    r?   c                   C   s   dS )a  The 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

    >>> reset_Breakpoint()
    >>> with PdbTestInput(['break test_gen',
    ...                    'continue',
    ...                    'next',
    ...                    'next',
    ...                    'next',
    ...                    'continue']):
    ...     test_function()
    > <doctest test.test_pdb.test_pdb_next_command_in_generator_for_loop[1]>(3)test_function()
    -> for i in test_gen():
    (Pdb) break test_gen
    Breakpoint 1 at <doctest test.test_pdb.test_pdb_next_command_in_generator_for_loop[0]>:1
    (Pdb) continue
    > <doctest test.test_pdb.test_pdb_next_command_in_generator_for_loop[0]>(2)test_gen()
    -> yield 0
    (Pdb) next
    value 0
    > <doctest test.test_pdb.test_pdb_next_command_in_generator_for_loop[0]>(3)test_gen()
    -> return 1
    (Pdb) next
    Internal StopIteration: 1
    > <doctest test.test_pdb.test_pdb_next_command_in_generator_for_loop[1]>(3)test_function()
    -> for i in test_gen():
    (Pdb) next
    > <doctest test.test_pdb.test_pdb_next_command_in_generator_for_loop[1]>(5)test_function()
    -> x = 123
    (Pdb) continue
    Nr   r   r   r   r   +test_pdb_next_command_in_generator_for_loop}  s    r@   c                   C   s   dS )a  The 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()
    > <doctest test.test_pdb.test_pdb_next_command_subiterator[2]>(3)test_function()
    -> for i in test_gen():
    (Pdb) step
    --Call--
    > <doctest test.test_pdb.test_pdb_next_command_subiterator[1]>(1)test_gen()
    -> def test_gen():
    (Pdb) step
    > <doctest test.test_pdb.test_pdb_next_command_subiterator[1]>(2)test_gen()
    -> x = yield from test_subgenerator()
    (Pdb) next
    value 0
    > <doctest test.test_pdb.test_pdb_next_command_subiterator[1]>(3)test_gen()
    -> return x
    (Pdb) next
    Internal StopIteration: 1
    > <doctest test.test_pdb.test_pdb_next_command_subiterator[2]>(3)test_function()
    -> for i in test_gen():
    (Pdb) next
    > <doctest test.test_pdb.test_pdb_next_command_subiterator[2]>(5)test_function()
    -> x = 123
    (Pdb) continue
    Nr   r   r   r   r   !test_pdb_next_command_subiterator  s    rA   c                   C   s   dS )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()
    > <doctest test.test_pdb.test_pdb_issue_20766[0]>(6)test_function()
    -> print('pdb %d: %s' % (i, sess._previous_sigint_handler))
    (Pdb) continue
    pdb 1: <built-in function default_int_handler>
    > <doctest test.test_pdb.test_pdb_issue_20766[0]>(5)test_function()
    -> sess.set_trace(sys._getframe())
    (Pdb) continue
    pdb 2: <built-in function default_int_handler>
    Nr   r   r   r   r   test_pdb_issue_20766  s    rB   c                   C   s   dS )aN  echo breakpoints cleared with filename:lineno

    >>> def test_function():
    ...     import pdb; pdb.Pdb(nosigint=True, readrc=False).set_trace()
    ...     print(1)
    ...     print(2)
    ...     print(3)
    ...     print(4)
    >>> reset_Breakpoint()
    >>> with PdbTestInput([  # doctest: +NORMALIZE_WHITESPACE
    ...     'break 3',
    ...     'clear <doctest test.test_pdb.test_pdb_issue_43318[0]>:3',
    ...     'continue'
    ... ]):
    ...     test_function()
    > <doctest test.test_pdb.test_pdb_issue_43318[0]>(3)test_function()
    -> print(1)
    (Pdb) break 3
    Breakpoint 1 at <doctest test.test_pdb.test_pdb_issue_43318[0]>:3
    (Pdb) clear <doctest test.test_pdb.test_pdb_issue_43318[0]>:3
    Deleted breakpoint 1 at <doctest test.test_pdb.test_pdb_issue_43318[0]>:3
    (Pdb) continue
    1
    2
    3
    4
    Nr   r   r   r   r   test_pdb_issue_43318  s    rC   c                   @   s  e Zd Zdd Zdd Zdd Zdd Zd	d
 Zdd Zdd Z	dd Z
dd Zdd Zdd Zdd Zdd Zdd Zdd Zdd  Zd!d" Zd#d$ Zd%d& Zd'd( Zd)d* Zd+d, Zd-d. Zd/d0 Zd1d2 Zd3d4 Zd5d6 Zd7d8 Zd9d: Zd;d< Z d=d> Z!e"j#d?d@ Z$dAdB Z%dCS )DPdbTestCasec                 C   s   t t j d S r   r   unlinkTESTFNr   r   r   r   tearDown  s    zPdbTestCase.tearDownc                 C   s   |  tjd tjddg| }tj|tjtjtji t	j
ddid$}|t|\}}W d    n1 sp0    Y  |ot|}|ot|}||fS )N__pycache__-mr1   PYTHONIOENCODINGutf-8stdoutr   stderrenv)
addCleanupr   rmtreer   
executable
subprocessPopenPIPESTDOUTosenvironcommunicatestrencodebytesdecode)r   Zpdb_argscommandscmdprocrN   rO   r   r   r   _run_pdb  s    2zPdbTestCase._run_pdbc                 C   sZ   d}t |d }|t| W d   n1 s40    Y  | tj| | |g|S )z3Run 'script' lines with pdb and the pdb 'commands'.main.pywN)openwritetextwrapdedentrQ   r   rF   rb   )r   scriptr_   filenamefr   r   r   run_pdb_script  s
    .zPdbTestCase.run_pdb_scriptc                 C   s   d| _ t| j  | j d }| j d }t| j  t|d}W d   n1 sR0    Y  t|d }|t| W d   n1 s0    Y  | 	tj| j  | 
d| j g|S )z(Runs the script code as part of a modulet_main/__main__.py/__init__.pyrd   NrJ   )module_namer   rR   rX   mkdirre   rf   rg   rh   rQ   rb   )r   ri   r_   	main_file	init_filerk   r   r   r   run_pdb_module'  s    

.zPdbTestCase.run_pdb_modulec                 C   sl   t tjd}|| W d    n1 s,0    Y  |s>d n|d tj|d f}| |t|tj d S )Nwbr   r	   )re   r   rG   rf   assertEqualr1   Zfind_function)r   Zfile_content	func_nameexpectedrk   r   r   r   _assert_find_function5  s    (z!PdbTestCase._assert_find_functionc                 C   s   |  ddd  d S )N    Zfoo)ry   r   r   r   r   test_find_function_empty_file>  s    z)PdbTestCase.test_find_function_empty_filec                 C   s   |  d dd d S )Nu@   def foo():
    pass

def bœr():
    pass

def quux():
    pass
   bœr)r|      ry   r\   r   r   r   r   test_find_function_foundA  s
    
z$PdbTestCase.test_find_function_foundc                 C   s   |  dddd d S )NuV   # coding: iso-8859-15
def foo():
    pass

def bœr():
    pass

def quux():
    pass
ziso-8859-15r|   )r|      r~   r   r   r   r   -test_find_function_found_with_encoding_cookieQ  s    
z9PdbTestCase.test_find_function_found_with_encoding_cookiec                 C   s   |  tjd  dd d S )Nu   def bœr():
    pass
r|   )r|   r	   )ry   codecsBOM_UTF8r\   r   r   r   r   !test_find_function_found_with_bomb  s
    z-PdbTestCase.test_find_function_found_with_bomc                 C   s   t tjd}|d W d    n1 s,0    Y  tjddtjg}tj|tjtjtj	d}| 
|jj |d\}}| d|d d S )	Nru   s   print("testing my pdb")
rJ   r1   rN   r   rO   s   quit
s   SyntaxErrorz0Got a syntax error running test script under PDB)re   r   rG   rf   r   rS   rT   rU   rV   rW   rQ   rN   closerZ   assertNotInr   rk   r`   ra   rN   rO   r   r   r   test_issue7964l  s    (zPdbTestCase.test_issue7964c                 C   sH   d}d}|  ||\}}|pd|p$d }| d|d | d|d d S )Nz
            def do_testcmdwithnodocs(self, arg):
                pass

            import pdb
            pdb.Pdb.do_testcmdwithnodocs = do_testcmdwithnodocs
        zA
            continue
            help testcmdwithnodocs
         AttributeErrorzCCalling help on a command with no docs should be handled gracefullyz;*** No help for 'testcmdwithnodocs'; __doc__ string missingz<Calling help on a command with no docs should print an error)rl   r   assertIn)r   ri   r_   rN   rO   outputr   r   r   test_issue46434{  s    zPdbTestCase.test_issue46434c                 C   s   d}d}d}t dd }|t| W d    n1 s<0    Y  | tjd | ||\}}| t	dd |
 D d d S )	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
        zbar.pyrd   c                 s   s   | ]}d |v V  qdS )zmain.py(5)foo()->NoneNr   .0lr   r   r   	<genexpr>  rz   z.PdbTestCase.test_issue13183.<locals>.<genexpr>z+Fail to step into the caller after a return)re   rf   rg   rh   rQ   r   rF   rl   
assertTrueany
splitlines)r   ri   r_   Zbarrk   rN   rO   r   r   r   test_issue13183  s    .zPdbTestCase.test_issue13183c                 C   s   t tjd&}|tdd W d    n1 s80    Y  tjdtjg}t	j
|t	jt	jt	ji tjddid}| |jj |d\}}| d	|d
 d S )Nru   a  
                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()ascii-urK   rL   rM   s   cont
   Error*Got an error running test script under PDBre   r   rG   rf   rg   rh   r\   r   rS   rT   rU   rV   rW   rX   rY   rQ   rN   r   rZ   r   r   r   r   r   test_issue13120  s     
$zPdbTestCase.test_issue13120c                 C   s   t tjd&}|tdd W d    n1 s80    Y  tjdtjg}t	j
|t	jt	jt	ji tjddid}| |jj |d\}}| d	|d
 d S )Nru   a  
                import threading
                import pdb

                evt = threading.Event()

                def start_pdb():
                    evt.wait()
                    pdb.Pdb(readrc=False).set_trace()

                t = threading.Thread(target=start_pdb)
                t.start()
                pdb.Pdb(readrc=False).set_trace()
                evt.set()
                t.join()r   r   rK   rL   rM   s
   cont
cont
r   r   r   r   r   r   r   test_issue36250  s     $zPdbTestCase.test_issue36250c                 C   s6   d}d}d}|  ||\}}| ||d|| d S )Nzdef f: pass
r   zSyntaxError:zE

Expected:
{}
Got:
{}
Fail to handle a syntax error in the debuggee.)rl   r   format)r   ri   r_   rx   rN   rO   r   r   r   test_issue16180  s    zPdbTestCase.test_issue16180c                 C   sL   d}d}|  ||\}}ddd | D }| |d | |d d S )Nzprint('hello')z[
            continue
            run a b c
            run d e f
            quit
        
c                 S   s   g | ]}|  qS r   )strip)r   xr   r   r   
<listcomp>  rz   z/PdbTestCase.test_issue26053.<locals>.<listcomp>z#Restarting .* with arguments:
a b cz#Restarting .* with arguments:
d e f)rl   joinr   ZassertRegex)r   ri   r_   rN   rO   resr   r   r   test_issue26053  s    zPdbTestCase.test_issue26053c              	   C   s@  t d}tjdd }z
t  tdd}|d W d    n1 sP0    Y  tdd}|| W d    n1 s0    Y  t	j
dg}tj|tjtjtjd}|* |d\}}| d	| W d    n1 s0    Y  W d    n1 s0    Y  W |d ur<|tjd< n|d ur:|tjd< 0 d S )
Nz_
            import pdb; pdb.Pdb(readrc=False).set_trace()

            print('hello')
        HOME.pdbrcrd   zinvalid
rc   r   s   q
s(   NameError: name 'invalid' is not defined)rg   rh   rX   rY   popr   temp_cwdre   rf   r   rS   rT   rU   rV   rZ   r   )r   ri   	save_homerk   r`   ra   rN   rO   r   r   r   test_readrc_kwarg  s0    

((
D

zPdbTestCase.test_readrc_kwargc                 C   s   t jdd }t }td t j|d}|t jj_	z`t
|d}|d W d    n1 sf0    Y  | t jd d W |d ur|t jd< n|d ur|t jd< 0 W d    n1 s0    Y  W d    n1 s0    Y  d S )Nr   zos.path.expanduserr   rd   invalidr   )rX   rY   r   r   temp_dirr   pathr   
expanduserZreturn_valuere   rf   rv   r1   r2   ZrcLines)r   r   r   Zrc_pathrk   r   r   r   test_readrc_homedir'  s    
(zPdbTestCase.test_readrc_homedirc                 C   sx   t  }d}t @}|td| |ttjd tj|d W d    n1 sV0    Y  | |	 |d  d S )Nz"Nobody expects... blah, blah, blahz
sys.stdout	set_trace)headerr   )
r   r   enter_contextr   objectr1   r2   r   rv   getvalue)r   rN   r   Z	resourcesr   r   r   test_header4  s    *zPdbTestCase.test_headerc                 C   s:   d}d}|  ||\}}| tdd | D | d S )Nzprint("SUCCESS")/
            continue
            quit
        c                 s   s   | ]}d |v V  qdS SUCCESSNr   r   r   r   r   r   D  rz   z.PdbTestCase.test_run_module.<locals>.<genexpr>rt   r   r   r   r   ri   r_   rN   rO   r   r   r   test_run_module=  s    zPdbTestCase.test_run_modulec                 C   s:   d}d}|  ||\}}| tdd | D | d S )NzQ
            if __name__ == '__main__':
                print("SUCCESS")
        r   c                 s   s   | ]}d |v V  qdS r   r   r   r   r   r   r   P  rz   z9PdbTestCase.test_module_is_run_as_main.<locals>.<genexpr>r   r   r   r   r   test_module_is_run_as_mainF  s    z&PdbTestCase.test_module_is_run_as_mainc                 C   sX   d}d}|  ||\}}| tdd | D | | tdd | D | d S )Nz{
            if __name__ == '__main__':
                pass
                print("SUCCESS")
                pass
        z*
            b 3
            quit
        c                 s   s   | ]}d |v V  qdS )zBreakpoint 1 atNr   r   r   r   r   r   ^  rz   z.PdbTestCase.test_breakpoint.<locals>.<genexpr>c                 s   s   | ]}d |vV  qdS r   r   r   r   r   r   r   _  rz   )rt   r   r   r   allr   r   r   r   test_breakpointR  s
    zPdbTestCase.test_breakpointc                 C   s2   d}|  ddg|\}}| tj|dd d S )Nz(
            c
            quit
        rJ   r1   r   )rb   r   r1   Z_usagereplace)r   r_   rN   rO   r   r   r   test_run_pdb_with_pdba  s    
z!PdbTestCase.test_run_pdb_with_pdbc                 C   s   d}t | |d }t| t|d W d    n1 s@0    Y  | t j| | d|gd\}}| d|  d S )Nrm   ro   rd   rJ   r   z,ImportError: No module named t_main.__main__)	r   rR   rX   rq   re   rQ   rb   r   r   )r   rp   rs   rN   rO   r   r   r   test_module_without_a_mainl  s    

z&PdbTestCase.test_module_without_a_mainc                 C   s   d}d}t | |d | }t| t|d d W d    n1 sL0    Y  | t j| | d|ddgd\}}| d	| d S )
NZt_pkgrm   /ro   rd   rJ   .r   z;'t_pkg.t_main' is a package and cannot be directly executed)	r   rR   rX   makedirsre   rQ   rb   r   r   )r   Zpkg_namerp   modpathrN   rO   r   r   r   test_package_without_a_mainx  s    

z'PdbTestCase.test_package_without_a_mainc                 C   s:   d}d}|  ||\}}| tdd | D | d S )NzY
                #This is a comment, on line 2

                print("SUCCESS")
        z
            quit
        c                 s   s   | ]}d |v V  qdS )z__main__.py(4)<module>()Nr   r   r   r   r   r     s   z=PdbTestCase.test_blocks_at_first_code_line.<locals>.<genexpr>r   r   r   r   r   test_blocks_at_first_code_line  s    z*PdbTestCase.test_blocks_at_first_code_linec                 C   sn  d| _ t| j  | j d }| j d }| j d }| tj| j  t| j  t|d }|t	d W d    n1 s|0    Y  t|d }|t	d W d    n1 s0    Y  t|d }|t	d W d    n1 s0    Y  d	}| 
d
| j g|\}}| tdd | D | | tdd | D  | tdd | D  d S )Nrm   rn   ro   
/module.pyrd   6
                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
        rJ   c                 s   s   | ]}d |v V  qdS zVAR from moduleNr   r   r   r   r   r     rz   z4PdbTestCase.test_relative_imports.<locals>.<genexpr>c                 s   s   | ]}d |v V  qdS )zVAR from topNr   r   r   r   r   r     rz   c                 s   s   | ]}d |v V  qdS )z
second varNr   r   r   r   r   r     rz   rp   r   rR   rQ   rX   rq   re   rf   rg   rh   rb   r   r   r   r   rr   rs   Zmodule_filerk   r_   rN   _r   r   r   test_relative_imports  s$    


...z!PdbTestCase.test_relative_importsc                 C   s:  d| _ t| j  | j d }| j d }| j d }| tj| j  t| j  t|d }|t	d W d    n1 s|0    Y  t|d }|t	d W d    n1 s0    Y  t|d }|t	d W d    n1 s0    Y  d	}| 
d
| j d g|\}}| tdd | D | d S )Nrm   z	/runme.pyro   r   rd   r   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
        rJ   z.runmec                 s   s   | ]}d |v V  qdS r   r   r   r   r   r   r     rz   zDPdbTestCase.test_relative_imports_on_plain_module.<locals>.<genexpr>r   r   r   r   r   %test_relative_imports_on_plain_module  s     


...z1PdbTestCase.test_relative_imports_on_plain_modulec                 C   sB   d g d}| d|d \}}| | dd  g d d S )Nr   )zprint(zdebug print(zdebug doesnotexistcr   r	   )	z3(Pdb) *** SyntaxError: unexpected EOF while parsing!(Pdb) ENTERING RECURSIVE DEBUGGERz-*** SyntaxError: unexpected EOF while parsingLEAVING RECURSIVE DEBUGGERr   z> <string>(1)<module>()z9((Pdb)) *** NameError: name 'doesnotexist' is not definedr   (Pdb) r   rl   rv   r   )r   r_   rN   r   r   r   r   test_errors_in_command  s    z"PdbTestCase.test_errors_in_commandc                    s$    fdd}|dd |dd dS )z.do_run handles exceptions from parsing its argc                    sV   d d|  dg} d|d \}} | dd  dd|  d| d	g d S )
Nr   zrun qpassr	   z-> passz(Pdb) *** Cannot run z: r   r   )Zbad_argmsgr_   rN   r   r   r   r   check  s    z*PdbTestCase.test_issue34266.<locals>.check\zNo escaped character"zNo closing quotationNr   )r   r   r   r   r   test_issue34266  s    
zPdbTestCase.test_issue34266c                 C   sx   t d}d}t N}dtj| }| ||\}}| |	dd 
d| W d   n1 sj0    Y  dS )zkWhen running `python foo.py` sys.path[0] is an absolute path. `python -m pdb foo.py` should behave the sameQ
            import sys
            print('sys.path[0] is', sys.path[0])
        c
q(Pdb) sys.path[0] is r      r   N)rg   rh   r   r   rX   r   realpathrl   rv   splitrstrip)r   ri   r_   cwdrx   rN   rO   r   r   r   test_issue42384  s    

zPdbTestCase.test_issue42384c           
   	   C   s  t d}d}t }tj|}tj|d}tj|d}d| }t| t	tj|dd}|
| W d   n1 s0    Y  t| ttj|dtj|d | tjddg|\}}	| |d	d
 d| W d   n1 s
0    Y  dS )ziWhen running `python foo.py` sys.path[0] resolves symlinks. `python -m pdb foo.py` should behave the samer   r   dir_onedir_twor   foo.pyrd   Nr   r   r   )rg   rh   r   r   rX   r   r   r   rq   re   rf   symlinkrb   rv   r   r   )
r   ri   r_   r   r   r   rx   rk   rN   rO   r   r   r   test_issue42384_symlink  s    



(
 z#PdbTestCase.test_issue42384_symlinkc           	   	   C   s
  t  }tdd$}td}|| W d    n1 s>0    Y  tj|d}t	| t	tj|d tj|d}t|d}|d W d    n1 s0    Y  | 
dgd\}}d}| |dd	 d
| W d    n1 s0    Y  d S )Nr   rd   z
                    print('The correct file was executed')

                    import os
                    os.chdir("subdir")
                subdirz$print("The wrong file was executed")zc
c
qz#(Pdb) The correct file was executedr      r   )r   r   re   rg   rh   rf   rX   r   r   rq   rb   rv   r   r   )	r   r   rk   sr   Z
wrong_filerN   rO   rx   r   r   r   test_issue42383&  s    

(
(zPdbTestCase.test_issue42383N)&r   r   r    rH   rb   rl   rt   ry   r{   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   Zskip_unless_symlinkr   r   r   r   r   r   rD     sD   	
$ 		%
rD   c                   @   s4   e Zd Zdd Zdd Zdd Zdd Zd	d
 ZdS )ChecklineTestsc                 C   s   t   d S r   )	linecache
clearcacher   r   r   r   setUp?  s    zChecklineTests.setUpc                 C   s   t t j d S r   rE   r   r   r   r   rH   B  s    zChecklineTests.tearDownc                 C   sX   t tjd}|d W d    n1 s,0    Y  t }| |tjdd d S Nrd   z
print(123)r	   )re   r   rG   rf   r1   r2   rv   	checkliner   rk   dbr   r   r   test_checkline_before_debuggingE  s    (z.ChecklineTests.test_checkline_before_debuggingc                 C   s`   t tjd}|d W d    n1 s,0    Y  t }|  | |tjdd d S r   )	re   r   rG   rf   r1   r2   resetrv   r   r   r   r   r   test_checkline_after_resetK  s
    (z)ChecklineTests.test_checkline_after_resetc                 C   s   t d}ttjd}|| W d    n1 s60    Y  t| d }tt	 : t
 }t|D ]}| |tj| qlW d    n1 s0    Y  d S )Nz\
            # Comment
            """ docstring """
            ''' docstring '''

        rd   r   )rg   rh   re   r   rG   rf   lenr   r   r   r1   r2   rangeZassertFalser   )r   r   rk   Z	num_linesr   linenor   r   r    test_checkline_is_not_executableR  s    
(z/ChecklineTests.test_checkline_is_not_executableN)r   r   r    r   rH   r   r   r   r   r   r   r   r   >  s
   r   c                  G   s2   ddl m} ttttt|g}t|S )Nr   )test_pdb)	testr  unittestZ	makeSuiterD   r   doctestZDocTestSuiteZ	TestSuite)argsr  Zsuitesr   r   r   
load_testsc  s    r  __main__)?r  rX   r1   r   typesr   r  rT   rg   r   
contextlibr   r   ior   r  r   Ztest.test_doctestr   Zunittest.mockr   Zbdbr   r   r   r   r"   r#   r$   r%   r&   r)   r*   r+   r,   r-   
ModuleTypemodexec__dict__r.   r/   r5   r6   r7   r8   r9   r:   r;   r<   r=   r>   r?   r@   rA   rB   rC   ZTestCaserD   r   r  r   mainr   r   r   r   <module>   sp    k_*>
+*@;8=7*,/*/    6%

