test: add repro for #10584

Automattic · Aug 24, 2021 · 0921b59 · 0921b59
1 parent bf2c102
commit 0921b59
Showing 1 changed file with 37 additions and 0 deletions.
diff --git a/test/types.map.test.js b/test/types.map.test.js
@@ -1035,4 +1035,41 @@ describe('Map', function() {
       assert.deepEqual(res.toObject().budgeted.get('2020'), [100, 200, 10]);
     });
   });
+
+  it('can populate map of subdocs with doc array using ref function (gh-10584)', async function() {
+    const Person = db.model('Person', Schema({ name: String }));
+    const Book = db.model('Book', Schema({ title: String }));
+
+    const schema = new Schema({
+      myMap: {
+        type: Map,
+        of: {
+          modelId: String,
+          data: [{
+            _id: {
+              type: mongoose.ObjectId,
+              ref: doc => doc.$parent().modelId
+            }
+          }]
+        }
+      }
+    });
+    const Test = db.model('Test', schema);
+
+    const people = await Person.create({ name: 'John' });
+    const book = await Book.create({ title: 'Intro to CS' });
+
+    await Test.create({
+      myMap: {
+        key1: { modelId: 'Person', data: [{ _id: people._id }] },
+        key2: { modelId: 'Book', data: [{ _id: book._id }] }
+      }
+    });
+
+    const res = await Test.findOne().populate('myMap.$*.data._id');
+    assert.equal(res.myMap.get('key1').data.length, 1);
+    assert.equal(res.myMap.get('key1').data[0]._id.name, 'John');
+    assert.equal(res.myMap.get('key2').data.length, 1);
+    assert.equal(res.myMap.get('key2').data[0]._id.title, 'Intro to CS');
+  });
 });