@@ -302,6 +302,7 @@ def sb(request):
302302 from selenium import webdriver
303303 from seleniumbase import BaseCase
304304 from seleniumbase import config as sb_config
305+ from seleniumbase.core import session_helper
305306
306307 class BaseClass (BaseCase ):
307308 def get_new_driver (self , * args , ** kwargs ):
@@ -326,6 +327,11 @@ def sb(request):
326327 super ().tearDown()
327328
328329 if request.cls:
330+ if sb_config.reuse_class_session:
331+ the_class = str (request.cls).split(" ." )[- 1 ].split(" '" )[0 ]
332+ if the_class != sb_config._sb_class:
333+ session_helper.end_reused_class_session_as_needed()
334+ sb_config._sb_class = the_class
329335 request.cls.sb = BaseClass(" base_method" )
330336 request.cls.sb.setUp()
331337 request.cls.sb._needs_tearDown = True
@@ -877,18 +883,18 @@ This pure Python format gives you a raw <code translate="no">webdriver</code> in
877883from seleniumbase import DriverContext
878884
879885with DriverContext() as driver:
880- driver.get( " https:// seleniumbase.github.io/" )
886+ driver.open( " seleniumbase.github.io/" )
881887 driver.highlight(' img[alt="SeleniumBase"]' , loops = 6 )
882888
883889with DriverContext(browser = " chrome" , incognito = True ) as driver:
884- driver.get( " https:// seleniumbase.io/apps/calculator" )
890+ driver.open( " seleniumbase.io/apps/calculator" )
885891 driver.click(' [id="4"]' )
886892 driver.click(' [id="2"]' )
887893 driver.assert_text(" 42" , " #output" )
888894 driver.highlight(" #output" , loops = 6 )
889895
890896with DriverContext() as driver:
891- driver.get( " https:// seleniumbase.github.io/demo_page" )
897+ driver.open( " seleniumbase.github.io/demo_page" )
892898 driver.highlight(" h2" )
893899 driver.type(" #myTextInput" , " Automation" )
894900 driver.click(" #checkBox1" )
@@ -908,7 +914,7 @@ from seleniumbase import Driver
908914
909915driver = Driver(browser = " chrome" , headless = False )
910916try :
911- driver.get( " https:// seleniumbase.io/apps/calculator" )
917+ driver.open( " seleniumbase.io/apps/calculator" )
912918 driver.click(' [id="4"]' )
913919 driver.click(' [id="2"]' )
914920 driver.assert_text(" 42" , " #output" )
@@ -918,7 +924,7 @@ finally:
918924
919925driver = Driver()
920926try :
921- driver.get( " https:// seleniumbase.github.io/demo_page" )
927+ driver.open( " seleniumbase.github.io/demo_page" )
922928 driver.highlight(" h2" )
923929 driver.type(" #myTextInput" , " Automation" )
924930 driver.click(" #checkBox1" )
0 commit comments