class TestExceptionHandler(unittest.TestCase):
EXCEPT_START = 'A "Exception" exception was found'
def setUp(self):
self.exception_handler = ExceptionHandler()
self.exception_handler.clear()
self.status = FakeStatus(None)
self.status.set_running_plugin('phase', 'plugin')
self.status.set_current_fuzzable_request('phase',
'http://www.w3af.org/')
@attr('smoke')
def test_handle_one(self):
try:
raise Exception('unittest')
except Exception, e:
exec_info = sys.exc_info()
enabled_plugins = ''
self.exception_handler.handle(self.status,
e,
exec_info,
enabled_plugins)
scan_id = self.exception_handler.get_scan_id()
self.assertTrue(scan_id)
all_edata = self.exception_handler.get_all_exceptions()
self.assertEqual(1, len(all_edata))
edata = all_edata[0]
self.assertTrue(edata.get_summary().startswith(self.EXCEPT_START))
self.assertTrue('traceback' in edata.get_details())
self.assertEquals(edata.plugin, 'plugin')
self.assertEquals(edata.phase, 'phase')
self.assertEquals(edata.fuzzable_request, 'http://www.w3af.org/')
self.assertEquals(edata.filename, 'test_exception_handler.py')
self.assertEquals(edata.exception_msg, str(e))
self.assertEquals(edata.exception_class, e.__class__.__name__)
# This is very very very dependant on changes to this file, but it was
# the only way to do it without much effort
self.assertEquals(edata.lineno, 50)
class TestExceptionHandler(unittest.TestCase):
EXCEPT_START = 'A "Exception" exception was found'
def setUp(self):
self.exception_handler = ExceptionHandler()
self.exception_handler.clear()
self.status = fake_status(None)
self.status.set_running_plugin('phase', 'plugin')
self.status.set_current_fuzzable_request('phase',
'http://www.w3af.org/')
@attr('smoke')
def test_handle_one(self):
try:
raise Exception('unittest')
except Exception, e:
exec_info = sys.exc_info()
enabled_plugins = ''
self.exception_handler.handle(self.status, e, exec_info,
enabled_plugins)
scan_id = self.exception_handler.get_scan_id()
self.assertTrue(scan_id)
all_edata = self.exception_handler.get_all_exceptions()
self.assertEqual(1, len(all_edata))
edata = all_edata[0]
self.assertTrue(edata.get_summary().startswith(self.EXCEPT_START))
self.assertTrue('traceback' in edata.get_details())
self.assertEquals(edata.plugin, 'plugin')
self.assertEquals(edata.phase, 'phase')
self.assertEquals(edata.fuzzable_request, 'http://www.w3af.org/')
self.assertEquals(edata.filename, 'test_exception_handler.py')
self.assertEquals(edata.exception, e)
# This is very very very dependant on changes to this file, but it was
# the only way to do it without much effort
self.assertEquals(edata.lineno, 50)
class TestExceptionHandler(unittest.TestCase):
EXCEPT_START = 'A "Exception" exception was found'
def setUp(self):
self.exception_handler = ExceptionHandler()
self.exception_handler.clear()
self.status = FakeStatus(None)
self.status.set_running_plugin('phase', 'plugin')
self.status.set_current_fuzzable_request('phase',
'http://www.w3af.org/')
@attr('smoke')
def test_handle_one(self):
try:
raise Exception('unittest')
except Exception as e:
exec_info = sys.exc_info()
enabled_plugins = ''
self.exception_handler.handle(self.status, e, exec_info,
enabled_plugins)
scan_id = self.exception_handler.get_scan_id()
self.assertTrue(scan_id)
all_edata = self.exception_handler.get_all_exceptions()
self.assertEqual(1, len(all_edata))
edata = all_edata[0]
self.assertTrue(edata.get_summary().startswith(self.EXCEPT_START))
self.assertTrue('traceback' in edata.get_details())
self.assertEquals(edata.plugin, 'plugin')
self.assertEquals(edata.phase, 'phase')
self.assertEquals(edata.fuzzable_request, 'http://www.w3af.org/')
self.assertEquals(edata.filename, 'test_exception_handler.py')
self.assertEquals(edata.exception_msg, str(e))
self.assertEquals(edata.exception_class, e.__class__.__name__)
# This is very very very dependant on changes to this file, but it was
# the only way to do it without much effort
self.assertEquals(edata.lineno, 50)
@attr('smoke')
def test_handle_multiple(self):
for _ in xrange(10):
try:
raise Exception('unittest')
except Exception as e:
exec_info = sys.exc_info()
enabled_plugins = ''
self.exception_handler.handle(self.status, e, exec_info,
enabled_plugins)
self.exception_handler.get_scan_id()
all_edata = self.exception_handler.get_all_exceptions()
self.assertEqual(self.exception_handler.MAX_EXCEPTIONS_PER_PLUGIN,
len(all_edata))
edata = all_edata[0]
self.assertTrue(edata.get_summary().startswith(self.EXCEPT_START))
self.assertTrue('traceback' in edata.get_details())
self.assertEquals(edata.plugin, 'plugin')
self.assertEquals(edata.phase, 'phase')
self.assertEquals(edata.fuzzable_request, 'http://www.w3af.org/')
self.assertEquals(edata.filename, 'test_exception_handler.py')
def test_get_unique_exceptions(self):
for _ in xrange(10):
try:
raise Exception('unittest')
except Exception as e:
exec_info = sys.exc_info()
enabled_plugins = ''
self.exception_handler.handle(self.status, e, exec_info,
enabled_plugins)
all_edata = self.exception_handler.get_all_exceptions()
self.assertEqual(self.exception_handler.MAX_EXCEPTIONS_PER_PLUGIN,
len(all_edata))
unique_edata = self.exception_handler.get_unique_exceptions()
self.assertEqual(1, len(unique_edata))
edata = unique_edata[0]
self.assertTrue(edata.get_summary().startswith(self.EXCEPT_START))
self.assertTrue('traceback' in edata.get_details())
self.assertEquals(edata.plugin, 'plugin')
self.assertEquals(edata.phase, 'phase')
self.assertEquals(edata.fuzzable_request, 'http://www.w3af.org/')
self.assertEquals(edata.filename, 'test_exception_handler.py')
def test_handle_threads_calls(self):
def test2():
raise Exception('unittest')
def test(ehandler):
try:
test2()
except Exception as e:
exec_info = sys.exc_info()
enabled_plugins = ''
ehandler.handle(self.status, e, exec_info, enabled_plugins)
th = threading.Thread(target=test, args=(self.exception_handler, ))
th.start()
th.join()
all_edata = self.exception_handler.get_all_exceptions()
self.assertEqual(1, len(all_edata))
edata = all_edata[0]
self.assertTrue(edata.get_summary().startswith(self.EXCEPT_START))
self.assertTrue('traceback' in edata.get_details())
self.assertEquals(edata.plugin, 'plugin')
self.assertEquals(edata.phase, 'phase')
self.assertEquals(edata.fuzzable_request, 'http://www.w3af.org/')
self.assertEquals(edata.filename, 'test_exception_handler.py')
# This is very very very dependant on changes to this file, but it was
# the only way to do it without much effort
self.assertEquals(edata.lineno, 137)
def test_handle_multi_calls(self):
def test3():
raise Exception('unittest')
def test2():
test3()
def test(ehandler):
try:
test2()
except Exception as e:
exec_info = sys.exc_info()
enabled_plugins = ''
ehandler.handle(self.status, e, exec_info, enabled_plugins)
test(self.exception_handler)
all_edata = self.exception_handler.get_all_exceptions()
self.assertEqual(1, len(all_edata))
edata = all_edata[0]
# This is very very very dependant on changes to this file, but it was
# the only way to do it without much effort
self.assertEquals(edata.lineno, 170)