chore: fix tests

Author: lukasbach

packages/core/src/features/checkboxes/checkboxes.spec.ts

@@ -102,6 +102,7 @@ describe("core-feature/checkboxes", () => {
     it("should turn folder checked if all children are checked", async () => {
       const testTree = await tree
         .with({
+          isItemFolder: (item: any) => item.getItemData().length < 4,
           propagateCheckedState: true,
           canCheckFolders: false,
         })

packages/core/src/features/tree/tree.spec.ts

@@ -327,15 +327,22 @@ describe("core-feature/tree", () => {
         expect(tree.instance.getItemInstance("x1").isFolder()).toBe(true);
       });
 
-      it("returns correct isFolder for hidden items", () => {
+      it("returns correct isFolder for hidden items", async () => {
         if (title.toLocaleLowerCase().includes("async")) {
           // async test tree defaults to "loading" item names
           return;
         }
+        const testTree = await tree
+          .with({
+            isItemFolder: (item: any) => item.getItemData().length < 4,
+          })
+          .createTestCaseTree();
 
         // Reference: https://github.com/lukasbach/headless-tree/issues/166
-        expect(tree.instance.getItemInstance("x44").isFolder()).toBe(true);
-        expect(tree.instance.getItemInstance("x444").isFolder()).toBe(false);
+        expect(testTree.instance.getItemInstance("x44").isFolder()).toBe(true);
+        expect(testTree.instance.getItemInstance("x444").isFolder()).toBe(
+          false,
+        );
       });
 
       it("returns isFolder=true for root item", () => {

packages/core/src/test-utils/test-tree.ts

@@ -122,7 +122,7 @@ export class TestTree<T = string> {
       },
       getItemName: (item) => item.getItemData(),
       indent: 20,
-      isItemFolder: (item) => item.getItemData().length < 4,
+      isItemFolder: (item) => item.getItemMeta().level < 2,
       initialState: {
         expandedItems: ["x1", "x11"],
       },