Search in sources :

Example 36 with Alert

use of org.parosproxy.paros.core.scanner.Alert in project zaproxy by zaproxy.

the class ExtensionAlertUnitTest method shouldAppendAlertNameCorrectly.

@Test
public void shouldAppendAlertNameCorrectly() {
    extAlert.setAlertOverrideProperty("1.name", "+" + NEW_NAME);
    Alert alert1 = newAlert(1);
    extAlert.applyOverrides(alert1);
    // When/Then
    assertEquals(ORIGINAL_NAME + NEW_NAME, alert1.getName());
    assertEquals(ORIGINAL_DESC, alert1.getDescription());
    assertEquals(ORIGINAL_SOLN, alert1.getSolution());
    assertEquals(ORIGINAL_OTHER, alert1.getOtherInfo());
    assertEquals(ORIGINAL_REF, alert1.getReference());
    // Check other alerts are not affected
    Alert alert2 = newAlert(2);
    extAlert.applyOverrides(alert2);
    // When/Then
    assertEquals(ORIGINAL_NAME, alert2.getName());
    assertEquals(ORIGINAL_DESC, alert2.getDescription());
    assertEquals(ORIGINAL_SOLN, alert2.getSolution());
    assertEquals(ORIGINAL_OTHER, alert2.getOtherInfo());
    assertEquals(ORIGINAL_REF, alert2.getReference());
}
Also used : Alert(org.parosproxy.paros.core.scanner.Alert) Test(org.junit.Test)

Example 37 with Alert

use of org.parosproxy.paros.core.scanner.Alert in project zaproxy by zaproxy.

the class AlertAddDialog method getBtnOk.

/**
	 * This method initializes btnStart	
	 * 	
	 * @return javax.swing.JButton	
	 */
private JButton getBtnOk() {
    if (btnOk == null) {
        btnOk = new JButton();
        btnOk.setText(Constant.messages.getString("alert.add.button.save"));
        btnOk.addActionListener(new java.awt.event.ActionListener() {

            @Override
            public void actionPerformed(java.awt.event.ActionEvent e) {
                Alert alert = alertViewPanel.getAlert();
                try {
                    ExtensionAlert extAlert = (ExtensionAlert) Control.getSingleton().getExtensionLoader().getExtension(ExtensionAlert.NAME);
                    if (alert.getAlertId() >= 0) {
                        // Its an existing alert so save it
                        if (extAlert != null) {
                            extAlert.updateAlert(alert);
                        } else if (historyRef != null) {
                            // Update history tree
                            historyRef.updateAlert(alert);
                            extension.notifyHistoryItemChanged(historyRef);
                        }
                    } else {
                        if (httpMessage != null) {
                            historyRef = new HistoryReference(Model.getSingleton().getSession(), historyType, httpMessage);
                        }
                        alert.setSource(Alert.Source.MANUAL);
                        // Raise it
                        if (extAlert != null) {
                            extAlert.alertFound(alert, historyRef);
                        } else {
                            historyRef.addAlert(alert);
                            extension.notifyHistoryItemChanged(historyRef);
                        }
                    }
                } catch (Exception ex) {
                    logger.error(ex.getMessage(), ex);
                }
                clearAndCloseDialog();
            }
        });
    }
    return btnOk;
}
Also used : HistoryReference(org.parosproxy.paros.model.HistoryReference) JButton(javax.swing.JButton) Alert(org.parosproxy.paros.core.scanner.Alert) ExtensionAlert(org.zaproxy.zap.extension.alert.ExtensionAlert) ExtensionAlert(org.zaproxy.zap.extension.alert.ExtensionAlert) HeadlessException(java.awt.HeadlessException)

Aggregations

Alert (org.parosproxy.paros.core.scanner.Alert)37 Test (org.junit.Test)15 ArrayList (java.util.ArrayList)7 RecordAlert (org.parosproxy.paros.db.RecordAlert)7 TableAlert (org.parosproxy.paros.db.TableAlert)7 HistoryReference (org.parosproxy.paros.model.HistoryReference)6 SiteNode (org.parosproxy.paros.model.SiteNode)6 ExtensionAlert (org.zaproxy.zap.extension.alert.ExtensionAlert)5 DatabaseException (org.parosproxy.paros.db.DatabaseException)4 TreePath (javax.swing.tree.TreePath)3 DefaultMutableTreeNode (javax.swing.tree.DefaultMutableTreeNode)2 Session (org.parosproxy.paros.model.Session)2 SiteMap (org.parosproxy.paros.model.SiteMap)2 AlertNode (org.zaproxy.zap.extension.alert.AlertNode)2 Component (java.awt.Component)1 Dimension (java.awt.Dimension)1 HeadlessException (java.awt.HeadlessException)1 Point (java.awt.Point)1 MouseEvent (java.awt.event.MouseEvent)1 Enumeration (java.util.Enumeration)1