obsidian-plugin/andy-stack_vaultkeeper-ai
1
0
Fork 0
mirror of https://github.com/andy-stack/vaultkeeper-ai.git synced 2026-07-22 06:42:03 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

Update unit tests.

Browse source
This commit is contained in:
Andrew Beal 2026-01-05 22:46:51 +00:00
parent aa1e34185a
commit c5a3a4a5fb
7 changed files with 28 additions and 24 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

4
__tests__/AIClasses/Claude.test.ts
View file

@ -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();

12
__tests__/AIClasses/Gemini.test.ts
View file

@ -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)

32
__tests__/AIClasses/OpenAI.test.ts
View file

@ -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];

1
__tests__/Services/VaultCacheService.performance.test.ts
View file

@ -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',

1
__tests__/Services/VaultService.pdf.test.ts
View file

@ -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',

1
__tests__/Services/VaultService.performance.test.ts
View file

@ -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',

1
__tests__/Services/VaultService.test.ts
View file

@ -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',