2323import  pytest 
2424
2525from  selenium .common .exceptions  import  SessionNotCreatedException 
26- from  selenium .webdriver .edge .options  import  Options 
2726from  selenium .webdriver .edge .service  import  Service 
2827
2928
3029@pytest .mark .no_driver_after_test  
31- def  test_uses_edgedriver_logging (clean_driver , driver_executable ) ->  None :
30+ def  test_uses_edgedriver_logging (clean_driver , clean_options ,  driver_executable ) ->  None :
3231    log_file  =  "msedgedriver.log" 
3332    service_args  =  ["--append-log" ]
3433
@@ -47,10 +46,10 @@ def test_uses_edgedriver_logging(clean_driver, driver_executable) -> None:
4746    driver1  =  None 
4847    driver2  =  None 
4948    try :
50-         driver1  =  clean_driver (service = service1 )
49+         driver1  =  clean_driver (options = clean_options ,  service = service1 )
5150        with  open (log_file ) as  fp :
5251            lines  =  len (fp .readlines ())
53-         driver2  =  clean_driver (service = service2 )
52+         driver2  =  clean_driver (options = clean_options ,  service = service2 )
5453        with  open (log_file ) as  fp :
5554            assert  len (fp .readlines ()) >=  2  *  lines 
5655    finally :
@@ -62,12 +61,12 @@ def test_uses_edgedriver_logging(clean_driver, driver_executable) -> None:
6261
6362
6463@pytest .mark .no_driver_after_test  
65- def  test_log_output_as_filename (clean_driver , driver_executable ) ->  None :
64+ def  test_log_output_as_filename (clean_driver , clean_options ,  driver_executable ) ->  None :
6665    log_file  =  "msedgedriver.log" 
6766    service  =  Service (log_output = log_file , executable_path = driver_executable )
6867    try :
6968        assert  "--log-path=msedgedriver.log"  in  service .service_args 
70-         driver  =  clean_driver (service = service )
69+         driver  =  clean_driver (options = clean_options ,  service = service )
7170        with  open (log_file ) as  fp :
7271            assert  "Starting Microsoft Edge WebDriver"  in  fp .readline ()
7372    finally :
@@ -76,12 +75,12 @@ def test_log_output_as_filename(clean_driver, driver_executable) -> None:
7675
7776
7877@pytest .mark .no_driver_after_test  
79- def  test_log_output_as_file (clean_driver , driver_executable ) ->  None :
78+ def  test_log_output_as_file (clean_driver , clean_options ,  driver_executable ) ->  None :
8079    log_name  =  "msedgedriver.log" 
8180    log_file  =  open (log_name , "w" , encoding = "utf-8" )
8281    service  =  Service (log_output = log_file , executable_path = driver_executable )
8382    try :
84-         driver  =  clean_driver (service = service )
83+         driver  =  clean_driver (options = clean_options ,  service = service )
8584        time .sleep (1 )
8685        with  open (log_name ) as  fp :
8786            assert  "Starting Microsoft Edge WebDriver"  in  fp .readline ()
@@ -92,9 +91,9 @@ def test_log_output_as_file(clean_driver, driver_executable) -> None:
9291
9392
9493@pytest .mark .no_driver_after_test  
95- def  test_log_output_as_stdout (clean_driver , capfd , driver_executable ) ->  None :
94+ def  test_log_output_as_stdout (clean_driver , clean_options ,  capfd , driver_executable ) ->  None :
9695    service  =  Service (log_output = subprocess .STDOUT , executable_path = driver_executable )
97-     driver  =  clean_driver (service = service )
96+     driver  =  clean_driver (options = clean_options ,  service = service )
9897
9998    out , err  =  capfd .readouterr ()
10099    assert  "Starting Microsoft Edge WebDriver"  in  out 
@@ -109,12 +108,11 @@ def test_log_output_null_default(driver, capfd) -> None:
109108
110109
111110@pytest .mark .no_driver_after_test  
112- def  test_driver_is_stopped_if_browser_cant_start (clean_driver ) ->  None :
113-     options  =  Options ()
114-     options .add_argument ("--user-data-dir=/no/such/location" )
115-     service  =  Service ()
111+ def  test_driver_is_stopped_if_browser_cant_start (clean_driver , clean_options , clean_service , driver_executable ) ->  None :
112+     clean_options .add_argument ("--user-data-dir=/no/such/location" )
113+     service  =  Service (executable_path = driver_executable )
116114    with  pytest .raises (SessionNotCreatedException ):
117-         clean_driver (options = options , service = service )
115+         clean_driver (options = clean_options , service = service )
118116    assert  not  service .is_connectable ()
119117    assert  service .process .poll () is  not None 
120118
0 commit comments