1313
1414def event_listener (driver ) -> EventFiringWebDriver :
1515 """Attach the event listener to the driver."""
16- e_driver : EventFiringWebDriver = EventFiringWebDriver (
17- driver , EventListener ()
18- )
16+ e_driver : EventFiringWebDriver = EventFiringWebDriver (driver , EventListener ())
1917 return e_driver
2018
2119
@@ -29,16 +27,15 @@ def driver_types(request):
2927@pytest .fixture
3028def make_driver (request ) -> EventFiringWebDriver :
3129 load_dotenv (dotenv_path = f"{ request .config .getoption ('--env' )} " )
32- env = request .config .getoption (' --env' )
33- dr_type = request .config .getoption (' --type' )
30+ env = request .config .getoption (" --env" )
31+ dr_type = request .config .getoption (" --type" )
3432 driver = None
3533
3634 def _make_driver () -> EventFiringWebDriver :
3735 nonlocal driver
3836 # Create WebDriver instance
3937 driver = WebDriverFactory ().create_driver (
40- environment = env ,
41- driver_type = dr_type
38+ environment = env , driver_type = dr_type
4239 )
4340 # Attach event listener
4441 driver_with_listener = event_listener (driver )
@@ -54,23 +51,31 @@ def _make_driver() -> EventFiringWebDriver:
5451
5552
5653def pytest_addoption (parser ):
57- parser .addoption ("--browser-version" , action = "store" , default = "129" ,
58- help = "Specify the browser version" )
59- parser .addoption ("--env" , action = "store" , default = 'stage' ,
60- help = "Run browser in headless mode" )
61- parser .addoption ("--type" , action = "store" , default = 'local' ,
62- help = "Run browser in os type" )
54+ parser .addoption (
55+ "--browser-version" ,
56+ action = "store" ,
57+ default = "129" ,
58+ help = "Specify the browser version" ,
59+ )
60+ parser .addoption (
61+ "--env" , action = "store" , default = "stage" , help = "Run browser in headless mode"
62+ )
63+ parser .addoption (
64+ "--type" , action = "store" , default = "local" , help = "Run browser in os type"
65+ )
6366
6467
6568def pytest_runtest_makereport (item , call ):
6669 """Capture screenshot on test failure."""
6770 if call .excinfo is not None :
6871 # Make sure the driver is being captured correctly
69- driver = item .funcargs .get (' make_driver' , None )
72+ driver = item .funcargs .get (" make_driver" , None )
7073
7174 if driver is not None :
7275 screenshot_dir = "reports/screenshots"
73- os .makedirs (screenshot_dir , exist_ok = True ) # Create directory if it does not exist
76+ os .makedirs (
77+ screenshot_dir , exist_ok = True
78+ ) # Create directory if it does not exist
7479 screenshot_path = os .path .join (screenshot_dir , f"{ item .name } .png" )
7580
7681 try :
0 commit comments