fix(model): allow virtual ref function to return arrays

Automattic · Nov 18, 2017 · 754db14 · 754db14
1 parent 4348fa1
commit 754db14
Showing 1 changed file with 9 additions and 1 deletion.
diff --git a/lib/model.js b/lib/model.js
@@ -3580,7 +3580,15 @@ function getModelsMapForPopulate(model, docs, options) {
           if (typeof ref === 'function') {
             ref = ref.call(doc, doc);
           }
-          modelNames = [ref];
+
+          // When referencing nested arrays, the ref should be an Array 
+          // of modelNames.
+          if (Array.isArray(ref)) {
+            modelNames = ref;
+          } else {
+            modelNames = [ref];
+          }
+
           isVirtual = true;
         } else {
           // We may have a discriminator, in which case we don't want to