28 lines
1.0 KiB
Python
28 lines
1.0 KiB
Python
from selenium import webdriver
|
|
from seleniumbase import BaseCase
|
|
BaseCase.main(__name__, __file__)
|
|
|
|
|
|
class OverrideDriverTest(BaseCase):
|
|
def get_new_driver(self, *args, **kwargs):
|
|
"""This method overrides get_new_driver() from BaseCase."""
|
|
options = webdriver.ChromeOptions()
|
|
options.add_argument("--disable-3d-apis")
|
|
options.add_argument("--disable-notifications")
|
|
if self.headless:
|
|
options.add_argument("--headless=new")
|
|
options.add_argument("--disable-gpu")
|
|
options.add_experimental_option(
|
|
"excludeSwitches", ["enable-automation", "enable-logging"],
|
|
)
|
|
prefs = {
|
|
"credentials_enable_service": False,
|
|
"profile.password_manager_enabled": False,
|
|
}
|
|
options.add_experimental_option("prefs", prefs)
|
|
return webdriver.Chrome(options=options)
|
|
|
|
def test_driver_override(self):
|
|
self.open("https://seleniumbase.io/demo_page")
|
|
self.assert_text("Demo Page", "h1")
|