feat(ElementArrayFinder): keep a reference to the original locator

angular · Aug 20, 2014 · 25cf88c · 25cf88c
1 parent 0da1e0c
commit 25cf88c
Show file tree

Hide file tree

Showing 2 changed files with 16 additions and 0 deletions.
diff --git a/lib/protractor.js b/lib/protractor.js
@@ -98,6 +98,13 @@ var buildElementHelper = function(ptor) {
     this.parentElementFinder_ = opt_parentElementFinder || null; 
   };
 
+  /**
+   * @return {webdriver.Locator}
+   */
+  ElementArrayFinder.prototype.locator = function() {
+    return this.locator_;
+  };
+
   /**
    * Returns the array of WebElements represented by this ElementArrayFinder. 
    *

diff --git a/spec/basic/elements_spec.js b/spec/basic/elements_spec.js
@@ -172,6 +172,15 @@ describe('ElementFinder', function() {
     });
   });
 
+  it('should keep a reference to the array original locator', function() {
+    var byCss = by.css('.menu li a');
+    var byModel = by.model('color');
+    browser.get('index.html#/form');
+
+    expect(element.all(byCss).locator()).toEqual(byCss);
+    expect(element.all(byModel).locator()).toEqual(byModel);
+  });
+
   it('should map each element on array and with promises', function() {
     browser.get('index.html#/form');
     var labels = element.all(by.css('.menu li a')).map(function(elm, index) {