Merge pull request #1714 from seleniumbase/choose-binary-location-and-more

Choose binary location and more

Author: mdmintz

README.md

@@ -524,6 +524,7 @@ pytest my_first_test.py --pdb
 --firefox-pref=SET  # (Set a Firefox preference:value set, comma-separated.)
 --extension-zip=ZIP  # (Load a Chrome Extension .zip|.crx, comma-separated.)
 --extension-dir=DIR  # (Load a Chrome Extension directory, comma-separated.)
+--binary-location=PATH  # (Set path of the Chromium browser binary to use.)
 --sjw  # (Skip JS Waits for readyState to be "complete" or Angular to load.)
 --pls=PLS  # (Set pageLoadStrategy on Chrome: "normal", "eager", or "none".)
 --headless  # (Run tests in headless mode. The default arg on Linux OS.)

examples/coffee_cart_tests.py

@@ -65,6 +65,7 @@ def test_remove_added_coffee(self):
         self.open("https://seleniumbase.io/coffee/")
         self.assert_exact_text("cart (0)", 'a[aria-label="Cart page"]')
         self.assert_exact_text("Total: $0.00", "button.pay")
+        self.wait_for_element('div[class="cup-body"]')
         self.click_visible_elements('div[class="cup-body"]', limit=6)
         self.assert_exact_text("cart (6)", 'a[aria-label="Cart page"]')
         self.assert_exact_text("Total: $74.00", 'button[data-test="checkout"]')

examples/offline_examples/handle_alert_test.py

@@ -1,5 +1,6 @@
 import pytest
 from seleniumbase import BaseCase
+BaseCase.main(__name__, __file__)
 
 
 @pytest.mark.offline  # Can be run with: "pytest -m offline"

examples/offline_examples/load_html_test.py

@@ -1,5 +1,6 @@
 import pytest
 from seleniumbase import BaseCase
+BaseCase.main(__name__, __file__)
 
 
 @pytest.mark.offline  # Can be run with: "pytest -m offline"

examples/offline_examples/test_demo_page.py

@@ -1,6 +1,7 @@
 import os
 import pytest
 from seleniumbase import BaseCase
+BaseCase.main(__name__, __file__)
 
 
 @pytest.mark.offline  # Can be run with: "pytest -m offline"

examples/offline_examples/user_agent_test.py

@@ -1,5 +1,6 @@
 import pytest
 from seleniumbase import BaseCase
+BaseCase.main(__name__, __file__)
 
 
 @pytest.mark.offline  # Can be run with: "pytest -m offline"
@@ -8,3 +9,31 @@ def test_get_user_agent(self):
         self.open("data:,")
         user_agent = self.get_user_agent()
         print('\nUser Agent = "%s"' % user_agent)
+
+        # Now change the user-agent using "execute_cdp_cmd()"
+        if not self.is_chromium():
+            msg = "\n* execute_cdp_cmd() is only for Chromium browsers"
+            print(msg)
+            self.skip(msg)
+        print("\n--------------------------")
+        try:
+            self.driver.execute_cdp_cmd(
+                "Network.setUserAgentOverride",
+                {
+                    "userAgent": "Mozilla/5.0 "
+                    "(Nintendo Switch; WifiWebAuthApplet) "
+                    "AppleWebKit/606.4 (KHTML, like Gecko) "
+                    "NF/6.0.1.15.4 NintendoBrowser/5.1.0.20393"
+                },
+            )
+            new_user_agent = self.get_user_agent()
+            print('\nOverrided User Agent = "%s"' % new_user_agent)
+        finally:
+            # Reset the user-agent back to the original
+            self.driver.execute_cdp_cmd(
+                "Network.setUserAgentOverride",
+                {"userAgent": user_agent},
+            )
+        print("\n--------------------------")
+        user_agent = self.get_user_agent()
+        print('\nUser Agent = "%s"' % user_agent)

examples/raw_parameter_script.py

@@ -102,6 +102,7 @@
     sb.dashboard = False
     sb._dash_initialized = False
     sb.message_duration = None
+    sb.binary_location = None
     sb.enable_3d_apis = False
     sb.block_images = False
     sb.do_not_track = False

examples/test_agent.py

@@ -10,6 +10,7 @@
 class CoffeeCartTest(BaseCase):
     def test_coffee_cart(self):
         self.open("https://seleniumbase.io/coffee/")
+        self.assert_title("Coffee Cart")
         self.click('div[data-sb="Cappuccino"]')
         self.click('div[data-sb="Flat-White"]')
         self.click('div[data-sb="Cafe-Latte"]')

examples/test_coffee_cart.py

@@ -113,6 +113,6 @@ def test_demo_site(self):
         self.highlight("h2")
 
         # Actions with Demo Mode enabled
-        self.demo_mode = True
+        self.activate_demo_mode()
         self.type("input", "Have a Nice Day!")
         self.assert_text("SeleniumBase", "h2")