Dismiss google account suspicious activity

Test accounts sometimes are flagged for suspicious activity which adds a
popup on most workspace pages when the account is logged in. Dismiss
this flag during login if present.

Change-Id: I266bd2d7c2fa2a0f29c49919d9ca29aeac69b533
Reviewed-on: https://chromium-review.googlesource.com/c/crossbench/+/6650142
Commit-Queue: Kameron Lutes <[email protected]>
Reviewed-by: Charles Dick <[email protected]>

Author: kalutes

crossbench/benchmarks/loading/config/login/google.py

@@ -62,6 +62,27 @@
     "source": "js",
 })
 
+SUSPICIOUS_ACTIVITY_URL: str = "https://myaccount.google.com/notifications"
+CHCEK_SUSPICIOUS_ACTIVITY: ClickAction = ClickAction.parse({
+    "action": "click",
+    "pos": {
+        "selector": "[aria-label='Check activity']",
+        "required": True,
+    },
+    "source": "js",
+})
+
+CLICK_YES_IT_WAS_ME: ClickAction = ClickAction.parse({
+    "action": "click",
+    "pos": {
+        "selector": "xpath///button[./span[text()='Yes, it was me']]",
+        "required": True,
+        "wait": True,
+    },
+    "source": "js",
+    "verify": "xpath///body[not(./button[./span[text()='Yes, it was me']])]",
+})
+
 
 class GoogleLogin(PresetLoginBlock):
   """Google-specific login steps."""
@@ -131,10 +152,25 @@ def run_with(self, runner: ActionRunner, run: Run,
           self._dismiss_login_page(action, runner, run, SKIP_HOME_ADDRESS,
                                    ADD_HOME_ADDRESS_REDIRECT, time_left)
 
+      self._clear_suspicious_activity(action, runner, run)
+
+
   def _dismiss_login_page(self, action: Actions, runner: ActionRunner, run: Run,
                           click_action: ClickAction, current_url: str,
                           timeout: dt.timedelta) -> None:
     runner.click(run, click_action)
     action.wait_js_condition(
         f"return !document.URL.startsWith('{current_url}');", 0.2, timeout)
     action.wait_for_ready_state(ReadyState.COMPLETE, timeout)
+
+  def _clear_suspicious_activity(self, action: Actions, runner: ActionRunner,
+                                 run: Run):
+    has_suspicious_activity = action.js(
+        "return document.querySelector("
+        "\"[aria-label='Check activity']\") != null;")
+
+    if not has_suspicious_activity:
+      return
+
+    runner.click(run, CHCEK_SUSPICIOUS_ACTIVITY)
+    runner.click(run, CLICK_YES_IT_WAS_ME)

tests/crossbench/benchmarks/loading/config/test_login.py

@@ -81,6 +81,8 @@ def expect_successful_google_login(self):
     self.browser.expect_js(result=True)
     # Return successful login URL
     self.browser.expect_js(result="https://myaccount.google.com")
+    # Check for suspicious activity
+    self.browser.expect_js(result=False)
 
   def test_google_account(self):
     config = PagesConfig.parse(self._CONFIG_DATA)
@@ -111,7 +113,7 @@ def test_logged_in_non_google_account(self):
     self.run.story_secrets = page[0].secrets
     config.pages[0].login.run_with(self.action_runner, self.run, page[0])
 
-  def test_account_maintenance_redirects_dismissed(self):
+  def test_full_account_maintenance_flow(self):
     config = PagesConfig.parse(self._CONFIG_DATA)
     page = LoadingPageFilter.stories_from_config(self.mock_args(), config)
 
@@ -168,6 +170,18 @@ def test_account_maintenance_redirects_dismissed(self):
     # Return successful login URL
     self.browser.expect_js(result="https://myaccount.google.com")
 
+    # Return suspicious activity is present
+    self.browser.expect_js(result=True)
+    # Click suspicious activity button
+    self.browser.expect_js(result=1)
+    # Wait for 'yes' button
+    self.browser.expect_js(result=1)
+    # Click 'yes' button
+    self.browser.expect_js(result=1)
+    # Wait 'yes' button not present.
+    self.browser.expect_js(result=1)
+
+
     self.run.story_secrets = page[0].secrets
     config.pages[0].login.run_with(self.action_runner, self.run, page[0])
 

tests/crossbench/benchmarks/loading/test_loading.py

@@ -511,6 +511,8 @@ def setup_expected_google_login_js(self):
         JsInvocation(True),
         # Return successful login URL
         JsInvocation("https://myaccount.google.com", re.compile(r".*URL.*")),
+        # No suspicious activity
+        JsInvocation(False),
     ]
     for browser in self.browsers:
       for script in expected_scripts: