Merge 46a0f8b into 270511e

faboulaws · Aug 29, 2019 · 0702cde · 0702cde
2 parents 270511e + 46a0f8b
commit 0702cde
Show file tree

Hide file tree

Showing 4 changed files with 110 additions and 4 deletions.
diff --git a/README.md b/README.md
@@ -131,7 +131,7 @@ const mock = blogFactory.generate({
 }
  ```
 
- # Define options for nested properties
+# Define options for nested properties
 
  To define options for nested a property use the nested property path (property names sperated by a dot).
  Example: 

diff --git a/lib/mocker.js b/lib/mocker.js
@@ -88,9 +88,29 @@ const generators = {
   },
 };
 
+function shouldSkip(path, options) {
+  const fieldOptions = options[path] || {};
+  if (fieldOptions.skip) {
+    return true;
+  }
+  const pathProps = path.split('.');
+  if (pathProps.length > 1) {
+    let parentPath = '';
+    // eslint-disable-next-line no-restricted-syntax
+    for (const pathProp of pathProps) {
+      parentPath = parentPath.length === 0 ? pathProp : `${parentPath}.${pathProp}`;
+      if (get(options, `${parentPath}.skip`, false) || get(options, `['${parentPath}'].skip`, false)) {
+        return true;
+      }
+    }
+  }
+
+  return false;
+}
+
 function populateField(mockObject, schema, path, options, staticValue) {
   const fieldOptions = options[path] || {};
-  if (!fieldOptions.skip) {
+  if (!shouldSkip(path, options)) {
     let value;
     const pathDef = schema.path(path);
 

diff --git a/package.json b/package.json
@@ -57,6 +57,7 @@
     "eslint": "^4.18.1",
     "eslint-config-airbnb-base": "^12.1.0",
     "eslint-plugin-import": "^2.9.0",
+    "flat": "^4.1.0",
     "husky": "^3.0.3",
     "mocha": "^6.2.0",
     "mocha-lcov-reporter": "^1.3.0",

diff --git a/test/mocker.test.js b/test/mocker.test.js
@@ -1,6 +1,7 @@
 const mongoose = require('mongoose');
 const { expect } = require('chai');
 const get = require('lodash.get');
+const { unflatten } = require('flat');
 const mocker = require('../');
 
 const { Schema } = mongoose;
@@ -228,7 +229,7 @@ describe('mocker test', () => {
         });
     });
 
-    describe('value overrides via options', () => {
+    describe('options > value', () => {
         const stringShema = new Schema({ firstName: String, username: String, lastName: String });
         const StringThing = mongoose.model('StrThing', stringShema);
 
@@ -256,6 +257,15 @@ describe('mocker test', () => {
             expect(mock.firstName).to.eql('John');
             expect(mock.username).to.eql('John.Doe');
         });
+    });
+
+    describe('options > skip', () => {
+        it('value at root', () => {
+            const stringShema = new Schema({ firstName: String, username: String, lastName: String });
+            const thingMocker = mocker(stringShema, { username: { skip: true } });
+            const mock = thingMocker.generate();
+            expect(mock).not.to.have.property('username');
+        });
 
         it('should use function value - nested', () => {
             const theShema = new Schema({ user: { info: { firstName: String, username: String, lastName: String } } });
@@ -271,7 +281,8 @@ describe('mocker test', () => {
         });
 
         it('skip value - root', () => {
-            const thingMocker = mocker(StringThing, { username: { skip: true } });
+            const theSchema = new Schema({ firstName: String, lastName: String, username: String });
+            const thingMocker = mocker(theSchema, { username: { skip: true } });
             const mock = thingMocker.generate();
             expect(mock).not.to.have.property('username');
         });
@@ -286,6 +297,80 @@ describe('mocker test', () => {
             expect(mock.user).to.have.property('lastName');
             expect(mock.user).not.to.have.property('username');
         });
+
+        it('values from root level', () => {
+            const sschema = new Schema({ root: { name: String, nickname: String }, other: { name: String } });
+            const thingMocker = mocker(sschema, { root: { skip: true } });
+            const mock = thingMocker.generate();
+            expect(mock).not.to.have.property('root');
+            expect(mock).to.have.property('other');
+        });
+
+        describe('value leaf level', () => {
+            const directOptions = { "root.name": { skip: true } };
+            it('nested path options', () => {
+                const sschema = new Schema({ root: { name: String, nickname: String }, other: { name: String } });
+                const options = unflatten(directOptions);
+                const thingMocker = mocker(sschema, options);
+                const mock = thingMocker.generate();
+                expect(mock).to.have.property('root');
+                expect(mock.root).to.have.property('nickname');
+                expect(mock.root).not.to.have.property('name');
+            });
+
+            it('direct path options', () => {
+                const sschema = new Schema({ root: { name: String, nickname: String }, other: { name: String } });
+                const options = directOptions;
+                const thingMocker = mocker(sschema, options);
+                const mock = thingMocker.generate();
+                expect(mock).to.have.property('root');
+                expect(mock.root).to.have.property('nickname');
+                expect(mock.root).not.to.have.property('name');
+            });
+        });
+
+        describe('value sub level nested', () => {
+            const sschema = new Schema({
+                root: {
+                    levelOne: {
+                        name: String,
+                        nickname: String
+                    },
+                    level1: {
+                        other: {
+                            name: String
+                        }
+                    }
+                }
+            });
+
+            const directOptions = {
+                "root.level1": { skip: true },
+                "root.levelOne.name": { skip: true },
+            }
+
+            it('nested options', () => {
+                const options = unflatten(directOptions);
+                const thingMocker = mocker(sschema, options);
+                const mock = thingMocker.generate();
+                expect(mock, 'Must have root').to.have.property('root');
+                expect(mock.root, 'Root must have levelOne').to.have.property('levelOne');
+                expect(mock.root, 'Root must not have level1').not.to.have.property('level1');
+                expect(mock.root.levelOne).not.to.have.property('name');
+                expect(mock.root.levelOne).to.have.property('nickname');
+            });
+
+            it('direct options', () => {
+                const options = directOptions;
+                const thingMocker = mocker(sschema, options);
+                const mock = thingMocker.generate();
+                expect(mock, 'Must have root').to.have.property('root');
+                expect(mock.root, 'Root must have levelOne').to.have.property('levelOne');
+                expect(mock.root, 'Root must not have level1').not.to.have.property('level1');
+                expect(mock.root.levelOne).not.to.have.property('name');
+                expect(mock.root.levelOne).to.have.property('nickname');
+            });
+        });
     });
 
     describe('options', () => {