adding sample to get node details

Author: nicegraham

pom.xml

@@ -12,7 +12,7 @@
         <dependency>
             <groupId>org.seleniumhq.selenium</groupId>
             <artifactId>selenium-server</artifactId>
-            <version>2.33.0</version>
+            <version>2.32.0</version>
         </dependency>
         <dependency>
             <groupId>junit</groupId>
@@ -27,7 +27,12 @@
         <dependency>
             <groupId>org.apache.httpcomponents</groupId>
             <artifactId>httpclient</artifactId>
-            <version>4.1</version>
+            <version>4.2.5</version>
+        </dependency>
+        <dependency>
+            <groupId>org.apache.httpcomponents</groupId>
+            <artifactId>httpcore</artifactId>
+            <version>4.2.4</version>
         </dependency>
     </dependencies>
 

src/main/java/com/seleniumgrid2api/api/GridInfo.java

@@ -0,0 +1,71 @@
+package com.seleniumgrid2api.api;
+
+import org.json.JSONException;
+import org.json.JSONObject;
+
+import java.net.MalformedURLException;
+import java.net.URL;
+
+
+public class GridInfo {
+
+  private URL proxyId;
+  private String host;
+  private int port;
+  private String internalKey;
+  private String session;
+  private String inactivityTime;
+  private String msg;
+  private String success;
+
+  public URL getProxyId() {
+    return proxyId;
+  }
+
+  public String getHost() {
+    return host;
+  }
+
+  public int getPort() {
+    return port;
+  }
+
+  public String getInternalKey() {
+    return internalKey;
+  }
+
+  public String getSession() {
+    return session;
+  }
+
+  public String getInactivityTime() {
+    return inactivityTime;
+  }
+
+  public String getMsg() {
+    return msg;
+  }
+
+  public String getSuccess() {
+    return success;
+  }
+
+  public GridInfo(JSONObject object) {
+    try {
+      proxyId = new URL(object.getString("proxyId"));
+      if ((proxyId.getHost() != null) && (proxyId.getPort() != -1)) {
+        host = proxyId.getHost();
+        port = proxyId.getPort();
+      }
+      internalKey = object.getString("internalKey");
+      session = object.getString("session");
+      inactivityTime = object.getString("inactivityTime");
+      msg = object.getString("msg");
+      success = object.getString("success");
+
+    } catch (MalformedURLException | JSONException e) {
+      System.out.println("Error Parsing Grid Info.");
+    }
+
+  }
+}

src/main/java/com/seleniumgrid2api/api/GridInfoExtractor.java

@@ -0,0 +1,58 @@
+package com.seleniumgrid2api.api;
+
+import org.apache.http.HttpHost;
+import org.apache.http.HttpResponse;
+import org.apache.http.HttpStatus;
+import org.apache.http.impl.client.DefaultHttpClient;
+import org.apache.http.message.BasicHttpEntityEnclosingRequest;
+import org.json.JSONException;
+import org.json.JSONObject;
+import org.openqa.selenium.remote.SessionId;
+
+import java.io.BufferedReader;
+import java.io.IOException;
+import java.io.InputStreamReader;
+import java.net.URL;
+
+public final class GridInfoExtractor {
+
+  private GridInfoExtractor() {
+  }
+
+  public static GridInfo getHostNameAndPort(String hubHost, int hubPort, SessionId session) {
+
+    GridInfo retVal = null;
+
+    try {
+      HttpHost host = new HttpHost(hubHost, hubPort);
+      DefaultHttpClient client = new DefaultHttpClient();
+      URL sessionURL = new URL("http://" + hubHost + ":" + hubPort + "/grid/api/testsession?session=" + session);
+      BasicHttpEntityEnclosingRequest basicHttpEntityEnclosingRequest = new BasicHttpEntityEnclosingRequest("POST", sessionURL.toExternalForm());
+      HttpResponse response = client.execute(host, basicHttpEntityEnclosingRequest);
+      if (response.getStatusLine().getStatusCode() == HttpStatus.SC_OK) {
+        JSONObject object = extractObject(response);
+        retVal = new GridInfo(object);
+      } else {
+        System.out.println("Problem connecting to Grid Server");
+      }
+    } catch (JSONException | IOException  e) {
+      throw new RuntimeException("Failed to acquire remote webdriver node and port info", e);
+    }
+    return retVal;
+  }
+
+  private static JSONObject extractObject(HttpResponse resp) throws IOException, JSONException {
+    try (BufferedReader bufferedReader = new BufferedReader(new InputStreamReader(resp.getEntity().getContent()))) {
+      StringBuilder stringBuilder = new StringBuilder();
+      String line;
+      while ((line = bufferedReader.readLine()) != null) {
+        stringBuilder.append(line);
+      }
+      bufferedReader.close();
+      return new JSONObject(stringBuilder.toString());
+    } catch (Exception e) {
+      System.out.println("error" + e.toString());
+    }
+    return null;
+  }
+}

src/test/java/com/seleniumgrid2api/GetNodeDetails.java

@@ -0,0 +1,30 @@
+package com.seleniumgrid2api;
+
+import com.seleniumgrid2api.api.GridInfo;
+import com.seleniumgrid2api.api.GridInfoExtractor;
+import org.junit.Test;
+import org.openqa.selenium.WebDriver;
+import org.openqa.selenium.remote.DesiredCapabilities;
+import org.openqa.selenium.remote.RemoteWebDriver;
+
+import java.net.URL;
+
+public class GetNodeDetails {
+
+  @Test
+  public void getNodeDetailsViaTestSession() throws Exception {
+
+    DesiredCapabilities capability = DesiredCapabilities.firefox();
+
+    URL hub = new URL("http://localhost:4444/wd/hub");
+
+    WebDriver driver = new RemoteWebDriver(hub, capability);
+
+    System.out.println("Session Id: " + ((RemoteWebDriver) driver).getSessionId());
+
+    GridInfo gridInfo = GridInfoExtractor.getHostNameAndPort(hub.getHost(), hub.getPort(), ((RemoteWebDriver) driver).getSessionId());
+
+    System.out.println("I'm running on: " + gridInfo.getHost() + ":" + gridInfo.getPort());
+
+  }
+}