External/nuclei
1
0
Fork 0
mirror of https://github.com/projectdiscovery/nuclei.git synced 2025-12-18 04:55:28 +00:00
Code Issues Packages Projects Releases Wiki Activity

Fix parseUrl test (#3426)

Browse Source 
* Fix parseUrl test

* fix lint error
This commit is contained in:
Shubham Rasal 2023-03-15 19:43:10 +05:30 committed by GitHub
parent 0bf8fc027d
commit 5e20c66215
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 37 additions and 24 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

61
v2/pkg/templates/compile_test.go
View File

@ -4,15 +4,18 @@ import (
"context"
"fmt"
"log"
netHttp "net/http"
"net/http/httptest"
"os"
"testing"
"time"
"github.com/julienschmidt/httprouter"
"github.com/projectdiscovery/nuclei/v2/pkg/catalog/disk"
"github.com/projectdiscovery/nuclei/v2/pkg/model"
"github.com/projectdiscovery/nuclei/v2/pkg/model/types/severity"
"github.com/projectdiscovery/nuclei/v2/pkg/model/types/stringslice"
"github.com/projectdiscovery/nuclei/v2/pkg/operators"
"github.com/projectdiscovery/nuclei/v2/pkg/operators/extractors"
"github.com/projectdiscovery/nuclei/v2/pkg/operators/matchers"
"github.com/projectdiscovery/nuclei/v2/pkg/parsers"
"github.com/projectdiscovery/nuclei/v2/pkg/progress"
@ -20,7 +23,6 @@ import (
"github.com/projectdiscovery/nuclei/v2/pkg/protocols/common/generators"
"github.com/projectdiscovery/nuclei/v2/pkg/protocols/common/variables"
"github.com/projectdiscovery/nuclei/v2/pkg/protocols/http"
"github.com/projectdiscovery/nuclei/v2/pkg/protocols/ssl"
"github.com/projectdiscovery/nuclei/v2/pkg/templates"
"github.com/projectdiscovery/nuclei/v2/pkg/testutils"
"github.com/projectdiscovery/nuclei/v2/pkg/workflows"
@ -54,42 +56,53 @@ func setup() {
}
func Test_ParseFromURL(t *testing.T) {
filePath := "https://api.nuclei.sh/api/v1/templates/raw/ssl/tls-version.yaml"
expectedTemplate := &templates.Template{
ID: "tls-version",
router := httprouter.New()
router.GET("/match-1.yaml", func(w netHttp.ResponseWriter, r *netHttp.Request, _ httprouter.Params) {
b, err := os.ReadFile("tests/match-1.yaml")
if err != nil {
w.Write([]byte(err.Error())) // nolint: errcheck
}
w.Write(b) // nolint: errcheck
})
ts := httptest.NewServer(router)
defer ts.Close()
var expectedTemplate = &templates.Template{
ID: "basic-get",
Info: model.Info{
Name: "Basic GET Request",
Authors: stringslice.StringSlice{Value: []string{"pdteam"}},
SeverityHolder: severity.Holder{Severity: severity.Info},
Name: "TLS Version",
Tags: stringslice.StringSlice{Value: []string{"ssl"}},
},
RequestsSSL: []*ssl.Request{
{
Address: "{{Host}}:{{Port}}",
Operators: operators.Operators{
Extractors: []*extractors.Extractor{
{
Type: extractors.ExtractorTypeHolder{ExtractorType: extractors.JSONExtractor},
JSON: []string{".tls_version"},
},
RequestsHTTP: []*http.Request{{
Operators: operators.Operators{
Matchers: []*matchers.Matcher{{
Type: matchers.MatcherTypeHolder{
MatcherType: matchers.WordsMatcher,
},
},
}},
Words: []string{"This is test matcher text"},
}},
},
Path: []string{"{{BaseURL}}"},
AttackType: generators.AttackTypeHolder{},
Method: http.HTTPMethodTypeHolder{
MethodType: http.HTTPGet,
},
}},
TotalRequests: 1,
Executer: nil,
Path: "https://api.nuclei.sh/api/v1/templates/raw/ssl/tls-version.yaml",
Path: ts.URL + "/match-1.yaml",
}
setup()
got, err := templates.Parse(filePath, nil, executerOpts)
got, err := templates.Parse(ts.URL+"/match-1.yaml", nil, executerOpts)
require.Nilf(t, err, "could not parse template (%s)", fmt.Sprint(err))
require.Nil(t, err, "could not parse template")
require.Equal(t, expectedTemplate.ID, got.ID)
require.Equal(t, expectedTemplate.Info, got.Info)
require.Equal(t, expectedTemplate.TotalRequests, got.TotalRequests)
require.Equal(t, expectedTemplate.Path, got.Path)
require.Equal(t, expectedTemplate.RequestsSSL[0].Address, got.RequestsSSL[0].Address)
require.Equal(t, expectedTemplate.RequestsSSL[0].Extractors[0].Type, got.RequestsSSL[0].Extractors[0].Type)
require.Equal(t, expectedTemplate.RequestsSSL[0].Extractors[0].JSON, got.RequestsSSL[0].Extractors[0].JSON)
require.Equal(t, len(expectedTemplate.RequestsSSL), len(got.RequestsSSL))
require.Equal(t, expectedTemplate.RequestsHTTP[0].Path, got.RequestsHTTP[0].Path)
require.Equal(t, expectedTemplate.RequestsHTTP[0].Operators.Matchers[0].Words, got.RequestsHTTP[0].Operators.Matchers[0].Words)
require.Equal(t, len(expectedTemplate.RequestsHTTP), len(got.RequestsHTTP))
}
func Test_ParseFromFile(t *testing.T) {