test: add mindmap collapse test (#9336)

toeverything · Dec 26, 2024 · 0de4f7a · 0de4f7a
1 parent 1afdb68
commit 0de4f7a
Show file tree

Hide file tree

Showing 5 changed files with 485 additions and 5 deletions.
diff --git a/blocksuite/affine/block-surface/src/view/mindmap.ts b/blocksuite/affine/block-surface/src/view/mindmap.ts
@@ -25,7 +25,12 @@ export class MindMapView extends GfxElementModelView<MindmapElementModel> {
     }
   >();
 
-  private _getCollapseButton(node: MindmapNode | string) {
+  /**
+   *
+   * @param node The mindmap node or its id to get the collapse button
+   * @returns
+   */
+  getCollapseButton(node: MindmapNode | string) {
     const id = typeof node === 'string' ? node : node.id;
     return this._collapseButtons.get(`collapse-btn-${id}`);
   }
@@ -149,7 +154,7 @@ export class MindMapView extends GfxElementModelView<MindmapElementModel> {
           elm?.id &&
           this.model.children.has(elm.id)
         ) {
-          const button = this._getCollapseButton(elm.id);
+          const button = this.getCollapseButton(elm.id);
 
           if (!button) {
             return;
@@ -164,7 +169,7 @@ export class MindMapView extends GfxElementModelView<MindmapElementModel> {
 
   private _updateButtonVisibility(node: string) {
     const latestNode = this.model.getNode(node);
-    const buttonModel = this._getCollapseButton(node);
+    const buttonModel = this.getCollapseButton(node);
 
     if (!buttonModel) {
       return;

diff --git a/blocksuite/presets/package.json b/blocksuite/presets/package.json
@@ -4,7 +4,7 @@
   "type": "module",
   "scripts": {
     "build": "tsc",
-    "test:unit": "nx vite:test --browser.headless --run",
+    "test:unit": "vitest --browser.headless --run",
     "test:debug": "PWDEBUG=1 npx vitest"
   },
   "sideEffects": false,
@@ -37,5 +37,8 @@
     "themes",
     "!src/__tests__",
     "!dist/__tests__"
-  ]
+  ],
+  "devDependencies": {
+    "vitest": "^2.1.8"
+  }
 }