[bidi][java] Add method to get browsing context tree with root

Author: pujagani

java/src/org/openqa/selenium/bidi/browsingcontext/BrowsingContext.java

@@ -148,6 +148,12 @@ public List<BrowsingContextInfo> getTree() {
             "browsingContext.getTree", Map.of("root", id), browsingContextInfoListMapper));
   }
 
+  public List<BrowsingContextInfo> getTree(String root) {
+    return this.bidi.send(
+      new Command<>(
+        "browsingContext.getTree", Map.of("root", root), browsingContextInfoListMapper));
+  }
+
   public List<BrowsingContextInfo> getTree(int maxDepth) {
     return this.bidi.send(
         new Command<>(
@@ -158,6 +164,16 @@ public List<BrowsingContextInfo> getTree(int maxDepth) {
             browsingContextInfoListMapper));
   }
 
+  public List<BrowsingContextInfo> getTree(String root, int maxDepth) {
+    return this.bidi.send(
+      new Command<>(
+        "browsingContext.getTree",
+        Map.of(
+          "root", root,
+          "maxDepth", maxDepth),
+        browsingContextInfoListMapper));
+  }
+
   public List<BrowsingContextInfo> getTopLevelContexts() {
     return this.bidi.send(
         new Command<>("browsingContext.getTree", new HashMap<>(), browsingContextInfoListMapper));

java/test/org/openqa/selenium/bidi/browsingcontext/BrowsingContextTest.java

@@ -24,6 +24,7 @@
 import static org.openqa.selenium.support.ui.ExpectedConditions.visibilityOfElementLocated;
 
 import java.util.List;
+
 import org.junit.jupiter.api.Test;
 import org.openqa.selenium.By;
 import org.openqa.selenium.JavascriptExecutor;
@@ -164,6 +165,49 @@ void canGetTreeWithDepth() {
     assertThat(info.getUserContext()).isEqualTo("default");
   }
 
+  @Test
+  @NeedsFreshDriver
+  void canGetTreeWithRootAndDepth() {
+    String referenceContextId = driver.getWindowHandle();
+    BrowsingContext parentWindow = new BrowsingContext(driver, referenceContextId);
+
+    String url = appServer.whereIs("iframes.html");
+
+    parentWindow.navigate(url, ReadinessState.COMPLETE);
+
+    List<BrowsingContextInfo> contextInfoList = parentWindow.getTree(referenceContextId, 1);
+
+    assertThat(contextInfoList.size()).isEqualTo(1);
+    BrowsingContextInfo info = contextInfoList.get(0);
+    assertThat(info.getChildren()).isNotNull(); // since depth is 1
+    assertThat(info.getId()).isEqualTo(referenceContextId);
+    assertThat(info.getOriginalOpener()).isNull();
+    assertThat(info.getClientWindow()).isNotNull();
+    assertThat(info.getUserContext()).isEqualTo("default");
+  }
+
+  @Test
+  @NeedsFreshDriver
+  void canGetTreeWithRoot() {
+    String referenceContextId = driver.getWindowHandle();
+    BrowsingContext parentWindow = new BrowsingContext(driver, referenceContextId);
+
+    String url = appServer.whereIs("iframes.html");
+
+    parentWindow.navigate(url, ReadinessState.COMPLETE);
+
+    BrowsingContext tab = new BrowsingContext(driver, WindowType.TAB);
+
+    List<BrowsingContextInfo> contextInfoList = parentWindow.getTree(tab.getId());
+
+    assertThat(contextInfoList.size()).isEqualTo(1);
+    BrowsingContextInfo info = contextInfoList.get(0);
+    assertThat(info.getId()).isEqualTo(tab.getId());
+    assertThat(info.getOriginalOpener()).isNull();
+    assertThat(info.getClientWindow()).isNotNull();
+    assertThat(info.getUserContext()).isEqualTo("default");
+  }
+
   @Test
   @NeedsFreshDriver
   void canGetAllTopLevelContexts() {