Add functions for reset password workflow

pages/webdriver/sign_in.py

@@ -21,6 +21,8 @@ class SignIn(Base):
     _sign_in_returning_user_locator = (By.ID, 'signInButton')
     _verify_email_locator = (By.ID, 'verify_user')
     _use_another_email_address_locator = (By.ID, 'back')
+    _forgot_password_locator = (By.ID, 'forgotPassword')
+    _reset_password_locator = (By.ID, 'password_reset')
 
     def __init__(self, selenium, timeout, expect='new'):
         Base.__init__(self, selenium, timeout)
@@ -120,6 +122,20 @@ def click_verify_email(self):
             lambda s: s.find_element(
                 *self._use_another_email_address_locator).is_displayed())
 
+    def click_forgot_password(self):
+        """Clicks 'forgot password' link (visible after entering a valid email)"""
+        self.selenium.find_element(*self._forgot_password_locator).click()
+        WebDriverWait(self.selenium, self.timeout).until(
+            lambda s: s.find_element(
+                *self._reset_password_locator).is_displayed())
+
+    def click_reset_password(self):
+        """Clicks 'reset password' after forgot password and new passwords entered"""
+        self.selenium.find_element(*self._reset_password_locator).click()
+        WebDriverWait(self.selenium, self.timeout).until(
+            lambda s: s.find_element(
+                *self._use_another_email_address_locator).is_displayed())
+
     def sign_in(self, email, password):
         """Signs in using the specified email address and password."""
         self.email = email