From b3dde830833e8f7d26eba2d9959d25d69554f51f Mon Sep 17 00:00:00 2001
From: Juan Cruz Viotti <jviottidc@gmail.com>
Date: Tue, 19 Jan 2016 09:42:57 -0400
Subject: [PATCH] Implement removeDrive and removeImage in selection service

---
 build/browser/app.js                          | 24 +++++++++++++++++++
 lib/browser/modules/selection-state.js        | 24 +++++++++++++++++++
 tests/browser/modules/selection-state.spec.js | 21 ++++++++++++++++
 3 files changed, 69 insertions(+)

diff --git a/build/browser/app.js b/build/browser/app.js
index 806df0db..cd95ccf5 100644
--- a/build/browser/app.js
+++ b/build/browser/app.js
@@ -669,6 +669,30 @@ selectionState.service('SelectionStateService', function() {
     return !!self.getImage();
   };
 
+  /**
+   * @summary Remove drive
+   * @function
+   * @public
+   *
+   * @example
+   * SelectionStateService.removeDrive();
+   */
+  this.removeDrive = function() {
+    self.setDrive(undefined);
+  };
+
+  /**
+   * @summary Remove image
+   * @function
+   * @public
+   *
+   * @example
+   * SelectionStateService.removeImage();
+   */
+  this.removeImage = function() {
+    self.setImage(undefined);
+  };
+
   /**
    * @summary Clear all selections
    * @function
diff --git a/lib/browser/modules/selection-state.js b/lib/browser/modules/selection-state.js
index 8b7b1788..567db922 100644
--- a/lib/browser/modules/selection-state.js
+++ b/lib/browser/modules/selection-state.js
@@ -123,6 +123,30 @@ selectionState.service('SelectionStateService', function() {
     return !!self.getImage();
   };
 
+  /**
+   * @summary Remove drive
+   * @function
+   * @public
+   *
+   * @example
+   * SelectionStateService.removeDrive();
+   */
+  this.removeDrive = function() {
+    self.setDrive(undefined);
+  };
+
+  /**
+   * @summary Remove image
+   * @function
+   * @public
+   *
+   * @example
+   * SelectionStateService.removeImage();
+   */
+  this.removeImage = function() {
+    self.setImage(undefined);
+  };
+
   /**
    * @summary Clear all selections
    * @function
diff --git a/tests/browser/modules/selection-state.spec.js b/tests/browser/modules/selection-state.spec.js
index 6f7a2140..f409ad50 100644
--- a/tests/browser/modules/selection-state.spec.js
+++ b/tests/browser/modules/selection-state.spec.js
@@ -78,6 +78,17 @@ describe('Browser: SelectionState', function() {
 
       });
 
+      describe('.removeDrive()', function() {
+
+        it('should clear the drive', function() {
+          SelectionStateService.removeDrive();
+          var drive = SelectionStateService.getDrive();
+          m.chai.expect(drive).to.be.undefined;
+        });
+
+      });
+
+
     });
 
     describe('given no drive', function() {
@@ -128,6 +139,16 @@ describe('Browser: SelectionState', function() {
 
       });
 
+      describe('.removeImage()', function() {
+
+        it('should clear the image', function() {
+          SelectionStateService.removeImage();
+          var image = SelectionStateService.getImage();
+          m.chai.expect(image).to.be.undefined;
+        });
+
+      });
+
     });
 
     describe('given no image', function() {