Added tests for page actions

2021-02-22 12:50:49 +05:30 · 2021-02-22 12:50:49 +05:30 · 5c50a5c079
parent 58237f103a
commit 5c50a5c079
1 changed files with 265 additions and 0 deletions
--- a/v2/pkg/protocols/headless/engine/page_actions_test.go
+++ b/v2/pkg/protocols/headless/engine/page_actions_test.go
@ -0,0 +1,265 @@
+package engine
+
+import (
+	"fmt"
+	"net/http"
+	"net/http/httptest"
+	"net/url"
+	"strings"
+	"testing"
+
+	"github.com/projectdiscovery/nuclei/v2/pkg/types"
+	"github.com/stretchr/testify/require"
+)
+
+func TestActionNavigate(t *testing.T) {
+	browser, err := New(&types.Options{ShowBrowser: false})
+	require.Nil(t, err, "could not create browser")
+	defer browser.Close()
+
+	instance, err := browser.NewInstance()
+	require.Nil(t, err, "could not create browser instance")
+
+	ts := httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
+		fmt.Fprintln(w, `
+		<html>
+		<head>
+			<title>Nuclei Test Page</title>
+		</head>
+		<body>
+			<h1>Nuclei Test</h1>
+		</body>
+	</html>`)
+	}))
+	defer ts.Close()
+
+	parsed, err := url.Parse(ts.URL)
+	require.Nil(t, err, "could not parse URL")
+
+	actions := []*Action{{ActionType: "navigate", Data: map[string]string{"url": "{{BaseURL}}"}}, {ActionType: "waitload"}}
+	_, page, err := instance.Run(parsed, actions)
+	require.Nil(t, err, "could not run page actions")
+	defer page.Close()
+
+	require.Equal(t, "Nuclei Test Page", page.Page().MustInfo().Title, "could not navigate correctly")
+}
+
+func TestActionScript(t *testing.T) {
+	browser, err := New(&types.Options{ShowBrowser: false})
+	require.Nil(t, err, "could not create browser")
+	defer browser.Close()
+
+	instance, err := browser.NewInstance()
+	require.Nil(t, err, "could not create browser instance")
+
+	t.Run("run-and-results", func(t *testing.T) {
+		ts := httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
+			fmt.Fprintln(w, `
+			<html>
+			<head>
+				<title>Nuclei Test Page</title>
+			</head>
+			<body>Nuclei Test Page</body>
+			<script>window.test = 'some-data';</script>
+		</html>`)
+		}))
+		defer ts.Close()
+
+		parsed, err := url.Parse(ts.URL)
+		require.Nil(t, err, "could not parse URL")
+
+		actions := []*Action{
+			{ActionType: "navigate", Data: map[string]string{"url": "{{BaseURL}}"}},
+			{ActionType: "waitload"},
+			{ActionType: "script", Name: "test", Data: map[string]string{"code": "window.test"}},
+		}
+		out, page, err := instance.Run(parsed, actions)
+		require.Nil(t, err, "could not run page actions")
+		defer page.Close()
+
+		require.Equal(t, "Nuclei Test Page", page.Page().MustInfo().Title, "could not navigate correctly")
+		require.Equal(t, "some-data", out["test"], "could not run js and get results correctly")
+	})
+
+	t.Run("hook", func(t *testing.T) {
+		ts := httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
+			fmt.Fprintln(w, `
+			<html>
+			<head>
+				<title>Nuclei Test Page</title>
+			</head>
+			<body>Nuclei Test Page</body>
+		</html>`)
+		}))
+		defer ts.Close()
+
+		parsed, err := url.Parse(ts.URL)
+		require.Nil(t, err, "could not parse URL")
+
+		actions := []*Action{
+			{ActionType: "script", Data: map[string]string{"code": "window.test = 'some-data';", "hook": "true"}},
+			{ActionType: "navigate", Data: map[string]string{"url": "{{BaseURL}}"}},
+			{ActionType: "waitload"},
+			{ActionType: "script", Name: "test", Data: map[string]string{"code": "window.test"}},
+		}
+		out, page, err := instance.Run(parsed, actions)
+		require.Nil(t, err, "could not run page actions")
+		defer page.Close()
+
+		require.Equal(t, "Nuclei Test Page", page.Page().MustInfo().Title, "could not navigate correctly")
+		require.Equal(t, "some-data", out["test"], "could not run js and get results correctly with js hook")
+	})
+}
+
+func TestActionClick(t *testing.T) {
+	browser, err := New(&types.Options{ShowBrowser: false})
+	require.Nil(t, err, "could not create browser")
+	defer browser.Close()
+
+	instance, err := browser.NewInstance()
+	require.Nil(t, err, "could not create browser instance")
+
+	ts := httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
+		fmt.Fprintln(w, `
+		<html>
+		<head>
+			<title>Nuclei Test Page</title>
+		</head>
+		<body>Nuclei Test Page</body>
+		<button onclick='this.setAttribute("a", "ok")'>click me</button>
+	</html>`)
+	}))
+	defer ts.Close()
+
+	parsed, err := url.Parse(ts.URL)
+	require.Nil(t, err, "could not parse URL")
+
+	actions := []*Action{
+		{ActionType: "navigate", Data: map[string]string{"url": "{{BaseURL}}"}},
+		{ActionType: "waitload"},
+		{ActionType: "click", Data: map[string]string{"selector": "button"}}, // Use css selector for clicking
+	}
+	_, page, err := instance.Run(parsed, actions)
+	require.Nil(t, err, "could not run page actions")
+	defer page.Close()
+
+	require.Equal(t, "Nuclei Test Page", page.Page().MustInfo().Title, "could not navigate correctly")
+	el := page.Page().MustElement("button")
+	val := el.MustAttribute("a")
+	require.Equal(t, "ok", *val, "could not click button")
+}
+
+func TestActionRightClick(t *testing.T) {
+	browser, err := New(&types.Options{ShowBrowser: false})
+	require.Nil(t, err, "could not create browser")
+	defer browser.Close()
+
+	instance, err := browser.NewInstance()
+	require.Nil(t, err, "could not create browser instance")
+
+	ts := httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
+		fmt.Fprintln(w, `
+		<html>
+		<head>
+			<title>Nuclei Test Page</title>
+		</head>
+		<body>Nuclei Test Page</body>
+		<button id="test" onrightclick=''>click me</button>
+		<script>
+		elm = document.getElementById("test");
+		elm.onmousedown = function(event) {
+			if (event.which == 3) {
+				elm.setAttribute("a", "ok")
+			}
+		}
+		</script>
+	</html>`)
+	}))
+	defer ts.Close()
+
+	parsed, err := url.Parse(ts.URL)
+	require.Nil(t, err, "could not parse URL")
+
+	actions := []*Action{
+		{ActionType: "navigate", Data: map[string]string{"url": "{{BaseURL}}"}},
+		{ActionType: "waitload"},
+		{ActionType: "rightclick", Data: map[string]string{"selector": "button"}}, // Use css selector for clicking
+	}
+	_, page, err := instance.Run(parsed, actions)
+	require.Nil(t, err, "could not run page actions")
+	defer page.Close()
+
+	require.Equal(t, "Nuclei Test Page", page.Page().MustInfo().Title, "could not navigate correctly")
+	el := page.Page().MustElement("button")
+	val := el.MustAttribute("a")
+	require.Equal(t, "ok", *val, "could not click button")
+}
+
+func TestActionTextInput(t *testing.T) {
+	browser, err := New(&types.Options{ShowBrowser: false})
+	require.Nil(t, err, "could not create browser")
+	defer browser.Close()
+
+	instance, err := browser.NewInstance()
+	require.Nil(t, err, "could not create browser instance")
+
+	ts := httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
+		fmt.Fprintln(w, `
+		<html>
+		<head>
+			<title>Nuclei Test Page</title>
+		</head>
+		<body>Nuclei Test Page</body>
+		<input type="text" onchange="this.setAttribute('event', 'input-change')">
+	</html>`)
+	}))
+	defer ts.Close()
+
+	parsed, err := url.Parse(ts.URL)
+	require.Nil(t, err, "could not parse URL")
+
+	actions := []*Action{
+		{ActionType: "navigate", Data: map[string]string{"url": "{{BaseURL}}"}},
+		{ActionType: "waitload"},
+		{ActionType: "text", Data: map[string]string{"selector": "input", "value": "test"}},
+	}
+	_, page, err := instance.Run(parsed, actions)
+	require.Nil(t, err, "could not run page actions")
+	defer page.Close()
+
+	require.Equal(t, "Nuclei Test Page", page.Page().MustInfo().Title, "could not navigate correctly")
+	el := page.Page().MustElement("input")
+	val := el.MustAttribute("event")
+	require.Equal(t, "input-change", *val, "could not get input change")
+	require.Equal(t, "test", el.MustText(), "could not get input change value")
+}
+
+func TestActionHeadersChange(t *testing.T) {
+	browser, err := New(&types.Options{ShowBrowser: false})
+	require.Nil(t, err, "could not create browser")
+	defer browser.Close()
+
+	instance, err := browser.NewInstance()
+	require.Nil(t, err, "could not create browser instance")
+
+	ts := httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
+		if r.Header.Get("Test") == "Hello" {
+			fmt.Fprintln(w, `found`)
+		}
+	}))
+	defer ts.Close()
+
+	parsed, err := url.Parse(ts.URL)
+	require.Nil(t, err, "could not parse URL")
+
+	actions := []*Action{
+		{ActionType: "setheader", Data: map[string]string{"part": "request", "key": "Test", "value": "Hello"}},
+		{ActionType: "navigate", Data: map[string]string{"url": "{{BaseURL}}"}},
+		{ActionType: "waitload"},
+	}
+	_, page, err := instance.Run(parsed, actions)
+	require.Nil(t, err, "could not run page actions")
+	defer page.Close()
+
+	require.Equal(t, "found", strings.ToLower(strings.TrimSpace(page.Page().MustElement("html").MustText())), "could not set header correctly")
+}