Update example tests

Author: mdmintz

examples/boilerplates/samples/google_objects.py

@@ -4,7 +4,6 @@
 class HomePage(object):
     dialog_box = '[role="dialog"] div'
     search_box = 'input[title="Search"]'
-    list_box = '[role="listbox"]'
     search_button = 'input[value="Google Search"]'
     feeling_lucky_button = """input[value="I'm Feeling Lucky"]"""
 

examples/boilerplates/samples/google_test.py

@@ -8,7 +8,6 @@ class GoogleTests(BaseCase):
     def test_google_dot_com(self):
         self.open("https://google.com/ncr")
         self.type(HomePage.search_box, "github")
-        self.assert_element(HomePage.list_box)
         self.assert_element(HomePage.search_button)
         self.assert_element(HomePage.feeling_lucky_button)
         self.click(HomePage.search_button)

examples/context_scripts.py

@@ -0,0 +1,22 @@
+"""Context Manager Tests"""
+from seleniumbase import SB
+
+with SB(test=True) as sb:
+    sb.open("https://google.com/ncr")
+    sb.type('[name="q"]', "SeleniumBase on GitHub\n")
+    sb.click('a[href*="github.com/seleniumbase"]')
+    sb.highlight("div.Layout-main")
+    sb.highlight("div.Layout-sidebar")
+    sb.sleep(0.5)
+
+with SB(test=True, rtf=True, demo=True) as sb:
+    sb.open("seleniumbase.github.io/demo_page")
+    sb.type("#myTextInput", "This is Automated")
+    sb.assert_text("This is Automated", "#myTextInput")
+    sb.assert_text("This Text is Green", "#pText")
+    sb.click('button:contains("Click Me")')
+    sb.assert_text("This Text is Purple", "#pText")
+    sb.click("#checkBox1")
+    sb.assert_element_not_visible("div#drop2 img#logo")
+    sb.drag_and_drop("img#logo", "div#drop2")
+    sb.assert_element("div#drop2 img#logo")

examples/github_test.py

@@ -10,8 +10,10 @@ def test_github(self):
         self.open("https://github.com/search?q=SeleniumBase")
         self.slow_click('a[href="/seleniumbase/SeleniumBase"]')
         self.click_if_visible('[data-action="click:signup-prompt#dismiss"]')
+        self.highlight("div.Layout-main")
+        self.highlight("div.Layout-sidebar")
         self.assert_element("div.repository-content")
         self.assert_text("SeleniumBase", "strong a")
-        self.slow_click('a[title="seleniumbase"]')
+        self.click('a[title="seleniumbase"]')
         self.slow_click('a[title="fixtures"]')
         self.assert_element('a[title="base_case.py"]')

examples/raw_driver.py

@@ -0,0 +1,22 @@
+from seleniumbase import js_utils
+from seleniumbase import page_actions
+from seleniumbase import Driver
+
+with Driver() as driver:
+    driver.get("https://google.com/ncr")
+    js_utils.highlight_with_js(driver, 'img[alt="Google"]', 6, "")
+
+with Driver() as driver:  # By default, browser="chrome"
+    driver.get("https://seleniumbase.github.io/demo_page")
+    js_utils.highlight_with_js(driver, "h2", 5, "")
+    CSS = "css selector"
+    driver.find_element(CSS, "#myTextInput").send_keys("Automation")
+    driver.find_element(CSS, "#checkBox1").click()
+    js_utils.highlight_with_js(driver, "img", 5, "")
+
+with Driver(browser="chrome", incognito=True) as driver:
+    driver.get("https://seleniumbase.io/apps/calculator")
+    page_actions.wait_for_element_visible(driver, "4", "id").click()
+    page_actions.wait_for_element_visible(driver, "2", "id").click()
+    page_actions.wait_for_text_visible(driver, "42", "output", "id")
+    js_utils.highlight_with_js(driver, "#output", 6, "")

examples/raw_parameter_script.py

@@ -66,6 +66,7 @@
     sb.enable_sync = False
     sb.use_auto_ext = False
     sb.undetectable = False
+    sb.uc_subprocess = False
     sb.no_sandbox = False
     sb.disable_js = False
     sb.disable_gpu = False

examples/raw_sb.py

@@ -0,0 +1,14 @@
+"""Context Manager Tests"""
+from seleniumbase import SB
+
+with SB() as sb:  # By default, browser="chrome" if not set.
+    sb.open("https://seleniumbase.github.io/realworld/login")
+    sb.type("#username", "demo_user")
+    sb.type("#password", "secret_pass")
+    sb.enter_mfa_code("#totpcode", "GAXG2MTEOR3DMMDG")  # 6-digit
+    sb.assert_text("Welcome!", "h1")
+    sb.highlight("img#image1")  # A fancier assert_element() call
+    sb.click('a:contains("This Page")')  # Use :contains() on any tag
+    sb.click_link("Sign out")  # Link must be "a" tag. Not "button".
+    sb.assert_element('a:contains("Sign in")')
+    sb.assert_exact_text("You have been signed out!", "#top_message")

examples/test_mfa_login.py

@@ -9,8 +9,8 @@ def test_mfa_login(self):
         self.enter_mfa_code("#totpcode", "GAXG2MTEOR3DMMDG")  # 6-digit
         self.assert_text("Welcome!", "h1")
         self.highlight("img#image1")  # A fancier assert_element() call
-        self.click('a:contains("This Page")')
+        self.click('a:contains("This Page")')  # Use :contains() on any tag
         self.save_screenshot_to_logs()  # In "./latest_logs/" folder.
-        self.click_link("Sign out")  # Must be "a" tag. Not "button".
+        self.click_link("Sign out")  # Link must be "a" tag. Not "button".
         self.assert_element('a:contains("Sign in")')
         self.assert_exact_text("You have been signed out!", "#top_message")

examples/test_usefixtures.py

@@ -11,8 +11,8 @@ def test_usefixtures_on_class(self):
         sb.enter_mfa_code("#totpcode", "GAXG2MTEOR3DMMDG")  # 6-digit
         sb.assert_text("Welcome!", "h1")
         sb.highlight("img#image1")  # A fancier assert_element() call
-        sb.click('a:contains("This Page")')
+        sb.click('a:contains("This Page")')  # Use :contains() on any tag
         sb.save_screenshot_to_logs()  # In "./latest_logs/" folder.
-        sb.click_link("Sign out")  # Must be "a" tag. Not "button".
+        sb.click_link("Sign out")  # Link must be "a" tag. Not "button".
         sb.assert_element('a:contains("Sign in")')
         sb.assert_exact_text("You have been signed out!", "#top_message")

examples/wordle_test.py

@@ -61,9 +61,8 @@ def skip_if_incorrect_env(self):
     def test_wordle(self):
         self.skip_if_incorrect_env()
         self.open("https://www.nytimes.com/games/wordle/index.html")
-        self.remove_elements("div.ad")
-        self.click('svg[data-testid="icon-close"]')
-        self.remove_elements("div.ad")
+        self.click_if_visible('svg[data-testid="icon-close"]', timeout=2)
+        self.remove_elements("div.place-ad")
         self.initialize_word_list()
         word = random.choice(self.word_list)
         num_attempts = 0