From bb3812fbfb1709e309c4afc59986b24540aba2c2 Mon Sep 17 00:00:00 2001
From: Joshua Chaitin-Pollak <jbcpollak@users.noreply.github.com>
Date: Thu, 6 Apr 2017 12:02:34 -0400
Subject: [PATCH] Allow `after save` hook to see count of records changed
 (#1231)

* `after save` hook allows count of records changed

* Fix PR linter
---
 lib/dao.js                      |  4 ++++
 test/persistence-hooks.suite.js | 12 ++++++++++--
 2 files changed, 14 insertions(+), 2 deletions(-)

diff --git a/lib/dao.js b/lib/dao.js
index 00b54955..f542a46d 100644
--- a/lib/dao.js
+++ b/lib/dao.js
@@ -2267,6 +2267,7 @@ DataAccessObject.destroyAll = function destroyAll(where, options, cb) {
         where: where,
         hookState: hookState,
         options: options,
+        info: info,
       };
       Model.notifyObserversOf('after delete', context, function(err) {
         cb(err, info);
@@ -2686,6 +2687,7 @@ DataAccessObject.updateAll = function(where, data, options, cb) {
         data: data,
         hookState: hookState,
         options: options,
+        info: info,
       };
       Model.notifyObserversOf('after save', context, function(err, ctx) {
         return cb(err, info);
@@ -2803,6 +2805,7 @@ DataAccessObject.prototype.remove =
               instance: inst,
               hookState: hookState,
               options: options,
+              info: info,
             };
             Model.notifyObserversOf('after delete', context, function(err) {
               cb(err, info);
@@ -2829,6 +2832,7 @@ DataAccessObject.prototype.remove =
                 instance: inst,
                 hookState: hookState,
                 options: options,
+                info: info,
               };
               Model.notifyObserversOf('after delete', context, function(err) {
                 cb(err, info);
diff --git a/test/persistence-hooks.suite.js b/test/persistence-hooks.suite.js
index 6f3a8599..30dbfe08 100644
--- a/test/persistence-hooks.suite.js
+++ b/test/persistence-hooks.suite.js
@@ -2748,18 +2748,22 @@ module.exports = function(dataSource, should, connectorCapabilities) {
 
         TestModel.deleteAll(function(err) {
           if (err) return done(err);
-          ctxRecorder.records.should.eql(aCtxForModel(TestModel, {where: {}}));
+          ctxRecorder.records.should.eql(aCtxForModel(TestModel, {
+            where: {},
+            info: {count: 2},
+          }));
           done();
         });
       });
 
-      it('triggers `after delete` hook without query', function(done) {
+      it('triggers `after delete` hook with query', function(done) {
         TestModel.observe('after delete', ctxRecorder.recordAndNext());
 
         TestModel.deleteAll({name: existingInstance.name}, function(err) {
           if (err) return done(err);
           ctxRecorder.records.should.eql(aCtxForModel(TestModel, {
             where: {name: existingInstance.name},
+            info: {count: 1},
           }));
           done();
         });
@@ -2855,6 +2859,7 @@ module.exports = function(dataSource, should, connectorCapabilities) {
           ctxRecorder.records.should.eql(aCtxForModel(TestModel, {
             where: {id: existingInstance.id},
             instance: existingInstance,
+            info: {count: 1},
           }));
           done();
         });
@@ -2867,6 +2872,7 @@ module.exports = function(dataSource, should, connectorCapabilities) {
           if (err) return done(err);
           ctxRecorder.records.should.eql(aCtxForModel(TestModel, {
             where: {name: existingInstance.name},
+            info: {count: 1},
           }));
           done();
         });
@@ -2900,6 +2906,7 @@ module.exports = function(dataSource, should, connectorCapabilities) {
             }),
             aCtxForModel(TestModel, {
               hookState: {foo: 'BAR'},
+              info: {count: 1},
               where: {id: '1'},
               instance: existingInstance,
             }),
@@ -3056,6 +3063,7 @@ module.exports = function(dataSource, should, connectorCapabilities) {
             ctxRecorder.records.should.eql(aCtxForModel(TestModel, {
               where: {id: existingInstance.id},
               data: {name: 'updated name'},
+              info: {count: 1},
             }));
             done();
           });