mirror of https://github.com/daffainfo/nuclei.git
63 lines
2.0 KiB
Go
63 lines
2.0 KiB
Go
package offlinehttp
|
|
|
|
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/operators"
|
|
"github.com/projectdiscovery/nuclei/v3/pkg/testutils"
|
|
permissionutil "github.com/projectdiscovery/utils/permission"
|
|
)
|
|
|
|
func TestFindResponses(t *testing.T) {
|
|
options := testutils.DefaultOptions
|
|
|
|
testutils.Init(options)
|
|
templateID := "testing-offline"
|
|
request := &Request{}
|
|
executerOpts := testutils.NewMockExecuterOptions(options, &testutils.TemplateInfo{
|
|
ID: templateID,
|
|
Info: model.Info{SeverityHolder: severity.Holder{Severity: severity.Low}, Name: "test"},
|
|
})
|
|
executerOpts.Operators = []*operators.Operators{{}}
|
|
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.txt": "TEST",
|
|
"final.txt": "TEST",
|
|
"image_ignored.png": "TEST",
|
|
"test.txt": "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.txt", "final.txt", "test.txt"}
|
|
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")
|
|
}
|