Add close_all_tabs action

Add a new action type 'close_all_tabs' that will iterate through all
active tabs and close them.

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

Author: kalutes

crossbench/action_runner/action/action_type.py

@@ -29,6 +29,7 @@ class ActionType(ConfigEnum):
       "Only supported in chromium-based browsers."))
   SCREENSHOT = ("screenshot", "Take a screenshot")
   SWITCH_TAB = ("switch_tab", "Switch the tab that actions are sent to")
+  CLOSE_ALL_TABS = ("close_all_tabs", "Close all tabs")
   CLOSE_TAB = ("close_tab", "Close the specified tab")
   WAIT_FOR_DOWNLOAD = ("wait_for_download", "wait for a download to complete")
   WAIT_FOR_READY_STATE = ("wait_for_ready_state",

crossbench/action_runner/action/all.py

@@ -9,6 +9,7 @@
 from crossbench.action_runner.action.action import ACTIONS, Action
 from crossbench.action_runner.action.click import ClickAction
 from crossbench.action_runner.action.close_tab import CloseTabAction
+from crossbench.action_runner.action.close_all_tabs import CloseAllTabsAction
 from crossbench.action_runner.action.dump_html import DumpHtmlAction
 from crossbench.action_runner.action.get import GetAction
 from crossbench.action_runner.action.inject_new_document_script import \
@@ -35,6 +36,7 @@
 ACTIONS_TUPLE: tuple[Type[Action], ...] = (
     ClickAction,
     CloseTabAction,
+    CloseAllTabsAction,
     DumpHtmlAction,
     GetAction,
     InjectNewDocumentScriptAction,

crossbench/action_runner/action/close_all_tabs.py

@@ -0,0 +1,24 @@
+# Copyright 2025 The Chromium Authors
+# Use of this source code is governed by a BSD-style license that can be
+# found in the LICENSE file.
+
+from __future__ import annotations
+
+from typing import TYPE_CHECKING
+
+from typing_extensions import override
+
+from crossbench.action_runner.action.action import Action
+from crossbench.action_runner.action.action_type import ActionType
+
+if TYPE_CHECKING:
+  from crossbench.action_runner.base import ActionRunner
+  from crossbench.runner.run import Run
+
+
+class CloseAllTabsAction(Action):
+  TYPE: ActionType = ActionType.CLOSE_ALL_TABS
+
+  @override
+  def run_with(self, run: Run, action_runner: ActionRunner) -> None:
+    action_runner.close_all_tabs(run, self)

crossbench/action_runner/base.py

@@ -308,5 +308,8 @@ def switch_tab(self, run: Run, action: i_action.SwitchTabAction):
   def close_tab(self, run: Run, action: i_action.CloseTabAction):
     raise ActionNotImplementedError(self, action)
 
+  def close_all_tabs(self, run: Run, action: i_action.CloseAllTabsAction):
+    raise ActionNotImplementedError(self, action)
+
   def wait_for_download(self, run: Run, action: i_action.WaitForDownloadAction):
     raise ActionNotImplementedError(self, action)

crossbench/action_runner/default_action_runner.py

@@ -295,6 +295,13 @@ def close_tab(self, run: Run, action: i_action.CloseTabAction) -> None:
       run.browser.close_tab(action.title, action.url, action.tab_index,
                             action.relative_tab_index, action.timeout)
 
+  @override
+  def close_all_tabs(self, run: Run,
+                     action: i_action.CloseAllTabsAction) -> None:
+    del action
+    with run.actions("CloseAllTabsAction", measure=False):
+      run.browser.close_all_tabs()
+
   def _get_scroll_field(self, has_selector: bool) -> str:
     if has_selector:
       return "scrollTop"

crossbench/browsers/browser.py

@@ -441,6 +441,9 @@ def close_tab(
     del timeout
     raise NotImplementedError(f"Closing tabs is not supported by {self}")
 
+  def close_all_tabs(self) -> None:
+    raise NotImplementedError(f"Closing all tabs is not supported by {self}")
+
   @property
   def current_url(self) -> str:
     raise NotImplementedError(f"Getting current url is not supported by {self}")

crossbench/browsers/chromium_based/webdriver.py

@@ -299,6 +299,22 @@ def close_tab(
       # to switching to the first tab.
       driver.switch_to.window(driver.window_handles[0])
 
+  @override
+  def close_all_tabs(self) -> None:
+    driver = self._private_driver
+    current_handle = driver.current_window_handle
+
+    for handle in driver.window_handles:
+      driver.switch_to.window(handle)
+      if handle != current_handle:
+        driver.close()
+
+    # Closing every tab will cause the browser to exit.
+    # As a workaround navigate the final tab to about:blank.
+    driver.switch_to.window(current_handle)
+    self.show_url("about:blank")
+
+
   @property
   def current_url(self) -> str:
     return self._private_driver.current_url

tests/crossbench/benchmarks/loading/test_action.py

@@ -11,6 +11,7 @@
                                                     Action)
 from crossbench.action_runner.action.action_type import ActionType
 from crossbench.action_runner.action.click import ClickAction
+from crossbench.action_runner.action.close_all_tabs import CloseAllTabsAction
 from crossbench.action_runner.action.close_tab import CloseTabAction
 from crossbench.action_runner.action.enums import ReadyState, WindowTarget
 from crossbench.action_runner.action.get import GetAction
@@ -896,6 +897,13 @@ def test_parse_switch_tab_only_relative_tab_index(self):
 
     self.assertEqual(action.relative_tab_index, 17)
 
+  def test_parse_close_all_tabs(self):
+    config_dict = {"action": "close_all_tabs"}
+
+    action = CloseAllTabsAction.parse(config_dict)
+
+    self.assertEqual(action.TYPE, ActionType.CLOSE_ALL_TABS)
+
   def test_parse_close_tab_all_args(self):
     config_dict = {
         "action": "close_tab",