mirror of https://github.com/daffainfo/nuclei.git
94 lines
2.7 KiB
Go
94 lines
2.7 KiB
Go
package loader
|
|
|
|
import (
|
|
"reflect"
|
|
"testing"
|
|
|
|
"github.com/stretchr/testify/require"
|
|
)
|
|
|
|
func TestLoadTemplates(t *testing.T) {
|
|
store, err := New(&Config{
|
|
Templates: []string{"cves/CVE-2021-21315.yaml"},
|
|
})
|
|
require.Nil(t, err, "could not load templates")
|
|
require.Equal(t, []string{"cves/CVE-2021-21315.yaml"}, store.finalTemplates, "could not get correct templates")
|
|
|
|
templatesDirectory := "/test"
|
|
t.Run("blank", func(t *testing.T) {
|
|
store, err := New(&Config{
|
|
TemplatesDirectory: templatesDirectory,
|
|
})
|
|
require.Nil(t, err, "could not load templates")
|
|
require.Equal(t, []string{templatesDirectory}, store.finalTemplates, "could not get correct templates")
|
|
})
|
|
t.Run("only-tags", func(t *testing.T) {
|
|
store, err := New(&Config{
|
|
Tags: []string{"cves"},
|
|
TemplatesDirectory: templatesDirectory,
|
|
})
|
|
require.Nil(t, err, "could not load templates")
|
|
require.Equal(t, []string{templatesDirectory}, store.finalTemplates, "could not get correct templates")
|
|
})
|
|
t.Run("tags-with-path", func(t *testing.T) {
|
|
store, err := New(&Config{
|
|
Tags: []string{"cves"},
|
|
TemplatesDirectory: templatesDirectory,
|
|
})
|
|
require.Nil(t, err, "could not load templates")
|
|
require.Equal(t, []string{templatesDirectory}, store.finalTemplates, "could not get correct templates")
|
|
})
|
|
}
|
|
|
|
func TestRemoteTemplates(t *testing.T) {
|
|
var nilStringSlice []string
|
|
type args struct {
|
|
config *Config
|
|
}
|
|
tests := []struct {
|
|
name string
|
|
args args
|
|
want *Store
|
|
wantErr bool
|
|
}{
|
|
{
|
|
name: "remote-templates-positive",
|
|
args: args{
|
|
config: &Config{
|
|
TemplateURLs: []string{"https://raw.githubusercontent.com/projectdiscovery/nuclei-templates/master/technologies/tech-detect.yaml"},
|
|
RemoteTemplateDomainList: []string{"localhost","raw.githubusercontent.com"},
|
|
},
|
|
},
|
|
want: &Store{
|
|
finalTemplates: []string{"https://raw.githubusercontent.com/projectdiscovery/nuclei-templates/master/technologies/tech-detect.yaml"},
|
|
},
|
|
wantErr: false,
|
|
},
|
|
{
|
|
name: "remote-templates-negative",
|
|
args: args{
|
|
config: &Config{
|
|
TemplateURLs: []string{"https://raw.githubusercontent.com/projectdiscovery/nuclei-templates/master/technologies/tech-detect.yaml"},
|
|
RemoteTemplateDomainList: []string{"localhost"},
|
|
},
|
|
},
|
|
want: &Store{
|
|
finalTemplates: nilStringSlice,
|
|
},
|
|
wantErr: true,
|
|
},
|
|
}
|
|
for _, tt := range tests {
|
|
t.Run(tt.name, func(t *testing.T) {
|
|
got, err := New(tt.args.config)
|
|
if (err != nil) != tt.wantErr {
|
|
t.Errorf("New() error = %v, wantErr %v", err, tt.wantErr)
|
|
return
|
|
}
|
|
if !reflect.DeepEqual(got.finalTemplates, tt.want.finalTemplates) {
|
|
t.Errorf("New() = %v, want %v", got.finalTemplates, tt.want.finalTemplates)
|
|
}
|
|
})
|
|
}
|
|
}
|