@lobehub/chat/src/services/__tests__/sync.test.ts

Lobe Chat - an open-source, high-performance chatbot framework that supports speech synthesis, multimodal, and extensible Function Call plugin system. Supports one-click free deployment of your private ChatGPT/LLM web application.

import { afterEach, beforeEach, describe, expect, it, vi } from 'vitest';

import { dataSync } from '@/database/_deprecated/core';
import { StartDataSyncParams } from '@/types/sync';

import { syncService } from '../sync';

vi.mock('@/database/_deprecated/core', () => ({
  dataSync: {
    startDataSync: vi.fn(),
    disconnect: vi.fn(),
  },
}));

describe('SyncService', () => {
  afterEach(() => {
    vi.resetAllMocks();
  });

  describe('enabledSync', () => {
    it('should return false when running on server side', async () => {
      const params = { user: { id: '123' }, authToken: 'abc' } as unknown as StartDataSyncParams;

      const origin = global.window;

      // @ts-ignore
      global.window = undefined;

      const result = await syncService.enabledSync(params);

      expect(result).toBe(false);
      expect(dataSync.startDataSync).not.toHaveBeenCalled();

      // reset
      global.window = origin;
    });

    it('should start data sync and return true when running on client side', async () => {
      const params = { user: { id: '123' }, authToken: 'abc' } as unknown as StartDataSyncParams;

      const result = await syncService.enabledSync(params);

      expect(result).toBe(true);
      expect(dataSync.startDataSync).toHaveBeenCalledWith(params);
    });
  });

  describe('disableSync', () => {
    it('should disconnect data sync and return false', async () => {
      const result = await syncService.disableSync();

      expect(result).toBe(false);
      expect(dataSync.disconnect).toHaveBeenCalled();
    });
  });
});