Add methods for URL asserts

Author: mdmintz

help_docs/method_summary.md

@@ -467,6 +467,10 @@ self.assert_title(title)
 
 self.assert_title_contains(substring)
 
+self.assert_url(url)
+
+self.assert_url_contains(substring)
+
 self.assert_no_js_errors(exclude=[])
 
 self.inspect_html()

seleniumbase/fixtures/base_case.py

@@ -6737,6 +6737,85 @@ def assert_title_contains(self, substring):
                         self.__extra_actions.append(action)
         return True
 
+    def assert_url(self, url):
+        """Asserts that the web page URL matches the expected URL."""
+        self.wait_for_ready_state_complete()
+        expected = url.strip()
+        actual = self.get_current_url().strip()
+        error = "Expected URL [%s] does not match the actual URL [%s]!"
+        try:
+            self.assertEqual(expected, actual, error % (expected, actual))
+        except Exception:
+            self.wait_for_ready_state_complete()
+            time.sleep(2)
+            actual = self.get_current_url().strip()
+            try:
+                self.assertEqual(expected, actual, error % (expected, actual))
+            except Exception:
+                self.wait_for_ready_state_complete()
+                time.sleep(2)
+                actual = self.get_current_url().strip()
+                self.assertEqual(expected, actual, error % (expected, actual))
+        if self.demo_mode and not self.recorder_mode:
+            a_u = "ASSERT URL"
+            if self._language != "English":
+                from seleniumbase.fixtures.words import SD
+
+                a_u = SD.translate_assert_url(self._language)
+            messenger_post = "<b>%s</b>: {%s}" % (a_u, expected)
+            self.__highlight_with_assert_success(messenger_post, "html")
+        if self.recorder_mode:
+            url = self.get_current_url()
+            if url and len(url) > 0:
+                if ("http:") in url or ("https:") in url or ("file:") in url:
+                    if self.get_session_storage_item("pause_recorder") == "no":
+                        time_stamp = self.execute_script("return Date.now();")
+                        origin = self.get_origin()
+                        action = ["a_url", expected, origin, time_stamp]
+                        self.__extra_actions.append(action)
+        return True
+
+    def assert_url_contains(self, substring):
+        """Asserts that the URL substring appears in the full URL."""
+        self.wait_for_ready_state_complete()
+        expected = substring.strip()
+        actual = self.get_current_url().strip()
+        error = (
+            "Expected URL substring [%s] does not appear "
+            "in the full URL [%s]!"
+        )
+        try:
+            self.assertIn(expected, actual, error % (expected, actual))
+        except Exception:
+            self.wait_for_ready_state_complete()
+            time.sleep(2)
+            actual = self.get_current_url().strip()
+            try:
+                self.assertIn(expected, actual, error % (expected, actual))
+            except Exception:
+                self.wait_for_ready_state_complete()
+                time.sleep(2)
+                actual = self.get_current_url().strip()
+                self.assertIn(expected, actual, error % (expected, actual))
+        if self.demo_mode and not self.recorder_mode:
+            a_u = "ASSERT URL CONTAINS"
+            if self._language != "English":
+                from seleniumbase.fixtures.words import SD
+
+                a_u = SD.translate_assert_url_contains(self._language)
+            messenger_post = "<b>%s</b>: {%s}" % (a_u, expected)
+            self.__highlight_with_assert_success(messenger_post, "html")
+        if self.recorder_mode:
+            url = self.get_current_url()
+            if url and len(url) > 0:
+                if ("http:") in url or ("https:") in url or ("file:") in url:
+                    if self.get_session_storage_item("pause_recorder") == "no":
+                        time_stamp = self.execute_script("return Date.now();")
+                        origin = self.get_origin()
+                        action = ["a_u_c", expected, origin, time_stamp]
+                        self.__extra_actions.append(action)
+        return True
+
     def assert_no_js_errors(self, exclude=[]):
         """Asserts current URL has no "SEVERE"-level JavaScript errors.
         Works ONLY on Chromium browsers (Chrome or Edge).