fix(endpoint:controller): wrap saveUpdates in an if(entity) statement

fixes #2052, closes #2059
angular-fullstack · Sep 13, 2016 · 31c0a79 · 31c0a79
1 parent 1fbf24b
commit 31c0a79
Showing 1 changed file with 9 additions and 7 deletions.
diff --git a/templates/endpoint/basename.controller.js b/templates/endpoint/basename.controller.js
@@ -25,13 +25,15 @@ function respondWithResult(res, statusCode) {
 
 function saveUpdates(updates) {
   return function(entity) {
-    <% if (filters.mongooseModels) { %>var updated = _.merge(entity, updates);
-    return updated.save()
-      .then(updated => {<% }
-       if (filters.sequelizeModels) { %>return entity.updateAttributes(updates)
-      .then(updated => {<% } %>
-        return updated;
-      });
+    if(entity) {
+      <% if (filters.mongooseModels) { %>var updated = _.merge(entity, updates);
+      return updated.save()
+        .then(updated => {<% }
+         if (filters.sequelizeModels) { %>return entity.updateAttributes(updates)
+        .then(updated => {<% } %>
+          return updated;
+        });
+    }
   };
 }