- Fix unit-tests to support req.get

PerimeterX · Jan 11, 2024 · ce35968 · ce35968
1 parent 5b32446
commit ce35968
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 0 deletions.
diff --git a/test/pxcors.test.js b/test/pxcors.test.js
@@ -37,7 +37,11 @@ describe('PX Cors - pxCors.js', () => {
         req.protocol = 'http';
         req.ip = '1.2.3.4';
         req.hostname = 'example.com';
+        req.host = 'example.com';
         req.get = (key) => {
+            if (key === 'host') {
+                return req.host;
+            }
             return req.headers[key] || '';
         };
 

diff --git a/test/pxenforcer.test.js b/test/pxenforcer.test.js
@@ -40,6 +40,9 @@ describe('PX Enforcer - pxenforcer.js', () => {
         req.ip = '1.2.3.4';
         req.hostname = 'example.com';
         req.get = (key) => {
+            if (key === 'host') {
+                return req.host;
+            }
             return req.headers[key] || '';
         };