github.com/projectdiscovery/nuclei/v2@v2.9.15/pkg/protocols/ssl/ssl_test.go (about)

     1  package ssl
     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/output"
    11  	"github.com/projectdiscovery/nuclei/v2/pkg/protocols/common/contextargs"
    12  	"github.com/projectdiscovery/nuclei/v2/pkg/testutils"
    13  )
    14  
    15  func TestSSLProtocol(t *testing.T) {
    16  	options := testutils.DefaultOptions
    17  
    18  	testutils.Init(options)
    19  	templateID := "testing-ssl"
    20  	request := &Request{
    21  		Address: "{{Hostname}}",
    22  	}
    23  	executerOpts := testutils.NewMockExecuterOptions(options, &testutils.TemplateInfo{
    24  		ID:   templateID,
    25  		Info: model.Info{SeverityHolder: severity.Holder{Severity: severity.Low}, Name: "test"},
    26  	})
    27  	err := request.Compile(executerOpts)
    28  	require.Nil(t, err, "could not compile ssl request")
    29  
    30  	var gotEvent output.InternalEvent
    31  	ctxArgs := contextargs.NewWithInput("google.com:443")
    32  	err = request.ExecuteWithResults(ctxArgs, nil, nil, func(event *output.InternalWrappedEvent) {
    33  		gotEvent = event.InternalEvent
    34  	})
    35  	require.Nil(t, err, "could not run ssl request")
    36  	require.NotEmpty(t, gotEvent, "could not get event items")
    37  }
    38  
    39  func TestGetAddress(t *testing.T) {
    40  	address, _ := getAddress("https://google.com")
    41  	require.Equal(t, "google.com:443", address, "could not get correct address")
    42  }