Add e2e tests

e2e/multi_value_test.go

@@ -0,0 +1,226 @@
+package e2e
+
+import (
+	"net/http"
+	"testing"
+)
+
+// --- CLI: multiple same-key values are all sent in every request ---
+
+func TestMultipleHeadersSameKeyCLI(t *testing.T) {
+	t.Parallel()
+	cs := newCaptureServer()
+	defer cs.Close()
+
+	res := run("-U", cs.URL, "-r", "1", "-q", "-o", "json",
+		"-H", "X-Multi: value1", "-H", "X-Multi: value2")
+	assertExitCode(t, res, 0)
+
+	req := cs.lastRequest()
+	vals := req.Headers["X-Multi"]
+	if len(vals) < 2 {
+		t.Fatalf("expected 2 values for X-Multi, got %v", vals)
+	}
+	found := map[string]bool{}
+	for _, v := range vals {
+		found[v] = true
+	}
+	if !found["value1"] || !found["value2"] {
+		t.Errorf("expected both value1 and value2, got %v", vals)
+	}
+}
+
+func TestMultipleParamsSameKeyCLI(t *testing.T) {
+	t.Parallel()
+	cs := newCaptureServer()
+	defer cs.Close()
+
+	res := run("-U", cs.URL, "-r", "1", "-q", "-o", "json",
+		"-P", "color=red", "-P", "color=blue")
+	assertExitCode(t, res, 0)
+
+	req := cs.lastRequest()
+	vals := req.Query["color"]
+	if len(vals) < 2 {
+		t.Fatalf("expected 2 values for color param, got %v", vals)
+	}
+	found := map[string]bool{}
+	for _, v := range vals {
+		found[v] = true
+	}
+	if !found["red"] || !found["blue"] {
+		t.Errorf("expected both red and blue, got %v", vals)
+	}
+}
+
+func TestMultipleCookiesSameKeyCLI(t *testing.T) {
+	t.Parallel()
+	cs := newCaptureServer()
+	defer cs.Close()
+
+	res := run("-U", cs.URL, "-r", "1", "-q", "-o", "json",
+		"-C", "token=abc", "-C", "token=def")
+	assertExitCode(t, res, 0)
+
+	req := cs.lastRequest()
+	cookieHeader := ""
+	if v := req.Headers["Cookie"]; len(v) > 0 {
+		cookieHeader = v[0]
+	}
+	assertContains(t, cookieHeader, "token=abc")
+	assertContains(t, cookieHeader, "token=def")
+}
+
+// --- Config file: multiple values for same key cycle across requests ---
+
+func TestMultipleHeadersSameKeyYAMLCycle(t *testing.T) {
+	t.Parallel()
+	cs := newCaptureServer()
+	defer cs.Close()
+
+	config := `
+url: "` + cs.URL + `"
+requests: 20
+concurrency: 1
+quiet: true
+output: json
+headers:
+  - X-Multi: [val-a, val-b]
+`
+	configPath := writeTemp(t, "multi_header.yaml", config)
+
+	res := run("-f", configPath)
+	assertExitCode(t, res, 0)
+
+	seen := map[string]bool{}
+	for _, req := range cs.allRequests() {
+		if vals := req.Headers["X-Multi"]; len(vals) > 0 {
+			seen[vals[0]] = true
+		}
+	}
+	if !seen["val-a"] {
+		t.Error("expected val-a to appear in some requests")
+	}
+	if !seen["val-b"] {
+		t.Error("expected val-b to appear in some requests")
+	}
+}
+
+func TestMultipleParamsSameKeyYAMLCycle(t *testing.T) {
+	t.Parallel()
+	cs := newCaptureServer()
+	defer cs.Close()
+
+	config := `
+url: "` + cs.URL + `"
+requests: 20
+concurrency: 1
+quiet: true
+output: json
+params:
+  - tag: [go, rust]
+`
+	configPath := writeTemp(t, "multi_param.yaml", config)
+
+	res := run("-f", configPath)
+	assertExitCode(t, res, 0)
+
+	seen := map[string]bool{}
+	for _, req := range cs.allRequests() {
+		if vals := req.Query["tag"]; len(vals) > 0 {
+			seen[vals[0]] = true
+		}
+	}
+	if !seen["go"] {
+		t.Error("expected 'go' to appear in some requests")
+	}
+	if !seen["rust"] {
+		t.Error("expected 'rust' to appear in some requests")
+	}
+}
+
+// --- Multiple bodies cycle ---
+
+func TestMultipleBodiesCycle(t *testing.T) {
+	t.Parallel()
+	cs := newCaptureServer()
+	defer cs.Close()
+
+	res := run("-U", cs.URL, "-r", "10", "-c", "1", "-M", "POST", "-q", "-o", "json",
+		"-B", "body-alpha", "-B", "body-beta")
+	assertExitCode(t, res, 0)
+
+	bodies := map[string]bool{}
+	for _, req := range cs.allRequests() {
+		bodies[req.Body] = true
+	}
+	if !bodies["body-alpha"] {
+		t.Error("expected body-alpha to appear in requests")
+	}
+	if !bodies["body-beta"] {
+		t.Error("expected body-beta to appear in requests")
+	}
+}
+
+// --- Multiple methods cycling ---
+
+func TestMultipleMethodsCycleDistribution(t *testing.T) {
+	t.Parallel()
+	cs := newCaptureServer()
+	defer cs.Close()
+
+	res := run("-U", cs.URL, "-r", "20", "-c", "1", "-q", "-o", "json",
+		"-M", "GET", "-M", "POST", "-M", "PUT")
+	assertExitCode(t, res, 0)
+
+	methods := map[string]int{}
+	for _, req := range cs.allRequests() {
+		methods[req.Method]++
+	}
+	if methods["GET"] == 0 {
+		t.Error("expected GET to appear")
+	}
+	if methods["POST"] == 0 {
+		t.Error("expected POST to appear")
+	}
+	if methods["PUT"] == 0 {
+		t.Error("expected PUT to appear")
+	}
+}
+
+// --- Template in method ---
+
+func TestTemplateInMethod(t *testing.T) {
+	t.Parallel()
+	cs := newCaptureServer()
+	defer cs.Close()
+
+	res := run("-U", cs.URL, "-r", "1", "-q", "-o", "json",
+		"-M", `{{ strings_ToUpper "post" }}`)
+	assertExitCode(t, res, 0)
+
+	req := cs.lastRequest()
+	if req.Method != http.MethodPost {
+		t.Errorf("expected method POST from template, got %s", req.Method)
+	}
+}
+
+// --- Template in cookie value ---
+
+func TestTemplateInCookie(t *testing.T) {
+	t.Parallel()
+	cs := newCaptureServer()
+	defer cs.Close()
+
+	res := run("-U", cs.URL, "-r", "1", "-q", "-o", "json",
+		"-C", `session={{ fakeit_UUID }}`)
+	assertExitCode(t, res, 0)
+
+	req := cs.lastRequest()
+	if req.Cookies["session"] == "" {
+		t.Error("expected session cookie with UUID value, got empty")
+	}
+	if len(req.Cookies["session"]) < 10 {
+		t.Errorf("expected UUID-like session cookie, got %q", req.Cookies["session"])
+	}
+}