node-tdd 3.1.2 → 3.3.2

package/README.md

@@ -132,6 +132,13 @@ Default: `false`
 
 When set to true, all headers are stripped when requests are recorded.
 
+### nockReqHeaderOverwrite
+
+Type: `object`<br>
+Default: `{}`
+
+Can be used to overwrite `reqheaders` in recordings. Cassette files are only updated when changed.
+
 #### fixtureFolder
 
 Type: `string`<br>

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,6 +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.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')))
@@ -68,6 +69,18 @@ module.exports = opts => {
     return flags.some(flag => needleFlags.includes(flag));
   };
 
+  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 value;
+  };
+
   return {
     inject: async cassetteFile => {
       assert(nockDone === null);
@@ -126,7 +139,8 @@ module.exports = opts => {
             nockRecorder.clear();
             return recorded.map(record => Object.assign(record, {
               headers: opts.stripHeaders === true ? undefined : convertHeaders(record.rawHeaders),
-              rawHeaders: undefined
+              rawHeaders: undefined,
+              reqheaders: rewriteHeaders(record.reqheaders, overwriteHeaders)
             }));
           });
         } else if (anyFlagPresent(['stub'])) {
@@ -136,7 +150,7 @@ module.exports = opts => {
             path: options.path,
             body: tryParseJson(body),
             status: 200,
-            reqheaders: rewriteHeaders(options.headers),
+            reqheaders: rewriteHeaders(options.headers, overwriteHeaders),
             response: {},
             responseIsBinary: false
           });
@@ -194,9 +208,10 @@ module.exports = opts => {
 
             if (anyFlagPresent(['magic', 'headers'])) {
               // add new headers
-              pendingMocks[idx].record.reqheaders = { ...rewriteHeaders(req.headers),
+              const reqheaders = { ...rewriteHeaders(req.headers),
                 ...rewriteHeaders(pendingMocks[idx].record.reqheaders)
               };
+              pendingMocks[idx].record.reqheaders = rewriteHeaders(reqheaders, overwriteHeaders);
             }
 
             if (anyFlagPresent(['magic', 'response'])) {
@@ -218,7 +233,7 @@ module.exports = opts => {
         afterRecord: recordings => JSON.stringify(recordings.map(r => ({ ...r,
           body: tryParseJson(r.body),
           rawHeaders: opts.stripHeaders === true ? undefined : r.rawHeaders,
-          reqheaders: rewriteHeaders(r.reqheaders)
+          reqheaders: rewriteHeaders(r.reqheaders, overwriteHeaders)
         })), null, 2)
       }, resolve));
       requestInjector.inject();

package/lib/util/desc.js

@@ -10,6 +10,10 @@ const callsites = require('callsites');
 
 const get = require('lodash.get');
 
+const {
+  fileURLToPath
+} = require('url');
+
 const minimist = require('minimist');
 
 const tmp = require('tmp');
@@ -45,7 +49,11 @@ const mocha = {
 const desc = (suiteName, optsOrTests, testsOrNull = null) => {
   const opts = testsOrNull === null ? {} : optsOrTests;
   const tests = testsOrNull === null ? optsOrTests : testsOrNull;
-  const testFile = path.resolve(callsites()[1].getFileName());
+  const filenameOrUrl = callsites()[1].getFileName(); // eslint-disable-next-line @blackflux/rules/istanbul-prevent-ignore
+
+  /* istanbul ignore next */
+
+  const testFile = path.resolve(filenameOrUrl.startsWith('file://') ? fileURLToPath(filenameOrUrl) : filenameOrUrl);
 
   const resolve = name => path.join(path.dirname(testFile), name.replace(/\$FILENAME/g, path.basename(testFile)));
 
@@ -55,6 +63,7 @@ const desc = (suiteName, optsOrTests, testsOrNull = null) => {
     nockFolder: Joi.string().optional(),
     nockModifiers: Joi.object().optional().pattern(Joi.string(), Joi.function()),
     nockStripHeaders: Joi.boolean().optional(),
+    nockReqHeaderOverwrite: Joi.object().optional().pattern(Joi.string(), Joi.alternatives(Joi.function(), Joi.string())),
     fixtureFolder: Joi.string().optional(),
     envVarsFile: Joi.string().optional(),
     envVars: Joi.object().optional().unknown(true).pattern(Joi.string(), Joi.string()),
@@ -73,6 +82,7 @@ const desc = (suiteName, optsOrTests, testsOrNull = null) => {
   const nockFolder = resolve(get(opts, 'nockFolder', '$FILENAME__cassettes'));
   const nockModifiers = get(opts, 'nockModifiers', {});
   const nockStripHeaders = get(opts, 'nockStripHeaders', false);
+  const nockReqHeaderOverwrite = get(opts, 'nockReqHeaderOverwrite', {});
   const fixtureFolder = resolve(get(opts, 'fixtureFolder', '$FILENAME__fixtures'));
   const envVarsFile = resolve(get(opts, 'envVarsFile', '$FILENAME.env.yml'));
   const envVars = get(opts, 'envVars', null);
@@ -178,6 +188,7 @@ const desc = (suiteName, optsOrTests, testsOrNull = null) => {
             requestRecorder = RequestRecorder({
               cassetteFolder: `${nockFolder}/`,
               stripHeaders: nockStripHeaders,
+              reqHeaderOverwrite: nockReqHeaderOverwrite,
               strict: true,
               heal: nockHeal,
               modifiers: nockModifiers

package/package.json

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