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 ) def test_handle_multiple(self): for i in xrange(10): try: raise Exception('unittest') except Exception, e: exec_info = sys.exc_info() enabled_plugins = '' exception_handler.handle( self.status, e , exec_info, enabled_plugins ) exception_handler.get_scan_id() all_edata = exception_handler.get_all_exceptions() self.assertEqual(exception_handler.MAX_EXCEPTIONS_PER_PLUGIN, len(all_edata)) edata = all_edata[0] self.assertTrue( edata.get_summary().startswith('An exception was found') ) 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' ) class fake_status(w3af_core_status):
def __init__(self, title ): simple_base_window.__init__(self) exception_list = exception_handler.get_all_exceptions() scan_id = exception_handler.get_scan_id() trac_multi_bug_report.__init__(self, exception_list, scan_id ) # We got here because of an autogenerated bug, not because of the user # going to the Help menu and then clicking on "Report a bug" self.autogen = True # Set generic window settings self.set_modal(True) self.set_title(title) self.vbox = gtk.VBox() self.vbox.set_border_width(10) # the label for the title self.title_label = gtk.Label() self.title_label.set_line_wrap(True) label_text = _('<b>The following exceptions were raised and handled</b>') self.title_label.set_markup(label_text) self.title_label.show() # A gtk.TextView for the exception frame = gtk.Frame('Handled exceptions') sw = gtk.ScrolledWindow() sw.set_policy(gtk.POLICY_AUTOMATIC, gtk.POLICY_AUTOMATIC) sw.set_size_request(200, 280) # Create the TreeStore to display all exceptions self.treestore = gtk.TreeStore(str, str) for edata in exception_list: where = edata.get_where() exception = str(edata.exception) tdata = [ where, exception] self.treestore.append(None, tdata) # create the TreeView using treestore self.treeview = gtk.TreeView(self.treestore) # First column that holds the icon and the location tvcol = gtk.TreeViewColumn('Location') cell = gtk.CellRendererPixbuf() pb = self.treeview.render_icon(gtk.STOCK_DND, gtk.ICON_SIZE_SMALL_TOOLBAR, None) cell.set_property('pixbuf', pb) tvcol.pack_start(cell, expand=False) cell = gtk.CellRendererText() tvcol.pack_start(cell, expand=False) tvcol.add_attribute(cell, "text", 0) self.treeview.append_column(tvcol) # Second column that holds the exception tvcol = gtk.TreeViewColumn('Exception') cell = gtk.CellRendererText() tvcol.pack_start(cell, expand=True) tvcol.add_attribute(cell, "text", 1) self.treeview.append_column(tvcol) sw.add(self.treeview) frame.add(sw) # the label for the rest of the message self.label = gtk.Label() self.label.set_line_wrap(True) label_text = _("<i>All these exceptions were stored in '%s' for your later") label_text += _(' review.</i>\n\nReporting these is recommended and will') label_text += _(' help us improve w3af. <b>You can contribute</b> to the') label_text += _(' w3af project and submit these exceptions to our') label_text += _(' bug tracking system from within this window only using') label_text += _(' <i>two clicks</i>.\n\n') label_text += _('w3af will only send the exception traceback and the') label_text += _(' version information to Trac, no personal or ') label_text += _(' confidential information is collected.') self.label.set_markup( label_text % gettempdir() ) self.label.show() self.vbox.pack_start(self.title_label, True, True, 10) self.vbox.pack_start(frame, True, True) self.vbox.pack_start(self.label, True, True, 10) # the buttons self.hbox = gtk.HBox() self.butt_cancel = gtk.Button(stock=gtk.STOCK_CANCEL) self.butt_cancel.connect("clicked", self._handle_cancel) self.hbox.pack_start(self.butt_cancel, True, False) self.butt_send = gtk.Button(stock=gtk.STOCK_OK) self.butt_send.connect("clicked", self.report_bug) self.hbox.pack_start(self.butt_send, True, False) self.vbox.pack_start(self.hbox, True, False, 10) #self.resize(400,450) self.add(self.vbox) self.show_all() # This is a quick fix to get around the problem generated by "set_selectable" # that selects the text by default self.label.select_region(0, 0)