External/nuclei
1
0
Fork 0
mirror of https://github.com/projectdiscovery/nuclei.git synced 2025-12-22 10:05:29 +00:00
Code Issues Packages Projects Releases Wiki Activity

chore: add new test

Browse Source
This commit is contained in:
Pedro Lopez Mareque 2021-10-09 12:26:18 +02:00
parent 4e54a61b65
commit 3386c1c5bd
2 changed files with 42 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
v2/pkg/protocols/headless/engine/page_actions.go
View File

@ -91,7 +91,10 @@ func (p *Page) WaitVisible(act *Action, out map[string]string) error {
if err != nil { if err != nil {
return err return err
} }
return element.WaitVisible() if err = element.WaitVisible(); err != nil {
return errors.Wrap(err, "could not wait element")
}
return nil
} }
// ActionAddHeader executes a AddHeader action. // ActionAddHeader executes a AddHeader action.

38
v2/pkg/protocols/headless/engine/page_actions_test.go
View File

@ -278,3 +278,41 @@ func TestActionHeadersChange(t *testing.T) {
require.Equal(t, "found", strings.ToLower(strings.TrimSpace(page.Page().MustElement("html").MustText())), "could not set header correctly") require.Equal(t, "found", strings.ToLower(strings.TrimSpace(page.Page().MustElement("html").MustText())), "could not set header correctly")
} }
func TestActionWaitVisible(t *testing.T) {
_ = protocolstate.Init(&types.Options{})
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>
<button style="display:none" id="test">Wait for me!</button>
<script>
setTimeout(() => document.querySelector('#test').style.display = 'block', 1000);
</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: "waitvisible", Data: map[string]string{"by": "x", "xpath": "//button[@id='test']"}},
}
_, page, err := instance.Run(parsed, actions, 20*time.Second)
require.Nil(t, err, "could not run page actions")
defer page.Close()
page.Page().MustElement("button").MustWaitVisible()
}