@lobehub/chat/src/services/user/client.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 { eq } from 'drizzle-orm';
import { DeepPartial } from 'utility-types';
import { afterEach, beforeEach, describe, expect, it, vi } from 'vitest';

import { clientDB, initializeDB } from '@/database/client/db';
import { userSettings, users } from '@/database/schemas';
import { UserPreference } from '@/types/user';
import { UserSettings } from '@/types/user/settings';

import { ClientService } from './client';

const mockUser = {
  avatar: 'avatar.png',
  settings: { themeMode: 'light' } as unknown as UserSettings,
  uuid: 'user-id',
};

const mockPreference = {
  useCmdEnterToSend: true,
} as UserPreference;
const clientService = new ClientService(mockUser.uuid);

beforeEach(async () => {
  vi.clearAllMocks();

  await initializeDB();
  await clientDB.delete(users);

  await clientDB.insert(users).values({ id: mockUser.uuid, avatar: 'avatar.png' });
  await clientDB
    .insert(userSettings)
    .values({ id: mockUser.uuid, general: { themeMode: 'light' } });
});

describe('ClientService', () => {
  it('should get user state correctly', async () => {
    const spyOn = vi
      .spyOn(clientService['preferenceStorage'], 'getFromLocalStorage')
      .mockResolvedValue(mockPreference);

    const userState = await clientService.getUserState();

    expect(userState).toMatchObject({
      avatar: mockUser.avatar,
      isOnboard: true,
      canEnablePWAGuide: false,
      hasConversation: false,
      canEnableTrace: false,
      preference: mockPreference,
      settings: { general: { themeMode: 'light' } },
      userId: mockUser.uuid,
    });
    expect(spyOn).toHaveBeenCalledTimes(1);
  });

  it('should update user settings correctly', async () => {
    const settingsPatch: DeepPartial<UserSettings> = { general: { fontSize: 12 } };

    await clientService.updateUserSettings(settingsPatch);

    const result = await clientDB.query.userSettings.findFirst({
      where: eq(userSettings.id, mockUser.uuid),
    });

    expect(result).toMatchObject(settingsPatch);
  });

  it('should reset user settings correctly', async () => {
    await clientService.resetUserSettings();

    const result = await clientDB.query.userSettings.findFirst({
      where: eq(userSettings.id, mockUser.uuid),
    });

    expect(result).toBeUndefined();
  });

  it('should update user avatar correctly', async () => {
    const newAvatar = 'new-avatar.png';

    await clientService.updateAvatar(newAvatar);
  });

  it('should update user preference correctly', async () => {
    const newPreference = {
      useCmdEnterToSend: false,
    } as UserPreference;

    const spyOn = vi
      .spyOn(clientService['preferenceStorage'], 'saveToLocalStorage')
      .mockResolvedValue(undefined);

    await clientService.updatePreference(newPreference);

    expect(spyOn).toHaveBeenCalledWith(newPreference);
    expect(spyOn).toHaveBeenCalledTimes(1);
  });

  it('should return empty array for getUserSSOProviders', async () => {
    const providers = await clientService.getUserSSOProviders();
    expect(providers).toEqual([]);
  });

  it('should do nothing when unlinkSSOProvider is called', async () => {
    const result = await clientService.unlinkSSOProvider('google', '123');
    expect(result).toBeUndefined();
  });
});