mirror of
https://github.com/projectdiscovery/nuclei.git
synced 2025-12-20 19:15:27 +00:00
35 lines
777 B
Go
35 lines
777 B
Go
|
package swagger
|
||
|
|
|
||
|
|
import (
|
||
|
|
"testing"
|
||
|
|
|
||
|
|
"github.com/projectdiscovery/nuclei/v3/pkg/input/types"
|
||
|
|
"github.com/stretchr/testify/require"
|
||
|
|
)
|
||
|
|
|
||
|
|
func TestSwaggerAPIParser(t *testing.T) {
|
||
|
|
format := New()
|
||
|
|
|
||
|
|
proxifyInputFile := "../testdata/swagger.yaml"
|
||
|
|
|
||
|
|
var gotMethodsToURLs []string
|
||
|
|
|
||
|
|
err := format.Parse(proxifyInputFile, func(request *types.RequestResponse) bool {
|
||
|
|
gotMethodsToURLs = append(gotMethodsToURLs, request.URL.String())
|
||
|
|
return false
|
||
|
|
})
|
||
|
|
if err != nil {
|
||
|
|
t.Fatal(err)
|
||
|
|
}
|
||
|
|
|
||
|
|
if len(gotMethodsToURLs) != 2 {
|
||
|
|
t.Fatalf("invalid number of methods: %d", len(gotMethodsToURLs))
|
||
|
|
}
|
||
|
|
|
||
|
|
expectedURLs := []string{
|
||
|
|
"https://localhost/users",
|
||
|
|
"https://localhost/users/1?test=asc",
|
||
|
|
}
|
||
|
|
require.ElementsMatch(t, gotMethodsToURLs, expectedURLs, "could not get swagger urls")
|
||
|
|
}
|