Merge pull request #1681 from seleniumbase/even-more-bug-fixes

Even more bug fixes

Author: mdmintz

examples/github_test.py

@@ -1,4 +1,5 @@
 from seleniumbase import BaseCase
+BaseCase.main(__name__, __file__)
 
 
 class GitHubTests(BaseCase):

examples/hack_the_planet.py

@@ -1,5 +1,6 @@
 """ Video Link: https://youtu.be/1s-Tj65AKZA """
 from seleniumbase import BaseCase
+BaseCase.main(__name__, __file__)
 
 
 class HackTests(BaseCase):

examples/proxy_test.py

@@ -1,8 +1,14 @@
+from seleniumbase.config import settings
 from seleniumbase import BaseCase
+BaseCase.main(__name__, __file__)
 
 
 class ProxyTests(BaseCase):
     def test_proxy(self):
+        settings.SKIP_JS_WAITS = True
+        if not self.page_load_strategy == "none":
+            # This page takes too long to load otherwise
+            self.get_new_driver(page_load_strategy="none")
         self.open("https://ipinfo.io/")
         ip_address = self.get_text('#ip-string span[class*="primary"] span')
         print("\n\nMy IP Address = %s\n" % ip_address)

examples/raw_browser_launcher.py

@@ -16,10 +16,12 @@
 
 # Example 2 using default args or command-line options
 driver = Driver()
-driver.get("https://seleniumbase.github.io/demo_page")
-js_utils.highlight_with_js(driver, "h2", loops=5)
-by_css = "css selector"
-driver.find_element(by_css, "#myTextInput").send_keys("Automation")
-driver.find_element(by_css, "#checkBox1").click()
-js_utils.highlight_with_js(driver, "img", loops=5)
-driver.quit()  # If the script fails early, the driver still quits
+try:
+    driver.get("https://seleniumbase.github.io/demo_page")
+    js_utils.highlight_with_js(driver, "h2", loops=5)
+    by_css = "css selector"
+    driver.find_element(by_css, "#myTextInput").send_keys("Automation")
+    driver.find_element(by_css, "#checkBox1").click()
+    js_utils.highlight_with_js(driver, "img", loops=5)
+finally:
+    driver.quit()

examples/raw_driver_context.py

@@ -8,17 +8,17 @@
     driver.get("https://seleniumbase.github.io/")
     js_utils.highlight_with_js(driver, 'img[alt="SeleniumBase"]', loops=6)
 
+with DriverContext(browser="chrome", incognito=True) as driver:
+    driver.get("https://seleniumbase.io/apps/calculator")
+    page_actions.wait_for_element(driver, "4", by="id").click()
+    page_actions.wait_for_element(driver, "2", by="id").click()
+    page_actions.wait_for_text(driver, "42", "output", by="id")
+    js_utils.highlight_with_js(driver, "#output", loops=6)
+
 with DriverContext() as driver:
     driver.get("https://seleniumbase.github.io/demo_page")
     js_utils.highlight_with_js(driver, "h2", loops=5)
     by_css = "css selector"
     driver.find_element(by_css, "#myTextInput").send_keys("Automation")
     driver.find_element(by_css, "#checkBox1").click()
     js_utils.highlight_with_js(driver, "img", loops=5)
-
-with DriverContext(browser="chrome", incognito=True) as driver:
-    driver.get("https://seleniumbase.io/apps/calculator")
-    page_actions.wait_for_element(driver, "4", by="id").click()
-    page_actions.wait_for_element(driver, "2", by="id").click()
-    page_actions.wait_for_text(driver, "42", "output", by="id")
-    js_utils.highlight_with_js(driver, "#output", loops=6)

examples/test_docs_site.py

@@ -1,4 +1,5 @@
 from seleniumbase import BaseCase
+BaseCase.main(__name__, __file__)
 
 
 class DocsSiteTests(BaseCase):

examples/test_select_options.py

@@ -1,4 +1,5 @@
 from seleniumbase import BaseCase
+BaseCase.main(__name__, __file__)
 
 
 class SelectTestClass(BaseCase):

examples/test_show_file_choosers.py

@@ -4,6 +4,7 @@
 """
 import os
 from seleniumbase import BaseCase
+BaseCase.main(__name__, __file__)
 
 
 class FileUpload(BaseCase):

examples/test_tinymce.py

@@ -1,4 +1,5 @@
 from seleniumbase import BaseCase
+BaseCase.main(__name__, __file__)
 
 
 class TinyMceTests(BaseCase):

examples/test_todomvc.py

@@ -1,5 +1,6 @@
 from parameterized import parameterized
 from seleniumbase import BaseCase
+BaseCase.main(__name__, __file__)
 
 
 class TodoMVC(BaseCase):