Merge pull request #99 from pangara/MEM-527

MEM-527 check for main document while removing collections
bcgov · Sep 25, 2017 · 4a11253 · 4a11253
2 parents f424a89 + b65f38c
commit 4a11253
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/modules/documents/client/directives/documents.manager.directive.js b/modules/documents/client/directives/documents.manager.directive.js
@@ -798,11 +798,15 @@ angular.module('documents')
 						var removed = _.filter(original, function(o) {
 							return !_.find(collections, function(c) { return o._id === c._id; });
 						});
-
+						// Updating collections:
+						// Addition - only other documents
+						// Removal - check main and other documents
 						promises = _.union(_.map(added, function(c) {
 							return CollectionModel.addOtherDocument(c._id, documents._id);
 						}), _.map(removed, function(c) {
 							return CollectionModel.removeOtherDocument(c._id, documents._id);
+						}), _.map(removed, function(c) {
+							return CollectionModel.removeMainDocument(c._id, documents._id);
 						}));
 
 						return Promise.all(promises).then(function() {