gaschz/trilium
1
0
Fork 0
mirror of https://github.com/zadam/trilium.git synced 2025-11-22 00:24:24 +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
const mockChatPipelineExecute = vi.fn();
const MockChatPipeline = vi.fn().mockImplementation(() => ({
execute: mockChatPipelineExecute
}));
const MockChatPipeline = vi.fn().mockImplementation(function () {
this.execute = mockChatPipelineExecute;
});
vi.mock("../../services/llm/pipeline/chat_pipeline.js", () => ({
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', () => ({
AnthropicService: vi.fn().mockImplementation(() => ({
isAvailable: vi.fn().mockReturnValue(true),
generateChatCompletion: vi.fn()
}))
AnthropicService: vi.fn().mockImplementation(function () {
this.isAvailable = vi.fn().mockReturnValue(true);
this.generateChatCompletion = vi.fn();
})
}));
vi.mock('./providers/openai_service.js', () => ({
OpenAIService: vi.fn().mockImplementation(() => ({
isAvailable: vi.fn().mockReturnValue(true),
generateChatCompletion: vi.fn()
}))
OpenAIService: vi.fn().mockImplementation(function () {
this.isAvailable = vi.fn().mockReturnValue(true);
this.generateChatCompletion = vi.fn();
};
}));
vi.mock('./providers/ollama_service.js', () => ({
OllamaService: vi.fn().mockImplementation(() => ({
isAvailable: vi.fn().mockReturnValue(true),
generateChatCompletion: vi.fn()
}))
OllamaService: vi.fn().mockImplementation(function () {
this.isAvailable = vi.fn().mockReturnValue(true);
this.generateChatCompletion = vi.fn();
})
}));
vi.mock('./config/configuration_helpers.js', () => ({
@ -65,7 +65,7 @@ vi.mock('./config/configuration_helpers.js', () => ({
}));
vi.mock('./context/index.js', () => ({
ContextExtractor: vi.fn().mockImplementation(() => ({}))
ContextExtractor: vi.fn().mockImplementation(function () {})
}));
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', () => ({
ToolHandler: vi.fn().mockImplementation(() => ({
handleToolCalls: vi.fn()
}))
ToolHandler: vi.fn().mockImplementation(function () {
this.handleToolCalls = vi.fn()
})
}));
vi.mock('../chat_storage_service.js', () => ({

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

@ -36,20 +36,22 @@ vi.mock('./constants/llm_prompt_constants.js', () => ({
}));
vi.mock('./pipeline/chat_pipeline.js', () => ({
ChatPipeline: vi.fn().mockImplementation((config) => ({
config,
execute: vi.fn(),
getMetrics: vi.fn(),
resetMetrics: vi.fn(),
stages: {
contextExtraction: {
execute: vi.fn()
},
semanticContextExtraction: {
execute: vi.fn()
ChatPipeline: vi.fn().mockImplementation(function (config) {
Object.assign(this, {
config,
execute: vi.fn(),
getMetrics: vi.fn(),
resetMetrics: vi.fn(),
stages: {
contextExtraction: {
execute: vi.fn()
},
semanticContextExtraction: {
execute: vi.fn()
}
}
}
}))
});
});
}));
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', () => ({
ContextExtractor: vi.fn().mockImplementation(() => ({
findRelevantNotes: vi.fn().mockResolvedValue([])
}))
ContextExtractor: vi.fn().mockImplementation(function () {
this.findRelevantNotes = vi.fn().mockResolvedValue([])
});
}));
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(() => ({
messages: {
const mockAnthropic = vi.fn().mockImplementation(function () {
this.messages = {
create: vi.fn().mockImplementation((params) => {
if (params.stream) {
return Promise.resolve(mockStream);
@ -71,8 +71,8 @@ vi.mock('@anthropic-ai/sdk', () => {
}
});
})
}
}));
};
});
return { default: mockAnthropic };
});
@ -127,7 +127,9 @@ describe('AnthropicService', () => {
}
};
AnthropicMock.mockImplementation(() => mockAnthropicInstance);
AnthropicMock.mockImplementation(function () {
Object.assign(this, mockAnthropicInstance);
});
service = new AnthropicService();
});

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