added test case

projectdiscovery · Jun 16, 2024 · 3ba404a · 3ba404a
1 parent 2e3edd9
commit 3ba404a
Showing 1 changed file with 34 additions and 0 deletions.
diff --git a/pkg/reporting/trackers/jira/jira_test.go b/pkg/reporting/trackers/jira/jira_test.go
@@ -4,6 +4,10 @@ import (
 	"strings"
 	"testing"
 
+	"github.com/projectdiscovery/nuclei/v3/pkg/model"
+	"github.com/projectdiscovery/nuclei/v3/pkg/model/types/severity"
+	"github.com/projectdiscovery/nuclei/v3/pkg/output"
+	"github.com/projectdiscovery/nuclei/v3/pkg/reporting/trackers/filters"
 	"github.com/stretchr/testify/require"
 )
 
@@ -36,3 +40,33 @@ func TestTableCreation(t *testing.T) {
 `
 	require.Equal(t, expected, table)
 }
+
+func Test_ShouldFilter_Tracker(t *testing.T) {
+	jiraIntegration := &Integration{
+		options: &Options{AllowList: &filters.Filter{
+			Severities: severity.Severities{severity.Critical},
+		}},
+	}
+
+	require.False(t, jiraIntegration.ShouldFilter(&output.ResultEvent{Info: model.Info{
+		SeverityHolder: severity.Holder{Severity: severity.Info},
+	}}))
+	require.True(t, jiraIntegration.ShouldFilter(&output.ResultEvent{Info: model.Info{
+		SeverityHolder: severity.Holder{Severity: severity.Critical},
+	}}))
+
+	t.Run("deny-list", func(t *testing.T) {
+		jiraIntegration := &Integration{
+			options: &Options{DenyList: &filters.Filter{
+				Severities: severity.Severities{severity.Critical},
+			}},
+		}
+
+		require.True(t, jiraIntegration.ShouldFilter(&output.ResultEvent{Info: model.Info{
+			SeverityHolder: severity.Holder{Severity: severity.Info},
+		}}))
+		require.False(t, jiraIntegration.ShouldFilter(&output.ResultEvent{Info: model.Info{
+			SeverityHolder: severity.Holder{Severity: severity.Critical},
+		}}))
+	})
+}