Chore: Add tests covering usePrefix from @inquirer/core

packages/core/core.test.mts

@@ -1,19 +1,23 @@
+import { AsyncResource } from 'node:async_hooks';
 import { describe, it, expect, vi } from 'vitest';
 import { render } from '@inquirer/testing';
 import stripAnsi from 'strip-ansi';
 import ansiEscapes from 'ansi-escapes';
+import spinners from 'cli-spinners';
 import {
   createPrompt,
   useEffect,
   useKeypress,
   useState,
   useRef,
   useMemo,
+  usePrefix,
   isDownKey,
   isUpKey,
   isEnterKey,
   isSpaceKey,
   Separator,
+  makeTheme,
   type KeypressEvent,
 } from './src/index.mjs';
 
@@ -355,6 +359,55 @@ describe('createPrompt()', () => {
     await expect(answer).resolves.toEqual('b');
   });
 
+  it.only('usePrefix() renders loader and prefix', async () => {
+    vi.useFakeTimers();
+    const { interval } = spinners.dots;
+    const totalDuration = interval * spinners.dots.frames.length;
+
+    const Prompt = (config: { message: string }, done: (value: string) => void) => {
+      const [status, setStatus] = useState('loading');
+      const prefix = usePrefix({ isLoading: status === 'loading' });
+
+      useEffect(() => {
+        setTimeout(
+          AsyncResource.bind(() => {
+            setStatus('idle');
+          }),
+          totalDuration,
+        );
+      }, []);
+
+      useKeypress((event: KeypressEvent) => {
+        if (isEnterKey(event)) {
+          done('');
+        }
+      });
+
+      return `${prefix} ${config.message}`;
+    };
+
+    const prompt = createPrompt(Prompt);
+    const { answer, events, getScreen } = await render(prompt, { message: 'Question' });
+    expect(getScreen()).toMatchInlineSnapshot(`"⠋ Question"`);
+
+    vi.advanceTimersByTime(interval);
+    expect(getScreen()).toMatchInlineSnapshot(`"⠙ Question"`);
+
+    vi.advanceTimersByTime(interval);
+    expect(getScreen()).toMatchInlineSnapshot(`"⠹ Question"`);
+
+    vi.advanceTimersByTime(interval);
+    expect(getScreen()).toMatchInlineSnapshot(`"⠸ Question"`);
+
+    vi.advanceTimersByTime(totalDuration);
+    expect(getScreen()).toMatchInlineSnapshot(`"? Question"`);
+
+    vi.useRealTimers();
+
+    events.keypress('enter');
+    await expect(answer).resolves.toEqual('');
+  });
+
   it('allow cancelling the prompt', async () => {
     const Prompt = (config: { message: string }, done: (value: string) => void) => {
       useKeypress((key: KeypressEvent) => {

packages/core/src/lib/use-prefix.mts

@@ -1,3 +1,4 @@
+import { AsyncResource } from 'node:async_hooks';
 import { useState } from './use-state.mjs';
 import { useEffect } from './use-effect.mjs';
 import { makeTheme } from './make-theme.mjs';
@@ -15,9 +16,12 @@ export function usePrefix({
 
   useEffect((): void | (() => unknown) => {
     if (isLoading) {
-      const timeout = setTimeout(() => {
-        setTick(tick + 1);
-      }, spinner.interval);
+      const timeout = setTimeout(
+        AsyncResource.bind(() => {
+          setTick(tick + 1);
+        }),
+        spinner.interval,
+      );
 
       return () => clearTimeout(timeout);
     }