getDriver for Buttons too

baloise · Jun 13, 2024 · 3fb5789 · 3fb5789
1 parent 96c822c
commit 3fb5789
Showing 1 changed file with 10 additions and 0 deletions.
diff --git a/taf-browser/src/main/java/com/baloise/testautomation/taf/browser/elements/BrButton.java b/taf-browser/src/main/java/com/baloise/testautomation/taf/browser/elements/BrButton.java
@@ -1,9 +1,11 @@
 package com.baloise.testautomation.taf.browser.elements;
 
+import org.openqa.selenium.WebDriver;
 import org.openqa.selenium.WebElement;
 
 import com.baloise.testautomation.taf.base._base.AButton;
 import com.baloise.testautomation.taf.base._base.TafError;
+import com.baloise.testautomation.taf.common.interfaces.IFinder;
 
 public class BrButton extends AButton {
 
@@ -17,4 +19,12 @@ public WebElement find() {
         return (WebElement) brFind();
     }
 
+    public WebDriver getDriver() {
+      IFinder<?> finder = getFinder();
+      if (finder instanceof BrFinder) {
+        return ((BrFinder)finder).getDriver();
+      }
+      return null;
+    }
+
 }