From c5a3a4a5fb288d6ac1fa166f53093a0e4dca1f53 Mon Sep 17 00:00:00 2001 From: Andrew Beal Date: Mon, 5 Jan 2026 22:46:51 +0000 Subject: [PATCH] Update unit tests. --- __tests__/AIClasses/Claude.test.ts | 4 +-- __tests__/AIClasses/Gemini.test.ts | 12 +++---- __tests__/AIClasses/OpenAI.test.ts | 32 +++++++++---------- .../VaultCacheService.performance.test.ts | 1 + __tests__/Services/VaultService.pdf.test.ts | 1 + .../Services/VaultService.performance.test.ts | 1 + __tests__/Services/VaultService.test.ts | 1 + 7 files changed, 28 insertions(+), 24 deletions(-) diff --git a/__tests__/AIClasses/Claude.test.ts b/__tests__/AIClasses/Claude.test.ts index 65e1e5c..66fd17b 100644 --- a/__tests__/AIClasses/Claude.test.ts +++ b/__tests__/AIClasses/Claude.test.ts @@ -1163,7 +1163,7 @@ describe('Claude', () => { yield { content: 'response', isComplete: true }; }); - const generator = claude.streamRequest(conversation); + const generator = claude.streamRequest(conversation, false); // Consume the generator for await (const chunk of generator) { @@ -1209,7 +1209,7 @@ describe('Claude', () => { yield { content: 'done', isComplete: true }; }); - const generator = claude.streamRequest(conversation); + const generator = claude.streamRequest(conversation, false); // Start consuming await generator.next(); diff --git a/__tests__/AIClasses/Gemini.test.ts b/__tests__/AIClasses/Gemini.test.ts index 642a90a..cf6e265 100644 --- a/__tests__/AIClasses/Gemini.test.ts +++ b/__tests__/AIClasses/Gemini.test.ts @@ -344,7 +344,7 @@ describe('Gemini', () => { yield { content: 'done', isComplete: true }; }); - const generator = gemini.streamRequest(conversation); + const generator = gemini.streamRequest(conversation, false); for await (const chunk of generator) {} const callArgs = mockStreamingService.streamRequest.mock.calls[0]; @@ -372,7 +372,7 @@ describe('Gemini', () => { yield { content: 'done', isComplete: true }; }); - const generator = gemini.streamRequest(conversation); + const generator = gemini.streamRequest(conversation, false); for await (const chunk of generator) {} const callArgs = mockStreamingService.streamRequest.mock.calls[0]; @@ -393,7 +393,7 @@ describe('Gemini', () => { yield { content: 'done', isComplete: true }; }); - const generator = gemini.streamRequest(conversation); + const generator = gemini.streamRequest(conversation, false); for await (const chunk of generator) {} const callArgs = mockStreamingService.streamRequest.mock.calls[0]; @@ -416,7 +416,7 @@ describe('Gemini', () => { yield { content: 'done', isComplete: true }; }); - const generator = gemini.streamRequest(conversation); + const generator = gemini.streamRequest(conversation, false); for await (const chunk of generator) {} const callArgs = mockStreamingService.streamRequest.mock.calls[0]; @@ -1077,7 +1077,7 @@ describe('Gemini', () => { yield { content: 'done', isComplete: true }; }); - const generator = gemini.streamRequest(conversation); + const generator = gemini.streamRequest(conversation, false); for await (const chunk of generator) {} @@ -1108,7 +1108,7 @@ describe('Gemini', () => { yield { content: 'done', isComplete: true }; }); - const generator = gemini.streamRequest(conversation); + const generator = gemini.streamRequest(conversation, false); await generator.next(); // State should be reset (after checking web search mode) diff --git a/__tests__/AIClasses/OpenAI.test.ts b/__tests__/AIClasses/OpenAI.test.ts index 6fe96f0..ac38fd5 100644 --- a/__tests__/AIClasses/OpenAI.test.ts +++ b/__tests__/AIClasses/OpenAI.test.ts @@ -317,7 +317,7 @@ describe('OpenAI', () => { yield { content: 'response', isComplete: true }; }); - const generator = openai.streamRequest(conversation); + const generator = openai.streamRequest(conversation, false); // Consume generator for await (const chunk of generator) { @@ -351,7 +351,7 @@ describe('OpenAI', () => { yield { content: 'done', isComplete: true }; }); - const generator = openai.streamRequest(conversation); + const generator = openai.streamRequest(conversation, false); for await (const chunk of generator) {} const callArgs = mockStreamingService.streamRequest.mock.calls[0]; @@ -411,7 +411,7 @@ describe('OpenAI', () => { yield { content: 'done', isComplete: true }; }); - const generator = openai.streamRequest(conversation); + const generator = openai.streamRequest(conversation, false); for await (const chunk of generator) {} const callArgs = mockStreamingService.streamRequest.mock.calls[0]; @@ -440,7 +440,7 @@ describe('OpenAI', () => { yield { content: 'done', isComplete: true }; }); - const generator = openai.streamRequest(conversation); + const generator = openai.streamRequest(conversation, false); for await (const chunk of generator) {} const callArgs = mockStreamingService.streamRequest.mock.calls[0]; @@ -483,7 +483,7 @@ describe('OpenAI', () => { yield { content: 'done', isComplete: true }; }); - const generator = openai.streamRequest(conversation); + const generator = openai.streamRequest(conversation, false); for await (const chunk of generator) {} const callArgs = mockStreamingService.streamRequest.mock.calls[0]; @@ -506,7 +506,7 @@ describe('OpenAI', () => { yield { content: 'done', isComplete: true }; }); - const generator = openai.streamRequest(conversation); + const generator = openai.streamRequest(conversation, false); for await (const chunk of generator) {} const callArgs = mockStreamingService.streamRequest.mock.calls[0]; @@ -537,7 +537,7 @@ describe('OpenAI', () => { yield { content: 'done', isComplete: true }; }); - const generator = openai.streamRequest(conversation); + const generator = openai.streamRequest(conversation, false); for await (const chunk of generator) {} const callArgs = mockStreamingService.streamRequest.mock.calls[0]; @@ -582,7 +582,7 @@ describe('OpenAI', () => { yield { content: 'done', isComplete: true }; }); - const generator = openai.streamRequest(conversation); + const generator = openai.streamRequest(conversation, false); for await (const chunk of generator) {} const callArgs = mockStreamingService.streamRequest.mock.calls[0]; @@ -627,7 +627,7 @@ describe('OpenAI', () => { yield { content: 'done', isComplete: true }; }); - const generator = openai.streamRequest(conversation); + const generator = openai.streamRequest(conversation, false); for await (const chunk of generator) {} const callArgs = mockStreamingService.streamRequest.mock.calls[0]; @@ -681,7 +681,7 @@ describe('OpenAI', () => { yield { content: 'done', isComplete: true }; }); - const generator = openai.streamRequest(conversation); + const generator = openai.streamRequest(conversation, false); for await (const chunk of generator) {} const callArgs = mockStreamingService.streamRequest.mock.calls[0]; @@ -714,7 +714,7 @@ describe('OpenAI', () => { yield { content: 'done', isComplete: true }; }); - const generator = openai.streamRequest(conversation); + const generator = openai.streamRequest(conversation, false); for await (const chunk of generator) {} const callArgs = mockStreamingService.streamRequest.mock.calls[0]; @@ -747,7 +747,7 @@ describe('OpenAI', () => { yield { content: 'done', isComplete: true }; }); - const generator = openai.streamRequest(conversation); + const generator = openai.streamRequest(conversation, false); for await (const chunk of generator) {} const callArgs = mockStreamingService.streamRequest.mock.calls[0]; @@ -804,7 +804,7 @@ describe('OpenAI', () => { yield { content: 'done', isComplete: true }; }); - const generator = openai.streamRequest(conversation); + const generator = openai.streamRequest(conversation, false); for await (const chunk of generator) {} const callArgs = mockStreamingService.streamRequest.mock.calls[0]; @@ -870,7 +870,7 @@ describe('OpenAI', () => { yield { content: 'done', isComplete: true }; }); - const generator = openai.streamRequest(conversation); + const generator = openai.streamRequest(conversation, false); for await (const chunk of generator) {} const callArgs = mockStreamingService.streamRequest.mock.calls[0]; @@ -951,7 +951,7 @@ describe('OpenAI', () => { yield { content: 'response', isComplete: true }; }); - const generator = openai.streamRequest(conversation); + const generator = openai.streamRequest(conversation, false); for await (const chunk of generator) { // Just consume @@ -983,7 +983,7 @@ describe('OpenAI', () => { yield { content: 'done', isComplete: true }; }); - const generator = openai.streamRequest(conversation); + const generator = openai.streamRequest(conversation, false); for await (const chunk of generator) {} const callArgs = mockStreamingService.streamRequest.mock.calls[0]; diff --git a/__tests__/Services/VaultCacheService.performance.test.ts b/__tests__/Services/VaultCacheService.performance.test.ts index c25409d..00d543e 100644 --- a/__tests__/Services/VaultCacheService.performance.test.ts +++ b/__tests__/Services/VaultCacheService.performance.test.ts @@ -89,6 +89,7 @@ const mockPlugin = { const mockSettings: IVaultkeeperAISettings = { firstTimeStart: false, model: AIProviderModel.ClaudeSonnet_4_5, + planningModel: AIProviderModel.ClaudeSonnet_4_5, apiKeys: { claude: 'test-claude-key', openai: 'test-openai-key', diff --git a/__tests__/Services/VaultService.pdf.test.ts b/__tests__/Services/VaultService.pdf.test.ts index e7eb463..e5a2d30 100644 --- a/__tests__/Services/VaultService.pdf.test.ts +++ b/__tests__/Services/VaultService.pdf.test.ts @@ -50,6 +50,7 @@ const mockFileManager = { const mockSettings: IVaultkeeperAISettings = { firstTimeStart: false, model: AIProviderModel.ClaudeSonnet_4_5, + planningModel: AIProviderModel.ClaudeSonnet_4_5, apiKeys: { claude: 'test-claude-key', openai: 'test-openai-key', diff --git a/__tests__/Services/VaultService.performance.test.ts b/__tests__/Services/VaultService.performance.test.ts index a75300f..f4d432d 100644 --- a/__tests__/Services/VaultService.performance.test.ts +++ b/__tests__/Services/VaultService.performance.test.ts @@ -51,6 +51,7 @@ const mockPlugin = { const mockSettings: IVaultkeeperAISettings = { firstTimeStart: false, model: AIProviderModel.ClaudeSonnet_4_5, + planningModel: AIProviderModel.ClaudeSonnet_4_5, apiKeys: { claude: 'test-claude-key', openai: 'test-openai-key', diff --git a/__tests__/Services/VaultService.test.ts b/__tests__/Services/VaultService.test.ts index a84fc7b..1b254cc 100644 --- a/__tests__/Services/VaultService.test.ts +++ b/__tests__/Services/VaultService.test.ts @@ -49,6 +49,7 @@ const mockFileManager = { const mockSettings: IVaultkeeperAISettings = { firstTimeStart: false, model: AIProviderModel.ClaudeSonnet_4_5, + planningModel: AIProviderModel.ClaudeSonnet_4_5, apiKeys: { claude: 'test-claude-key', openai: 'test-openai-key',