fix(element): Fix typing of ElementFinder.then

angular · Dec 16, 2016 · 6a59293 · 6a59293
1 parent 76cb4b4
commit 6a59293
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/lib/element.ts b/lib/element.ts
@@ -792,7 +792,9 @@ export class ElementArrayFinder extends WebdriverWebElement {
 export class ElementFinder extends WebdriverWebElement {
   parentElementArrayFinder: ElementArrayFinder;
   elementArrayFinder_: ElementArrayFinder;
-  then: (fn: Function, errorFn?: Function) => wdpromise.Promise<any> = null;
+  then?:
+      (fn: (value: any) => any | wdpromise.IThenable<any>,
+       errorFn?: (error: any) => any) => wdpromise.Promise<any> = null;
 
   constructor(public browser_: ProtractorBrowser, elementArrayFinder: ElementArrayFinder) {
     super();
@@ -806,7 +808,7 @@ export class ElementFinder extends WebdriverWebElement {
     if (this.parentElementArrayFinder.actionResults_) {
       // Access the underlying actionResult of ElementFinder.
       this.then =
-          (fn: (value: any) => {} | wdpromise.IThenable<{}>, errorFn?: (error: any) => any) => {
+          (fn: (value: any) => any | wdpromise.IThenable<any>, errorFn?: (error: any) => any) => {
             return this.elementArrayFinder_.then((actionResults: any) => {
               if (!fn) {
                 return actionResults[0];