How to use reserve method of org.openqa.selenium.grid.distributor.GridModel class

Best Selenium code snippet using org.openqa.selenium.grid.distributor.GridModel.reserve

Source:GridModel.java

...48import static org.openqa.selenium.grid.data.Availability.DRAINING;49import static org.openqa.selenium.grid.data.Availability.UP;50public class GridModel {51  private static final Logger LOG = Logger.getLogger(GridModel.class.getName());52  private static final SessionId RESERVED = new SessionId("reserved");53  private final ReadWriteLock lock = new ReentrantReadWriteLock(/* fair */ true);54  private final Map<Availability, Set<NodeStatus>> nodes = new ConcurrentHashMap<>();55  private final EventBus events;56  public GridModel(EventBus events, Secret registrationSecret) {57    this.events = Require.nonNull("Event bus", events);58    events.addListener(NodeDrainStarted.listener(nodeId -> setAvailability(nodeId, DRAINING)));59    events.addListener(NodeDrainComplete.listener(this::remove));60    events.addListener(NodeRemovedEvent.listener(this::remove));61    events.addListener(NodeStatusEvent.listener(status -> refresh(registrationSecret, status)));62    events.addListener(SessionClosedEvent.listener(this::release));63  }64  public GridModel add(NodeStatus node) {65    Require.nonNull("Node", node);66    Lock writeLock = lock.writeLock();67    writeLock.lock();68    try {69      // If we've already added the node, remove it.70      for (Set<NodeStatus> nodes : nodes.values()) {71        Iterator<NodeStatus> iterator = nodes.iterator();72        while (iterator.hasNext()) {73          NodeStatus next = iterator.next();74          // If the ID is the same, we're re-adding a node. If the URI is the same a node probably restarted75          if (next.getId().equals(node.getId()) || next.getUri().equals(node.getUri())) {76            LOG.info(String.format("Re-adding node with id %s and URI %s.", node.getId(), node.getUri()));77            iterator.remove();78          }79        }80      }81      // Nodes are initially added in the "down" state until something changes their availability82      nodes(DOWN).add(node);83    } finally {84      writeLock.unlock();85    }86    return this;87  }88  public GridModel refresh(Secret registrationSecret, NodeStatus status) {89    Require.nonNull("Node status", status);90    Secret statusSecret = status.getRegistrationSecret() == null ? null : new Secret(status.getRegistrationSecret());91    if (!Objects.equals(registrationSecret, statusSecret)) {92      LOG.severe(String.format("Node at %s failed to send correct registration secret. Node NOT refreshed.", status.getUri()));93      events.fire(new NodeRejectedEvent(status.getUri()));94      return this;95    }96    Lock writeLock = lock.writeLock();97    writeLock.lock();98    try {99      AvailabilityAndNode availabilityAndNode = findNode(status.getId());100      if (availabilityAndNode == null) {101        return this;102      }103      // if the node was marked as "down", keep it down until a healthcheck passes:104      // just because the node can hit the event bus doesn't mean it's reachable105      if (DOWN.equals(availabilityAndNode.availability)) {106        nodes(DOWN).remove(availabilityAndNode.status);107        nodes(DOWN).add(status);108        return this;109      }110      // But do trust the node if it tells us it's draining111      nodes(availabilityAndNode.availability).remove(availabilityAndNode.status);112      nodes(status.getAvailability()).add(status);113      return this;114    } finally {115      writeLock.unlock();116    }117  }118  public GridModel remove(NodeId id) {119    Require.nonNull("Node ID", id);120    Lock writeLock = lock.writeLock();121    writeLock.lock();122    try {123      AvailabilityAndNode availabilityAndNode = findNode(id);124      if (availabilityAndNode == null) {125        return this;126      }127      nodes(availabilityAndNode.availability).remove(availabilityAndNode.status);128      return this;129    } finally {130      writeLock.unlock();131    }132  }133  public Availability setAvailability(NodeId id, Availability availability) {134    Require.nonNull("Node ID", id);135    Require.nonNull("Availability", availability);136    Lock writeLock = lock.writeLock();137    writeLock.lock();138    try {139      AvailabilityAndNode availabilityAndNode = findNode(id);140      if (availabilityAndNode == null) {141        return DOWN;142      }143      if (availability.equals(availabilityAndNode.availability)) {144        return availability;145      }146      nodes(availabilityAndNode.availability).remove(availabilityAndNode.status);147      nodes(availability).add(availabilityAndNode.status);148      LOG.info(String.format(149        "Switching node %s (uri: %s) from %s to %s",150        id,151        availabilityAndNode.status.getUri(),152        availabilityAndNode.availability,153        availability));154      return availabilityAndNode.availability;155    } finally {156      writeLock.unlock();157    }158  }159  public boolean reserve(SlotId slotId) {160    Lock writeLock = lock.writeLock();161    writeLock.lock();162    try {163      AvailabilityAndNode node = findNode(slotId.getOwningNodeId());164      if (node == null) {165        LOG.warning(String.format("Asked to reserve slot on node %s, but unable to find node", slotId.getOwningNodeId()));166        return false;167      }168      if (!UP.equals(node.availability)) {169        LOG.warning(String.format(170          "Asked to reserve a slot on node %s, but not is %s",171          slotId.getOwningNodeId(),172          node.availability));173        return false;174      }175      Optional<Slot> maybeSlot = node.status.getSlots().stream()176        .filter(slot -> slotId.equals(slot.getId()))177        .findFirst();178      if (!maybeSlot.isPresent()) {179        LOG.warning(String.format(180          "Asked to reserve slot on node %s, but no slot with id %s found",181          node.status.getId(),182          slotId));183        return false;184      }185      reserve(node.status, maybeSlot.get());186      return true;187    } finally {188      writeLock.unlock();189    }190  }191  public Set<NodeStatus> getSnapshot() {192    Lock readLock = this.lock.readLock();193    readLock.lock();194    try {195      ImmutableSet.Builder<NodeStatus> snapshot = ImmutableSet.builder();196      for (Map.Entry<Availability, Set<NodeStatus>> entry : nodes.entrySet()) {197        entry.getValue().stream()198          .map(status -> rewrite(status, entry.getKey()))199          .forEach(snapshot::add);200      }201      return snapshot.build();202    } finally {203      readLock.unlock();204    }205  }206  private Set<NodeStatus> nodes(Availability availability) {207    return nodes.computeIfAbsent(availability, ignored -> new HashSet<>());208  }209  private AvailabilityAndNode findNode(NodeId id) {210    for (Map.Entry<Availability, Set<NodeStatus>> entry : nodes.entrySet()) {211      for (NodeStatus nodeStatus : entry.getValue()) {212        if (id.equals(nodeStatus.getId())) {213          return new AvailabilityAndNode(entry.getKey(), nodeStatus);214        }215      }216    }217    return null;218  }219  private NodeStatus rewrite(NodeStatus status, Availability availability) {220    return new NodeStatus(221      status.getId(),222      status.getUri(),223      status.getMaxSessionCount(),224      status.getSlots(),225      availability,226      status.getRegistrationSecret() == null ? null : new Secret(status.getRegistrationSecret()));227  }228  private void release(SessionId id) {229    if (id == null) {230      return;231    }232    Lock writeLock = lock.writeLock();233    writeLock.lock();234    try {235      for (Map.Entry<Availability, Set<NodeStatus>> entry : nodes.entrySet()) {236        for (NodeStatus node : entry.getValue()) {237          for (Slot slot : node.getSlots()) {238            if (!slot.getSession().isPresent()) {239              continue;240            }241            if (id.equals(slot.getSession().get().getId())) {242              Slot released = new Slot(243                slot.getId(),244                slot.getStereotype(),245                slot.getLastStarted(),246                Optional.empty());247              amend(entry.getKey(), node, released);248              return;249            }250          }251        }252      }253    } finally {254      writeLock.unlock();255    }256  }257  private void reserve(NodeStatus status, Slot slot) {258    Instant now = Instant.now();259    Slot reserved = new Slot(260      slot.getId(),261      slot.getStereotype(),262      now,263      Optional.of(new Session(264        RESERVED,265        status.getUri(),266        slot.getStereotype(),267        slot.getStereotype(),268        now)));269    amend(UP, status, reserved);270  }271  public void setSession(SlotId slotId, Session session) {272    Require.nonNull("Slot ID", slotId);273    AvailabilityAndNode node = findNode(slotId.getOwningNodeId());274    if (node == null) {275      LOG.warning("Grid model and reality have diverged. Unable to find node " + slotId.getOwningNodeId());276      return;277    }278    Optional<Slot> maybeSlot = node.status.getSlots().stream()279      .filter(slot -> slotId.equals(slot.getId()))280      .findFirst();281    if (!maybeSlot.isPresent()) {282      LOG.warning("Grid model and reality have diverged. Unable to find slot " + slotId);283      return;284    }285    Slot slot = maybeSlot.get();286    Optional<Session> maybeSession = slot.getSession();287    if (!maybeSession.isPresent()) {288      LOG.warning("Grid model and reality have diverged. Slot is not reserved. " + slotId);289      return;290    }291    Session current = maybeSession.get();292    if (!RESERVED.equals(current.getId())) {293      LOG.warning("Gid model and reality have diverged. Slot has session and is not reserved. " + slotId);294      return;295    }296    Slot updated = new Slot(297      slot.getId(),298      slot.getStereotype(),299      session == null ? slot.getLastStarted() : session.getStartTime(),300      Optional.ofNullable(session));301    amend(node.availability, node.status, updated);302  }303  private void amend(Availability availability, NodeStatus status, Slot slot) {304    Set<Slot> newSlots = new HashSet<>(status.getSlots());305    newSlots.removeIf(s -> s.getId().equals(slot.getId()));306    newSlots.add(slot);307    nodes(availability).remove(status);...

Source:LocalDistributor.java

...237      readLock.unlock();238    }239  }240  @Override241  protected Supplier<CreateSessionResponse> reserve(SlotId slotId, CreateSessionRequest request) {242    Require.nonNull("Slot ID", slotId);243    Require.nonNull("New Session request", request);244    Lock writeLock = this.lock.writeLock();245    writeLock.lock();246    try {247      Node node = nodes.get(slotId.getOwningNodeId());248      if (node == null) {249        return () -> {250          throw new SessionNotCreatedException("Unable to find node");251        };252      }253      model.reserve(slotId);254      return () -> {255        Optional<CreateSessionResponse> response = node.newSession(request);256        if (!response.isPresent()) {257          model.setSession(slotId, null);258          throw new SessionNotCreatedException("Unable to create session for " + request);259        }260        model.setSession(slotId, response.get().getSession());261        return response.get();262      };263    } finally {264      writeLock.unlock();265    }266  }267}...

reserve

Using AI Code Generation

1import org.openqa.selenium.grid.distributor.GridModel;2import org.openqa.selenium.grid.distributor.local.LocalDistributor;3import org.openqa.selenium.grid.node.local.LocalNode;4import org.openqa.selenium.grid.server.BaseServerOptions;5import org.openqa.selenium.grid.server.Server;6import org.openqa.selenium.grid.web.Values;7import org.openqa.selenium.remote.http.HttpClient;8import org.openqa.selenium.remote.http.HttpRequest;9import org.openqa.selenium.remote.http.HttpResponse;10import org.openqa.selenium.remote.http.Route;11import org.openqa.selenium.remote.tracing.Tracer;12import org.openqa.selenium.remote.tracing.opentelemetry.OpenTelemetryTracer;13import java.net.MalformedURLException;14import java.net.URL;15import java.util.concurrent.TimeUnit;16public class ReserveMethod {17    public static void main(String[] args) throws MalformedURLException, InterruptedException {18        Tracer tracer = new OpenTelemetryTracer();19        HttpClient.Factory clientFactory = HttpClient.Factory.createDefault();20        LocalNode.Builder nodeBuilder = LocalNode.builder(tracer, clientFactory);21        LocalNode node = nodeBuilder.build();22        LocalDistributor.Builder distributorBuilder = LocalDistributor.builder(tracer, clientFactory);23        distributorBuilder.add(node);24        LocalDistributor distributor = distributorBuilder.build();25        BaseServerOptions serverOptions = new BaseServerOptions();26        serverOptions.port = 5555;27        Server<?> server = new Server<>(serverOptions, new Route("/").to(() -> req -> {28            GridModel model = distributor.getModel();29            Values values = model.reserve(req, 30, TimeUnit.SECONDS);30            return new HttpResponse().setContent(values.toJson());31        }), tracer, clientFactory);32        server.start();33        HttpRequest request = new HttpRequest("GET", "/");34        System.out.println(response.getContentString());35        server.stop();36    }37}38{"value":{"sessionId":"f7b9e9a0-7e9b-4e8f-abf4-4d4e4c1e2b81","capabilities":{"acceptInsecureCerts":false,"browserName":"chrome","browserVersion":"85.0.4183.102","chrome":{"chromedriverVersion":"85.0.4183

reserve

Using AI Code Generation

1GridModel model = new GridModel();2LocalDistributor distributor = new LocalDistributor(new LocalDistributorConfig(), new NullEventBus());3RemoteDistributor distributor = new RemoteDistributor(new RemoteDistributorConfig(), new NullEventBus());4RemoteDistributor distributor = new RemoteDistributor(new RemoteDistributorConfig(), new NullEventBus());5RemoteDistributor distributor = new RemoteDistributor(new RemoteDistributorConfig(), new NullEventBus());6RemoteDistributor distributor = new RemoteDistributor(new Remote

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.