From 0f28f7b8b991bf55d2e58eb23fad224a60e97fb1 Mon Sep 17 00:00:00 2001
From: "gru-agent[bot]" <185149714+gru-agent[bot]@users.noreply.github.com>
Date: Wed, 19 Feb 2025 17:10:36 +0000
Subject: [PATCH] Add unit tests for ServerService methods in user service.

---
 src/services/user/server.test.ts | 149 +++++++++++++++++++++++++++++++
 1 file changed, 149 insertions(+)
 create mode 100644 src/services/user/server.test.ts

diff --git a/src/services/user/server.test.ts b/src/services/user/server.test.ts
new file mode 100644
index 0000000000000..ea2b48795ae64
--- /dev/null
+++ b/src/services/user/server.test.ts
@@ -0,0 +1,149 @@
+import { DeepPartial } from 'utility-types';
+import { describe, expect, it, vi } from 'vitest';
+
+import { lambdaClient } from '@/libs/trpc/client';
+import { UserInitializationState, UserPreference } from '@/types/user';
+import { UserSettings } from '@/types/user/settings';
+
+import { ServerService } from './server';
+
+vi.mock('@/libs/trpc/client', () => ({
+  lambdaClient: {
+    user: {
+      getUserRegistrationDuration: {
+        query: vi.fn(),
+      },
+      getUserState: {
+        query: vi.fn(),
+      },
+      getUserSSOProviders: {
+        query: vi.fn(),
+      },
+      unlinkSSOProvider: {
+        mutate: vi.fn(),
+      },
+      makeUserOnboarded: {
+        mutate: vi.fn(),
+      },
+      updatePreference: {
+        mutate: vi.fn(),
+      },
+      updateGuide: {
+        mutate: vi.fn(),
+      },
+      updateSettings: {
+        mutate: vi.fn(),
+      },
+      resetSettings: {
+        mutate: vi.fn(),
+      },
+    },
+  },
+}));
+
+describe('ServerService', () => {
+  const service = new ServerService();
+
+  it('should get user registration duration', async () => {
+    const mockData = {
+      createdAt: '2023-01-01',
+      duration: 100,
+      updatedAt: '2023-01-02',
+    };
+    vi.mocked(lambdaClient.user.getUserRegistrationDuration.query).mockResolvedValue(mockData);
+
+    const result = await service.getUserRegistrationDuration();
+    expect(result).toEqual(mockData);
+  });
+
+  it('should get user state', async () => {
+    const mockState: UserInitializationState = {
+      isOnboard: true,
+      preference: {
+        telemetry: true,
+      },
+      settings: {},
+    };
+    vi.mocked(lambdaClient.user.getUserState.query).mockResolvedValue(mockState);
+
+    const result = await service.getUserState();
+    expect(result).toEqual(mockState);
+  });
+
+  it('should get user SSO providers', async () => {
+    const mockProviders = [
+      {
+        provider: 'google',
+        providerAccountId: '123',
+        userId: 'user1',
+        type: 'oauth' as const,
+        access_token: 'token',
+        token_type: 'bearer' as const,
+        expires_at: 123,
+        scope: 'email profile',
+      },
+    ];
+    vi.mocked(lambdaClient.user.getUserSSOProviders.query).mockResolvedValue(mockProviders);
+
+    const result = await service.getUserSSOProviders();
+    expect(result).toEqual(mockProviders);
+  });
+
+  it('should unlink SSO provider', async () => {
+    const provider = 'google';
+    const providerAccountId = '123';
+    await service.unlinkSSOProvider(provider, providerAccountId);
+
+    expect(lambdaClient.user.unlinkSSOProvider.mutate).toHaveBeenCalledWith({
+      provider,
+      providerAccountId,
+    });
+  });
+
+  it('should make user onboarded', async () => {
+    await service.makeUserOnboarded();
+    expect(lambdaClient.user.makeUserOnboarded.mutate).toHaveBeenCalled();
+  });
+
+  it('should update user preference', async () => {
+    const preference: Partial<UserPreference> = {
+      telemetry: true,
+      useCmdEnterToSend: true,
+    };
+    await service.updatePreference(preference);
+    expect(lambdaClient.user.updatePreference.mutate).toHaveBeenCalledWith(preference);
+  });
+
+  it('should update user guide', async () => {
+    const guide = {
+      moveSettingsToAvatar: true,
+      topic: false,
+      uploadFileInKnowledgeBase: true,
+    };
+    await service.updateGuide(guide);
+    expect(lambdaClient.user.updateGuide.mutate).toHaveBeenCalledWith(guide);
+  });
+
+  it('should update user settings', async () => {
+    const settings: DeepPartial<UserSettings> = {
+      defaultAgent: {
+        config: {
+          model: 'gpt-4',
+          provider: 'openai',
+        },
+        meta: {
+          avatar: 'avatar',
+          description: 'test agent',
+        },
+      },
+    };
+    const signal = new AbortController().signal;
+    await service.updateUserSettings(settings, signal);
+    expect(lambdaClient.user.updateSettings.mutate).toHaveBeenCalledWith(settings, { signal });
+  });
+
+  it('should reset user settings', async () => {
+    await service.resetUserSettings();
+    expect(lambdaClient.user.resetSettings.mutate).toHaveBeenCalled();
+  });
+});