From 55d116ec3f0715ed59a678058acdd71a549d9a2d Mon Sep 17 00:00:00 2001
From: Fabien Franzen <info@atelierfabien.be>
Date: Tue, 14 Oct 2014 22:47:59 +0200
Subject: [PATCH] Enable include scope for belongsTo

---
 lib/include.js             |  3 +-
 lib/relation-definition.js | 10 +++++
 test/include.test.js       | 76 +++++++++++++++++++++++++-------------
 3 files changed, 62 insertions(+), 27 deletions(-)

diff --git a/lib/include.js b/lib/include.js
index eddf7a39..5e62523d 100644
--- a/lib/include.js
+++ b/lib/include.js
@@ -167,7 +167,7 @@ Inclusion.include = function (objects, include, cb) {
       
       var related; // relation accessor function
       
-      if (relation.multiple && scope) {
+      if ((relation.multiple || relation.type === 'belongsTo') && scope) {
         var includeScope = {};
         var filter = scope.conditions();
         
@@ -183,6 +183,7 @@ Inclusion.include = function (objects, include, cb) {
         }
         
         utils.mergeQuery(filter, includeScope, {fields: false});
+        
         related = inst[relationName].bind(inst, filter);
       } else {
         related = inst[relationName].bind(inst);
diff --git a/lib/relation-definition.js b/lib/relation-definition.js
index 1fa299e4..94ec6a4c 100644
--- a/lib/relation-definition.js
+++ b/lib/relation-definition.js
@@ -1239,10 +1239,14 @@ BelongsTo.prototype.related = function (refresh, params) {
   var fk = this.definition.keyFrom;
   var modelInstance = this.modelInstance;
   var discriminator;
+  var scopeQuery = null;
   
   if (arguments.length === 1) {
     params = refresh;
     refresh = false;
+  } else if (arguments.length === 2
+    && typeof refresh === 'object' && typeof params === 'function') {
+    scopeQuery = refresh;
   } else if (arguments.length > 2) {
     throw new Error('Method can\'t be called with more than two arguments');
   }
@@ -1292,6 +1296,12 @@ BelongsTo.prototype.related = function (refresh, params) {
     
       this.definition.applyScope(modelInstance, query);
       
+      if (scopeQuery) mergeQuery(query, scopeQuery);
+      
+      if (Array.isArray(query.fields) && query.fields.indexOf(pk) === -1) {
+          query.fields.push(pk); // always include the pk
+      }
+      
       modelTo.findOne(query, function (err, inst) {
         if (err) {
           return cb(err);
diff --git a/test/include.test.js b/test/include.test.js
index 1773eb2b..7eced3ae 100644
--- a/test/include.test.js
+++ b/test/include.test.js
@@ -156,7 +156,24 @@ describe('include', function () {
     });
   });
   
-  it('should fetch Users with include scope on Posts', function (done) {
+  it('should fetch Users with include scope on Posts - belongsTo', function (done) {
+      Post.find({
+        include: { relation: 'author', scope:{ fields: ['name'] }}
+      }, function (err, posts) {
+        should.not.exist(err);
+        should.exist(posts);
+        posts.length.should.equal(5);
+        
+        var author = posts[0].author();
+        author.name.should.equal('User A');
+        author.should.have.property('id');
+        author.should.not.have.property('age');
+        
+        done();
+      });
+    });
+  
+  it('should fetch Users with include scope on Posts - hasMany', function (done) {
     User.find({
       include: {relation: 'posts', scope:{
         order: 'title DESC'
@@ -186,7 +203,7 @@ describe('include', function () {
     });
   });
   
-  it('should fetch Users with include scope on Passports', function (done) {
+  it('should fetch Users with include scope on Passports - hasMany', function (done) {
     User.find({
       include: {relation: 'passports', scope:{
         where: { number: '2' }
@@ -253,37 +270,44 @@ describe('include', function () {
   });
 
   it('should support hasAndBelongsToMany', function (done) {
-
-    Assembly.destroyAll(function(err) {
-      Part.destroyAll(function(err) {
-        Assembly.relations.parts.modelThrough.destroyAll(function(err) {
-        Assembly.create({name: 'car'}, function (err, assembly) {
-          Part.create({partNumber: 'engine'}, function (err, part) {
-            assembly.parts.add(part, function (err, data) {
-              assembly.parts(function (err, parts) {
-                should.not.exist(err);
-                should.exists(parts);
-                parts.length.should.equal(1);
-                parts[0].partNumber.should.equal('engine');
-
-                // Create a part
-                assembly.parts.create({partNumber: 'door'}, function (err, part4) {
-
-                  Assembly.find({include: 'parts'}, function (err, assemblies) {
-                    assemblies.length.should.equal(1);
-                    assemblies[0].parts.length.should.equal(2);
-                    done();
-                  });
-
-                });
+    Assembly.create({name: 'car'}, function (err, assembly) {
+      Part.create({partNumber: 'engine'}, function (err, part) {
+        assembly.parts.add(part, function (err, data) {
+          assembly.parts(function (err, parts) {
+            should.not.exist(err);
+            should.exists(parts);
+            parts.length.should.equal(1);
+            parts[0].partNumber.should.equal('engine');
+            
+            // Create a part
+            assembly.parts.create({partNumber: 'door'}, function (err, part4) {
+              
+              Assembly.find({include: 'parts'}, function (err, assemblies) {
+                assemblies.length.should.equal(1);
+                assemblies[0].parts().length.should.equal(2);
+                done();
               });
+              
             });
           });
         });
-        });
       });
     });
   });
+  
+  // Not implemented correctly, see: loopback-datasource-juggler/issues/166
+  //
+  // it('should support include scope on hasAndBelongsToMany', function (done) {
+  //   Assembly.find({include: { relation: 'parts', scope: {
+  //     where: { partNumber: 'engine' }
+  //   }}}, function (err, assemblies) {
+  //     assemblies.length.should.equal(1);
+  //     var parts = assemblies[0].parts();
+  //     parts.should.have.length(1);
+  //     parts[0].partNumber.should.equal('engine');
+  //     done();
+  //   });
+  // });
 
 });