mirror of
https://github.com/projectdiscovery/nuclei.git
synced 2025-12-18 17:25:28 +00:00
2d317884b5
* new sdk progress * nuclei v3 new sdk/library * fix TestActionGetResource broken link * fix clistats + clustering and more * fix lint error * fix missing ticker * update advanced library usage example * fix integration tests * misc update * add utm_source and fix lint error --------- Co-authored-by: sandeep <8293321+ehsandeep@users.noreply.github.com>
43 lines
1.3 KiB
Go
43 lines
1.3 KiB
Go
package nuclei_test
|
|
|
|
import (
|
|
"testing"
|
|
|
|
nuclei "github.com/projectdiscovery/nuclei/v2/lib"
|
|
"github.com/stretchr/testify/require"
|
|
)
|
|
|
|
func TestSimpleNuclei(t *testing.T) {
|
|
ne, err := nuclei.NewNucleiEngine(
|
|
nuclei.WithTemplateFilters(nuclei.TemplateFilters{ProtocolTypes: "dns"}),
|
|
nuclei.EnableStatsWithOpts(nuclei.StatsOptions{JSON: true}),
|
|
)
|
|
require.Nil(t, err)
|
|
ne.LoadTargets([]string{"scanme.sh"}, false) // probe non http/https target is set to false here
|
|
// when callback is nil it nuclei will print JSON output to stdout
|
|
err = ne.ExecuteWithCallback(nil)
|
|
require.Nil(t, err)
|
|
defer ne.Close()
|
|
}
|
|
|
|
func TestThreadSafeNuclei(t *testing.T) {
|
|
// create nuclei engine with options
|
|
ne, err := nuclei.NewThreadSafeNucleiEngine()
|
|
require.Nil(t, err)
|
|
|
|
// scan 1 = run dns templates on scanme.sh
|
|
t.Run("scanme.sh", func(t *testing.T) {
|
|
err = ne.ExecuteNucleiWithOpts([]string{"scanme.sh"}, nuclei.WithTemplateFilters(nuclei.TemplateFilters{ProtocolTypes: "dns"}))
|
|
require.Nil(t, err)
|
|
})
|
|
|
|
// scan 2 = run dns templates on honey.scanme.sh
|
|
t.Run("honey.scanme.sh", func(t *testing.T) {
|
|
err = ne.ExecuteNucleiWithOpts([]string{"honey.scanme.sh"}, nuclei.WithTemplateFilters(nuclei.TemplateFilters{ProtocolTypes: "dns"}))
|
|
require.Nil(t, err)
|
|
})
|
|
|
|
// wait for all scans to finish
|
|
defer ne.Close()
|
|
}
|