code: // Licensed to the Software Freedom Conservancy (SFC) under one
  // or more contributor license agreements.  See the NOTICE file
  // distributed with this work for additional information
  // regarding copyright ownership.  The SFC licenses this file
  // to you under the Apache License, Version 2.0 (the
  // "License"); you may not use this file except in compliance
  // with the License.  You may obtain a copy of the License at
  //
  //   http://www.apache.org/licenses/LICENSE-2.0
  //
  // Unless required by applicable law or agreed to in writing,
  // software distributed under the License is distributed on an
  // "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
  // KIND, either express or implied.  See the License for the
  // specific language governing permissions and limitations
  // under the License.
  
  package org.openqa.selenium.grid.router;
  
  import static org.assertj.core.api.AssertionsForClassTypes.assertThat;
  
  import java.io.StringReader;
  import java.util.Collections;
  import java.util.concurrent.CompletableFuture;
  import java.util.concurrent.ExecutionException;
  import java.util.concurrent.TimeUnit;
  import java.util.concurrent.TimeoutException;
  import org.junit.jupiter.api.Test;
  import org.openqa.selenium.By;
  import org.openqa.selenium.WebDriver;
  import org.openqa.selenium.WindowType;
  import org.openqa.selenium.bidi.BiDi;
  import org.openqa.selenium.bidi.BiDiSessionStatus;
  import org.openqa.selenium.bidi.Command;
  import org.openqa.selenium.bidi.HasBiDi;
  import org.openqa.selenium.bidi.LogInspector;
  import org.openqa.selenium.bidi.browsingcontext.BrowsingContext;
  import org.openqa.selenium.bidi.browsingcontext.NavigationResult;
  import org.openqa.selenium.bidi.log.ConsoleLogEntry;
  import org.openqa.selenium.bidi.log.LogLevel;
  import org.openqa.selenium.environment.webserver.AppServer;
  import org.openqa.selenium.environment.webserver.NettyAppServer;
  import org.openqa.selenium.firefox.FirefoxOptions;
  import org.openqa.selenium.grid.config.TomlConfig;
  import org.openqa.selenium.grid.router.DeploymentTypes.Deployment;
  import org.openqa.selenium.remote.Augmenter;
  import org.openqa.selenium.remote.RemoteWebDriver;
  import org.openqa.selenium.testing.drivers.Browser;
  
  class RemoteWebDriverBiDiTest {
  
    @Test
    void ensureBiDiSessionCreation() {
      Browser browser = Browser.FIREFOX;
  
      Deployment deployment =
          DeploymentTypes.STANDALONE.start(
              browser.getCapabilities(),
              new TomlConfig(
                  new StringReader(
                      "[node]\n"
                          + "selenium-manager = true\n"
                          + "driver-implementation = "
                          + browser.displayName())));
  
      FirefoxOptions options = createFirefoxOptions();
      // Enable BiDi
      options.setCapability("webSocketUrl", true);
      options.merge(Browser.FIREFOX.getCapabilities());
  
      WebDriver driver = new RemoteWebDriver(deployment.getServer().getUrl(), options);
      driver = new Augmenter().augment(driver);
  
      try (BiDi biDi = ((HasBiDi) driver).getBiDi()) {
        BiDiSessionStatus status =
            biDi.send(
                new Command<>("session.status", Collections.emptyMap(), BiDiSessionStatus.class));
        assertThat(status).isNotNull();
        assertThat(status.getMessage()).isEqualTo("Session already started");
      }
    }
  
    @Test
    void canListenToLogs() throws ExecutionException, InterruptedException, TimeoutException {
      Browser browser = Browser.FIREFOX;
  
      Deployment deployment =
          DeploymentTypes.STANDALONE.start(
              browser.getCapabilities(),
              new TomlConfig(
                  new StringReader(
                      "[node]\n"
                          + "selenium-manager = true\n"
                          + "driver-implementation = "
                          + browser.displayName())));
  
      FirefoxOptions options = createFirefoxOptions();
      // Enable BiDi
      options.setCapability("webSocketUrl", true);
  
      WebDriver driver = new RemoteWebDriver(deployment.getServer().getUrl(), options);
      driver = new Augmenter().augment(driver);
  
      AppServer server = new NettyAppServer();
      server.start();
  
      try (LogInspector logInspector = new LogInspector(driver)) {
        CompletableFuture<ConsoleLogEntry> future = new CompletableFuture<>();
        logInspector.onConsoleEntry(future::complete);
  
        String page = server.whereIs("/bidi/logEntryAdded.html");
        driver.get(page);
        driver.findElement(By.id("consoleLog")).click();
  
        ConsoleLogEntry logEntry = future.get(5, TimeUnit.SECONDS);
  
        assertThat(logEntry.getText()).isEqualTo("Hello, world!");
        assertThat(logEntry.getRealm()).isNull();
        assertThat(logEntry.getArgs().size()).isEqualTo(1);
        assertThat(logEntry.getType()).isEqualTo("console");
        assertThat(logEntry.getLevel()).isEqualTo(LogLevel.INFO);
        assertThat(logEntry.getMethod()).isEqualTo("log");
        assertThat(logEntry.getStackTrace()).isNull();
      }
    }
  
    @Test
    void canNavigateToUrl() {
      Browser browser = Browser.FIREFOX;
  
      Deployment deployment =
          DeploymentTypes.STANDALONE.start(
              browser.getCapabilities(),
              new TomlConfig(
                  new StringReader(
                      "[node]\n"
                          + "selenium-manager = true\n"
                          + "driver-implementation = "
                          + browser.displayName())));
  
      FirefoxOptions options = createFirefoxOptions();
      // Enable BiDi
      options.setCapability("webSocketUrl", true);
  
      WebDriver driver = new RemoteWebDriver(deployment.getServer().getUrl(), options);
      driver = new Augmenter().augment(driver);
  
      AppServer server = new NettyAppServer();
      server.start();
  
      BrowsingContext browsingContext = new BrowsingContext(driver, WindowType.TAB);
  
      String url = server.whereIs("/bidi/logEntryAdded.html");
      NavigationResult info = browsingContext.navigate(url);
  
      assertThat(browsingContext.getId()).isNotEmpty();
      assertThat(info.getNavigationId()).isNull();
      assertThat(info.getUrl()).contains("/bidi/logEntryAdded.html");
    }
  
    private FirefoxOptions createFirefoxOptions() {
      return (FirefoxOptions) Browser.FIREFOX.getCapabilities();
    }
  }
  