From 97c9a57e8c52297eceb7bdde09e49259b1dea6e5 Mon Sep 17 00:00:00 2001
From: hahwul <hahwul@gmail.com>
Date: Sat, 25 Dec 2021 01:01:27 +0900
Subject: [PATCH] Update test code

---
 pkg/optimization/optimization_test.go | 21 ++++++++++++++++++++-
 1 file changed, 20 insertions(+), 1 deletion(-)

diff --git a/pkg/optimization/optimization_test.go b/pkg/optimization/optimization_test.go
index a0dc2b5d..9c1b4aab 100644
--- a/pkg/optimization/optimization_test.go
+++ b/pkg/optimization/optimization_test.go
@@ -193,7 +193,26 @@ func TestMakeRequestQuery(t *testing.T) {
 				payload: "dalfox",
 				ptype:   "",
 				pAction: "",
-				pEncode: "",
+				pEncode: "htmlEncode",
+				options: model.Options{
+					Data:   "abcd=1234",
+					Cookie: "abcd=1234",
+					Header: []string{
+						"X-API-Key: 1234",
+					},
+					Method: "POST",
+				},
+			},
+		},
+		{
+			name: "TestMakeRequestQuery2",
+			args: args{
+				target:  "https://www.hahwul.com",
+				param:   "param",
+				payload: "dalfox",
+				ptype:   "FORM",
+				pAction: "toAppend",
+				pEncode: "urlEncode",
 				options: model.Options{
 					Data:   "abcd=1234",
 					Cookie: "abcd=1234",