Update Recorder Mode

Author: mdmintz

seleniumbase/behave/behave_helper.py

@@ -259,6 +259,18 @@ def generate_gherkin(srt_actions):
                 sb_actions.append('%s "%s"' % (method, action[1]))
             else:
                 sb_actions.append("%s '%s'" % (method, action[1]))
+        elif action[0] == "a_url":
+            method = "Assert URL"
+            if '"' not in action[1]:
+                sb_actions.append('%s "%s"' % (method, action[1]))
+            else:
+                sb_actions.append("%s '%s'" % (method, action[1]))
+        elif action[0] == "a_u_c":
+            method = "Assert URL contains"
+            if '"' not in action[1]:
+                sb_actions.append('%s "%s"' % (method, action[1]))
+            else:
+                sb_actions.append("%s '%s'" % (method, action[1]))
         elif action[0] == "as_df":
             method = "Assert downloaded file"
             if '"' not in action[1]:

seleniumbase/fixtures/base_case.py

@@ -4315,6 +4315,8 @@ def __process_recorded_actions(self):
         ext_actions.append("as_lt")
         ext_actions.append("as_ti")
         ext_actions.append("as_tc")
+        ext_actions.append("a_url")
+        ext_actions.append("a_u_c")
         ext_actions.append("as_df")
         ext_actions.append("do_fi")
         ext_actions.append("as_at")
@@ -4669,6 +4671,18 @@ def __process_recorded_actions(self):
                     sb_actions.append('self.%s("%s")' % (method, action[1]))
                 else:
                     sb_actions.append("self.%s('%s')" % (method, action[1]))
+            elif action[0] == "a_url":
+                method = "assert_url"
+                if '"' not in action[1]:
+                    sb_actions.append('self.%s("%s")' % (method, action[1]))
+                else:
+                    sb_actions.append("self.%s('%s')" % (method, action[1]))
+            elif action[0] == "a_u_c":
+                method = "assert_url_contains"
+                if '"' not in action[1]:
+                    sb_actions.append('self.%s("%s")' % (method, action[1]))
+                else:
+                    sb_actions.append("self.%s('%s')" % (method, action[1]))
             elif action[0] == "as_df":
                 method = "assert_downloaded_file"
                 if '"' not in action[1]: