mirror of
https://github.com/projectdiscovery/nuclei.git
synced 2025-12-18 16:25:24 +00:00
ef5b476c6d
This PR adds the support in http module to iterate over the dynamically extracted data from extractors and use it in other requests. This allows nuclei to follow links on pages, do operations with multiple versions of the same extracted value, etc.
40 lines
899 B
Go
40 lines
899 B
Go
package operators
|
|
|
|
import (
|
|
"testing"
|
|
|
|
"github.com/stretchr/testify/require"
|
|
)
|
|
|
|
func TestMakeDynamicValuesCallback(t *testing.T) {
|
|
input := map[string][]string{
|
|
"a": []string{"1", "2"},
|
|
"b": []string{"3"},
|
|
"c": []string{},
|
|
"d": []string{"A", "B", "C"},
|
|
}
|
|
|
|
count := 0
|
|
MakeDynamicValuesCallback(input, func(data map[string]interface{}) bool {
|
|
count++
|
|
require.Len(t, data, 3, "could not get correct output length")
|
|
return false
|
|
})
|
|
require.Equal(t, 3, count, "could not get correct result count")
|
|
|
|
t.Run("single", func(t *testing.T) {
|
|
input := map[string][]string{
|
|
"a": []string{"1"},
|
|
"b": []string{"2"},
|
|
"c": []string{"3"},
|
|
}
|
|
|
|
count := 0
|
|
MakeDynamicValuesCallback(input, func(data map[string]interface{}) {
|
|
count++
|
|
require.Len(t, data, 3, "could not get correct output length")
|
|
})
|
|
require.Equal(t, 1, count, "could not get correct result count")
|
|
})
|
|
}
|