@headless-tree/core 1.4.0 → 1.5.1

package/src/test-utils/test-tree-do.ts

@@ -135,4 +135,10 @@ export class TestTreeDo<T> {
       "function called with inconsistent parameters",
     ).toBeOneOf([0, 1]);
   }
+
+  async awaitNextTick() {
+    await new Promise((r) => {
+      setTimeout(r);
+    });
+  }
 }

package/src/test-utils/test-tree.ts

@@ -63,21 +63,22 @@ export class TestTree<T = string> {
     }),
   };
 
-  forSuits(runSuite: (tree: TestTree<T>) => void) {
+  forSuits(runSuite: (tree: TestTree<T>, title: string) => void) {
     describe.for([
       this.suits.sync(),
       this.suits.async(),
       this.suits.proxifiedSync(),
       this.suits.proxifiedAsync(),
-    ])("$title", ({ tree }) => {
+    ])("$title", ({ tree, title }) => {
       tree.resetBeforeEach();
-      runSuite(tree);
+      runSuite(tree, title);
     });
   }
 
   get instance() {
     if (!this.treeInstance) {
       this.treeInstance = createTree(this.config);
+      this.treeInstance.setMounted(true);
       this.treeInstance.rebuildTree();
     }
     return this.treeInstance;
@@ -87,10 +88,9 @@ export class TestTree<T = string> {
 
   static async resolveAsyncLoaders() {
     do {
+      await vi.advanceTimersToNextTimerAsync();
       TestTree.asyncLoaderResolvers.shift()?.();
-      await new Promise<void>((r) => {
-        setTimeout(r);
-      });
+      await vi.advanceTimersToNextTimerAsync();
     } while (TestTree.asyncLoaderResolvers.length);
   }
 
@@ -101,6 +101,17 @@ export class TestTree<T = string> {
     await TestTree.resolveAsyncLoaders();
   }
 
+  async runWhileResolvingItems(cb: () => Promise<void>) {
+    const interval = setInterval(() => {
+      TestTree.resolveAsyncLoaders();
+    }, 5);
+    try {
+      await cb();
+    } finally {
+      clearInterval(interval);
+    }
+  }
+
   static default(config: Partial<TreeConfig<string>>) {
     return new TestTree({
       rootItemId: "x",

package/src/types/core.ts

@@ -93,11 +93,11 @@ type MayReturnNull<T extends (...x: any[]) => any> = (
   ...args: Parameters<T>
 ) => ReturnType<T> | null;
 
-export type ItemInstanceOpts<Key extends keyof ItemInstance<any>> = {
-  item: ItemInstance<any>;
-  tree: TreeInstance<any>;
+export type ItemInstanceOpts<T, Key extends keyof ItemInstance<any>> = {
+  item: ItemInstance<T>;
+  tree: TreeInstance<T>;
   itemId: string;
-  prev?: MayReturnNull<ItemInstance<any>[Key]>;
+  prev?: MayReturnNull<ItemInstance<T>[Key]>;
 };
 
 export type TreeInstanceOpts<Key extends keyof TreeInstance<any>> = {
@@ -131,7 +131,7 @@ export type FeatureImplementation<T = any> = {
 
   itemInstance?: {
     [key in keyof ItemInstance<T>]?: (
-      opts: ItemInstanceOpts<key>,
+      opts: ItemInstanceOpts<T, key>,
       ...args: Parameters<ItemInstance<T>[key]>
     ) => void;
   };