How to use drain method of org.openqa.selenium.grid.distributor.remote.RemoteDistributor class

Best Selenium code snippet using org.openqa.selenium.grid.distributor.remote.RemoteDistributor.drain

Source:DistributorTest.java

...204        new PassthroughHttpClient.Factory(node),205        sessions,206        registrationSecret);207    distributor.add(node);208    distributor.drain(node.getId());209    assertTrue(node.isDraining());210    NewSessionPayload payload = NewSessionPayload.create(caps);211    distributor.newSession(createRequest(payload)).getSession();212  }213  @Test214  public void testDrainedNodeShutsDownOnceEmpty() throws URISyntaxException, InterruptedException {215    URI nodeUri = new URI("http://example:5678");216    URI routableUri = new URI("http://localhost:1234");217    SessionMap sessions = new LocalSessionMap(tracer, bus);218    LocalNode node = LocalNode.builder(tracer, bus, routableUri, routableUri, registrationSecret)219        .add(caps, new TestSessionFactory((id, c) -> new Session(id, nodeUri, stereotype, c, Instant.now())))220        .build();221    CountDownLatch latch = new CountDownLatch(1);222    bus.addListener(NodeRemovedEvent.listener(ignored -> latch.countDown()));223    Distributor distributor = new LocalDistributor(224        tracer,225        bus,226        new PassthroughHttpClient.Factory(node),227        sessions,228        registrationSecret);229    distributor.add(node);230    distributor.drain(node.getId());231    latch.await(5, SECONDS);232    assertThat(latch.getCount()).isEqualTo(0);233    assertThat(distributor.getAvailableNodes().size()).isEqualTo(0);234    try (NewSessionPayload payload = NewSessionPayload.create(caps)) {235      assertThatExceptionOfType(SessionNotCreatedException.class)236          .isThrownBy(() -> distributor.newSession(createRequest(payload)));237    }238  }239  @Test240  public void drainedNodeDoesNotShutDownIfNotEmpty() throws URISyntaxException, InterruptedException {241    URI nodeUri = new URI("http://example:5678");242    URI routableUri = new URI("http://localhost:1234");243    SessionMap sessions = new LocalSessionMap(tracer, bus);244    LocalNode node = LocalNode.builder(tracer, bus, routableUri, routableUri, registrationSecret)245        .add(caps, new TestSessionFactory((id, c) -> new Session(id, nodeUri, stereotype, c, Instant.now())))246        .build();247    CountDownLatch latch = new CountDownLatch(1);248    bus.addListener(NodeRemovedEvent.listener(ignored -> latch.countDown()));249    Distributor distributor = new LocalDistributor(250        tracer,251        bus,252        new PassthroughHttpClient.Factory(node),253        sessions,254        registrationSecret);255    distributor.add(node);256    NewSessionPayload payload = NewSessionPayload.create(caps);257    distributor.newSession(createRequest(payload));258    distributor.drain(node.getId());259    latch.await(5, SECONDS);260    assertThat(latch.getCount()).isEqualTo(1);261    assertThat(distributor.getAvailableNodes().size()).isEqualTo(1);262  }263  @Test264  public void drainedNodeShutsDownAfterSessionsFinish() throws URISyntaxException, InterruptedException {265    URI nodeUri = new URI("http://example:5678");266    URI routableUri = new URI("http://localhost:1234");267    SessionMap sessions = new LocalSessionMap(tracer, bus);268    LocalNode node = LocalNode.builder(tracer, bus, routableUri, routableUri, registrationSecret)269        .add(caps, new TestSessionFactory((id, c) -> new Session(id, nodeUri, stereotype, c, Instant.now())))270        .add(caps, new TestSessionFactory((id, c) -> new Session(id, nodeUri, stereotype, c, Instant.now())))271        .build();272    CountDownLatch latch = new CountDownLatch(1);273    bus.addListener(NodeRemovedEvent.listener(ignored -> latch.countDown()));274    Distributor distributor = new LocalDistributor(275        tracer,276        bus,277        new PassthroughHttpClient.Factory(node),278        sessions,279        registrationSecret);280    distributor.add(node);281    NewSessionPayload payload = NewSessionPayload.create(caps);282    CreateSessionResponse firstResponse = distributor.newSession(createRequest(payload));283    CreateSessionResponse secondResponse = distributor.newSession(createRequest(payload));284    distributor.drain(node.getId());285    assertThat(distributor.getAvailableNodes().size()).isEqualTo(1);286    node.stop(firstResponse.getSession().getId());287    node.stop(secondResponse.getSession().getId());288    latch.await(5, SECONDS);289    assertThat(latch.getCount()).isEqualTo(0);290    assertThat(distributor.getAvailableNodes().size()).isEqualTo(0);291  }292  @Test293  public void registeringTheSameNodeMultipleTimesOnlyCountsTheFirstTime()294      throws URISyntaxException {295    URI nodeUri = new URI("http://example:5678");296    URI routableUri = new URI("http://localhost:1234");297    LocalNode node = LocalNode.builder(tracer, bus, routableUri, routableUri, registrationSecret)298        .add(caps, new TestSessionFactory((id, c) -> new Session(id, nodeUri, stereotype, c, Instant.now())))...

Source:AddingNodesTest.java

...373    public HealthCheck getHealthCheck() {374      return () -> new HealthCheck.Result(UP, "tl;dr");375    }376    @Override377    public void drain() {378    }379  }380}...

Source:RemoteDistributor.java

...67    LOG.info(String.format("Added node %s.", node.getId()));68    return this;69  }70  @Override71  public boolean drain(NodeId nodeId) {72    Require.nonNull("Node ID", nodeId);73    HttpRequest request = new HttpRequest(POST, "/se/grid/distributor/node/" + nodeId + "/drain");74    HttpTracing.inject(tracer, tracer.getCurrentContext(), request);75    request.setContent(asJson(nodeId));76    HttpResponse response = client.with(addSecret).execute(request);77    return Values.get(response, Boolean.class);78  }79  @Override80  public void remove(NodeId nodeId) {81    Require.nonNull("Node ID", nodeId);82    HttpRequest request = new HttpRequest(DELETE, "/se/grid/distributor/node/" + nodeId);83    HttpTracing.inject(tracer, tracer.getCurrentContext(), request);84    HttpResponse response = client.with(addSecret).execute(request);85    Values.get(response, Void.class);86  }87  @Override...

drain

Using AI Code Generation

1import org.openqa.selenium.grid.distributor.remote.RemoteDistributor;2import org.openqa.selenium.grid.distributor.remote.RemoteDistributorOptions;3import org.openqa.selenium.grid.node.local.LocalNode;4import org.openqa.selenium.grid.node.local.LocalNodeOptions;5import org.openqa.selenium.grid.sessionmap.config.SessionMapOptions;6import org.openqa.selenium.grid.web.Routable;7import org.openqa.selenium.internal.Require;8import org.openqa.selenium.json.Json;9import org.openqa.selenium.remote.http.HttpClient;10import org.openqa.selenium.remote.http.HttpRequest;11import org.openqa.selenium.remote.http.HttpResponse;12import org.openqa.selenium.remote.tracing.Tracer;13import org.openqa.selenium.remote.tracing.opentelemetry.OpenTelemetryTracer;14import java.io.IOException;15import java.net.URI;16import java.util.Objects;17import java.util.concurrent.atomic.AtomicInteger;18import java.util.function.Predicate;19public class DrainExample {20  private static final Json JSON = new Json();21  public static void main(String[] args) {22    Tracer tracer = new OpenTelemetryTracer();23    HttpClient.Factory clientFactory = HttpClient.Factory.createDefault();24    LocalNodeOptions nodeOptions = LocalNodeOptions.create(tracer, clientFactory);25    SessionMapOptions sessionMapOptions = SessionMapOptions.create();26    LocalNode node = nodeOptions.createLocalNode(sessionMapOptions);27    RemoteDistributorOptions distributorOptions = RemoteDistributorOptions.create();28    RemoteDistributor distributor = distributorOptions.createDistributor(clientFactory);29    AtomicInteger count = new AtomicInteger(0);30    Predicate<HttpRequest> predicate = req -> {31      String path = req.getUri().getPath();32      return "/se/grid/node".equals(path) || path.startsWith("/se/grid/node/");33    };34    Routable drain = new Routable() {35      public void execute(HttpRequest req, HttpResponse resp) throws IOException {36        count.incrementAndGet();37        resp.setStatus(200);38      }39      public boolean matches(HttpRequest req) {40        return predicate.test(req);41      }42    };43    node.add(drain);44    URI uri = node.getUrl().toURI();45    distributor.add(uri);46    distributor.drain();47    System.out.println("Drained " + count.get() + " nodes.");48  }49}50import org.openqa.selenium.grid.distributor.local.LocalDistributor;51import org.openqa.selenium.grid.distributor.local.LocalDistributorOptions;52import org.openqa.selenium.grid.node.local.LocalNode;53import org

drain

Using AI Code Generation

1import java.net.URI;2import java.net.URISyntaxException;3import java.util.List;4import java.util.stream.Collectors;5import org.openqa.selenium.Capabilities;6import org.openqa.selenium.ImmutableCapabilities;7import org.openqa.selenium.grid.config.Config;8import org.openqa.selenium.grid.config.MapConfig;9import org.openqa.selenium.grid.distributor.Distributor;10import org.openqa.selenium.grid.distributor.local.LocalDistributor;11import org.openqa.selenium.grid.distributor.remote.RemoteDistributor;12import org.openqa.selenium.grid.node.local.LocalNode;13import org.openqa.selenium.grid.sessionmap.config.SessionMapOptions;14import org.openqa.selenium.grid.web.Routable;15import org.openqa.selenium.grid.web.Routes;16import org.openqa.selenium.internal.Require;17import org.openqa.selenium.json.Json;18import org.openqa.selenium.remote.http.HttpMethod;19import org.openqa.selenium.remote.tracing.Tracer;20import org.openqa.selenium.remote.tracing.zipkin.ZipkinTracer;21import org.openqa.selenium.remote.tracing.zipkin.ZipkinTracerFactory;22import org.openqa.selenium.remote.tracing.zipkin.ZipkinTracerOptions;23import org.openqa.selenium.remote.tracing.zipkin.ZipkinTracerOptions.TracingLevel;24import org.openqa.selenium.remote.tracing.zipkin.receivers.InMemoryReceiver;25import org.openqa.selenium.remote.tracing.zipkin.receivers.Receiver;26import org.openqa.selenium.remote.tracing.zipkin.reporters.Sender;27import org.openqa.selenium.remote.tracing.zipkin.reporters.ZipkinSenderFactory;28import org.openqa.selenium.remote.tracing.zipkin.reporters.ZipkinSenderOptions;29import org.openqa.selenium.remote.tracing.zipkin.reporters.ZipkinSenderOptions.SenderType;30import org.openqa.selenium.remote.tracing.zipkin.reporters.okhttp3.OkHttpSenderFactory;31import org.openqa.selenium.remote.tracing.zipkin.reporters.okhttp3.OkHttpSenderOptions;32import org.openqa.selenium.remote.tracing.zipkin.reporters.okhttp3.OkHttpSenderOptions.HttpClientType;33import org.openqa.selenium.remote.tracing.zipkin.reporters.okhttp3.OkHttpSenderOptions.HttpVersion;34import org.openqa.selenium.remote.tracing.zipkin.reporters.okhttp3.OkHttpSenderOptions.SslProvider;35import org.openqa.selenium.remote.tracing.zipkin.reporters.okhttp3.OkHttpSenderOptions.TrustManagerType;36import org.openqa.selenium.remote.tracing.zipkin.reporters.okhttp3.OkHttpSenderOptions.Zipping;37import org.openqa.selenium.remote.tracing.zipkin.reporters.okhttp3.OkHttpSenderOptions.ZippingType;38import org.openqa.selenium.support.ui.Duration;39import com

drain

Using AI Code Generation

1import org.openqa.selenium.grid.distributor.remote.RemoteDistributor;2import org.openqa.selenium.grid.node.local.LocalNode;3import org.openqa.selenium.grid.node.local.LocalNodeFactory;4import org.openqa.selenium.grid.node.remote.RemoteNode;5import org.openqa.selenium.remote.http.HttpClient;6import org.openqa.selenium.remote.http.HttpClient.Factory;7import org.openqa.selenium.remote.tracing.Tracer;8import org.openqa.selenium.remote.tracing.distributed.DistributedTracer;9import org.openqa.selenium.remote.tracing.opentelemetry.OpenTelemetryOptions;10import org.openqa.selenium.remote.tracing.opentelemetry.OpenTelemetryTracer;11import java.net.URI;12import java.net.URISyntaxException;13import java.util.HashMap;14import java.util.Map;15public class RemoveNodeFromGrid {16    public static void main(String[] args) throws URISyntaxException {17        Tracer tracer = new DistributedTracer(new OpenTelemetryTracer(OpenTelemetryOptions.builder().build()));18        Factory clientFactory = HttpClient.Factory.createDefault();19        Map<String, String> nodeConfig = new HashMap<>();20        nodeConfig.put("node-id", "node1");21        nodeConfig.put("max-sessions", "5");22        nodeConfig.put("register-cycle", "5000");23        nodeConfig.put("unregister-if-still-down-after", "60000");24        LocalNodeFactory nodeFactory = new LocalNodeFactory(tracer, clientFactory, nodeConfig);25        distributor.drain(remoteNode);26    }27}

drain

Using AI Code Generation

1import org.openqa.selenium.grid.distributor.Distributor;2import org.openqa.selenium.grid.distributor.local.LocalDistributor;3import org.openqa.selenium.grid.distributor.remote.RemoteDistributor;4import org.openqa.selenium.grid.sessionqueue.NewSessionQueue;5import org.openqa.selenium.grid.sessionqueue.local.LocalNewSessionQueue;6import org.openqa.selenium.grid.sessionqueue.remote.RemoteNewSessionQueue;7import org.openqa.selenium.net.UrlChecker;8import org.openqa.selenium.remote.http.HttpClient;9import org.openqa.selenium.remote.http.HttpClientFactory;10import org.openqa.selenium.remote.tracing.Tracer;11import org.openqa.selenium.remote.tracing.TracerBuilder;12import org.openqa.selenium.remote.tracing.distributed.DistributedTracer;13import org.openqa.selenium.remote.tracing.distributed.http.HttpDistributedTracer;14import org.openqa.selenium.remote.tracing.opentelemetry.OpenTelemetryTracer;15import org.openqa.selenium.remote.tracing.opentelemetry.config.OpenTelemetryConfig;16import org.openqa.selenium.remote.tracing.opentelemetry.config.OpenTelemetryConfigBuilder;17import java.net.URI;18import java.util.concurrent.TimeUnit;19public class DistributorDrainSession {20  public static void main(String[] args) throws Exception {21    Tracer tracer = new TracerBuilder()22        .addSpanExporter(new OpenTelemetryTracer(23            OpenTelemetryConfig.builder()24                .setServiceName("selenium-grid")25                .build()))26        .build();27    HttpClient client = HttpClientFactory.createDefault().createClient(uri);28    DistributedTracer distributedTracer = new HttpDistributedTracer(29        uri.resolve("/se/grid/distributor/traces"));30    NewSessionQueue queue = new RemoteNewSessionQueue(31        distributedTracer);32    Distributor distributor = new RemoteDistributor(

drain

Using AI Code Generation

1import org.openqa.selenium.grid.distributor.remote.RemoteDistributor;2import org.openqa.selenium.grid.node.remote.RemoteNode;3import org.openqa.selenium.remote.http.HttpClient;4public class RemoveNode {5  public static void main(String[] args) {6    RemoteDistributor distributor = new RemoteDistributor(client);7    distributor.drain(node);8  }9}

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.