issue js-data#118: Allow multiple relationships to same model

Author: chieffancypants

karma.start.js

@@ -2,7 +2,7 @@
 var $rootScope, $q, $log, $timeout, DSHttpAdapterProvider, DSProvider, DSLocalStorageAdapter, DS, DSUtils, DSHttpAdapter, app, $httpBackend, p1, p2, p3, p4, p5;
 
 var user1, organization2, comment3, profile4;
-var comment11, comment12, comment13, organization14, profile15, user10, user16, user17, user18, organization15, user20, comment19, user22, profile21;
+var comment11, comment12, comment13, organization14, profile15, user10, user16, user17, user18, organization15, user19, user20, comment19, user22, profile21;
 
 var lifecycle = {};
 
@@ -183,12 +183,20 @@ function startInjector() {
     DS.defineResource({
       name: 'comment',
       relations: {
-        belongsTo: {
-          user: {
-            localField: 'user',
-            localKey: 'userId'
+        belongsTo: [
+          {
+            user: {
+              localField: 'user',
+              localKey: 'userId'
+            }
+          },
+          {
+            user: {
+              localField: 'approvedByUser',
+              localKey: 'approvedBy'
+            }
           }
-        }
+        ]
       }
     });
     $log = _$log_;
@@ -294,13 +302,19 @@ function startInjector() {
         user18
       ]
     };
+    user19 = {
+      id: 19,
+      name: 'test user 19'
+    };
     user20 = {
       id: 20,
       name: 'test user 20'
     };
     comment19 = {
       content: 'test comment 19',
       id: 19,
+      approvedBy: 19,
+      approvedByUser: user19,
       userId: 20,
       user: user20
     };

src/datastore/async_methods/loadRelations.js

@@ -87,30 +87,35 @@ function loadRelations(resourceName, instance, relations, options) {
     var tasks = [];
     var fields = [];
 
-    DS.utils.forOwn(definition.relations, function (relation, type) {
-      DS.utils.forOwn(relation, function (def, relationName) {
-        if (DS.utils.contains(relations, relationName)) {
-          var task;
-          var params = {};
-          params[def.foreignKey] = instance[definition.idAttribute];
+    DS.utils.forOwn(definition.relations, function (definedRelations, type) {
+      if (!(definedRelations instanceof Array)) {
+        definedRelations = [definedRelations];
+      }
+      definedRelations.forEach(function (definedRelation) {
+        DS.utils.forOwn(definedRelation, function (def, relationName) {
+          if (DS.utils.contains(relations, relationName)) {
+            var task;
+            var params = {};
+            params[def.foreignKey] = instance[definition.idAttribute];
 
-          if (type === 'hasMany') {
-            task = DS.findAll(relationName, params, options);
-          } else if (type === 'hasOne') {
-            if (def.localKey && instance[def.localKey]) {
-              task = DS.find(relationName, instance[def.localKey], options);
-            } else if (def.foreignKey) {
+            if (type === 'hasMany') {
               task = DS.findAll(relationName, params, options);
+            } else if (type === 'hasOne') {
+              if (def.localKey && instance[def.localKey]) {
+                task = DS.find(relationName, instance[def.localKey], options);
+              } else if (def.foreignKey) {
+                task = DS.findAll(relationName, params, options);
+              }
+            } else {
+              task = DS.find(relationName, instance[def.localKey], options);
             }
-          } else {
-            task = DS.find(relationName, instance[def.localKey], options);
-          }
 
-          if (task) {
-            tasks.push(task);
-            fields.push(def.localField);
+            if (task) {
+              tasks.push(task);
+              fields.push(def.localField);
+            }
           }
-        }
+        });
       });
     });
 

src/datastore/sync_methods/inject.js

@@ -115,15 +115,20 @@ function _inject(definition, resource, attrs) {
 
 function _injectRelations(definition, injected) {
   var DS = this;
-  DS.utils.forOwn(definition.relations, function (relation, type) {
-    DS.utils.forOwn(relation, function (def, relationName) {
-      if (DS.definitions[relationName] && injected[def.localField]) {
-        try {
-          injected[def.localField] = DS.inject(relationName, injected[def.localField]);
-        } catch (err) {
-          DS.$log.error(errorPrefix(definition.name) + 'Failed to inject ' + type + ' relation: "' + relationName + '"!', err);
+  DS.utils.forOwn(definition.relations, function (definedRelations, type) {
+    if (!(definedRelations instanceof Array)) {
+      definedRelations = [definedRelations];
+    }
+    definedRelations.forEach(function (definedRelation) {
+      DS.utils.forOwn(definedRelation, function (def, relationName) {
+        if (DS.definitions[relationName] && injected[def.localField]) {
+          try {
+            injected[def.localField] = DS.inject(relationName, injected[def.localField]);
+          } catch (err) {
+            DS.$log.error(errorPrefix(definition.name) + 'Failed to inject ' + type + ' relation: "' + relationName + '"!', err);
+          }
         }
-      }
+      });
     });
   });
 }

test/integration/datastore/sync_methods/inject.test.js

@@ -185,6 +185,7 @@ describe('DS.inject(resourceName, attrs)', function () {
 
     // comment19 relations
     assert.deepEqual(DS.get('user', 20), user20);
+    assert.deepEqual(DS.get('user', 19), user19);
 
     // profile21 relations
     assert.deepEqual(DS.get('user', 22), user22);