Adding "server" and "port" run options for pytest (for Selenium Grid)

Author: mdmintz

seleniumbase/fixtures/base_case.py

@@ -1384,6 +1384,8 @@ def setUp(self):
             self.with_basic_test_info = (
                 pytest.config.option.with_basic_test_info)
             self.with_page_source = pytest.config.option.with_page_source
+            self.servername = pytest.config.option.servername
+            self.port = pytest.config.option.port
             self.database_env = pytest.config.option.database_env
             self.log_path = pytest.config.option.log_path
             self.browser = pytest.config.option.browser
@@ -1393,6 +1395,10 @@ def setUp(self):
             self.highlights = pytest.config.option.highlights
             self.verify_delay = pytest.config.option.verify_delay
             self.timeout_multiplier = pytest.config.option.timeout_multiplier
+            self.use_grid = False
+            if self.servername != "localhost":
+                # Use Selenium Grid (Use --server=127.0.0.1 for localhost Grid)
+                self.use_grid = True
             if self.with_db_reporting:
                 self.execution_guid = str(uuid.uuid4())
                 self.testcase_guid = None

seleniumbase/plugins/pytest_plugin.py

@@ -70,6 +70,16 @@ def pytest_addoption(parser):
                      dest='with_page_source',
                      default=False,
                      help="Use to save page source on test failure.")
+    parser.addoption('--server', action='store',
+                     dest='servername',
+                     default='localhost',
+                     help="""Designates the server used by the test.
+                          Default: localhost.""")
+    parser.addoption('--port', action='store',
+                     dest='port',
+                     default='4444',
+                     help="""Designates the port used by the test.
+                          Default: 4444.""")
     parser.addoption('--headless', action="store_true",
                      dest='headless',
                      default=False,