Sleep based on x-set-response-delay-ms header

mendhak · mendhak · commit 5613125d126a · 2021-02-18T23:03:34.000Z
diff --git a/index.js b/index.js
@@ -6,6 +6,8 @@ var app = express()
 const os = require('os');
 const jwt = require('jsonwebtoken');
 var concat = require('concat-stream');
+const { promisify } = require('util');
+const sleep = promisify(setTimeout);
 
 app.set('json spaces', 2);
 app.set('trust proxy', ['loopback', 'linklocal', 'uniquelocal']);
@@ -61,17 +63,24 @@ app.all('*', (req, res) => {
     res.status(setResponseStatusCode)
   }
 
-  res.json(echo);
-  if (process.env.LOG_IGNORE_PATH != req.path) {
-    console.log('-----------------')
+  const sleepTime = parseInt(req.headers["x-set-response-delay-ms"], 0)
+  sleep(sleepTime).then(() => {
+    
+    res.json(echo);
 
-    let spacer = 4;
-    if(process.env.LOG_WITHOUT_NEWLINE){
-      spacer = null;
+    if (process.env.LOG_IGNORE_PATH != req.path) {
+      console.log('-----------------')
+  
+      let spacer = 4;
+      if(process.env.LOG_WITHOUT_NEWLINE){
+        spacer = null;
+      }
+  
+      console.log(JSON.stringify(echo, null, spacer));
     }
+  });
 
-    console.log(JSON.stringify(echo, null, spacer));
-  }
+  
 });
 
 const sslOpts = {
diff --git a/tests.sh b/tests.sh
@@ -64,14 +64,23 @@ else
     exit 1
 fi
 
+REQUEST_WITH_SLEEP_MS=$(curl -o /dev/null -Ss -H "x-set-response-delay-ms: 6000" -k https://localhost:8443/ -w '%{time_total}')
+if [[ $(echo "$REQUEST_WITH_SLEEP_MS>5" | bc -l) == 1 ]]; then 
+    passed "Request with response delay passed"
+else 
+    failed "Request with response delay failed"
+    echo $REQUEST_WITH_SLEEP_MS
+    exit 1
+fi
+
 REQUEST=$(curl -s -X PUT -H "Arbitrary:Header" -d aaa=bbb http://localhost:8080/hello-world)
 if [ $(echo $REQUEST | jq -r '.path') == '/hello-world' ] && \
    [ $(echo $REQUEST | jq -r '.method') == 'PUT' ] && \
    [ $(echo $REQUEST | jq -r '.headers.arbitrary') == 'Header' ]
 then
-    passed "HTTP request passed."
+    passed "HTTP request with arbitrary header passed."
 else
-    failed "HTTP request failed."
+    failed "HTTP request with arbitrary header failed."
     echo $REQUEST | jq
     exit 1
 fi
