Add --headed parameter to override Linux headless mode

Author: mdmintz

seleniumbase/fixtures/base_case.py

@@ -3170,6 +3170,7 @@ def setUp(self, masterqa_mode=False):
             self.with_selenium = sb_config.with_selenium  # Should be True
             self.headless = sb_config.headless
             self.headless_active = False
+            self.headed = sb_config.headed
             self.log_path = sb_config.log_path
             self.with_testing_base = sb_config.with_testing_base
             self.with_basic_test_info = sb_config.with_basic_test_info

seleniumbase/plugins/pytest_plugin.py

@@ -159,8 +159,18 @@ def pytest_addoption(parser):
                      action="store_true",
                      dest='headless',
                      default=False,
-                     help="""Using this makes Webdriver run headlessly,
-                          which is required on headless machines.""")
+                     help="""Using this makes Webdriver run web browsers
+                          headlessly, which is required on headless machines.
+                          Default: False on Mac/Windows. True on Linux.""")
+    parser.addoption('--headed', '--gui',
+                     action="store_true",
+                     dest='headed',
+                     default=False,
+                     help="""Using this makes Webdriver run web browsers with
+                          a GUI when running tests on Linux machines.
+                          (The default setting on Linux is headless.)
+                          (The default setting on Mac or Windows is headed.)
+                          """)
     parser.addoption('--is_pytest', '--is-pytest',
                      action="store_true",
                      dest='is_pytest',
@@ -257,6 +267,7 @@ def pytest_configure(config):
     sb_config.with_selenium = config.getoption('with_selenium')
     sb_config.user_agent = config.getoption('user_agent')
     sb_config.headless = config.getoption('headless')
+    sb_config.headed = config.getoption('headed')
     sb_config.extension_zip = config.getoption('extension_zip')
     sb_config.extension_dir = config.getoption('extension_dir')
     sb_config.with_testing_base = config.getoption('with_testing_base')

seleniumbase/plugins/selenium_plugin.py

@@ -25,6 +25,7 @@ class SeleniumBrowser(Plugin):
     self.options.extension_zip -- load a Chrome Extension ZIP (--extension_zip)
     self.options.extension_dir -- load a Chrome Extension DIR (--extension_dir)
     self.options.headless -- the option to run headlessly (--headless)
+    self.options.headed -- the option to run with a GUI on Linux (--headed)
     self.options.demo_mode -- the option to slow down Selenium (--demo_mode)
     self.options.demo_sleep -- Selenium action delay in DemoMode (--demo_sleep)
     self.options.highlights -- # of highlight animations shown (--highlights)
@@ -128,8 +129,18 @@ def options(self, parser, env):
             action="store_true",
             dest='headless',
             default=False,
-            help="""Using this makes Webdriver run headlessly,
-                    which is required on headless machines.""")
+            help="""Using this makes Webdriver run web browsers headlessly,
+                    which is required on headless machines.
+                    Default: False on Mac/Windows. True on Linux.""")
+        parser.add_option(
+            '--headed', '--gui',
+            action="store_true",
+            dest='headed',
+            default=False,
+            help="""Using this makes Webdriver run web browsers with
+                    a GUI when running tests on Linux machines.
+                    (The default setting on Linux is headless.)
+                    (The default setting on Mac or Windows is headed.)""")
         parser.add_option(
             '--demo_mode', '--demo-mode', '--demo',
             action="store_true",
@@ -233,6 +244,7 @@ def beforeTest(self, test):
         test.test.browser = self.options.browser
         test.test.cap_file = self.options.cap_file
         test.test.headless = self.options.headless
+        test.test.headed = self.options.headed
         test.test.servername = self.options.servername
         test.test.port = self.options.port
         test.test.user_data_dir = self.options.user_data_dir