feat(element): element.all now has 'first' and 'last' methods

Return the first or last element that matches a locator. Closes #171.

Author: juliemr

lib/protractor.js

@@ -109,6 +109,23 @@ var buildElementHelper = function(ptor) {
       return new webdriver.WebElement(ptor.driver, id);
     };
 
+    elementArrayFinder.first = function() {
+      var id = ptor.findElements(locator).then(function(arr) {
+        if (!arr.length) {
+          throw new Error('No element found using locator: ' + locator.message);
+        }
+        return arr[0];
+      });
+      return new webdriver.WebElement(ptor.driver, id);
+    };
+
+    elementArrayFinder.last = function() {
+      var id = ptor.findElements(locator).then(function(arr) {
+        return arr[arr.length - 1];
+      });
+      return new webdriver.WebElement(ptor.driver, id);
+    };
+
     elementArrayFinder.then = function(fn) {
       return ptor.findElements(locator).then(fn);
     };

spec/basic/findelements_spec.js

@@ -328,6 +328,20 @@ describe('global element function', function() {
     expect(colorList.get(2).getAttribute('value')).toEqual('red');
   });
 
+  it('should get the first element from an array', function() {
+    var colorList = element.all(by.model('color'));
+    browser.get('index.html#/form');
+
+    expect(colorList.first(0).getAttribute('value')).toEqual('blue');
+  });
+
+  it('should get the last element from an array', function() {
+    var colorList = element.all(by.model('color'));
+    browser.get('index.html#/form');
+
+    expect(colorList.last(0).getAttribute('value')).toEqual('red');
+  });
+
   it('should export an isPresent helper', function() {
     expect(element(by.binding('greet')).isPresent()).toBe(true);
     expect(element(by.binding('nopenopenope')).isPresent()).toBe(false);