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

chore(deps): adapt most of the tests
Some checks failed
Checks / main (push) Has been cancelled
Details

Browse Source
This commit is contained in:
Elian Doran 2025-11-01 19:21:57 +02:00
parent fa30bfc04b
commit 35f244cf50
No known key found for this signature in database
7 changed files with 77 additions and 71 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
apps/server/src/routes/api/llm.spec.ts
View File

@ -52,9 +52,9 @@ vi.mock("../../services/llm/ai_service_manager.js", () => ({
// Mock chat pipeline // Mock chat pipeline
const mockChatPipelineExecute = vi.fn(); const mockChatPipelineExecute = vi.fn();
const MockChatPipeline = vi.fn().mockImplementation(() => ({ const MockChatPipeline = vi.fn().mockImplementation(function () {
execute: mockChatPipelineExecute this.execute = mockChatPipelineExecute;
})); });
vi.mock("../../services/llm/pipeline/chat_pipeline.js", () => ({ vi.mock("../../services/llm/pipeline/chat_pipeline.js", () => ({
ChatPipeline: MockChatPipeline ChatPipeline: MockChatPipeline
})); }));

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

@ -35,24 +35,24 @@ vi.mock('../log.js', () => ({
})); }));
vi.mock('./providers/anthropic_service.js', () => ({ vi.mock('./providers/anthropic_service.js', () => ({
AnthropicService: vi.fn().mockImplementation(() => ({ AnthropicService: vi.fn().mockImplementation(function () {
isAvailable: vi.fn().mockReturnValue(true), this.isAvailable = vi.fn().mockReturnValue(true);
generateChatCompletion: vi.fn() this.generateChatCompletion = vi.fn();
})) })
})); }));
vi.mock('./providers/openai_service.js', () => ({ vi.mock('./providers/openai_service.js', () => ({
OpenAIService: vi.fn().mockImplementation(() => ({ OpenAIService: vi.fn().mockImplementation(function () {
isAvailable: vi.fn().mockReturnValue(true), this.isAvailable = vi.fn().mockReturnValue(true);
generateChatCompletion: vi.fn() this.generateChatCompletion = vi.fn();
})) };
})); }));
vi.mock('./providers/ollama_service.js', () => ({ vi.mock('./providers/ollama_service.js', () => ({
OllamaService: vi.fn().mockImplementation(() => ({ OllamaService: vi.fn().mockImplementation(function () {
isAvailable: vi.fn().mockReturnValue(true), this.isAvailable = vi.fn().mockReturnValue(true);
generateChatCompletion: vi.fn() this.generateChatCompletion = vi.fn();
})) })
})); }));
vi.mock('./config/configuration_helpers.js', () => ({ vi.mock('./config/configuration_helpers.js', () => ({
@ -65,7 +65,7 @@ vi.mock('./config/configuration_helpers.js', () => ({
})); }));
vi.mock('./context/index.js', () => ({ vi.mock('./context/index.js', () => ({
ContextExtractor: vi.fn().mockImplementation(() => ({})) ContextExtractor: vi.fn().mockImplementation(function () {})
})); }));
vi.mock('./context_extractors/index.js', () => ({ vi.mock('./context_extractors/index.js', () => ({

6
apps/server/src/services/llm/chat/rest_chat_service.spec.ts
View File

@ -39,9 +39,9 @@ vi.mock('../pipeline/chat_pipeline.js', () => ({
})); }));
vi.mock('./handlers/tool_handler.js', () => ({ vi.mock('./handlers/tool_handler.js', () => ({
ToolHandler: vi.fn().mockImplementation(() => ({ ToolHandler: vi.fn().mockImplementation(function () {
handleToolCalls: vi.fn() this.handleToolCalls = vi.fn()
})) })
})); }));
vi.mock('../chat_storage_service.js', () => ({ vi.mock('../chat_storage_service.js', () => ({

6
apps/server/src/services/llm/chat_service.spec.ts
View File

@ -36,7 +36,8 @@ vi.mock('./constants/llm_prompt_constants.js', () => ({
})); }));
vi.mock('./pipeline/chat_pipeline.js', () => ({ vi.mock('./pipeline/chat_pipeline.js', () => ({
ChatPipeline: vi.fn().mockImplementation((config) => ({ ChatPipeline: vi.fn().mockImplementation(function (config) {
Object.assign(this, {
config, config,
execute: vi.fn(), execute: vi.fn(),
getMetrics: vi.fn(), getMetrics: vi.fn(),
@ -49,7 +50,8 @@ vi.mock('./pipeline/chat_pipeline.js', () => ({
execute: vi.fn() execute: vi.fn()
} }
} }
})) });
});
})); }));
vi.mock('./ai_service_manager.js', () => ({ vi.mock('./ai_service_manager.js', () => ({

6
apps/server/src/services/llm/context/services/context_service.spec.ts
View File

@ -47,9 +47,9 @@ vi.mock('../../ai_service_manager.js', () => ({
})); }));
vi.mock('../index.js', () => ({ vi.mock('../index.js', () => ({
ContextExtractor: vi.fn().mockImplementation(() => ({ ContextExtractor: vi.fn().mockImplementation(function () {
findRelevantNotes: vi.fn().mockResolvedValue([]) this.findRelevantNotes = vi.fn().mockResolvedValue([])
})) });
})); }));
describe('ContextService', () => { describe('ContextService', () => {

12
apps/server/src/services/llm/providers/anthropic_service.spec.ts
View File

@ -48,8 +48,8 @@ vi.mock('@anthropic-ai/sdk', () => {
} }
}; };
const mockAnthropic = vi.fn().mockImplementation(() => ({ const mockAnthropic = vi.fn().mockImplementation(function () {
messages: { this.messages = {
create: vi.fn().mockImplementation((params) => { create: vi.fn().mockImplementation((params) => {
if (params.stream) { if (params.stream) {
return Promise.resolve(mockStream); return Promise.resolve(mockStream);
@ -71,8 +71,8 @@ vi.mock('@anthropic-ai/sdk', () => {
} }
}); });
}) })
} };
})); });
return { default: mockAnthropic }; return { default: mockAnthropic };
}); });
@ -127,7 +127,9 @@ describe('AnthropicService', () => {
} }
}; };
AnthropicMock.mockImplementation(() => mockAnthropicInstance); AnthropicMock.mockImplementation(function () {
Object.assign(this, mockAnthropicInstance);
});
service = new AnthropicService(); service = new AnthropicService();
}); });

28
apps/server/src/services/llm/providers/ollama_service.spec.ts
View File

@ -30,11 +30,11 @@ vi.mock('./providers.js', () => ({
})); }));
vi.mock('../formatters/ollama_formatter.js', () => ({ vi.mock('../formatters/ollama_formatter.js', () => ({
OllamaMessageFormatter: vi.fn().mockImplementation(() => ({ OllamaMessageFormatter: vi.fn().mockImplementation(function () {
formatMessages: vi.fn().mockReturnValue([ this.formatMessages = vi.fn().mockReturnValue([
{ role: 'user', content: 'Hello' } { role: 'user', content: 'Hello' }
]), ]);
formatResponse: vi.fn().mockReturnValue({ this.formatResponse = vi.fn().mockReturnValue({
text: 'Hello! How can I help you today?', text: 'Hello! How can I help you today?',
provider: 'Ollama', provider: 'Ollama',
model: 'llama2', model: 'llama2',
@ -44,8 +44,8 @@ vi.mock('../formatters/ollama_formatter.js', () => ({
totalTokens: 15 totalTokens: 15
}, },
tool_calls: null tool_calls: null
});
}) })
}))
})); }));
vi.mock('../tools/tool_registry.js', () => ({ vi.mock('../tools/tool_registry.js', () => ({
@ -83,8 +83,8 @@ vi.mock('ollama', () => {
} }
}; };
const mockOllama = vi.fn().mockImplementation(() => ({ const mockOllama = vi.fn().mockImplementation(function () {
chat: vi.fn().mockImplementation((params) => { this.chat = vi.fn().mockImplementation((params) => {
if (params.stream) { if (params.stream) {
return Promise.resolve(mockStream); return Promise.resolve(mockStream);
} }
@ -97,8 +97,8 @@ vi.mock('ollama', () => {
model: 'llama2', model: 'llama2',
done: true done: true
}); });
}), });
show: vi.fn().mockResolvedValue({ this.show = vi.fn().mockResolvedValue({
modelfile: 'FROM llama2', modelfile: 'FROM llama2',
parameters: {}, parameters: {},
template: '', template: '',
@ -109,8 +109,8 @@ vi.mock('ollama', () => {
parameter_size: '7B', parameter_size: '7B',
quantization_level: 'Q4_0' quantization_level: 'Q4_0'
} }
}), });
list: vi.fn().mockResolvedValue({ this.list = vi.fn().mockResolvedValue({
models: [ models: [
{ {
name: 'llama2:latest', name: 'llama2:latest',
@ -119,7 +119,7 @@ vi.mock('ollama', () => {
} }
] ]
}) })
})); });
return { Ollama: mockOllama }; return { Ollama: mockOllama };
}); });
@ -196,7 +196,9 @@ describe('OllamaService', () => {
}) })
}; };
OllamaMock.mockImplementation(() => mockOllamaInstance); OllamaMock.mockImplementation(function () {
Object.assign(this, mockOllamaInstance);
});
service = new OllamaService(); service = new OllamaService();