Add ability to set response code via querystring

mendhak · mendhak · commit 0bf71dcea860 · 2021-11-21T19:20:26.000Z
Issue mendhak#32
diff --git a/index.js b/index.js
@@ -58,7 +58,7 @@ app.all('*', (req, res) => {
       echo.jwt = decoded;
     }
   }
-  const setResponseStatusCode = parseInt(req.headers["x-set-response-status-code"], 10)
+  const setResponseStatusCode = parseInt(req.headers["x-set-response-status-code"] || req.query["x-set-response-status-code"], 10)
   if (100 <= setResponseStatusCode && setResponseStatusCode < 600) {
     res.status(setResponseStatusCode)
   }
diff --git a/tests.sh b/tests.sh
@@ -57,9 +57,20 @@ REQUEST_WITH_STATUS_CODE=$(curl -s -k -o /dev/null -w "%{http_code}" -H "x-set-r
 REQUEST_WITH_STATUS_CODE_V=$(curl -v -k -o /dev/null -w "%{http_code}" -H "x-set-response-status-code: 404" https://localhost:8443/hello-world)
 if [ $(echo $REQUEST_WITH_STATUS_CODE == '404') ]
 then
-    passed "HTTPS status code request passed."
+    passed "HTTPS status code header passed."
 else
-    failed "HTTPS status code request failed."
+    failed "HTTPS status code header failed."
+    echo $REQUEST_WITH_STATUS_CODE_V
+    exit 1
+fi
+
+REQUEST_WITH_STATUS_CODE=$(curl -s -k -o /dev/null -w "%{http_code}" https://localhost:8443/status/test?x-set-response-status-code=419)
+REQUEST_WITH_STATUS_CODE_V=$(curl -v -k -o /dev/null -w "%{http_code}" https://localhost:8443/hello-world?x-set-response-status-code=419)
+if [ $(echo $REQUEST_WITH_STATUS_CODE == '419') ]
+then
+    passed "HTTPS status code querystring passed."
+else
+    failed "HTTPS status code querystring failed."
     echo $REQUEST_WITH_STATUS_CODE_V
     exit 1
 fi

Original file line number	Diff line number	Diff line change
`@@ -58,7 +58,7 @@ app.all('*', (req, res) => {`
`58`	`58`	`echo.jwt = decoded;`
`59`	`59`	`}`
`60`	`60`	`}`
`61`		`- const setResponseStatusCode = parseInt(req.headers["x-set-response-status-code"], 10)`
	`61`	`+ const setResponseStatusCode = parseInt(req.headers["x-set-response-status-code"] \|\| req.query["x-set-response-status-code"], 10)`
`62`	`62`	`if (100 <= setResponseStatusCode && setResponseStatusCode < 600) {`
`63`	`63`	`res.status(setResponseStatusCode)`
`64`	`64`	`}`