Simplify the Selenium Grid code.

Author: mdmintz

integrations/selenium_grid/ReadMe.md

@@ -1,11 +1,48 @@
-""" Download the selenium server jar file """
+""" Downloads the Selenium Server JAR file and renames it. """
 
 import os
-from seleniumbase.core import selenium_launcher
+import sys
+import urllib
 
-if not selenium_launcher.is_available_locally():
-    selenium_launcher.download_selenium()
+SELENIUM_JAR = ("http://selenium-release.storage.googleapis.com"
+                "/2.53/selenium-server-standalone-2.53.0.jar")
+JAR_FILE = "selenium-server-standalone-2.53.0.jar"
+try:
+    import selenium
+    if selenium.__version__[0] == '3':
+        SELENIUM_JAR = ("http://selenium-release.storage.googleapis.com"
+                        "/3.8/selenium-server-standalone-3.8.1.jar")
+        JAR_FILE = "selenium-server-standalone-3.8.1.jar"
+except Exception:
+    pass
 
-for filename in os.listdir("."):
-    if filename.startswith("selenium-server-standalone-"):
-        os.rename(filename, "selenium-server-standalone.jar")
+RENAMED_JAR_FILE = "selenium-server-standalone.jar"
+
+
+def download_selenium():
+    """
+    Downloads the Selenium Server JAR file from its
+    online location and stores it locally.
+    """
+    try:
+        local_file = open(JAR_FILE, 'wb')
+        remote_file = urllib.urlopen(SELENIUM_JAR)
+        print('Downloading Selenium Server JAR File...\n')
+        local_file.write(remote_file.read())
+        local_file.close()
+        remote_file.close()
+        print('Download Complete!\n')
+    except Exception:
+        raise Exception("Error downloading the Selenium Server JAR file.\n"
+                        "Details: %s" % sys.exc_info()[1])
+
+
+def is_available_locally():
+    return os.path.isfile(RENAMED_JAR_FILE)
+
+
+if not is_available_locally():
+    download_selenium()
+    for filename in os.listdir("."):
+        if filename.startswith("selenium-server-standalone-"):
+            os.rename(filename, RENAMED_JAR_FILE)

integrations/selenium_grid/download_selenium.py

@@ -21,7 +21,7 @@ fi
 ################################################################################
 
 WEBDRIVER_SERVER_JAR=./selenium-server-standalone.jar
-WEBDRIVER_NODE_PARAMS="-role webdriver -hubHost 127.0.0.1 -hubPort 4444 -host 127.0.0.1 -browserName=firefox"
+WEBDRIVER_NODE_PARAMS="-role webdriver -hubHost 127.0.0.1 -hubPort 4444 -host 127.0.0.1 -browser browserName=firefox,maxInstances=5,version=ANY,platform=ANY -browser browserName=chrome,maxInstances=5,version=ANY,platform=ANY"
 WEBDRIVER_NODE_PIDFILE="/tmp/webdriver_node.pid"
 
 if [ ! -f $WEBDRIVER_SERVER_JAR ]; then

integrations/selenium_grid/grid-node

@@ -0,0 +1,2 @@
+cd c:\
+java -jar selenium-server-standalone.jar -role node -hub http://localhost:4444/grid/register -browser browserName=chrome,maxInstances=5,version=ANY,platform=ANY -browser browserName=firefox,maxInstances=5,version=ANY,platform=ANY -browser browserName="internet explorer",maxInstances=1,version=ANY,platform=ANY

integrations/selenium_grid/register-node.bat

@@ -0,0 +1,2 @@
+#!/bin/bash
+java -jar selenium-server-standalone.jar -role node -hub http://localhost:4444/grid/register -browser browserName=chrome,maxInstances=5,version=ANY,platform=ANY -browser browserName=firefox,maxInstances=5,version=ANY,platform=ANY

integrations/selenium_grid/register-node.sh

@@ -0,0 +1,2 @@
+cd c:\
+java -jar selenium-server-standalone.jar -role hub