mirror of https://github.com/daffainfo/nuclei.git
68 lines
2.1 KiB
Go
68 lines
2.1 KiB
Go
package file
|
|
|
|
import (
|
|
"os"
|
|
"path/filepath"
|
|
"testing"
|
|
|
|
"github.com/stretchr/testify/require"
|
|
|
|
"github.com/projectdiscovery/nuclei/v3/pkg/model"
|
|
"github.com/projectdiscovery/nuclei/v3/pkg/model/types/severity"
|
|
"github.com/projectdiscovery/nuclei/v3/pkg/testutils"
|
|
permissionutil "github.com/projectdiscovery/utils/permission"
|
|
)
|
|
|
|
func TestFindInputPaths(t *testing.T) {
|
|
options := testutils.DefaultOptions
|
|
|
|
testutils.Init(options)
|
|
templateID := "testing-file"
|
|
request := &Request{
|
|
ID: templateID,
|
|
MaxSize: "1Gb",
|
|
NoRecursive: false,
|
|
Extensions: []string{"all", ".lock"},
|
|
DenyList: []string{".go"},
|
|
Operators: newMockOperator(),
|
|
}
|
|
executerOpts := testutils.NewMockExecuterOptions(options, &testutils.TemplateInfo{
|
|
ID: templateID,
|
|
Info: model.Info{SeverityHolder: severity.Holder{Severity: severity.Low}, Name: "test"},
|
|
})
|
|
err := request.Compile(executerOpts)
|
|
require.Nil(t, err, "could not compile file request")
|
|
|
|
tempDir, err := os.MkdirTemp("", "test-*")
|
|
require.Nil(t, err, "could not create temporary directory")
|
|
defer os.RemoveAll(tempDir)
|
|
|
|
files := map[string]string{
|
|
"test.go": "TEST",
|
|
"config.yaml": "TEST",
|
|
"final.yaml": "TEST",
|
|
"image_ignored.png": "TEST",
|
|
"test.js": "TEST",
|
|
}
|
|
for k, v := range files {
|
|
err = os.WriteFile(filepath.Join(tempDir, k), []byte(v), permissionutil.TempFilePermission)
|
|
require.Nil(t, err, "could not write temporary file")
|
|
}
|
|
expected := []string{"config.yaml", "final.yaml", "test.js"}
|
|
got := []string{}
|
|
err = request.getInputPaths(tempDir+"/*", func(item string) {
|
|
base := filepath.Base(item)
|
|
got = append(got, base)
|
|
})
|
|
require.Nil(t, err, "could not get input paths for glob")
|
|
require.ElementsMatch(t, expected, got, "could not get correct file matches for glob")
|
|
|
|
got = []string{}
|
|
err = request.getInputPaths(tempDir, func(item string) {
|
|
base := filepath.Base(item)
|
|
got = append(got, base)
|
|
})
|
|
require.Nil(t, err, "could not get input paths for directory")
|
|
require.ElementsMatch(t, expected, got, "could not get correct file matches for directory")
|
|
}
|