From 806f38113c675a26171776a559a20bf3899aa2cc Mon Sep 17 00:00:00 2001
From: Julie <ju.ralph@gmail.com>
Date: Tue, 10 Sep 2013 10:23:15 -0700
Subject: [PATCH] Fix: findElements() and isElementPresent() now work for
 protractor.By.input. Closes #79.

---
 lib/protractor.js         | 25 +++++++++++++++++++++++++
 spec/findelements_spec.js |  6 ++++++
 2 files changed, 31 insertions(+)

diff --git a/lib/protractor.js b/lib/protractor.js
index 0473f8f4d..b4c935c72 100644
--- a/lib/protractor.js
+++ b/lib/protractor.js
@@ -212,6 +212,27 @@ clientSideScripts.findRepeaterColumn = function() {
  * arguments[0] {Element} The scope of the search.
  * arguments[1] {string} The model name.
  *
+ * @return {Array.<Element?} The matching input elements.
+*/
+clientSideScripts.findInputs = function() {
+  var using = arguments[0] || document;
+  var model = arguments[1];
+  var prefixes = ['ng-', 'ng_', 'data-ng-', 'x-ng-', 'ng\\:'];
+  for (var p = 0; p < prefixes.length; ++p) {
+    var selector = 'input[' + prefixes[p] + 'model="' + model + '"]';
+    var inputs = using.querySelectorAll(selector);
+    if (inputs.length) {
+      return inputs;
+    }
+  }
+};
+
+/**
+ * Find input elements by model name.
+ *
+ * arguments[0] {Element} The scope of the search.
+ * arguments[1] {string} The model name.
+ *
  * @return {Element} The first matching input element.
 */
 clientSideScripts.findInput = function() {
@@ -701,6 +722,10 @@ ProtractorBy.prototype.input = function(model) {
     findOverride: function(driver, using) {
       return driver.findElement(
           webdriver.By.js(clientSideScripts.findInput), using, model);
+    },
+    findArrayOverride: function(driver, using) {
+      return driver.findElements(
+          webdriver.By.js(clientSideScripts.findInputs), using, model);
     }
   };
 };
diff --git a/spec/findelements_spec.js b/spec/findelements_spec.js
index fc624c045..a281ba9b7 100644
--- a/spec/findelements_spec.js
+++ b/spec/findelements_spec.js
@@ -65,6 +65,12 @@ describe('finding elements', function() {
       expect(letterList.getText()).toBe('wxyz');
     });
 
+    it('should find multiple inputs', function() {
+      ptor.findElements(protractor.By.input('color')).then(function(arr) {
+        expect(arr.length).toEqual(3);
+      });
+    });
+
     it('should find a repeater by partial match', function() {
       var fullMatch = ptor.findElement(
           protractor.By.repeater('baz in days | filter:\'T\'').