github.com/projectdiscovery/nuclei/v2@v2.9.15/pkg/protocols/network/network_test.go (about)

     1  package network
     2  
     3  import (
     4  	"testing"
     5  
     6  	"github.com/stretchr/testify/require"
     7  
     8  	"github.com/projectdiscovery/nuclei/v2/pkg/model"
     9  	"github.com/projectdiscovery/nuclei/v2/pkg/model/types/severity"
    10  	"github.com/projectdiscovery/nuclei/v2/pkg/testutils"
    11  )
    12  
    13  func TestNetworkCompileMake(t *testing.T) {
    14  	options := testutils.DefaultOptions
    15  
    16  	testutils.Init(options)
    17  	templateID := "testing-network"
    18  	request := &Request{
    19  		ID:       templateID,
    20  		Address:  []string{"tls://{{Host}}:443"},
    21  		ReadSize: 1024,
    22  		Inputs:   []*Input{{Data: "test-data"}},
    23  	}
    24  	executerOpts := testutils.NewMockExecuterOptions(options, &testutils.TemplateInfo{
    25  		ID:   templateID,
    26  		Info: model.Info{SeverityHolder: severity.Holder{Severity: severity.Low}, Name: "test"},
    27  	})
    28  	err := request.Compile(executerOpts)
    29  	require.Nil(t, err, "could not compile network request")
    30  
    31  	require.Equal(t, 1, len(request.addresses), "could not get correct number of input address")
    32  	t.Run("check-tls-with-port", func(t *testing.T) {
    33  		require.True(t, request.addresses[0].tls, "could not get correct port for host")
    34  	})
    35  }