From 3fb57898a0b16d074002a81d903faaa0aac78b56 Mon Sep 17 00:00:00 2001
From: Christian Josephy <christian.josephy@baloise.ch>
Date: Thu, 13 Jun 2024 17:13:40 +0200
Subject: [PATCH] getDriver for Buttons too

---
 .../testautomation/taf/browser/elements/BrButton.java  | 10 ++++++++++
 1 file changed, 10 insertions(+)

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
index b4f9450..414678d 100644
--- 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;
+    }
+
 }