[grid] Displaying the real stereotype config at startup

Author: diemol

java/src/org/openqa/selenium/grid/node/SessionFactory.java

@@ -29,4 +29,5 @@ public interface SessionFactory extends
   Function<CreateSessionRequest, Either<WebDriverException, ActiveSession>>,
   Predicate<Capabilities> {
 
+  Capabilities getStereotype();
 }

java/src/org/openqa/selenium/grid/node/config/DriverServiceSessionFactory.java

@@ -95,6 +95,11 @@ public DriverServiceSessionFactory(
     this.sessionCapabilitiesMutator = new SessionCapabilitiesMutator(this.stereotype);
   }
 
+  @Override
+  public Capabilities getStereotype() {
+    return stereotype;
+  }
+
   @Override
   public boolean test(Capabilities capabilities) {
     return predicate.test(capabilities);

java/src/org/openqa/selenium/grid/node/config/NodeOptions.java

@@ -651,7 +651,12 @@ private void report(Map.Entry<WebDriverInfo, Collection<SessionFactory>> entry)
     StringBuilder caps = new StringBuilder();
     try (JsonOutput out = JSON.newOutput(caps)) {
       out.setPrettyPrint(false);
-      out.write(entry.getKey().getCanonicalCapabilities());
+      Optional<SessionFactory> optionalSessionFactory = entry.getValue().stream().findFirst();
+      if (optionalSessionFactory.isPresent()) {
+        out.write(optionalSessionFactory.get().getStereotype());
+      } else {
+        out.write(entry.getKey().getCanonicalCapabilities());
+      }
     }
 
     LOG.info(String.format(

java/src/org/openqa/selenium/grid/node/docker/DockerSessionFactory.java

@@ -134,6 +134,11 @@ public DockerSessionFactory(
     this.slotMatcher = new DefaultSlotMatcher();
   }
 
+  @Override
+  public Capabilities getStereotype() {
+    return stereotype;
+  }
+
   @Override
   public boolean test(Capabilities capabilities) {
     return slotMatcher.matches(stereotype, capabilities);

java/src/org/openqa/selenium/grid/node/relay/RelaySessionFactory.java

@@ -97,6 +97,11 @@ public RelaySessionFactory(
       .copyOf(Require.nonNull("Stereotype", stereotype));
   }
 
+  @Override
+  public Capabilities getStereotype() {
+    return stereotype;
+  }
+
   @Override
   public boolean test(Capabilities capabilities) {
     // If a request reaches this point is because the basic match of W3C caps has already been done.

java/test/org/openqa/selenium/grid/graphql/GraphqlHandlerTest.java

@@ -254,6 +254,11 @@ void shouldBeAbleToGetUrlsOfAllNodes() throws URISyntaxException {
     String nodeUri = "http://localhost:5556";
     Node node = LocalNode.builder(tracer, bus, new URI(nodeUri), publicUri, registrationSecret)
       .add(stereotype, new SessionFactory() {
+        @Override
+        public Capabilities getStereotype() {
+          return null;
+        }
+
         @Override
         public Either<WebDriverException, ActiveSession> apply(
           CreateSessionRequest createSessionRequest) {

java/test/org/openqa/selenium/grid/node/config/NodeOptionsTest.java

@@ -613,6 +613,11 @@ public static class HelperFactory {
 
     public static SessionFactory create(Config config, Capabilities caps) {
       return new SessionFactory() {
+        @Override
+        public Capabilities getStereotype() {
+          return null;
+        }
+
         @Override
         public Either<WebDriverException, ActiveSession> apply(
           CreateSessionRequest createSessionRequest) {

java/test/org/openqa/selenium/grid/testing/TestSessionFactory.java

@@ -57,6 +57,11 @@ public TestSessionFactory(Capabilities stereotype, BiFunction<SessionId, Capabil
     this.sessionGenerator = sessionGenerator;
   }
 
+  @Override
+  public Capabilities getStereotype() {
+    return stereotype;
+  }
+
   @Override
   public Either<WebDriverException, ActiveSession> apply(CreateSessionRequest sessionRequest) {
     SessionId id = new SessionId(UUID.randomUUID());