mirror of
https://github.com/projectdiscovery/nuclei.git
synced 2025-12-18 21:15:29 +00:00
03c7e89d9e
* Added workflow names based condition * Added conditional filtering to workflow executor * Replaced names with single name stringslice * Fixed workflow matcher condition + tests * Fixed workflow item name
44 lines
2.2 KiB
Go
44 lines
2.2 KiB
Go
package workflows
|
|
|
|
import (
|
|
"testing"
|
|
|
|
"github.com/projectdiscovery/nuclei/v2/pkg/model/types/stringslice"
|
|
"github.com/projectdiscovery/nuclei/v2/pkg/operators"
|
|
"github.com/stretchr/testify/require"
|
|
)
|
|
|
|
func TestWorkflowMatchAndCompile(t *testing.T) {
|
|
t.Run("name", func(t *testing.T) {
|
|
matcher := &Matcher{Name: stringslice.StringSlice{Value: "sphinx"}}
|
|
matched := matcher.Match(&operators.Result{Matches: map[string][]string{"sphinx": {}}, Extracts: map[string][]string{}})
|
|
require.True(t, matched, "could not match value")
|
|
})
|
|
t.Run("name-negative", func(t *testing.T) {
|
|
matcher := &Matcher{Name: stringslice.StringSlice{Value: "tomcat"}}
|
|
matched := matcher.Match(&operators.Result{Matches: map[string][]string{"apache": {}}, Extracts: map[string][]string{}})
|
|
require.False(t, matched, "could not match value")
|
|
})
|
|
t.Run("names-or", func(t *testing.T) {
|
|
matcher := &Matcher{Name: stringslice.StringSlice{Value: []string{"sphinx", "elastic"}}, Condition: "or"}
|
|
_ = matcher.Compile()
|
|
matched := matcher.Match(&operators.Result{Matches: map[string][]string{"elastic": {}}, Extracts: map[string][]string{}})
|
|
require.True(t, matched, "could not match value")
|
|
matched = matcher.Match(&operators.Result{Matches: map[string][]string{"sphinx": {}}, Extracts: map[string][]string{}})
|
|
require.True(t, matched, "could not match value")
|
|
matched = matcher.Match(&operators.Result{Matches: map[string][]string{"random": {}}, Extracts: map[string][]string{}})
|
|
require.False(t, matched, "could not match value")
|
|
})
|
|
t.Run("names-and", func(t *testing.T) {
|
|
matcher := &Matcher{Name: stringslice.StringSlice{Value: []string{"sphinx", "elastic"}}, Condition: "and"}
|
|
_ = matcher.Compile()
|
|
matched := matcher.Match(&operators.Result{Matches: map[string][]string{"elastic": {}, "sphinx": {}}, Extracts: map[string][]string{}})
|
|
require.True(t, matched, "could not match value")
|
|
|
|
matched = matcher.Match(&operators.Result{Matches: map[string][]string{"sphinx": {}}, Extracts: map[string][]string{}})
|
|
require.False(t, matched, "could not match value")
|
|
matched = matcher.Match(&operators.Result{Matches: map[string][]string{"random": {}}, Extracts: map[string][]string{}})
|
|
require.False(t, matched, "could not match value")
|
|
})
|
|
}
|