Update example tests

Author: mdmintz

examples/test_agent.py

@@ -4,8 +4,41 @@
 class UserAgentTests(BaseCase):
     def test_user_agent(self):
         self.open("http://whatsmyuseragent.org/")
-        user_agent = self.get_text(".user-agent p")
-        print("\n\nUser-Agent:\n%s\n" % user_agent)
+        user_agent_detected = self.get_text(".user-agent p")
+        original_user_agent = user_agent_detected
+        if not self.user_agent:
+            # Using the built-in user-agent string
+            print("\n\nUser-Agent:\n%s\n" % user_agent_detected)
+        else:
+            # User-agent was overridden using: --agent=STRING
+            print("\n\nUser-Agent override:\n%s\n" % user_agent_detected)
         print(self.get_text(".ip-address p"))
-        print("\nThe browser will close automatically in 7 seconds...")
-        self.sleep(7)
+        self.sleep(3)
+
+        # 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"
+                }
+            )
+            self.open("http://whatsmyuseragent.org/")
+            user_agent_detected = self.get_text(".user-agent p")
+            print("\nUser-Agent (after override):\n%s\n" % user_agent_detected)
+            print(self.get_text(".ip-address p"))
+            self.sleep(3)
+        finally:
+            # Reset the user-agent back to the original
+            self.driver.execute_cdp_cmd(
+                "Network.setUserAgentOverride", {
+                    "userAgent": original_user_agent
+                }
+            )

examples/test_demo_site.py

@@ -31,7 +31,7 @@ def test_demo_site(self):
 
         # Verify that a button click changes text on the page
         self.assert_text("This Text is Green", "#pText")
-        self.click("#myButton")
+        self.click('button:contains("Click Me")')
         self.assert_text("This Text is Purple", "#pText")
 
         # Assert that the given SVG is visible on the page

examples/user_agent_test.py

@@ -4,9 +4,39 @@
 class UserAgentTests(BaseCase):
     def test_user_agent(self):
         self.open("https://www.whatsmyua.info/")
-        user_agent = self.get_text("#custom-ua-string")
-        print("\n\nUser-Agent = %s\n" % user_agent)
-        print("Displaying User-Agent Info:")
-        print(self.get_text("#useragent"))
-        print("\nThe browser will close automatically in 7 seconds...")
-        self.sleep(7)
+        user_agent_detected = self.get_text("#custom-ua-string")
+        original_user_agent = user_agent_detected
+        if not self.user_agent:
+            # Using the built-in user-agent string
+            print("\n\nUser-Agent = %s\n" % user_agent_detected)
+        else:
+            # User-agent was overridden using: --agent=STRING
+            print("\n\nUser-Agent override = %s\n" % user_agent_detected)
+        self.sleep(3)
+
+        try:
+            # 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("--------------------------")
+            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"
+                }
+            )
+            self.open("https://www.whatsmyua.info/")
+            user_agent_detected = self.get_text("#custom-ua-string")
+            print("\nUser-Agent (after override) = %s" % user_agent_detected)
+            self.sleep(3)
+        finally:
+            # Reset the user-agent back to the original
+            self.driver.execute_cdp_cmd(
+                "Network.setUserAgentOverride", {
+                    "userAgent": original_user_agent
+                }
+            )