angular-protractor: Additional signature for protractor.ElementArrayF…

…inder.map
snipervld · Jan 27, 2016 · 17fa1e5 · 17fa1e5
1 parent 7ce80da
commit 17fa1e5
Show file tree

Hide file tree

Showing 2 changed files with 12 additions and 1 deletion.
diff --git a/angular-protractor/angular-protractor-tests.ts b/angular-protractor/angular-protractor-tests.ts
@@ -406,9 +406,19 @@ function TestElementArrayFinder() {
     elementArrayFinder.each(function(element: protractor.ElementFinder){
         // nothing
     });
+
     stringPromise = elementArrayFinder.map(function(element: protractor.ElementFinder, index: number){
         return 'abc';
-    })
+    });
+
+    stringPromise = elementArrayFinder.map<string>(function(element: protractor.ElementFinder, index: number): string {
+        return 'abc';
+    });
+
+    stringPromise = elementArrayFinder.map<string, webdriver.promise.Promise<string>>(function(element: protractor.ElementFinder, index: number): webdriver.promise.Promise<string> {
+        return element.getText();
+    });
+
     elementArrayFinder = elementArrayFinder.filter(function(element: protractor.ElementFinder, index: number){
         return element.getText().then((text: string) => {
             return text === "foo";

diff --git a/angular-protractor/angular-protractor.d.ts b/angular-protractor/angular-protractor.d.ts
@@ -992,6 +992,7 @@ declare module protractor {
         *     of values returned by the map function.
         */
         map<T>(mapFn: (element: ElementFinder, index: number) => T): webdriver.promise.Promise<T[]>;
+        map<T, T2>(mapFn: (element: ElementFinder, index: number) => T2): webdriver.promise.Promise<T[]>;
 
         /**
         * Apply a filter function to each element within the ElementArrayFinder. Returns