fix: Fix issue with unwind and empty arrays creating an extra column (#…

…496) * fix: Fix issue with unwind and empty arrays creating an extra column * fix: bring text coverage back to 100%
zemirco · Nov 10, 2020 · 0b331fc · 0b331fc
1 parent b37e242
commit 0b331fc
Show file tree

Hide file tree

Showing 11 changed files with 173 additions and 37 deletions.
diff --git a/lib/transforms/unwind.js b/lib/transforms/unwind.js
@@ -1,6 +1,6 @@
 
 const lodashGet = require('lodash.get');
-const { setProp, flattenReducer } = require('../utils');
+const { setProp, unsetProp, flattenReducer } = require('../utils');
 
 function getUnwindablePaths(obj, currentPath) {
   return Object.keys(obj).reduce((unwindablePaths, key) => {
@@ -42,7 +42,7 @@ function unwind({ paths = undefined, blankOut = false } = {}) {
         }
 
         if (!unwindArray.length) {
-          return setProp(row, unwindPath, undefined);
+          return unsetProp(row, unwindPath);
         }
 
         return unwindArray.map((unwindRow, index) => {

diff --git a/lib/utils.js b/lib/utils.js
@@ -6,11 +6,29 @@ function getProp(obj, path, defaultValue) {
 
 function setProp(obj, path, value) {
   const pathArray = Array.isArray(path) ? path : path.split('.');
-  const key = pathArray[0];
-  const newValue = pathArray.length > 1 ? setProp(obj[key] || {}, pathArray.slice(1), value) : value;
+  const [key, ...restPath] = pathArray;
+  const newValue = pathArray.length > 1 ? setProp(obj[key] || {}, restPath, value) : value;
   return Object.assign({}, obj, { [key]: newValue });
 }
 
+function unsetProp(obj, path) {
+  const pathArray = Array.isArray(path) ? path : path.split('.');
+  const [key, ...restPath] = pathArray;
+
+  if (typeof obj[key] !== 'object') {
+    // This will never be hit in the current code because unwind does the check before calling unsetProp
+    /* istanbul ignore next */
+    return obj;
+  }
+
+  if (pathArray.length === 1) {
+    delete obj[key];
+    return obj;
+  }
+
+  return unsetProp(obj[key], restPath);
+}
+
 function flattenReducer(acc, arr) {
   try {
     // This is faster but susceptible to `RangeError: Maximum call stack size exceeded`
@@ -41,6 +59,7 @@ function fastJoin(arr, separator) {
 module.exports = {
   getProp,
   setProp,
+  unsetProp,
   fastJoin,
   flattenReducer
 };
diff --git a/package-lock.json b/package-lock.json
diff --git a/test/CLI.js b/test/CLI.js
@@ -830,4 +830,16 @@ module.exports = (testRunner, jsonFixtures, csvFixtures) => {
       t.end();
     });
   });
+
+  testRunner.add('should unwind complex objects using the unwind transform', (t) => {
+    const opts = '--fields carModel,price,extras.items.name,extras.items.items.position,extras.items.items.color,extras.items.items,name,color,extras.items.color'
+      + ' --unwind extras.items,extras.items.items --flatten-objects --flatten-arrays';
+
+    exec(`${cli} -i "${getFixturePath('/json/unwindComplexObject.json')}" ${opts}`, (err, stdout, stderr) => {
+      t.notOk(stderr);
+      const csv = stdout;
+      t.equal(csv, csvFixtures.unwindComplexObject);
+      t.end();
+    });
+  });
 };
diff --git a/test/JSON2CSVAsyncParser.js b/test/JSON2CSVAsyncParser.js
@@ -1260,6 +1260,23 @@ module.exports = (testRunner, jsonFixtures, csvFixtures, inMemoryJsonFixtures) =
     t.end();
   });
 
+  testRunner.add('should unwind complex objects using the unwind transform', async (t) => {
+    const opts = {
+      fields: ["carModel", "price", "extras.items.name", "extras.items.items.position", "extras.items.items.color", "extras.items.items", "name", "color", "extras.items.color"],
+      transforms: [unwind({ paths: ['extras.items', 'extras.items.items'] }), flatten()],
+    };
+
+    const parser = new AsyncParser(opts);
+    try {
+      const csv = await parser.fromInput(jsonFixtures.unwindComplexObject()).promise();
+      t.equal(csv, csvFixtures.unwindComplexObject);
+    } catch(err) {
+      t.fail(err.message);
+    }
+    t.end();
+  });
+
+
   testRunner.add('should support custom transforms', async (t) => {
     const opts = {
       transforms: [row => ({

diff --git a/test/JSON2CSVParser.js b/test/JSON2CSVParser.js
@@ -781,6 +781,18 @@ module.exports = (testRunner, jsonFixtures, csvFixtures) => {
     t.end();
   });
 
+  testRunner.add('should unwind complex objects using the unwind transform', (t) => {
+    const opts = {
+      transforms: [unwind({ paths: ['extras.items', 'extras.items.items'] }), flatten()],
+    };
+
+    const parser = new Json2csvParser(opts);
+    const csv = parser.parse(jsonFixtures.unwindComplexObject);
+
+    t.equal(csv, csvFixtures.unwindComplexObject);
+    t.end();
+  });
+
   testRunner.add('should support custom transforms', (t) => {
     const opts = {
       transforms: [row => ({

diff --git a/test/JSON2CSVTransform.js b/test/JSON2CSVTransform.js
@@ -1295,6 +1295,29 @@ module.exports = (testRunner, jsonFixtures, csvFixtures, inMemoryJsonFixtures) =
       });
   });
 
+
+  testRunner.add('should unwind complex objects using the unwind transform', async (t) => {
+    const opts = {
+      fields: ["carModel", "price", "extras.items.name", "extras.items.items.position", "extras.items.items.color", "extras.items.items", "name", "color", "extras.items.color"],
+      transforms: [unwind({ paths: ['extras.items', 'extras.items.items'] }), flatten()],
+    };
+
+    const transform = new Json2csvTransform(opts);
+    const processor = jsonFixtures.unwindComplexObject().pipe(transform);
+
+    let csv = '';
+    processor
+      .on('data', chunk => (csv += chunk.toString()))
+      .on('end', () => {
+        t.equal(csv, csvFixtures.unwindComplexObject);
+        t.end();
+      })
+      .on('error', err => {
+        t.fail(err.message);
+        t.end();  
+      });
+  });
+
   testRunner.add('should support custom transforms', (t) => {
     const opts = {
       transforms: [row => ({

diff --git a/test/fixtures/csv/unwindAndFlatten.csv b/test/fixtures/csv/unwindAndFlatten.csv
@@ -2,4 +2,5 @@
 "BMW",15000,"airbag","white"
 "BMW",15000,"dashboard","black"
 "Porsche",30000,"airbag","gray"
-"Porsche",30000,"dashboard","red"
+"Porsche",30000,"dashboard","red"
+"Mercedes",20000,,
diff --git a/test/fixtures/csv/unwindComplexObject.csv b/test/fixtures/csv/unwindComplexObject.csv
@@ -0,0 +1,6 @@
+"carModel","price","extras.items.name","extras.items.items.position","extras.items.items.color","extras.items.items","name","color","extras.items.color"
+"Porsche",30000,"airbag","left","white",,,,
+"Porsche",30000,"airbag","right","gray",,,,
+"Porsche",30000,"dashboard",,,"none",,,
+,,,,,,"airbag","white",
+"BMW",15000,"dashboard",,,,,,"black"
diff --git a/test/fixtures/json/unwindAndFlatten.json b/test/fixtures/json/unwindAndFlatten.json
@@ -1,28 +1,33 @@
 [
-    {
-        "carModel": "BMW",
-        "price": 15000,
-        "items": [
-            {
-                "name": "airbag",
-                "color": "white"
-            }, {
-                "name": "dashboard",
-                "color": "black"
-            }
-        ]
-    }, {
-        "carModel": "Porsche",
-        "price": 30000,
-        "items": [
-            {
-                "name": "airbag",
-                "color": "gray"
-            },
-            {
-                "name": "dashboard", 
-                "color": "red"
-            }
-        ]
-    }
+  {
+    "carModel": "BMW",
+    "price": 15000,
+    "items": [
+      {
+        "name": "airbag",
+        "color": "white"
+      }, {
+        "name": "dashboard",
+        "color": "black"
+      }
+    ]
+  }, {
+    "carModel": "Porsche",
+    "price": 30000,
+    "items": [
+      {
+        "name": "airbag",
+        "color": "gray"
+      },
+      {
+        "name": "dashboard", 
+        "color": "red"
+      }
+    ]
+  },
+  {
+    "carModel": "Mercedes", 
+    "price": 20000,  
+    "items": []
+  }
 ]
diff --git a/test/fixtures/json/unwindComplexObject.json b/test/fixtures/json/unwindComplexObject.json
@@ -0,0 +1,41 @@
+[
+  {
+    "carModel": "Porsche",
+    "price": 30000,
+    "extras": {
+      "items": [
+        {
+          "name": "airbag",
+          "items": [
+            {
+              "position": "left",
+              "color": "white"
+            }, {
+              "position": "right",
+              "color": "gray"
+            }
+          ]
+        },
+        {
+          "name": "dashboard",
+          "items": "none"
+        }
+      ]
+    }
+  }, {
+    "carModel": "BMW",
+    "price": 15000,
+    "extras": {
+      "items": [
+        {
+          "name": "airbag",
+          "color": "white",
+          "items": []
+        }, {
+          "name": "dashboard",
+          "color": "black"
+        }
+      ]
+    }
+  }
+]