Make updates to Multi-Factor Authentication methods

Author: mdmintz

seleniumbase/fixtures/base_case.py

@@ -5278,9 +5278,10 @@ def is_chromedriver_too_old(self):
             return True  # chromedriver is too old! Please upgrade!
         return False
 
-    def get_totp_code(self, totp_key=None):
-        """Returns a time-based one-time password based on the
-        Google Authenticator algorithm. Works with Authy and Okta.
+    def get_mfa_code(self, totp_key=None):
+        """Same as get_totp_code() and get_google_auth_password().
+        Returns a time-based one-time password based on the
+        Google Authenticator algorithm for multi-factor authentication.
         If the "totp_key" is not specified, this method defaults
         to using the one provided in [seleniumbase/config/settings.py].
         Google Authenticator codes expire & change at 30-sec intervals.
@@ -5307,6 +5308,36 @@ def get_totp_code(self, totp_key=None):
         totp = pyotp.TOTP(totp_key)
         return str(totp.now())
 
+    def enter_mfa_code(
+        self, selector, totp_key=None, by=By.CSS_SELECTOR, timeout=None
+    ):
+        """Enters into the field a Multi-Factor Authentication TOTP Code.
+        If the "totp_key" is not specified, this method defaults
+        to using the one provided in [seleniumbase/config/settings.py].
+        The TOTP code is generated by the Google Authenticator Algorithm.
+        This method will automatically press ENTER after typing the code."""
+        self.__check_scope()
+        if not timeout:
+            timeout = settings.SMALL_TIMEOUT
+        self.wait_for_element_visible(selector, by=by, timeout=timeout)
+        if self.recorder_mode:
+            css_selector = self.convert_to_css_selector(selector, by=by)
+            url = self.get_current_url()
+            if url and len(url) > 0:
+                if ("http:") in url or ("https:") in url or ("file:") in url:
+                    origin = self.get_origin()
+                    if self.get_session_storage_item("pause_recorder") == "no":
+                        time_stamp = self.execute_script("return Date.now();")
+                        sel_key = [css_selector, totp_key]
+                        action = ["e_mfa", sel_key, origin, time_stamp]
+                        self.__extra_actions.append(action)
+                    # Sometimes Sign-In leaves the origin... Save work first.
+                    self.__origins_to_save.append(origin)
+                    tab_actions = self.__get_recorded_actions_on_active_tab()
+                    self.__actions_to_save.append(tab_actions)
+        mfa_code = self.get_mfa_code(totp_key)
+        self.update_text(selector, mfa_code + "\n", by=by, timeout=timeout)
+
     def convert_css_to_xpath(self, css):
         return css_to_xpath.convert_css_to_xpath(css)
 
@@ -6247,12 +6278,24 @@ def assert_element_not_present(
         return True
 
     def get_google_auth_password(self, totp_key=None):
-        """ Same as self.get_totp_code() """
-        return self.get_totp_code(totp_key=totp_key)
+        """ Same as self.get_mfa_code() """
+        return self.get_mfa_code(totp_key=totp_key)
 
     def get_google_auth_code(self, totp_key=None):
-        """ Same as self.get_totp_code() """
-        return self.get_totp_code(totp_key=totp_key)
+        """ Same as self.get_mfa_code() """
+        return self.get_mfa_code(totp_key=totp_key)
+
+    def get_totp_code(self, totp_key=None):
+        """ Same as self.get_mfa_code() """
+        return self.get_mfa_code(totp_key=totp_key)
+
+    def enter_totp_code(
+        self, selector, totp_key=None, by=By.CSS_SELECTOR, timeout=None
+    ):
+        """ Same as self.enter_mfa_code() """
+        return self.enter_mfa_code(
+            selector=selector, totp_key=totp_key, by=by, timeout=timeout
+        )
 
     def assert_no_broken_links(self, multithreaded=True):
         """ Same as self.assert_no_404_errors() """