Add option to reuse the browser session between tests

Author: mdmintz

examples/raw_parameter_script.py

@@ -42,6 +42,7 @@
     sb.archive_logs = False
     sb.disable_csp = False
     sb.enable_sync = False
+    sb._reuse_session = False
     sb.visual_baseline = False
     sb.maximize_option = False
     sb.save_screenshot_after_test = False

seleniumbase/fixtures/base_case.py

@@ -4122,6 +4122,7 @@ def setUp(self, masterqa_mode=False):
             self.extension_zip = sb_config.extension_zip
             self.extension_dir = sb_config.extension_dir
             self.maximize_option = sb_config.maximize_option
+            self._reuse_session = sb_config.reuse_session
             self.save_screenshot_after_test = sb_config.save_screenshot
             self.visual_baseline = sb_config.visual_baseline
             self.timeout_multiplier = sb_config.timeout_multiplier
@@ -4192,21 +4193,50 @@ def setUp(self, masterqa_mode=False):
         if self.settings_file:
             settings_parser.set_settings(self.settings_file)
 
-        # Launch WebDriver for both Pytest and Nosetests
-        self.driver = self.get_new_driver(browser=self.browser,
-                                          headless=self.headless,
-                                          servername=self.servername,
-                                          port=self.port,
-                                          proxy=self.proxy_string,
-                                          agent=self.user_agent,
-                                          switch_to=True,
-                                          cap_file=self.cap_file,
-                                          disable_csp=self.disable_csp,
-                                          enable_sync=self.enable_sync,
-                                          user_data_dir=self.user_data_dir,
-                                          extension_zip=self.extension_zip,
-                                          extension_dir=self.extension_dir)
-        self._default_driver = self.driver
+        has_url = False
+        if self._reuse_session:
+            if not hasattr(sb_config, 'shared_driver'):
+                sb_config.shared_driver = None
+            if sb_config.shared_driver:
+                try:
+                    self._default_driver = sb_config.shared_driver
+                    self.driver = sb_config.shared_driver
+                    self._drivers_list = [sb_config.shared_driver]
+                    url = self.get_current_url()
+                    if len(url) > 3:
+                        has_url = True
+                except Exception:
+                    pass
+
+        if self._reuse_session and sb_config.shared_driver and has_url:
+            if self.start_page and len(self.start_page) >= 4:
+                if page_utils.is_valid_url(self.start_page):
+                    self.open(self.start_page)
+                else:
+                    new_start_page = "http://" + self.start_page
+                    if page_utils.is_valid_url(new_start_page):
+                        self.open(new_start_page)
+            else:
+                if self.get_current_url() != "data:,":
+                    self.open("data:,")
+        else:
+            # Launch WebDriver for both Pytest and Nosetests
+            self.driver = self.get_new_driver(browser=self.browser,
+                                              headless=self.headless,
+                                              servername=self.servername,
+                                              port=self.port,
+                                              proxy=self.proxy_string,
+                                              agent=self.user_agent,
+                                              switch_to=True,
+                                              cap_file=self.cap_file,
+                                              disable_csp=self.disable_csp,
+                                              enable_sync=self.enable_sync,
+                                              user_data_dir=self.user_data_dir,
+                                              extension_zip=self.extension_zip,
+                                              extension_dir=self.extension_dir)
+            self._default_driver = self.driver
+            if self._reuse_session:
+                sb_config.shared_driver = self.driver
 
     def __set_last_page_screenshot(self):
         """ self.__last_page_screenshot is only for pytest html report logs
@@ -4280,6 +4310,16 @@ def __add_pytest_html_extra(self):
             pass
 
     def __quit_all_drivers(self):
+        if self._reuse_session and sb_config.shared_driver:
+            if len(self._drivers_list) > 0:
+                sb_config.shared_driver = self._drivers_list[0]
+                self._default_driver = self._drivers_list[0]
+                self.switch_to_default_driver()
+            if len(self._drivers_list) > 1:
+                self._drivers_list = self._drivers_list[1:]
+            else:
+                self._drivers_list = []
+
         # Close all open browser windows
         self._drivers_list.reverse()  # Last In, First Out
         for driver in self._drivers_list:
@@ -4290,6 +4330,7 @@ def __quit_all_drivers(self):
             except Exception:
                 pass
         self.driver = None
+        self._default_driver = None
         self._drivers_list = []
 
     def tearDown(self):

seleniumbase/plugins/pytest_plugin.py

@@ -39,6 +39,7 @@ def pytest_addoption(parser):
     --verify_delay=SECONDS  (The delay before MasterQA verification checks.)
     --disable_csp  (This disables the Content Security Policy of websites.)
     --enable_sync  (The option to enable "Chrome Sync".)
+    --reuse_session  (The option to reuse the browser session between tests.)
     --maximize_window  (The option to start with the web browser maximized.)
     --save_screenshot  (The option to save a screenshot after each test.)
     --visual_baseline  (Set the visual baseline for Visual/Layout tests.)
@@ -292,6 +293,12 @@ def pytest_addoption(parser):
                      dest='enable_sync',
                      default=False,
                      help="""Using this enables the "Chrome Sync" feature.""")
+    parser.addoption('--reuse_session', '--reuse-session',
+                     action="store_true",
+                     dest='reuse_session',
+                     default=False,
+                     help="""The option to reuse the selenium browser window
+                          session between tests.""")
     parser.addoption('--maximize_window', '--maximize-window', '--maximize',
                      '--fullscreen',
                      action="store_true",
@@ -360,12 +367,18 @@ def pytest_configure(config):
     sb_config.verify_delay = config.getoption('verify_delay')
     sb_config.disable_csp = config.getoption('disable_csp')
     sb_config.enable_sync = config.getoption('enable_sync')
+    sb_config.reuse_session = config.getoption('reuse_session')
+    sb_config.shared_driver = None  # The default driver for session reuse
     sb_config.maximize_option = config.getoption('maximize_option')
     sb_config.save_screenshot = config.getoption('save_screenshot')
     sb_config.visual_baseline = config.getoption('visual_baseline')
     sb_config.timeout_multiplier = config.getoption('timeout_multiplier')
     sb_config.pytest_html_report = config.getoption('htmlpath')  # --html=FILE
 
+    if sb_config.reuse_session:
+        if "".join(sys.argv) == "-c":  # Can't "reuse_session" if multithreaded
+            sb_config.reuse_session = False
+
     if "linux" in sys.platform and (
             not sb_config.headed and not sb_config.headless):
         print(
@@ -383,6 +396,16 @@ def pytest_configure(config):
 def pytest_unconfigure():
     """ This runs after all tests have completed with pytest. """
     proxy_helper.remove_proxy_zip_if_present()
+    if sb_config.reuse_session:
+        # Close the shared browser session
+        if sb_config.shared_driver:
+            try:
+                sb_config.shared_driver.quit()
+            except AttributeError:
+                pass
+            except Exception:
+                pass
+        sb_config.shared_driver = None
 
 
 def pytest_runtest_setup():

seleniumbase/plugins/selenium_plugin.py

@@ -287,6 +287,7 @@ def beforeTest(self, test):
         test.test.visual_baseline = self.options.visual_baseline
         test.test.timeout_multiplier = self.options.timeout_multiplier
         test.test.use_grid = False
+        test.test._reuse_session = False
         if test.test.servername != "localhost":
             # Use Selenium Grid (Use --server=127.0.0.1 for localhost Grid)
             test.test.use_grid = True