mirror of
https://github.com/projectdiscovery/nuclei.git
synced 2025-12-22 20:35:27 +00:00
628628893c
* fix make http request inconsistencies * remove parameters from http vars * fix trailingslash unit test * naming conventions: best practices * fix publish docs action * remove branch ref from push
64 lines
2.9 KiB
Go
64 lines
2.9 KiB
Go
package utils
|
|
|
|
import (
|
|
"testing"
|
|
|
|
"github.com/projectdiscovery/nuclei/v2/pkg/protocols/common/contextargs"
|
|
urlutil "github.com/projectdiscovery/utils/url"
|
|
"github.com/stretchr/testify/require"
|
|
)
|
|
|
|
func TestVariables(t *testing.T) {
|
|
baseURL := "http://localhost:9001/test/123"
|
|
parsed, _ := urlutil.Parse(baseURL)
|
|
// trailingslash is only true when both target/inputURL and payload {{BaseURL}}/xyz both have slash
|
|
values := GenerateVariablesWithURL(parsed, false, nil)
|
|
|
|
require.Equal(t, values["BaseURL"], parsed.String(), "incorrect baseurl")
|
|
require.Equal(t, values["RootURL"], "http://localhost:9001", "incorrect rootURL")
|
|
require.Equal(t, values["Host"], "localhost", "incorrect domain name")
|
|
require.Equal(t, values["Path"], "/test", "incorrect path")
|
|
require.Equal(t, values["File"], "123", "incorrect file")
|
|
require.Equal(t, values["Port"], "9001", "incorrect port number")
|
|
require.Equal(t, values["Scheme"], "http", "incorrect scheme")
|
|
require.Equal(t, values["Hostname"], "localhost:9001", "incorrect hostname")
|
|
|
|
baseURL = "https://example.com"
|
|
parsed, _ = urlutil.Parse(baseURL)
|
|
values = GenerateVariablesWithURL(parsed, false, nil)
|
|
|
|
require.Equal(t, values["BaseURL"], parsed.String(), "incorrect baseurl")
|
|
require.Equal(t, values["Host"], "example.com", "incorrect domain name")
|
|
require.Equal(t, values["RootURL"], "https://example.com", "incorrect rootURL")
|
|
require.Equal(t, values["Path"], "", "incorrect path")
|
|
require.Equal(t, values["Port"], "443", "incorrect port number")
|
|
require.Equal(t, values["Scheme"], "https", "incorrect scheme")
|
|
require.Equal(t, values["Hostname"], "example.com", "incorrect hostname")
|
|
|
|
baseURL = "ftp://foobar.com/"
|
|
parsed, _ = urlutil.Parse(baseURL)
|
|
values = GenerateVariablesWithURL(parsed, false, nil)
|
|
|
|
require.Equal(t, values["BaseURL"], parsed.String(), "incorrect baseurl")
|
|
require.Equal(t, values["Host"], "foobar.com", "incorrect domain name")
|
|
require.Equal(t, values["RootURL"], "ftp://foobar.com", "incorrect rootURL")
|
|
require.Equal(t, values["Path"], "/", "incorrect path")
|
|
require.Equal(t, values["Port"], "", "incorrect port number") // Unsupported protocol results in a blank port
|
|
require.Equal(t, values["Scheme"], "ftp", "incorrect scheme")
|
|
require.Equal(t, values["Hostname"], "foobar.com", "incorrect hostname")
|
|
|
|
baseURL = "http://scanme.sh"
|
|
ctxArgs := contextargs.NewWithInput(baseURL)
|
|
ctxArgs.MetaInput.CustomIP = "1.2.3.4"
|
|
values = GenerateVariablesWithContextArgs(ctxArgs, true)
|
|
|
|
require.Equal(t, values["BaseURL"], baseURL, "incorrect baseurl")
|
|
require.Equal(t, values["Host"], "scanme.sh", "incorrect domain name")
|
|
require.Equal(t, values["RootURL"], "http://scanme.sh", "incorrect rootURL")
|
|
require.Equal(t, values["Path"], "", "incorrect path")
|
|
require.Equal(t, values["Port"], "80", "incorrect port number")
|
|
require.Equal(t, values["Scheme"], "http", "incorrect scheme")
|
|
require.Equal(t, values["Hostname"], "scanme.sh", "incorrect hostname")
|
|
require.Equal(t, values["ip"], "1.2.3.4", "incorrect ip")
|
|
}
|