node-tdd 6.0.3 → 6.1.0

package/lib/modules/request-recorder/heal-response-headers.js

@@ -0,0 +1,14 @@
+export default (interceptor) => {
+  const result = [];
+  for (let i = 0; i < interceptor.rawHeaders.length; i += 2) {
+    const name = interceptor.rawHeaders[i];
+    let value = interceptor.rawHeaders[i + 1];
+    if (name.toLowerCase() === 'content-length') {
+      value = interceptor.body[
+        interceptor.body instanceof Buffer ? 'byteLength' : 'length'
+      ].toString();
+    }
+    result.push(name, value);
+  }
+  return result;
+};

package/lib/modules/request-recorder.js

@@ -14,14 +14,14 @@ import nockMock from './request-recorder/nock-mock.js';
 import healSqs from './request-recorder/heal-sqs.js';
 import applyModifiers from './request-recorder/apply-modifiers.js';
 import requestInjector from './request-recorder/request-injector.js';
-
+import updateAndRestoreModifiers from './request-recorder/update-and-restore-modifiers.js';
+import healResponseHeaders from './request-recorder/heal-response-headers.js';
 import {
   buildKey,
   tryParseJson,
   nullAsString,
   rewriteHeaders
 } from './request-recorder/util.js';
-import updateAndRestoreModifiers from './request-recorder/update-and-restore-modifiers.js';
 
 const nockBack = nock.back;
 const nockRecorder = nock.recorder;
@@ -231,6 +231,16 @@ export default (opts) => {
                   ...rewriteHeaders(pendingMocks[idx].record.reqheaders)
                 };
                 pendingMocks[idx].record.reqheaders = rewriteHeaders(reqheaders, overwriteHeaders);
+
+                // heal response headers
+                const newHeaders = healResponseHeaders(interceptor);
+                if (newHeaders.length === 0) {
+                  delete pendingMocks[idx].record.rawHeaders;
+                } else {
+                  pendingMocks[idx].record.rawHeaders = newHeaders;
+                }
+                // eslint-disable-next-line no-param-reassign
+                interceptor.rawHeaders = newHeaders; // ensure response handled correctly
               }
 
               if (anyFlagPresent(['magic', 'response'])) {

package/package.json

@@ -1,7 +1,7 @@
 {
   "name": "node-tdd",
   "type": "module",
-  "version": "6.0.3",
+  "version": "6.1.0",
   "description": "Drop in extension for mocha to abstract commonly used test setups",
   "main": "lib/index.js",
   "scripts": {