fix(populate): handle using count option with document arrays

Re: Automattic#7573
Fonger · Aug 3, 2019 · dc2e87c · dc2e87c
1 parent 4ebbcf2
commit dc2e87c
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/lib/helpers/populate/assignVals.js b/lib/helpers/populate/assignVals.js
@@ -118,6 +118,11 @@ module.exports = function assignVals(o) {
 
 function numDocs(v) {
   if (Array.isArray(v)) {
+    // If setting underneath an array of populated subdocs, we may have an
+    // array of arrays. See gh-7573
+    if (v.some(el => Array.isArray(el))) {
+      return v.map(el => numDocs(el));
+    }
     return v.length;
   }
   return v == null ? 0 : 1;