feat(locators) Add map() function to element.all

Added a map function to element.all to apply a function to each element and return the result of the transformation.

Closes #392

Author: andresdominguez

lib/protractor.js

@@ -139,6 +139,24 @@ var buildElementHelper = function(ptor) {
       });
     };
 
+    elementArrayFinder.map = function(fn) {
+      return ptor.findElements(locator).then(function(arr) {
+        var list = [];
+        arr.forEach(function(webElem) {
+          var mapResult = fn(webElem);
+          // Is the result a promise?
+          if (mapResult.then) {
+            mapResult.then(function(value) {
+              list.push(value);
+            });
+          } else {
+            list.push(mapResult);
+          }
+        });
+        return list;
+      });
+    };
+
     return elementArrayFinder;
   }
 

spec/basic/findelements_spec.js

@@ -414,6 +414,26 @@ describe('global element function', function() {
     });
   });
 
+  it('should map each element on array and with promises', function() {
+    var labels = element.all(by.css('.menu li a')).map(function(elm) {
+      return elm.getText();
+    });
+    browser.get('index.html#/form');
+
+    expect(labels).toEqual(
+        ['repeater', 'bindings', 'form', 'async', 'conflict', 'polling']);
+  });
+
+  it('should map each element from a literal and promise array', function() {
+    var i = 1;
+    var labels = element.all(by.css('.menu li a')).map(function(elm) {
+      return i++;
+    });
+    browser.get('index.html#/form');
+
+    expect(labels).toEqual([1, 2, 3, 4, 5, 6]);
+  });
+
   it('should export an isPresent helper', function() {
     expect(element(by.binding('greet')).isPresent()).toBe(true);
     expect(element(by.binding('nopenopenope')).isPresent()).toBe(false);