NapCatQQ source snapshot

packages/napcat-test/offlineRecovery.test.ts

@@ -0,0 +1,98 @@
+import { afterEach, beforeEach, describe, expect, test, vi } from 'vitest';
+import { createOfflineRecovery } from '../napcat-core/helper/offline-recovery';
+
+function setup (restartResult: boolean | Error = true) {
+  const state = { online: false };
+  const onOffline = vi.fn();
+  const onRecovered = vi.fn();
+  const restart = vi.fn(async () => {
+    if (restartResult instanceof Error) throw restartResult;
+    return restartResult;
+  });
+  const handleOffline = createOfflineRecovery({
+    isOnline: () => state.online,
+    onOffline,
+    onRecovered,
+    restart,
+  });
+  return { state, onOffline, onRecovered, restart, handleOffline };
+}
+
+describe('offline recovery', () => {
+  beforeEach(() => {
+    vi.useFakeTimers();
+  });
+
+  afterEach(() => {
+    vi.useRealTimers();
+  });
+
+  test('kicked offline restarts after a short delay', async () => {
+    const { restart, onOffline, handleOffline } = setup();
+    handleOffline('kicked', false);
+
+    expect(onOffline).toHaveBeenCalledWith('kicked');
+    await vi.advanceTimersByTimeAsync(3_000);
+    expect(restart).not.toHaveBeenCalled();
+    await vi.advanceTimersByTimeAsync(500);
+    expect(restart).toHaveBeenCalledTimes(1);
+    expect(restart).toHaveBeenCalledWith('kicked');
+  });
+
+  test('silent offline waits for the account to reconnect before restarting', async () => {
+    const { restart, handleOffline } = setup();
+    handleOffline('silent', true);
+
+    await vi.advanceTimersByTimeAsync(3_500);
+    expect(restart).not.toHaveBeenCalled();
+    await vi.advanceTimersByTimeAsync(60_000);
+    expect(restart).toHaveBeenCalledTimes(1);
+  });
+
+  test('silent offline that recovers on its own does not restart', async () => {
+    const { state, restart, onRecovered, handleOffline } = setup();
+    handleOffline('silent', true);
+
+    await vi.advanceTimersByTimeAsync(10_000);
+    state.online = true;
+    await vi.advanceTimersByTimeAsync(60_000);
+
+    expect(restart).not.toHaveBeenCalled();
+    expect(onRecovered).toHaveBeenCalledTimes(1);
+  });
+
+  test('a kick during the silent-offline wait restarts without waiting the full minute', async () => {
+    const { restart, handleOffline } = setup();
+    handleOffline('silent', true);
+    await vi.advanceTimersByTimeAsync(1_000);
+    handleOffline('kicked', false);
+
+    await vi.advanceTimersByTimeAsync(3_500);
+    expect(restart).toHaveBeenCalledTimes(1);
+    expect(restart).toHaveBeenCalledWith('kicked');
+    await vi.advanceTimersByTimeAsync(120_000);
+    expect(restart).toHaveBeenCalledTimes(1);
+  });
+
+  test('repeated offline events schedule only one restart', async () => {
+    const { restart, onOffline, handleOffline } = setup();
+    handleOffline('kicked', false);
+    handleOffline('kicked again', false);
+    handleOffline('silent', true);
+
+    await vi.advanceTimersByTimeAsync(120_000);
+    expect(restart).toHaveBeenCalledTimes(1);
+    expect(onOffline).toHaveBeenCalledTimes(3);
+  });
+
+  test.each([false, new Error('boom')])('a failed restart (%s) can be retried by the next offline event', async (result) => {
+    const { restart, handleOffline } = setup(result);
+    handleOffline('kicked', false);
+    await vi.advanceTimersByTimeAsync(3_500);
+    expect(restart).toHaveBeenCalledTimes(1);
+
+    handleOffline('kicked', false);
+    await vi.advanceTimersByTimeAsync(3_500);
+    expect(restart).toHaveBeenCalledTimes(2);
+  });
+});