Update example tests

Author: mdmintz

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/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")

examples/test_demo_site.py

@@ -4,9 +4,7 @@
 
 class DocsSiteTests(BaseCase):
     def test_docs(self):
-        self.open("https://seleniumbase.io/examples/ReadMe/")
-        self.assert_text("Running Example Tests", "h1")
-        self.js_click('a[href="../../help_docs/customizing_test_runs/"]')
+        self.open("https://seleniumbase.io/help_docs/customizing_test_runs/")
         self.assert_text("Command Line Options", "h1")
         self.js_click('a[href="../../examples/example_logs/ReadMe/"]')
         self.assert_text("Dashboard / Reports", "h1")

examples/test_docs_site.py

@@ -1,5 +1,6 @@
 """Test drag_and_drop() on different pages."""
 from seleniumbase import BaseCase
+BaseCase.main(__name__, __file__)
 
 
 class DragAndDropTests(BaseCase):
@@ -12,6 +13,7 @@ def test_drag_and_drop(self):
 
     def test_w3schools_drag_and_drop(self):
         self.open("https://seleniumbase.io/w3schools/drag_drop")
+        self.assert_url_contains("drag_drop")
         self.click("button#runbtn")
         self.switch_to_frame("iframeResult")
         self.assert_element_not_visible("#div1 img#drag1")