Add test for nested private paths

divyeshdhokiya · Dec 7, 2020 · a243caa · a243caa
1 parent 6cc324c
commit a243caa
Show file tree

Hide file tree

Showing 2 changed files with 19 additions and 1 deletion.
diff --git a/src/models/plugins/toJSON.plugin.js b/src/models/plugins/toJSON.plugin.js
@@ -6,7 +6,6 @@
  *  - replaces _id with id
  */
 
-
 const deleteAtPath = (obj, path, index) => {
   if (index === path.length - 1) {
     delete obj[path[index]];

diff --git a/tests/unit/models/plugins/toJSON.plugin.test.js b/tests/unit/models/plugins/toJSON.plugin.test.js
@@ -46,6 +46,25 @@ describe('toJSON plugin', () => {
     expect(doc.toJSON()).toHaveProperty('public');
   });
 
+  it('should remove any nested paths set as private', () => {
+    const schema = mongoose.Schema({
+      public: { type: String },
+      nested: {
+        private: { type: String, private: true },
+      },
+    });
+    schema.plugin(toJSON);
+    const Model = connection.model('Model', schema);
+    const doc = new Model({
+      public: 'some public value',
+      nested: {
+        private: 'some nested private value',
+      },
+    });
+    expect(doc.toJSON()).not.toHaveProperty('nested.private');
+    expect(doc.toJSON()).toHaveProperty('public');
+  });
+
   it('should also call the schema toJSON transform function', () => {
     const schema = mongoose.Schema(
       {