behave-framework/browserdriver/__init__.py

48 lines
1.4 KiB
Python
Raw Normal View History

2024-07-23 14:00:35 +00:00
# pylint: disable=too-few-public-methods
from selenium import webdriver
from selenium.webdriver.chrome.options import Options as ChromeOptions
from selenium.webdriver.firefox.options import Options as FirefoxOptions
from selenium.webdriver.edge.options import Options as EdgeOptions
class BrowserDriver:
2024-07-23 14:00:35 +00:00
@staticmethod
def get(browser=None, headless=True):
2024-07-23 14:00:35 +00:00
valid_browsers = ['chrome', 'firefox', 'safari', 'edge']
if browser in valid_browsers:
browser_method = globals()[browser]
return browser_method(headless)
raise ValueError(f"'{browser}' is not a supported browser")
def chrome(headless=True):
options = ChromeOptions()
if headless:
options.add_argument("--headless")
options.add_argument('--ignore-certificate-errors')
2024-07-23 14:00:35 +00:00
return webdriver.Chrome(options=options)
def firefox(headless=True):
options = FirefoxOptions()
if headless:
options.add_argument("--headless")
options.accept_insecure_certs = True
2024-07-23 14:00:35 +00:00
return webdriver.Firefox(options=options)
2020-10-10 22:27:46 +00:00
def edge(headless=True):
options = EdgeOptions()
2020-10-10 22:27:46 +00:00
options.use_chromium = True
options.add_argument('disable-gpu')
if headless:
options.add_argument('headless')
return webdriver.Edge(options=options)
def safari():
# Because safari driver is bundled with safari
return webdriver.Safari()