daffainfo
/
nuclei
mirror of https://github.com/daffainfo/nuclei.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Added TLS protocol test

dev
Ice3man543 2021-02-23 18:17:24 +05:30
parent a502847621
commit 37ce0a72f4
1 changed files with 10 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

15
v2/pkg/protocols/network/network_test.go
View File

@ -14,7 +14,7 @@ func TestNetworkCompileMake(t *testing.T) {
templateID := "testing-network" templateID := "testing-network"
request := &Request{ request := &Request{
ID: templateID, ID: templateID,
Address: []string{"{{Hostname}}", "{{Hostname}}:8082"}, Address: []string{"{{Hostname}}", "{{Hostname}}:8082", "tls://{{Hostname}}:443"},
ReadSize: 1024, ReadSize: 1024,
Inputs: []*Input{{Data: "test-data"}}, Inputs: []*Input{{Data: "test-data"}},
} }
@ -25,12 +25,17 @@ func TestNetworkCompileMake(t *testing.T) {
err := request.Compile(executerOpts) err := request.Compile(executerOpts)
require.Nil(t, err, "could not compile network request") require.Nil(t, err, "could not compile network request")
require.Equal(t, 2, len(request.addresses), "could not get correct number of input address") require.Equal(t, 3, len(request.addresses), "could not get correct number of input address")
t.Run("check-host", func(t *testing.T) { t.Run("check-host", func(t *testing.T) {
require.Equal(t, "{{Hostname}}", request.addresses[0].key, "could not get correct host") require.Equal(t, "{{Hostname}}", request.addresses[0].ip, "could not get correct host")
}) })
t.Run("check-host-with-port", func(t *testing.T) { t.Run("check-host-with-port", func(t *testing.T) {
require.Equal(t, "{{Hostname}}", request.addresses[1].key, "could not get correct host with port") require.Equal(t, "{{Hostname}}", request.addresses[1].ip, "could not get correct host with port")
require.Equal(t, "8082", request.addresses[1].value, "could not get correct port for host") require.Equal(t, "8082", request.addresses[1].port, "could not get correct port for host")
})
t.Run("check-tls-with-port", func(t *testing.T) {
require.Equal(t, "{{Hostname}}", request.addresses[2].ip, "could not get correct host with port")
require.Equal(t, "443", request.addresses[2].port, "could not get correct port for host")
require.True(t, request.addresses[2].tls, "could not get correct port for host")
}) })
} }