-def setup():
+from contextlib import contextmanager
+
+@contextmanager
+def managed_driver():
 driver = webdriver.Chrome()
- driver.get("https://www.selenium.dev/selenium/web/web-form.html")
- return driver
+ try:
+ driver.get("https://www.selenium.dev/selenium/web/web-form.html")
+ yield driver
+ finally:
+ driver.quit()
 
-def teardown(driver):
- driver.quit()
-

+import logging
+
 def setup():
- driver = webdriver.Chrome()
- driver.get("https://www.selenium.dev/selenium/web/web-form.html")
- return driver
+ try:
+ driver = webdriver.Chrome()
+ driver.get("https://www.selenium.dev/selenium/web/web-form.html")
+ return driver
+ except Exception as e:
+ logging.error(f"Error during setup: {e}")
+ raise
 
 def teardown(driver):
- driver.quit()
+ try:
+ driver.quit()
+ except Exception as e:
+ logging.error(f"Error during teardown: {e}")
 

+from selenium.webdriver.support.ui import WebDriverWait
+from selenium.webdriver.support import expected_conditions as EC
+from selenium.webdriver.common.by import By
+
 def setup():
 driver = webdriver.Chrome()
 driver.get("https://www.selenium.dev/selenium/web/web-form.html")
+ WebDriverWait(driver, 10).until(
+ EC.presence_of_element_located((By.ID, "message"))
+ )
 return driver
 

-def setup():
- driver = webdriver.Chrome()
- driver.get("https://www.selenium.dev/selenium/web/web-form.html")
+def setup(browser_type='chrome', url='https://www.selenium.dev/selenium/web/web-form.html'):
+ if browser_type.lower() == 'chrome':
+ driver = webdriver.Chrome()
+ elif browser_type.lower() == 'firefox':
+ driver = webdriver.Firefox()
+ else:
+ raise ValueError(f"Unsupported browser type: {browser_type}")
+ driver.get(url)
 return driver