diff options
Diffstat (limited to 'Lib')
| -rw-r--r-- | Lib/test/test_trace.py | 92 |
1 files changed, 83 insertions, 9 deletions
diff --git a/Lib/test/test_trace.py b/Lib/test/test_trace.py index ee847b6..eca8c16 100644 --- a/Lib/test/test_trace.py +++ b/Lib/test/test_trace.py @@ -69,6 +69,62 @@ no_pop_blocks.events = [(0, 'call'), (3, 'line'), (3, 'return')] +def called(): # line -3 + x = 1 + +def call(): # line 0 + called() + +call.events = [(0, 'call'), + (1, 'line'), + (-3, 'call'), + (-2, 'line'), + (-2, 'return'), + (1, 'return')] + +def raises(): + raise Exception + +def test_raise(): + try: + raises() + except Exception, exc: + x = 1 + +test_raise.events = [(0, 'call'), + (1, 'line'), + (2, 'line'), + (-3, 'call'), + (-2, 'line'), + (-2, 'exception'), + (2, 'exception'), + (3, 'line'), + (4, 'line'), + (4, 'return')] + +def _settrace_and_return(tracefunc): + sys.settrace(tracefunc) + sys._getframe().f_back.f_trace = tracefunc +def settrace_and_return(tracefunc): + _settrace_and_return(tracefunc) + +settrace_and_return.events = [(1, 'return')] + +def _settrace_and_raise(tracefunc): + sys.settrace(tracefunc) + sys._getframe().f_back.f_trace = tracefunc + raise RuntimeError +def settrace_and_raise(tracefunc): + try: + _settrace_and_raise(tracefunc) + except RuntimeError, exc: + pass + +settrace_and_raise.events = [(2, 'exception'), + (3, 'line'), + (4, 'line'), + (4, 'return')] + class Tracer: def __init__(self): self.events = [] @@ -77,18 +133,29 @@ class Tracer: return self.trace class TraceTestCase(unittest.TestCase): + def compare_events(self, line_offset, events, expected_events): + events = [(l - line_offset, e) for (l, e) in events] + if events != expected_events: + self.fail( + "events did not match expectation:\n" + + "\n".join(difflib.ndiff(map(str, expected_events), + map(str, events)))) + + def run_test(self, func): tracer = Tracer() sys.settrace(tracer.trace) func() sys.settrace(None) - fl = func.func_code.co_firstlineno - events = [(l - fl, e) for (l, e) in tracer.events] - if events != func.events: - self.fail( - "events did not match expectation:\n" + - "\n".join(difflib.ndiff(map(str, func.events), - map(str, events)))) + self.compare_events(func.func_code.co_firstlineno, + tracer.events, func.events) + + def run_test2(self, func): + tracer = Tracer() + func(tracer.trace) + sys.settrace(None) + self.compare_events(func.func_code.co_firstlineno, + tracer.events, func.events) def test_1_basic(self): self.run_test(basic) @@ -100,8 +167,15 @@ class TraceTestCase(unittest.TestCase): self.run_test(no_pop_blocks) def test_5_no_pop_tops(self): self.run_test(no_pop_tops) - - + def test_6_call(self): + self.run_test(call) + def test_7_raise(self): + self.run_test(test_raise) + + def test_8_settrace_and_return(self): + self.run_test2(settrace_and_return) + def test_9_settrace_and_raise(self): + self.run_test2(settrace_and_raise) def test_main(): test_support.run_unittest(TraceTestCase) |