Add collapse all functionality to test coverage view (microsoft#258906)

- ✨ Implement collapseAll method in TestCoverageView
- ✨ Add collapseAll method in TestCoverageTree
- 🆕 Register Collapse All action in the testing menu

Author: timheuer

src/vs/workbench/contrib/testing/browser/testCoverageView.ts

@@ -102,6 +102,10 @@ export class TestCoverageView extends ViewPane {
 		super.layoutBody(height, width);
 		this.tree.value?.layout(height, width);
 	}
+
+	public collapseAll(): void {
+		this.tree.value?.collapseAll();
+	}
 }
 
 let fnNodeId = 0;
@@ -381,6 +385,10 @@ class TestCoverageTree extends Disposable {
 		this.tree.layout(height, width);
 	}
 
+	public collapseAll() {
+		this.tree.collapseAll();
+	}
+
 	private updateWithDetails(el: IPrefixTreeNode<FileCoverage>, details: readonly CoverageDetails[]) {
 		if (!this.tree.hasElement(el)) {
 			return; // avoid any issues if the tree changes in the meanwhile
@@ -753,6 +761,7 @@ registerAction2(class TestCoverageChangeSortingAction extends ViewAction<TestCov
 				id: MenuId.ViewTitle,
 				when: ContextKeyExpr.equals('view', Testing.CoverageViewId),
 				group: 'navigation',
+				order: 1,
 			}
 		});
 	}
@@ -781,3 +790,24 @@ registerAction2(class TestCoverageChangeSortingAction extends ViewAction<TestCov
 		}));
 	}
 });
+
+registerAction2(class TestCoverageCollapseAllAction extends ViewAction<TestCoverageView> {
+	constructor() {
+		super({
+			id: TestCommandId.CoverageViewCollapseAll,
+			viewId: Testing.CoverageViewId,
+			title: localize2('testing.coverageCollapseAll', 'Collapse All Coverage'),
+			icon: Codicon.collapseAll,
+			menu: {
+				id: MenuId.ViewTitle,
+				when: ContextKeyExpr.equals('view', Testing.CoverageViewId),
+				group: 'navigation',
+				order: 2,
+			}
+		});
+	}
+
+	override runInView(_accessor: ServicesAccessor, view: TestCoverageView) {
+		view.collapseAll();
+	}
+});

src/vs/workbench/contrib/testing/common/constants.ts

@@ -72,6 +72,7 @@ export const enum TestCommandId {
 	CoverageToggleInExplorer = 'testing.toggleCoverageInExplorer',
 	CoverageToggleToolbar = 'testing.coverageToggleToolbar',
 	CoverageViewChangeSorting = 'testing.coverageViewChangeSorting',
+	CoverageViewCollapseAll = 'testing.coverageViewCollapseAll',
 	DebugAction = 'testing.debug',
 	DebugAllAction = 'testing.debugAll',
 	DebugAtCursor = 'testing.debugAtCursor',