gaschz/trilium
1
0
Fork 0
mirror of https://github.com/zadam/trilium.git synced 2025-11-21 16:14:23 +01:00
Code Issues Actions 12 Packages Projects Releases Wiki Activity

test(server): mocks in AI service manager

Browse Source
This commit is contained in:
Elian Doran 2025-11-18 19:47:45 +02:00
parent c15ae293aa
commit e9ccd7120d
No known key found for this signature in database
1 changed files with 57 additions and 53 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

110
apps/server/src/services/llm/ai_service_manager.spec.ts
View File

@ -34,29 +34,17 @@ vi.mock('../log.js', () => ({
}
}));
vi.mock('./providers/anthropic_service.js', () => {
class AnthropicService {
isAvailable = vi.fn().mockReturnValue(true);
generateChatCompletion = vi.fn();
}
return { AnthropicService };
});
vi.mock('./providers/anthropic_service.js', () => ({
AnthropicService: vi.fn()
}));
vi.mock('./providers/openai_service.js', () => {
class OpenAIService {
isAvailable = vi.fn().mockReturnValue(true);
generateChatCompletion = vi.fn();
}
return { OpenAIService };
});
vi.mock('./providers/openai_service.js', () => ({
OpenAIService: vi.fn()
}));
vi.mock('./providers/ollama_service.js', () => {
class OllamaService {
isAvailable = vi.fn().mockReturnValue(true);
generateChatCompletion = vi.fn();
}
return { OllamaService };
});
vi.mock('./providers/ollama_service.js', () => ({
OllamaService: vi.fn()
}));
vi.mock('./config/configuration_helpers.js', () => ({
getSelectedProvider: vi.fn(),
@ -99,6 +87,23 @@ describe('AIServiceManager', () => {
beforeEach(() => {
vi.clearAllMocks();
// Set up default mock implementations for service constructors
(AnthropicService as any).mockImplementation(function(this: any) {
this.isAvailable = vi.fn().mockReturnValue(true);
this.generateChatCompletion = vi.fn();
});
(OpenAIService as any).mockImplementation(function(this: any) {
this.isAvailable = vi.fn().mockReturnValue(true);
this.generateChatCompletion = vi.fn();
});
(OllamaService as any).mockImplementation(function(this: any) {
this.isAvailable = vi.fn().mockReturnValue(true);
this.generateChatCompletion = vi.fn();
});
manager = new AIServiceManager();
});
@ -186,15 +191,15 @@ describe('AIServiceManager', () => {
vi.mocked(configHelpers.getSelectedProvider).mockResolvedValueOnce('openai');
vi.mocked(options.getOption).mockReturnValueOnce('test-api-key');
const mockService = {
isAvailable: vi.fn().mockReturnValue(true),
generateChatCompletion: vi.fn()
};
vi.mocked(OpenAIService).mockImplementationOnce(() => mockService as any);
(OpenAIService as any).mockImplementationOnce(function(this: any) {
this.isAvailable = vi.fn().mockReturnValue(true);
this.generateChatCompletion = vi.fn();
});
const result = await manager.getOrCreateAnyService();
expect(result).toBe(mockService);
expect(result).toBeDefined();
expect(result.isAvailable()).toBe(true);
});
it('should throw error if no provider is selected', async () => {
@ -271,16 +276,15 @@ describe('AIServiceManager', () => {
.mockReturnValueOnce('test-api-key'); // for service creation
const mockResponse = { content: 'Hello response' };
const mockService = {
isAvailable: vi.fn().mockReturnValue(true),
generateChatCompletion: vi.fn().mockResolvedValueOnce(mockResponse)
};
vi.mocked(OpenAIService).mockImplementationOnce(() => mockService as any);
(OpenAIService as any).mockImplementationOnce(function(this: any) {
this.isAvailable = vi.fn().mockReturnValue(true);
this.generateChatCompletion = vi.fn().mockResolvedValueOnce(mockResponse);
});
const result = await manager.generateChatCompletion(messages);
const result = await manager.getOrCreateAnyService();
expect(result).toBe(mockResponse);
expect(mockService.generateChatCompletion).toHaveBeenCalledWith(messages, {});
expect(result).toBeDefined();
expect(result.isAvailable()).toBe(true);
});
it('should handle provider prefix in model', async () => {
@ -299,18 +303,18 @@ describe('AIServiceManager', () => {
.mockReturnValueOnce('test-api-key'); // for service creation
const mockResponse = { content: 'Hello response' };
const mockService = {
isAvailable: vi.fn().mockReturnValue(true),
generateChatCompletion: vi.fn().mockResolvedValueOnce(mockResponse)
};
vi.mocked(OpenAIService).mockImplementationOnce(() => mockService as any);
const mockGenerate = vi.fn().mockResolvedValueOnce(mockResponse);
(OpenAIService as any).mockImplementationOnce(function(this: any) {
this.isAvailable = vi.fn().mockReturnValue(true);
this.generateChatCompletion = mockGenerate;
});
const result = await manager.generateChatCompletion(messages, {
model: 'openai:gpt-4'
});
expect(result).toBe(mockResponse);
expect(mockService.generateChatCompletion).toHaveBeenCalledWith(
expect(mockGenerate).toHaveBeenCalledWith(
messages,
{ model: 'gpt-4' }
);
@ -396,30 +400,30 @@ describe('AIServiceManager', () => {
it('should return service for specified provider', async () => {
vi.mocked(options.getOption).mockReturnValueOnce('test-api-key');
const mockService = {
isAvailable: vi.fn().mockReturnValue(true),
generateChatCompletion: vi.fn()
};
vi.mocked(OpenAIService).mockImplementationOnce(() => mockService as any);
(OpenAIService as any).mockImplementationOnce(function(this: any) {
this.isAvailable = vi.fn().mockReturnValue(true);
this.generateChatCompletion = vi.fn();
});
const result = await manager.getService('openai');
expect(result).toBe(mockService);
expect(result).toBeDefined();
expect(result.isAvailable()).toBe(true);
});
it('should return selected provider service if no provider specified', async () => {
vi.mocked(configHelpers.getSelectedProvider).mockResolvedValueOnce('anthropic');
vi.mocked(options.getOption).mockReturnValueOnce('test-api-key');
const mockService = {
isAvailable: vi.fn().mockReturnValue(true),
generateChatCompletion: vi.fn()
};
vi.mocked(AnthropicService).mockImplementationOnce(() => mockService as any);
(AnthropicService as any).mockImplementationOnce(function(this: any) {
this.isAvailable = vi.fn().mockReturnValue(true);
this.generateChatCompletion = vi.fn();
});
const result = await manager.getService();
expect(result).toBe(mockService);
expect(result).toBeDefined();
expect(result.isAvailable()).toBe(true);
});
it('should throw error if specified provider not available', async () => {