CodePlex Issue #34386 Plain Text Attachments

test_traceback.py

import unittest, sys, traceback

class TestTraceback(unittest.TestCase):
	def test_truncate_traceback_with_typeerror(self):
		def task3():
			return (1, 2) + [3, 4] # TypeError

		def task2():
			task3()

		def task1():
			try:
				task2()
			except: 
				raise

		try:
			task1()
			self.fail()
		except:
			exc_type, exc_value, exc_traceback = sys.exc_info()
			methods = [function_name for filename, line_number, function_name, text in traceback.extract_tb(exc_traceback)]
			self.assertEqual(methods, ['test_truncate_traceback_with_typeerror', 'task1', 'task2', 'task3'])

	def test_keep_traceback_with_exception(self):
		def task3():
			raise Exception('ahhh')

		def task2():
			task3()

		def task1():
			try:
				task2()
			except: 
				raise

		try:
			task1()
			self.fail()
		except:
			exc_type, exc_value, exc_traceback = sys.exc_info()
			methods = [function_name for filename, line_number, function_name, text in traceback.extract_tb(exc_traceback)]
			self.assertEqual(methods, ['test_keep_traceback_with_exception', 'task1', 'task2', 'task3'])