How to use drain method of org.openqa.selenium.grid.node.local.LocalNode class

Best Selenium code snippet using org.openqa.selenium.grid.node.local.LocalNode.drain

Source:NodeTest.java

...358    assertThat(baseDir).doesNotExist();359  }360  @Test361  public void shouldNotCreateSessionIfDraining() {362    node.drain();363    assertThat(local.isDraining()).isTrue();364    assertThat(node.isDraining()).isTrue();365    Optional<CreateSessionResponse> sessionResponse = node.newSession(createSessionRequest(caps));366    assertThat(sessionResponse.isPresent()).isFalse();367  }368  @Test369  public void shouldNotShutdownDuringOngoingSessionsIfDraining() throws InterruptedException {370    Optional<Session> firstSession =371        node.newSession(createSessionRequest(caps)).map(CreateSessionResponse::getSession);372    Optional<Session> secondSession =373        node.newSession(createSessionRequest(caps)).map(CreateSessionResponse::getSession);374    CountDownLatch latch = new CountDownLatch(1);375    bus.addListener(NODE_DRAIN_COMPLETE, e -> latch.countDown());376    node.drain();377    assertThat(local.isDraining()).isTrue();378    assertThat(node.isDraining()).isTrue();379    Optional<CreateSessionResponse> sessionResponse = node.newSession(createSessionRequest(caps));380    assertThat(sessionResponse.isPresent()).isFalse();381    assertThat(firstSession.isPresent()).isTrue();382    assertThat(secondSession.isPresent()).isTrue();383    assertThat(local.getCurrentSessionCount()).isEqualTo(2);384    latch.await(1, SECONDS);385    assertThat(latch.getCount()).isEqualTo(1);386  }387  @Test388  public void shouldShutdownAfterSessionsCompleteIfDraining() throws InterruptedException {389    CountDownLatch latch = new CountDownLatch(1);390    bus.addListener(NODE_DRAIN_COMPLETE, e -> latch.countDown());391    Optional<Session> firstSession =392        node.newSession(createSessionRequest(caps)).map(CreateSessionResponse::getSession);393    Optional<Session> secondSession =394        node.newSession(createSessionRequest(caps)).map(CreateSessionResponse::getSession);395    node.drain();396    assertThat(firstSession.isPresent()).isTrue();397    assertThat(secondSession.isPresent()).isTrue();398    node.stop(firstSession.get().getId());399    node.stop(secondSession.get().getId());400    latch.await(5, SECONDS);401    assertThat(latch.getCount()).isEqualTo(0);402  }403  @Test404  public void shouldAllowsWebDriverCommandsForOngoingSessionIfDraining() throws InterruptedException {405    CountDownLatch latch = new CountDownLatch(1);406    bus.addListener(NODE_DRAIN_COMPLETE, e -> latch.countDown());407    Optional<Session> session =408        node.newSession(createSessionRequest(caps)).map(CreateSessionResponse::getSession);409    node.drain();410    SessionId sessionId = session.get().getId();411    HttpRequest req = new HttpRequest(POST, String.format("/session/%s/url", sessionId));412    HttpResponse response = node.execute(req);413    assertThat(response.getStatus()).isEqualTo(200);414    assertThat(latch.getCount()).isEqualTo(1);415  }416  private File createTmpFile(String content) {417    try {418      File f = File.createTempFile("webdriver", "tmp");419      f.deleteOnExit();420      Files.write(f.toPath(), content.getBytes(StandardCharsets.UTF_8));421      return f;422    } catch (IOException e) {423      throw new UncheckedIOException(e);...

Source:AddingNodesTest.java

...277    public HealthCheck getHealthCheck() {278      return () -> new HealthCheck.Result(UP, "tl;dr");279    }280    @Override281    public void drain() {282    }283  }284}...

Source:LocalDistributorTest.java

...176    Set<NodeStatus> nodesBefore = statusBefore.getNodes();177    assertThat(nodesBefore.size()).isEqualTo(1);178    NodeStatus nodeBefore = nodesBefore.iterator().next();179    assertThat(nodeBefore.getAvailability()).isNotEqualTo(DRAINING);180    distributor.drain(localNode.getId());181    assertThat(localNode.isDraining()).isTrue();182    //Recheck the status - there should still be no node, it is removed183    DistributorStatus statusAfter = distributor.getStatus();184    Set<NodeStatus> nodesAfter = statusAfter.getNodes();185    assertThat(nodesAfter.size()).isEqualTo(0);186  }187  @Test188  public void testDrainNodeFromNode() {189    assertThat(localNode.isDraining()).isFalse();190    Distributor191        distributor =192        new LocalDistributor(tracer, bus, clientFactory, new LocalSessionMap(tracer, bus), null);193    distributor.add(localNode);194    localNode.drain();195    assertThat(localNode.isDraining()).isTrue();196  }197  private class Handler extends Session implements HttpHandler {198    private Handler(Capabilities capabilities) {199      super(new SessionId(UUID.randomUUID()), uri, new ImmutableCapabilities(), capabilities, Instant.now());200    }201    @Override202    public HttpResponse execute(HttpRequest req) throws UncheckedIOException {203      return new HttpResponse();204    }205  }206}...

Source:LocalNodeTest.java

...96    assertThat(node.isSessionOwner(session.getId())).isFalse();97  }98  @Test99  public void cannotAcceptNewSessionsWhileDraining() {100    node.drain();101    assertThat(node.isDraining()).isTrue();102    node.stop(session.getId()); //stop the default session103    Capabilities stereotype = new ImmutableCapabilities("cheese", "brie");104    Optional<CreateSessionResponse> sessionResponse = node.newSession(105        new CreateSessionRequest(106            ImmutableSet.of(W3C),107            stereotype,108            ImmutableMap.of()));109    assertThat(sessionResponse).isEmpty();110  }111  @Test112  public void canReturnStatusInfo() {113    NodeStatus status = node.getStatus();114    assertThat(status.getSlots().stream()...

drain

Using AI Code Generation

1import org.openqa.selenium.grid.config.Config;2import org.openqa.selenium.grid.config.MapConfig;3import org.openqa.selenium.grid.config.TomlConfig;4import org.openqa.selenium.grid.node.local.LocalNode;5import org.openqa.selenium.remote.http.HttpClient;6import java.io.BufferedReader;7import java.io.IOException;8import java.io.InputStreamReader;9import java.nio.file.Paths;10import java.util.HashMap;11import java.util.Map;12public class DrainNode {13    public static void main(String[] args) throws IOException {14        Config config = new TomlConfig(Paths.get("config.toml"));15        Config nodeConfig = new MapConfig(new HashMap<>(config.getAll()));16        LocalNode node = new LocalNode(nodeConfig, HttpClient.Factory.createDefault());17        BufferedReader bufferedReader = new BufferedReader(new InputStreamReader(System.in));18        System.out.println("Press enter to drain the node");19        bufferedReader.readLine();20        node.drain();21        System.out.println("Node drained");22    }23}24import org.openqa.selenium.By;25import org.openqa.selenium.WebDriver;26import org.openqa.selenium.WebElement;27import org.openqa.selenium.chrome.ChromeDriver;28import org.openqa.selenium.chrome.ChromeOptions;29import org.openqa.selenium.remote.RemoteWebDriver;30import java.net.MalformedURLException;31import java.net.URL;32public class CreateSession {33    public static void main(String[] args) throws MalformedURLException, InterruptedException {34        ChromeOptions options = new ChromeOptions();35        options.setExperimentalOption("w3c", true);36        WebElement searchBox = driver.findElement(By.name("q"));37        searchBox.sendKeys("selenium");38        searchBox.submit();39        Thread.sleep(300000);40    }41}

drain

Using AI Code Generation

1import org.openqa.selenium.grid.config.Config;2import org.openqa.selenium.grid.config.MemoizedConfig;3import org.openqa.selenium.grid.node.local.LocalNode;4import org.openqa.selenium.grid.sessionmap.config.SessionMapOptions;5import org.openqa.selenium.grid.sessionmap.remote.RemoteSessionMap;6import org.openqa.selenium.grid.web.Routable;7import org.openqa.selenium.grid.web.Routes;8import org.openqa.selenium.grid.web.Values;9import org.openqa.selenium.internal.Require;10import org.openqa.selenium.net.PortProber;11import org.openqa.selenium.remote.http.HttpClient;12import org.openqa.selenium.remote.http.HttpMethod;13import org.openqa.selenium.remote.http.HttpRequest;14import org.openqa.selenium.remote.http.HttpResponse;15import org.openqa.selenium.remote.tracing.Tracer;16import org.openqa.selenium.remote.tracing.opentelemetry.OpenTelemetryTracer;17import java.io.IOException;18import java.net.URI;19import java.net.URISyntaxException;20import java.time.Duration;21import java.util.Objects;22import java.util.concurrent.atomic.AtomicBoolean;23import java.util.function.Supplier;24public class LocalNodeDrain {25  public static void main(String[] args) throws URISyntaxException {26    LocalNode node = LocalNode.builder(27      new OpenTelemetryTracer(),28        new Config(),29      new RemoteSessionMap(30        new OpenTelemetryTracer(),31        HttpClient.Factory.createDefault(),32      PortProber.findFreePort())33        public void execute(HttpRequest req, HttpResponse resp) throws IOException {34          resp.setStatus(200);35        }36        public boolean matches(HttpRequest req) {37          return req.getMethod() == HttpMethod.GET;38        }39      }))40      .build();41    AtomicBoolean drained = new AtomicBoolean(false);42    while (!drained.get()) {43      try {44        Thread.sleep(1000);45      } catch (InterruptedException e) {46        e.printStackTrace();47      }48    }49    node.stop();50  }51}

drain

Using AI Code Generation

1import java.net.URL;2import java.util.Optional;3import java.util.concurrent.TimeUnit;4import java.util.logging.Level;5import java.util.logging.Logger;6import org.openqa.selenium.Capabilities;7import org.openqa.selenium.ImmutableCapabilities;8import org.openqa.selenium.SessionNotCreatedException;9import org.openqa.selenium.WebDriver;10import org.openqa.selenium.WebDriverException;11import org.openqa.selenium.grid.config.Config;12import org.openqa.selenium.grid.config.MapConfig;13import org.openqa.selenium.grid.config.TomlConfig;14import org.openqa.selenium.grid.data.CreateSessionResponse;15import org.openqa.selenium.grid.data.Session;16import org.openqa.selenium.grid.distributor.local.LocalDistributor;17import org.openqa.selenium.grid.node.local.LocalNode;18import org.openqa.selenium.grid.security.Secret;19import org.openqa.selenium.grid.server.BaseServerOptions;20import org.openqa.selenium.grid.server.Server;21import org.openqa.selenium.grid.server.ServerFlags;22import org.openqa.selenium.grid.server.StandaloneServer;23import org.openqa.selenium.grid.sessionmap.local.LocalSessionMap;24import org.openqa.selenium.grid.web.Routable;25import org.openqa.selenium.grid.web.Routes;26import org.openqa.selenium.internal.Require;27import org.openqa.selenium.remote.NewSessionPayload;28import org.openqa.selenium.remote.RemoteWebDriver;29import org.openqa.selenium.remote.http.HttpMethod;30import org.openqa.selenium.remote.http.HttpRequest;31import org.openqa.selenium.remote.http.HttpResponse;32import org.openqa.selenium.remote.tracing.DefaultTestTracer;33import org.openqa.selenium.remote.tracing.Tracer;34import com.google.common.collect.ImmutableMap;35public class MyGrid {36  private static final Logger LOG = Logger.getLogger(MyGrid.class.getName());37  private final Tracer tracer;38  private final LocalNode node;39  private final LocalDistributor distributor;40  private final LocalSessionMap sessions;41  private final Server<?> server;42  public MyGrid(Tracer tracer, LocalNode node, LocalDistributor distributor, LocalSessionMap sessions, Server<?> server) {43    this.tracer = Require.nonNull("Tracer", tracer);44    this.node = Require.nonNull("Node", node);45    this.distributor = Require.nonNull("Distributor", distributor);46    this.sessions = Require.nonNull("Session map", sessions);47    this.server = Require.nonNull("Server", server);48  }49  public static MyGrid create(Config config) {50    Tracer tracer = DefaultTestTracer.createTracer();51    LocalNode node = LocalNode.create(tracer, config);52    LocalDistributor distributor = LocalDistributor.create(tracer, config);

drain

Using AI Code Generation

1import org.openqa.selenium.By2import org.openqa.selenium.WebDriver3import org.openqa.selenium.chrome.ChromeDriver4import org.openqa.selenium.chrome.ChromeOptions5import org.openqa.selenium.devtools.DevTools6import org.openqa.selenium.devtools.v90.browser.Browser7import org.openqa.selenium.devtools.v90.browser.model.BrowserContextID8import org.openqa.selenium.devtools.v90.browser.model.Bounds9import org.openqa.selenium.devtools.v90.browser.model.WindowID10import org.openqa.selenium.devtools.v90.log.Log11import org.openqa.selenium.devtools.v90.network.Network12import org.openqa.selenium.devtools.v90.network.model.*13import org.openqa.selenium.devtools.v90.page.Page14import org.openqa.selenium.devtools.v90.page.model.*15import org.openqa.selenium.grid.config.Config16import org.openqa.selenium.grid.config.MemoizedConfig17import org.openqa.selenium.grid.config.TomlConfig18import org.openqa.selenium.grid.data.Session19import org.openqa.selenium.grid.node.local.LocalNode20import org.openqa.selenium.grid.sessionmap.config.SessionMapOptions21import org.openqa.selenium.grid.web.Values22import org.openqa.selenium.grid.web.config.DistributorOptions23import org.openqa.selenium.grid.web.config.WebDriverFactory24import org.openqa.selenium.internal.Require25import org.openqa.selenium.remote.RemoteWebDriver26import org.openqa.selenium.remote.http.HttpClient27import org.openqa.selenium.remote.http.HttpMethod28import org.openqa.selenium.remote.http.HttpRequest29import org.openqa.selenium.remote.http.HttpResponse30import org.openqa.selenium.support.ui.ExpectedConditions31import org.openqa.selenium.support.ui.WebDriverWait32import java.io.File33import java.io.IOException34import java.io.PrintWriter35import java.net.URI36import java.nio.file.Files37import java.nio.file.Path38import java.nio.file.Paths39import java.util.*40import java.util.concurrent.TimeUnit41import java.util.logging.Level42import java.util.logging.Logger43class CustomWebDriverFactory : WebDriverFactory {44    override fun apply(config: Config): WebDriver {45        val options = ChromeOptions()46        val chromeDriverPath = Paths.get("C:\\Users\\v-akumar\\Downloads\\chromedriver_win32\\chromedriver.exe")47        System.setProperty("webdriver.chrome.driver", chromeDriverPath.toString())48        options.addArguments("start-maximized")49        options.addArguments("enable-automation")50        options.addArguments("--no-sandbox")

Selenium 4 Tutorial:

LambdaTest’s Selenium 4 tutorial is covering every aspects of Selenium 4 testing with examples and best practices. Here you will learn basics, such as how to upgrade from Selenium 3 to Selenium 4, to some advanced concepts, such as Relative locators and Selenium Grid 4 for Distributed testing. Also will learn new features of Selenium 4, such as capturing screenshots of specific elements, opening a new tab or window on the browser, and new protocol adoptions.

Chapters:

Upgrading From Selenium 3 To Selenium 4?: In this chapter, learn in detail how to update Selenium 3 to Selenium 4 for Java binding. Also, learn how to upgrade while using different build tools such as Maven or Gradle and get comprehensive guidance for upgrading Selenium.
What’s New In Selenium 4 & What’s Being Deprecated? : Get all information about new implementations in Selenium 4, such as W3S protocol adaption, Optimized Selenium Grid, and Enhanced Selenium IDE. Also, learn what is deprecated for Selenium 4, such as DesiredCapabilites and FindsBy methods, etc.
Selenium 4 With Python: Selenium supports all major languages, such as Python, C#, Ruby, and JavaScript. In this chapter, learn how to install Selenium 4 for Python and the features of Python in Selenium 4, such as Relative locators, Browser manipulation, and Chrom DevTool protocol.
Selenium 4 Is Now W3C Compliant: JSON Wireframe protocol is retiring from Selenium 4, and they are adopting W3C protocol to learn in detail about the advantages and impact of these changes.
How To Use Selenium 4 Relative Locator? : Selenium 4 came with new features such as Relative Locators that allow constructing locators with reference and easily located constructors nearby. Get to know its different use cases with examples.
Selenium Grid 4 Tutorial For Distributed Testing: Selenium Grid 4 allows you to perform tests over different browsers, OS, and device combinations. It also enables parallel execution browser testing, reads up on various features of Selenium Grid 4 and how to download it, and runs a test on Selenium Grid 4 with best practices.
Selenium Video Tutorials: Binge on video tutorials on Selenium by industry experts to get step-by-step direction from automating basic to complex test scenarios with Selenium.