1 files changed, 49 insertions, 0 deletions
diff --git a/pkg/html/html_test.go b/pkg/html/html_test.go
new file mode 100644
index 000000000..d824e0246
--- /dev/null
+++ b/pkg/html/html_test.go
@@ -0,0 +1,49 @@
+// Copyright 2023 syzkaller project authors. All rights reserved.
+// Use of this source code is governed by Apache 2 LICENSE that can be found in the LICENSE file.
+
+package html
+
+import (
+	"testing"
+
+	"github.com/stretchr/testify/assert"
+)
+
+func TestDropParam(t *testing.T) {
+	tests := []struct {
+		in    string
+		key   string
+		value string
+		out   string
+	}{
+		{
+			in:    `/upstream?first=a&second=b`,
+			key:   `first`,
+			value: ``,
+			out:   `/upstream?second=b`,
+		},
+		{
+			in:    `/upstream?first=a&first=b&second=c`,
+			key:   `second`,
+			value: ``,
+			out:   `/upstream?first=a&first=b`,
+		},
+		{
+			in:    `/upstream?first=a&first=b&second=c`,
+			key:   `first`,
+			value: ``,
+			out:   `/upstream?second=c`,
+		},
+		{
+			in:    `/upstream?first=a&first=b&second=c`,
+			key:   `first`,
+			value: `b`,
+			out:   `/upstream?first=a&second=c`,
+		},
+	}
+
+	for _, test := range tests {
+		got := DropParam(test.in, test.key, test.value)
+		assert.Equal(t, test.out, got)
+	}
+}