Merge pull request #5606 from thc202/automation/move-job-alert-tests

automation: let job report alert tests support
zaproxy · Jul 25, 2024 · 0132d38 · 0132d38
2 parents e8fb7cb + dfe2fc2
commit 0132d38
Show file tree

Hide file tree

Showing 4 changed files with 15 additions and 3 deletions.
diff --git a/addOns/automation/src/main/java/org/zaproxy/addon/automation/AutomationJob.java b/addOns/automation/src/main/java/org/zaproxy/addon/automation/AutomationJob.java
@@ -493,6 +493,10 @@ public void logTestsToProgress(AutomationProgress progress) {
         tests.forEach(t -> t.logToProgress(progress));
     }
 
+    public boolean supportsAlertTests() {
+        return false;
+    }
+
     public boolean supportsMonitorTests() {
         return false;
     }

diff --git a/addOns/automation/src/main/java/org/zaproxy/addon/automation/gui/AddTestDialog.java b/addOns/automation/src/main/java/org/zaproxy/addon/automation/gui/AddTestDialog.java
@@ -25,8 +25,6 @@
 import org.parosproxy.paros.view.View;
 import org.zaproxy.addon.automation.AutomationEventPublisher;
 import org.zaproxy.addon.automation.AutomationJob;
-import org.zaproxy.addon.automation.jobs.ActiveScanJob;
-import org.zaproxy.addon.automation.jobs.PassiveScanWaitJob;
 import org.zaproxy.addon.automation.tests.AbstractAutomationTest;
 import org.zaproxy.addon.automation.tests.AutomationAlertTest;
 import org.zaproxy.addon.automation.tests.AutomationMonitorTest;
@@ -60,7 +58,7 @@ public AddTestDialog(AutomationJob job) {
         }
         testNames.add(Constant.messages.getString(URL_PRESENCE_TEST_NAME));
 
-        if (job instanceof PassiveScanWaitJob || job instanceof ActiveScanJob) {
+        if (job.supportsAlertTests()) {
             testNames.add(Constant.messages.getString(ALERT_TEST_NAME));
         }
 

diff --git a/addOns/automation/src/main/java/org/zaproxy/addon/automation/jobs/ActiveScanJob.java b/addOns/automation/src/main/java/org/zaproxy/addon/automation/jobs/ActiveScanJob.java
@@ -67,6 +67,11 @@ public ActiveScanJob() {
         data = new Data(this, this.parameters, this.policyDefinition);
     }
 
+    @Override
+    public boolean supportsAlertTests() {
+        return true;
+    }
+
     private ExtensionActiveScan getExtAScan() {
         if (extAScan == null) {
             extAScan =

diff --git a/addOns/automation/src/main/java/org/zaproxy/addon/automation/jobs/PassiveScanWaitJob.java b/addOns/automation/src/main/java/org/zaproxy/addon/automation/jobs/PassiveScanWaitJob.java
@@ -47,6 +47,11 @@ public PassiveScanWaitJob() {
         this.data = new Data(this, parameters);
     }
 
+    @Override
+    public boolean supportsAlertTests() {
+        return true;
+    }
+
     @Override
     public void runJob(AutomationEnvironment env, AutomationProgress progress) {
         ExtensionPassiveScan extPScan = getExtPassiveScan();