How to use retryAddToQueue method of org.openqa.selenium.grid.sessionqueue.NewSessionQueue class

Best Selenium code snippet using org.openqa.selenium.grid.sessionqueue.NewSessionQueue.retryAddToQueue

Source:NewSessionQueuerTest.java

...205    assertFalse(httpRequest.isPresent());206  }207  @Test208  public void shouldBeAbleToAddAgainToQueue() {209    boolean added = local.retryAddToQueue(request, new RequestId(UUID.randomUUID()));210    assertTrue(added);211  }212  @Test213  public void shouldBeAbleToAddAgainToQueueRemotely() {214    HttpRequest request = createRequest(payload, POST, "/se/grid/newsessionqueuer/session");215    boolean added = remote.retryAddToQueue(request, new RequestId(UUID.randomUUID()));216    assertTrue(added);217  }218  @Test219  public void shouldBeAbleToRetryRequest() {220    AtomicBoolean isPresent = new AtomicBoolean(false);221    AtomicBoolean retrySuccess = new AtomicBoolean(false);222    bus.addListener(NewSessionRequestEvent.listener(reqId -> {223      // Keep a count of event fired224      count++;225      Optional<HttpRequest> sessionRequest = this.remote.remove();226      isPresent.set(sessionRequest.isPresent());227      if (count == 1) {228        retrySuccess.set(remote.retryAddToQueue(sessionRequest.get(), reqId));229      }230      // Only if it was retried after an interval, the count is 2231      if (count == 2) {232        ImmutableCapabilities capabilities = new ImmutableCapabilities("browserName", "chrome");233        try {234          SessionId sessionId = new SessionId("123");235          Session session =236              new Session(237                  sessionId,238                  new URI("http://example.com"),239                  caps,240                  capabilities,241                  Instant.now());242          CreateSessionResponse sessionResponse = new CreateSessionResponse(243              session,244              JSON.toJson(245                  ImmutableMap.of(246                      "value", ImmutableMap.of(247                          "sessionId", sessionId,248                          "capabilities", capabilities)))249                  .getBytes(UTF_8));250          NewSessionResponse newSessionResponse =251              new NewSessionResponse(reqId, sessionResponse.getSession(),252                                     sessionResponse.getDownstreamEncodedResponse());253          bus.fire(new NewSessionResponseEvent(newSessionResponse));254        } catch (URISyntaxException e) {255          bus.fire(256              new NewSessionRejectedEvent(257                  new NewSessionErrorResponse(new RequestId(UUID.randomUUID()), "Error")));258        }259      }260    }));261    HttpResponse httpResponse = remote.addToQueue(request);262    assertThat(isPresent.get()).isTrue();263    assertThat(retrySuccess.get()).isTrue();264    assertEquals(httpResponse.getStatus(), HTTP_OK);265  }266  @Test267  public void shouldBeAbleToHandleMultipleSessionRequestsAtTheSameTime() {268    bus.addListener(NewSessionRequestEvent.listener(reqId -> {269      Optional<HttpRequest> sessionRequest = this.local.remove();270      ImmutableCapabilities capabilities = new ImmutableCapabilities("browserName", "chrome");271      try {272        SessionId sessionId = new SessionId(UUID.randomUUID());273        Session session =274            new Session(275                sessionId,276                new URI("http://example.com"),277                caps,278                capabilities,279                Instant.now());280        CreateSessionResponse sessionResponse = new CreateSessionResponse(281            session,282            JSON.toJson(283                ImmutableMap.of(284                    "value", ImmutableMap.of(285                        "sessionId", sessionId,286                        "capabilities", capabilities)))287                .getBytes(UTF_8));288        NewSessionResponse newSessionResponse =289            new NewSessionResponse(reqId, sessionResponse.getSession(),290                                   sessionResponse.getDownstreamEncodedResponse());291        bus.fire(new NewSessionResponseEvent(newSessionResponse));292      } catch (URISyntaxException e) {293        bus.fire(294            new NewSessionRejectedEvent(295                new NewSessionErrorResponse(new RequestId(UUID.randomUUID()), "Error")));296      }297    }));298    ExecutorService executor = Executors.newFixedThreadPool(2);299    Callable<HttpResponse> callable = () -> remote.addToQueue(request);300    Future<HttpResponse> firstRequest = executor.submit(callable);301    Future<HttpResponse> secondRequest = executor.submit(callable);302    try {303      HttpResponse firstResponse = firstRequest.get(30, TimeUnit.SECONDS);304      HttpResponse secondResponse = secondRequest.get(30, TimeUnit.SECONDS);305      String firstResponseContents = Contents.string(firstResponse);306      String secondResponseContents = Contents.string(secondResponse);307      assertEquals(firstResponse.getStatus(), HTTP_OK);308      assertEquals(secondResponse.getStatus(), HTTP_OK);309      assertNotEquals(firstResponseContents, secondResponseContents);310    } catch (InterruptedException | ExecutionException | TimeoutException e) {311      fail("Could not create session");312    }313    executor.shutdown();314  }315  @Test316  public void shouldBeAbleToTimeoutARequestOnRetry() {317    Tracer tracer = DefaultTestTracer.createTracer();318    LocalNewSessionQueue sessionQueue = new LocalNewSessionQueue(319        tracer,320        bus,321        Duration.ofSeconds(4),322        Duration.ofSeconds(2));323    local = new LocalNewSessionQueuer(tracer, bus, sessionQueue);324    HttpClient client = new PassthroughHttpClient(local);325    remote = new RemoteNewSessionQueuer(tracer, client);326    HttpRequest request = createRequest(payload, POST, "/session");327    request.addHeader(SESSIONREQUEST_TIMESTAMP_HEADER,328                      Long.toString(1539091064));329    AtomicInteger count = new AtomicInteger();330    bus.addListener(NewSessionRequestEvent.listener(reqId -> {331      // Add to front of queue, when retry is triggered it will check if request timed out332      count.incrementAndGet();333      remote.retryAddToQueue(request, reqId);334    }));335    HttpResponse httpResponse = remote.addToQueue(request);336    assertEquals(count.get(),1);337    assertEquals(httpResponse.getStatus(), HTTP_INTERNAL_ERROR);338  }339  @Test340  public void shouldBeAbleToTimeoutARequestOnPoll() {341    Tracer tracer = DefaultTestTracer.createTracer();342    LocalNewSessionQueue sessionQueue = new LocalNewSessionQueue(343        tracer,344        bus,345        Duration.ofSeconds(4),346        Duration.ofSeconds(0));347    local = new LocalNewSessionQueuer(tracer, bus, sessionQueue);...

Source:RemoteNewSessionQueue.java

...84    upstream.setContent(Contents.asJson(request));85    return client.with(addSecret).execute(upstream);86  }87  @Override88  public boolean retryAddToQueue(SessionRequest request) {89    Require.nonNull("Session request", request);90    HttpRequest upstream =91      new HttpRequest(POST, String.format("/se/grid/newsessionqueue/session/%s/retry", request.getRequestId()));92    HttpTracing.inject(tracer, tracer.getCurrentContext(), upstream);93    upstream.setContent(Contents.asJson(request));94    HttpResponse response = client.with(addSecret).execute(upstream);95    return Values.get(response, Boolean.class);96  }97  @Override98  public Optional<SessionRequest> remove(RequestId reqId) {99    HttpRequest upstream = new HttpRequest(POST, "/se/grid/newsessionqueue/session/" + reqId);100    HttpTracing.inject(tracer, tracer.getCurrentContext(), upstream);101    HttpResponse response = client.with(addSecret).execute(upstream);102    if (response.isSuccessful()) {...

Source:RemoteNewSessionQueuer.java

...76    upstream.setContent(request.getContent());77    return client.execute(upstream);78  }79  @Override80  public boolean retryAddToQueue(HttpRequest request, RequestId reqId) {81    HttpRequest upstream =82      new HttpRequest(POST, "/se/grid/newsessionqueuer/session/retry/" + reqId.toString());83    HttpTracing.inject(tracer, tracer.getCurrentContext(), upstream);84    upstream.setContent(request.getContent());85    upstream.setHeader(timestampHeader, request.getHeader(timestampHeader));86    upstream.setHeader(reqIdHeader, reqId.toString());87    HttpResponse response = client.with(addSecret).execute(upstream);88    return Values.get(response, Boolean.class);89  }90  @Override91  public Optional<HttpRequest> remove(RequestId reqId) {92    HttpRequest upstream =93      new HttpRequest(GET, "/se/grid/newsessionqueuer/session/" + reqId.toString());94    HttpTracing.inject(tracer, tracer.getCurrentContext(), upstream);...

Source:LocalNewSessionQueuer.java

...55        getNewSessionResponse = new GetNewSessionResponse(tracer, bus, sessionRequests);56    return getNewSessionResponse.add(request);57  }58  @Override59  public boolean retryAddToQueue(HttpRequest request, RequestId reqId) {60    return sessionRequests.offerFirst(request, reqId);61  }62  @Override63  public Optional<HttpRequest> remove() {64    return sessionRequests.poll();65  }66  @Override67  public int clearQueue() {68    return sessionRequests.clear();69  }70  @Override71  public boolean isReady() {72    return bus.isReady();73  }...

Source:AddBackToSessionQueue.java

...44    try (Span span = newSpanAsChildOf(tracer, req, "sessionqueue.retry")) {45      HTTP_REQUEST.accept(span, req);46      span.setAttribute(AttributeKey.REQUEST_ID.getKey(), id.toString());47      SessionRequest sessionRequest = Contents.fromJson(req, SessionRequest.class);48      boolean value = newSessionQueue.retryAddToQueue(sessionRequest);49      span.setAttribute("request.retry", value);50      HttpResponse response = new HttpResponse()51        .setContent(asJson(singletonMap("value", value)));52      HTTP_RESPONSE.accept(span, response);53      return response;54    }55  }56}...

retryAddToQueue

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.data.NewSessionRequest;5import org.openqa.selenium.grid.data.Session;6import org.openqa.selenium.grid.node.local.LocalNode;7import org.openqa.selenium.grid.sessionqueue.NewSessionQueue;8import org.openqa.selenium.grid.sessionqueue.config.SessionQueueOptions;9import org.openqa.selenium.internal.Require;10import org.openqa.selenium.remote.http.Contents;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 java.io.IOException;16import java.net.URI;17import java.util.Map;18import java.util.Objects;19import java.util.concurrent.TimeUnit;20import java.util.concurrent.atomic.AtomicReference;21import static org.openqa.selenium.remote.http.Contents.asJson;22import static org.openqa.selenium.remote.http.Contents.bytes;23import static org.openqa.selenium.remote.http.Contents.utf8String;24import static org.openqa.selenium.remote.http.HttpMethod.DELETE;25import static org.openqa.selenium.remote.http.HttpMethod.GET;26import static org.openqa.selenium.remote.http.HttpMethod.POST;27public class RetryAddToQueueExample {28  public static void main(String[] args) {29    Config config = new MapConfig();30    SessionQueueOptions sessionQueueOptions = new SessionQueueOptions(config);31    NewSessionQueue sessionQueue = sessionQueueOptions.getSessionQueue();32    NewSessionRequest newSessionRequest = new NewSessionRequest(33            new MapConfig(Map.of("capabilities", Map.of("alwaysMatch", Map.of("browserName", "chrome")))));34    sessionQueue.add(newSessionRequest);35    sessionQueue.retryAddToQueue(newSessionRequest, 10, TimeUnit.SECONDS);36  }37}38import org.openqa.selenium.grid.config.Config;39import org.openqa.selenium.grid.config.MapConfig;40import org.openqa.selenium.grid.config.TomlConfig;41import org.openqa.selenium.grid.data.NewSessionRequest;42import org.openqa.selenium.grid.data.Session;43import org.openqa.selenium.grid.node.local.LocalNode;44import org.openqa.selenium.grid.sessionqueue.NewSessionQueue;45import org.openqa.selenium.grid.sessionqueue.config

retryAddToQueue

Using AI Code Generation

1import org.openqa.selenium.grid.sessionqueue.NewSessionQueue;2import org.openqa.selenium.grid.sessionqueue.config.NewSessionQueueOptions;3import org.openqa.selenium.grid.sessionqueue.local.LocalNewSessionQueue;4import org.openqa.selenium.internal.Require;5import org.openqa.selenium.remote.http.HttpClient;6import org.openqa.selenium.remote.http.HttpRequest;7import org.openqa.selenium.remote.http.HttpResponse;8import org.openqa.selenium.remote.tracing.Tracer;9import java.time.Duration;10import java.util.concurrent.TimeUnit;11public class RetryAddToQueue {12  private final NewSessionQueue delegate;13  private final HttpClient.Factory clientFactory;14  private final Tracer tracer;15  private final NewSessionQueueOptions options;16  public RetryAddToQueue(NewSessionQueue delegate, HttpClient.Factory clientFactory, Tracer tracer, NewSessionQueueOptions options) {17    this.delegate = Require.nonNull("Delegate queue", delegate);18    this.clientFactory = Require.nonNull("HTTP client factory", clientFactory);19    this.tracer = Require.nonNull("Tracer", tracer);20    this.options = Require.nonNull("Options", options);21  }22  public void retryAddToQueue(HttpRequest request) {23    Duration retryInterval = Duration.ofSeconds(5);24    int retryCount = 3;25    for (int i = 0; i < retryCount; i++) {26      try {27        delegate.addToQueue(request);28        break;29      } catch (Exception e) {30        if (i == retryCount - 1) {31          throw e;32        }33        try {34          TimeUnit.SECONDS.sleep(retryInterval.getSeconds());35        } catch (InterruptedException interruptedException) {36          throw new RuntimeException(interruptedException);37        }38      }39    }40  }41  public static void main(String[] args) {42    HttpClient.Factory clientFactory = HttpClient.Factory.createDefault();43    Tracer tracer = Tracer.getDefault();44    NewSessionQueueOptions options = NewSessionQueueOptions.create();45    NewSessionQueue queue = new LocalNewSessionQueue(tracer, options);46    RetryAddToQueue retryAddToQueue = new RetryAddToQueue(queue, clientFactory, tracer, options);47    HttpRequest request = new HttpRequest("POST", "/session");48    retryAddToQueue.retryAddToQueue(request);49  }50}

retryAddToQueue

Using AI Code Generation

1import org.openqa.selenium.grid.config.Config;2import org.openqa.selenium.grid.config.ConfigException;3import org.openqa.selenium.grid.config.MemoizedConfig;4import org.openqa.selenium.grid.sessionqueue.NewSessionQueue;5import org.openqa.selenium.grid.sessionqueue.local.LocalNewSessionQueue;6import org.openqa.selenium.json.Json;7import org.openqa.selenium.remote.http.HttpClient;8import org.openqa.selenium.remote.tracing.Tracer;9import org.openqa.selenium.remote.tracing.opentelemetry.OpenTelemetryTracer;10import java.io.IOException;11import java.io.UncheckedIOException;12import java.net.URL;13import java.time.Duration;14import java.util.Optional;15public class RetryAddToQueue {16  public static void main(String[] args) throws IOException {17    Config config = new MemoizedConfig();18    Json json = new Json();19    HttpClient.Factory clientFactory = HttpClient.Factory.createDefault();20    Tracer tracer = OpenTelemetryTracer.create();21    NewSessionQueue sessionQueue = LocalNewSessionQueue.create(22      tracer);23    int retries = 5;24    int retryInterval = 5;25    Optional<NewSessionQueue.AddStatus> status = sessionQueue.retryAddToQueue(26      Duration.ofSeconds(retries),27      Duration.ofSeconds(retryInterval));28    if (status.isPresent()) {29      System.out.println("Session added successfully");30    } else {31      System.out.println("Session could not be added");32    }33  }34}35import org.openqa.selenium.grid.config.Config;36import org.openqa.selenium.grid.config.ConfigException;37import org.openqa.selenium.grid.config.MemoizedConfig;38import org.openqa.selenium.grid.sessionqueue.NewSessionQueue;39import org.openqa.selenium

retryAddToQueue

Using AI Code Generation

1package org.openqa.selenium.grid.sessionqueue;2import static org.openqa.selenium.grid.data.SessionClosedEvent.SESSION_CLOSED;3import static org.openqa.selenium.grid.data.SessionCreatedEvent.SESSION_CREATED;4import static org.openqa.selenium.grid.data.SessionQueuedEvent.SESSION_QUEUED;5import static org.openqa.selenium.grid.data.SessionRequestEvent.SESSION_REQUEST;6import static org.openqa.selenium.grid.data.SessionRequestEvent.SESSION_REQUEST_REJECTED;7import static org.openqa.selenium.grid.data.SessionRequestEvent.SESSION_REQUEST_TIMED_OUT;8import static org.openqa.selenium.grid.data.SessionRequestEvent.SESSION_REQUEST_UNFULFILLED;9import static org.openqa.selenium.grid.data.SessionTerminatedEvent.SESSION_TERMINATED;10import static org.openqa.selenium.grid.sessionqueue.local.LocalNewSessionQueue.MAX_CONCURRENCY;11import static org.openqa.selenium.grid.sessionqueue.local.LocalNewSessionQueue.MAX_RETRIES;12import static org.openqa.selenium.grid.sessionqueue.local.LocalNewSessionQueue.RETRY_TIMEOUT;13import static org.openqa.selenium.grid.sessionqueue.local.LocalNewSessionQueue.RETRY_TIMEOUT_UNIT;14import static org.openqa.selenium.grid.sessionqueue.local.LocalNewSessionQueue.RETRY_WAIT;15import static org.openqa.selenium.grid.sessionqueue.local.LocalNewSessionQueue.RETRY_WAIT_UNIT;16import static org.openqa.selenium.support.ui.ExpectedConditions.and;17import static org.openqa.selenium.support.ui.ExpectedConditions.or;18import static org.openqa.selenium.support.ui.ExpectedConditions.presenceOfElementLocated;19import com.google.common.collect.ImmutableMap;20import com.google.common.collect.ImmutableSet;21import com.google.common.collect.ImmutableSortedSet;22import com.google.common.collect.Maps;23import com.google.common.eventbus.EventBus;24import com.google.common.eventbus.Subscribe;25import com.google.common.util.concurrent.Uninterruptibles;26import org.openqa.selenium.Capabilities;27import org.openqa.selenium.ImmutableCapabilities;28import org.openqa.selenium.NoSuchSessionException;29import org.openqa.selenium.NoSuchSessionException.SessionNotReachableException;30import org.openqa.selenium.WebDriverException;31import org.openqa.selenium.concurrent.Regularly;32import org.openqa.selenium.events.EventBusOptions;33import org.openqa.selenium.events.EventBusOptions.Topic;34import org.openqa.selenium.events.zeromq.ZeroMqEventBus;35import org.openqa.selenium.grid.config.Config;36import org.openqa.selenium.grid.config.MapConfig;37import org.openqa.selenium.grid.config.TomlConfig;38import org.openqa.selenium.grid.data.CreateSessionRequest;39import org.openqa.selenium.grid.data.CreateSessionResponse;40import org.openqa.selenium.grid.data.RequestId;41import org.openqa.selenium.grid.data.Session;42import org.openqa.selenium.grid.data.SessionClosedEvent;

retryAddToQueue

Using AI Code Generation

1package org.openqa.selenium.grid.sessionqueue;2import com.google.common.collect.ImmutableList;3import com.google.common.collect.ImmutableSet;4import com.google.common.collect.Lists;5import com.google.common.collect.Sets;6import org.openqa.selenium.Capabilities;7import org.openqa.selenium.internal.Require;8import org.openqa.selenium.json.Json;9import org.openqa.selenium.remote.tracing.Tracer;10import org.openqa.selenium.remote.tracing.opentelemetry.OpenTelemetryTracer;11import org.openqa.selenium.remote.tracing.opentelemetry.OpenTelemetryTracerFactory;12import org.openqa.selenium.support.ui.Duration;13import java.util.ArrayList;14import java.util.Collection;15import java.util.List;16import java.util.Optional;17import java.util.Set;18import java.util.concurrent.ConcurrentHashMap;19import java.util.concurrent.ConcurrentMap;20import java.util.concurrent.CopyOnWriteArrayList;21import java.util.concurrent.CopyOnWriteArraySet;22import java.util.concurrent.TimeUnit;23import java.util.function.Predicate;24import java.util.logging.Logger;25import java.util.stream.Collectors;26public class NewSessionQueue {27  private static final Logger LOG = Logger.getLogger(NewSessionQueue.class.getName());28  private static final Json JSON = new Json();29      new ConcurrentHashMap<>();30      new ConcurrentHashMap<>();31      new ConcurrentHashMap<>();32      new ConcurrentHashMap<>();33      new ConcurrentHashMap<>();34      new ConcurrentHashMap<>();35  private final Tracer tracer;36  public NewSessionQueue(Tracer tracer) {37    this.tracer = Require.nonNull("Tracer", tracer);38  }39  public void add(Capabilities capabilities, Duration timeout, Predicate<Throwable> predicate) {40    Require.nonNull("Capabilities", capabilities);41    Require.nonNull("Timeout", timeout);42    Require.nonNull("Predicate", predicate);43    NewSessionRequest request = new NewSessionRequest(tracer);44    this.requests.computeIfAbsent(capabilities, k -> new CopyOnWriteArraySet<>()).add(request);45    this.capabilities.put(request,

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.