node-tdd 3.2.0 → 3.3.0

package/lib/modules/request-recorder/util.js

@@ -31,5 +31,10 @@ module.exports.rewriteHeaders = (headers, fn = (k, v) => v) => {
     return {};
   }
 
-  return Object.fromEntries(Object.entries(headers).sort(([a], [b]) => a > b ? 1 : -1).map(([k, v]) => [k.toLowerCase(), fn(k, v)]));
+  const headersLower = Object.fromEntries(Object.entries(headers).sort(([a], [b]) => a > b ? 1 : -1).map(([k, v]) => [k.toLowerCase(), v]));
+  const result = {};
+  Object.entries(headersLower).forEach(([k, v]) => {
+    result[k] = fn(k, v, headersLower);
+  });
+  return result;
 };

package/lib/modules/request-recorder.js

@@ -45,7 +45,7 @@ module.exports = opts => {
   Joi.assert(opts, Joi.object().keys({
     cassetteFolder: Joi.string(),
     stripHeaders: Joi.boolean(),
-    reqHeaderOverwrite: Joi.object().pattern(Joi.string().case('lower'), Joi.string()),
+    reqHeaderOverwrite: Joi.object().pattern(Joi.string().case('lower'), Joi.alternatives(Joi.string(), Joi.function())),
     strict: Joi.boolean(),
     heal: Joi.alternatives(Joi.boolean(), Joi.string()),
     modifiers: Joi.object().pattern(Joi.string(), Joi.alternatives(Joi.function(), Joi.link('#modifiers')))
@@ -69,12 +69,16 @@ module.exports = opts => {
     return flags.some(flag => needleFlags.includes(flag));
   };
 
-  const overwriteHeaders = (k, v) => {
-    if (k in opts.reqHeaderOverwrite) {
-      return opts.reqHeaderOverwrite[k];
+  const overwriteHeaders = (key, value, headers) => {
+    if (key in opts.reqHeaderOverwrite) {
+      return typeof opts.reqHeaderOverwrite[key] === 'function' ? opts.reqHeaderOverwrite[key]({
+        key,
+        value,
+        headers
+      }) : opts.reqHeaderOverwrite[key];
     }
 
-    return v;
+    return value;
   };
 
   return {

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "node-tdd",
-  "version": "3.2.0",
+  "version": "3.3.0",
   "description": "Drop in extension for mocha to abstract commonly used test setups",
   "main": "lib/index.js",
   "scripts": {