test(headless): add headless-dsl integration test

Signed-off-by: Dwi Siswanto <[email protected]>
projectdiscovery · Feb 2, 2025 · a7fbba2 · a7fbba2
1 parent 06c6a37
commit a7fbba2
Show file tree

Hide file tree

Showing 2 changed files with 21 additions and 1 deletion.
diff --git a/cmd/integration-test/headless.go b/cmd/integration-test/headless.go
@@ -1,6 +1,7 @@
 package main
 
 import (
+	"fmt"
 	"io"
 	"net/http"
 	"net/http/httptest"
@@ -13,6 +14,7 @@ import (
 var headlessTestcases = []TestCaseInfo{
 	{Path: "protocols/headless/headless-basic.yaml", TestCase: &headlessBasic{}},
 	{Path: "protocols/headless/headless-waitevent.yaml", TestCase: &headlessBasic{}},
+	{Path: "protocols/headless/headless-dsl.yaml", TestCase: &headlessBasic{}},
 	{Path: "protocols/headless/headless-self-contained.yaml", TestCase: &headlessSelfContained{}},
 	{Path: "protocols/headless/headless-header-action.yaml", TestCase: &headlessHeaderActions{}},
 	{Path: "protocols/headless/headless-extract-values.yaml", TestCase: &headlessExtractValues{}},
@@ -30,7 +32,7 @@ type headlessBasic struct{}
 func (h *headlessBasic) Execute(filePath string) error {
 	router := httprouter.New()
 	router.GET("/", func(w http.ResponseWriter, r *http.Request, _ httprouter.Params) {
-		_, _ = w.Write([]byte("<html><body></body></html>"))
+		_, _ = fmt.Fprintf(w, "<html><body>%s</body></html>", r.URL.Query().Get("_"))
 	})
 	ts := httptest.NewServer(router)
 	defer ts.Close()

diff --git a/integration_tests/protocols/headless/headless-dsl.yaml b/integration_tests/protocols/headless/headless-dsl.yaml
@@ -0,0 +1,18 @@
+id: headless-dsl
+
+info:
+  name: Headless DSL
+  author: dwisiswant0
+  severity: info
+  tags: headless
+
+headless:
+  - steps:
+      - action: navigate
+        args:
+          url: "{{BaseURL}}/?_={{urlencode(concat('foo', '-', 'bar'))}}"
+      - action: waitload
+    matchers:
+      - type: word
+        words:
+          - "foo-bar"