mirror of
https://github.com/projectdiscovery/nuclei.git
synced 2025-12-18 16:35:25 +00:00
29 lines
744 B
Go
29 lines
744 B
Go
|
package yaml
|
||
|
|
|
||
|
|
import (
|
||
|
|
"testing"
|
||
|
|
|
||
|
|
"github.com/projectdiscovery/nuclei/v3/pkg/input/types"
|
||
|
|
"github.com/stretchr/testify/require"
|
||
|
|
)
|
||
|
|
|
||
|
|
func TestYamlFormatterParse(t *testing.T) {
|
||
|
|
format := New()
|
||
|
|
|
||
|
|
proxifyInputFile := "../testdata/ginandjuice.proxify.yaml"
|
||
|
|
|
||
|
|
expectedUrls := []string{
|
||
|
|
"https://ginandjuice.shop/blog/post?postId=3&source=proxify",
|
||
|
|
"https://ginandjuice.shop/users/3",
|
||
|
|
}
|
||
|
|
|
||
|
|
var urls []string
|
||
|
|
err := format.Parse(proxifyInputFile, func(request *types.RequestResponse) bool {
|
||
|
|
urls = append(urls, request.URL.String())
|
||
|
|
return false
|
||
|
|
})
|
||
|
|
require.Nilf(t, err, "error parsing yaml file: %v", err)
|
||
|
|
require.Len(t, urls, len(expectedUrls), "invalid number of urls")
|
||
|
|
require.ElementsMatch(t, urls, expectedUrls, "invalid urls")
|
||
|
|
}
|