feat(protractor): add removeMockModule method

angular · Feb 15, 2014 · 642de06 · 642de06
1 parent 71c9a07
commit 642de06
Show file tree

Hide file tree

Showing 2 changed files with 22 additions and 0 deletions.
diff --git a/lib/protractor.js b/lib/protractor.js
@@ -639,6 +639,16 @@ Protractor.prototype.clearMockModules = function() {
   this.moduleScripts_ = [];
 };
 
+/**
+ * Remove a registered mock module.
+ * @param {!string} name The name of the module to remove.
+ */
+Protractor.prototype.removeMockModule = function (name) {
+  var index = this.moduleNames_.indexOf(name);
+  this.moduleNames_.splice(index, 1);
+  this.moduleScripts_.splice(index, 1);
+};
+
 /**
  * See webdriver.WebDriver.get
  *

diff --git a/spec/basic/mockmodule_spec.js b/spec/basic/mockmodule_spec.js
@@ -36,4 +36,16 @@ describe('mock modules', function() {
 
     expect(element(by.css('[app-version]')).getText()).toEqual('3');
   });
+
+  it('should have the version of the module A after deleting module B', function() {
+    browser.addMockModule('moduleA', mockModuleA);
+    browser.addMockModule('moduleB', mockModuleB);
+
+    browser.removeMockModule('moduleB');
+
+    browser.get('index.html');
+
+    expect(element(by.css('[app-version]')).getText()).toEqual('2');
+  });
+
 });