fix(core): fixes Collection.reference

rafamel · Nov 24, 2019 · 16d7690 · 16d7690
1 parent abc1dea
commit 16d7690
Showing 1 changed file with 9 additions and 4 deletions.
diff --git a/packages/core/src/classes/Collection/helpers/reference.ts b/packages/core/src/classes/Collection/helpers/reference.ts
@@ -33,12 +33,17 @@ export function reference(
   const isArray = Array.isArray(names);
   const arr = (isArray ? names : [names]) as string[];
 
-  for (const mode of modes) {
-    for (const name of arr) {
-      if (!containsKey(collection[mode], name)) {
-        throw Error(`Can't reference "${name}" on collection`);
+  for (const name of arr) {
+    let found = false;
+    for (const mode of modes) {
+      if (containsKey(collection[mode], name)) {
+        found = true;
+        break;
       }
     }
+    if (!found) {
+      throw Error(`Can't reference "${name}" on collection`);
+    }
   }
 
   return isArray ? arr : arr[0];