Skip to content

Commit e770ae7

Browse files
committed
Update the example tests
1 parent 8963b27 commit e770ae7

File tree

2 files changed

+22
-2
lines changed

2 files changed

+22
-2
lines changed

examples/test_override_driver.py

Lines changed: 20 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,20 @@
1+
from selenium import webdriver
2+
from seleniumbase import BaseCase
3+
4+
5+
class OverrideDriverTest(BaseCase):
6+
7+
def get_new_driver(self, *args, **kwargs):
8+
""" This method overrides get_new_driver() from BaseCase. """
9+
options = webdriver.ChromeOptions()
10+
options.add_experimental_option(
11+
"excludeSwitches", ["enable-automation"]
12+
)
13+
options.add_experimental_option("useAutomationExtension", False)
14+
if self.headless:
15+
options.add_argument("--headless")
16+
return webdriver.Chrome(options=options)
17+
18+
def test_simple(self):
19+
self.open("https://seleniumbase.io/demo_page")
20+
self.assert_text("Demo Page", "h1")

examples/test_shadow_dom.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -30,7 +30,7 @@ def test_shadow_dom(self):
3030

3131
# Download Python package files from PyPI
3232
file_name_1 = self.download_tar_file_from_pypi("sbase")
33-
file_name_2 = self.download_tar_file_from_pypi("tensorpy")
33+
file_name_2 = self.download_tar_file_from_pypi("seleniumbase")
3434
self.assert_downloaded_file(file_name_1, browser=True)
3535
self.assert_downloaded_file(file_name_2, browser=True)
3636

@@ -67,7 +67,7 @@ def test_shadow_dom(self):
6767
self.type(search_input, "sbase")
6868
self.assert_text(file_name_1, file_link)
6969
print("\n Download 1: %s" % self.get_text(file_link))
70-
self.type(search_input, "tensorpy")
70+
self.type(search_input, "seleniumbase")
7171
self.assert_text(file_name_2, file_link)
7272
print(" Download 2: %s" % self.get_text(file_link))
7373
self.click(clear_search_icon)

0 commit comments

Comments
 (0)