test: useIntervalFn

packages/unuse/src/useIntervalFn/index.spec.ts

@@ -1,7 +1,7 @@
 // @vitest-environment happy-dom
 
 import type { UnComputed } from 'unuse-reactivity';
-import { isUnComputed, unSignal } from 'unuse-reactivity';
+import { isUnComputed, unEffectScope, unSignal } from 'unuse-reactivity';
 import { beforeEach, describe, expect, it, vi } from 'vitest';
 import type { Pausable } from '.';
 import { useIntervalFn } from '.';
@@ -39,6 +39,28 @@
     expect(callback).toHaveBeenCalledTimes(2);
   }
 
+  async function execImmediateCallback({ isActive, pause, resume }: Pausable) {
+    expect(isActive).toSatisfy(isUnComputed);
+    expect((isActive as unknown as UnComputed<boolean>).get()).toBeTruthy();
+    expect(callback).toHaveBeenCalledTimes(1);
+
+    await vi.advanceTimersByTimeAsync(60);
+    expect(callback).toHaveBeenCalledTimes(2);
+
+    pause();
+    expect((isActive as unknown as UnComputed<boolean>).get()).toBeFalsy();
+
+    await vi.advanceTimersByTimeAsync(60);
+    expect(callback).toHaveBeenCalledTimes(2);
+
+    resume();
+    expect((isActive as unknown as UnComputed<boolean>).get()).toBeTruthy();
+    expect(callback).toHaveBeenCalledTimes(3);
+
+    await vi.advanceTimersByTimeAsync(60);
+    expect(callback).toHaveBeenCalledTimes(4);
+  }
+
   it('basic pause/resume', async () => {
     await exec(useIntervalFn(callback, 50));
 
@@ -53,6 +75,41 @@
     expect(callback).toHaveBeenCalledTimes(1);
   });
 
+  it('pause/resume with immediateCallback', async () => {
+    await execImmediateCallback(
+      useIntervalFn(callback, 50, { immediateCallback: true })
+    );
+
+    callback = vi.fn();
+
+    const interval = unSignal(50);
+    await execImmediateCallback(
+      useIntervalFn(callback, interval, { immediateCallback: true })
+    );
+
+    callback.mockClear();
+    interval.set(20);
+    // mimic nextTick
+    await vi.waitFor(() => {
+      expect(callback).toHaveBeenCalledTimes(1);
+    });
+  });
+
+  it('pause/resume in scope', async () => {
+    let promise!: Promise<void>;
+
+    const scope = unEffectScope(() => {
+      promise = exec(useIntervalFn(callback, 50));
+    });
+
+    await promise;
+
+    callback.mockClear();
+    scope();
+    await vi.advanceTimersByTimeAsync(60);
+    expect(callback).toHaveBeenCalledTimes(0);
+  });
+
   it('pause in callback', async () => {
     const pausable = useIntervalFn(
       () => {