storeOneResults fix

Author: bhunjadi

lib/query/lib/prepareForDelivery.js

@@ -177,7 +177,21 @@ export function storeOneResults(node, sameLevelResults) {
                 return;
             }
 
-            storeOneResults(collectionNode, dot.pick(collectionNode.linkName, result));
+            const [root, ...rest] = collectionNode.linkName.split('.');
+            if (rest.length === 0) {
+                storeOneResults(collectionNode, result[root]);
+            }
+            else {
+                const rootValue = result[root];
+                if (_.isArray(rootValue)) {
+                    rootValue.forEach(value => {
+                        storeOneResults(collectionNode, dot.pick(rest.join('.'), value));
+                    });
+                }
+                else if (_.isObject(rootValue)) {
+                    storeOneResults(collectionNode, dot.pick(rest.join('.'), rootValue));
+                }
+            }
         });
 
         if (collectionNode.isOneResult) {