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

test: remove unused imports and update type references

Browse source
This commit is contained in:
Andrew Beal 2025-11-01 11:50:11 +00:00
parent 72ddbba72e
commit f8f606bdf6
8 changed files with 32 additions and 33 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

3
__tests__/Services/AIFunctionService.test.ts
View file

@ -1,10 +1,9 @@
import { describe, it, expect, beforeEach, afterEach, vi } from 'vitest';
import { AIFunctionService } from '../../Services/AIFunctionService';
import { FileSystemService } from '../../Services/FileSystemService';
import { RegisterSingleton, DeregisterAllServices } from '../../Services/DependencyService';
import { Services } from '../../Services/Services';
import { AIFunction } from '../../Enums/AIFunction';
import { TFile, TFolder } from 'obsidian';
import { TFile } from 'obsidian';
/**
* INTEGRATION TESTS - AIFunctionService

2
__tests__/Services/ChatService.test.ts
View file

@ -168,7 +168,7 @@ describe('ChatService - Integration Tests (Sync Methods Only)', () => {
const conversation = new Conversation();
conversation.contents.push(new ConversationContent(Role.User, 'Regular message'));
conversation.contents.push(
new ConversationContent(Role.User, 'Function result', '', new Date(), false, true, 'tool-id')
new ConversationContent(Role.User, 'Function result', '', '', new Date(), false, true, 'tool-id')
);
await service.updateTokenDisplay(conversation);

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

@ -1,6 +1,5 @@
import { describe, it, expect, beforeEach, afterEach, vi } from 'vitest';
import { ConversationFileSystemService } from '../../Services/ConversationFileSystemService';
import { FileSystemService } from '../../Services/FileSystemService';
import { RegisterSingleton, DeregisterAllServices } from '../../Services/DependencyService';
import { Services } from '../../Services/Services';
import { Conversation } from '../../Conversations/Conversation';

3
__tests__/Services/InputService.test.ts
View file

@ -1,6 +1,5 @@
import { describe, it, expect, beforeEach, vi } from 'vitest';
import { describe, it, expect, beforeEach, vi, afterEach } from 'vitest';
import { InputService } from '../../Services/InputService';
import { SearchTrigger } from '../../Enums/SearchTrigger';
/**
* UNIT TESTS

2
__tests__/Services/SanitiserService.test.ts
View file

@ -1,5 +1,5 @@
import { describe, it, expect, beforeEach } from 'vitest';
import { SanitiserService, type SanitizeOptions } from '../../Services/SanitiserService';
import { SanitiserService } from '../../Services/SanitiserService';
describe('SanitiserService', () => {
let service: SanitiserService;

48
__tests__/Services/StreamingService.test.ts
View file

@ -1,5 +1,5 @@
import { describe, it, expect, vi, beforeEach, afterEach } from 'vitest';
import { StreamingService, StreamChunk } from '../../Services/StreamingService';
import { StreamingService, IStreamChunk } from '../../Services/StreamingService';
import { Selector } from '../../Enums/Selector';
/**
@ -44,7 +44,7 @@ describe('StreamingService', () => {
}
// Helper to create a simple parser
const simpleParser = (chunk: string): StreamChunk => {
const simpleParser = (chunk: string): IStreamChunk => {
try {
const data = JSON.parse(chunk);
return {
@ -70,7 +70,7 @@ describe('StreamingService', () => {
body: createMockStream(chunks)
});
const results: StreamChunk[] = [];
const results: IStreamChunk[] = [];
for await (const chunk of service.streamRequest(
'https://api.example.com/stream',
{ prompt: 'test' },
@ -155,7 +155,7 @@ describe('StreamingService', () => {
body: createMockStream(chunks)
});
const results: StreamChunk[] = [];
const results: IStreamChunk[] = [];
for await (const chunk of service.streamRequest(
'https://api.example.com/stream',
{},
@ -184,7 +184,7 @@ describe('StreamingService', () => {
body: createMockStream(chunks)
});
const results: StreamChunk[] = [];
const results: IStreamChunk[] = [];
for await (const chunk of service.streamRequest(
'https://api.example.com/stream',
{},
@ -210,7 +210,7 @@ describe('StreamingService', () => {
body: createMockStream(chunks)
});
const results: StreamChunk[] = [];
const results: IStreamChunk[] = [];
for await (const chunk of service.streamRequest(
'https://api.example.com/stream',
{},
@ -239,7 +239,7 @@ describe('StreamingService', () => {
body: createMockStream(chunks)
});
const results: StreamChunk[] = [];
const results: IStreamChunk[] = [];
for await (const chunk of service.streamRequest(
'https://api.example.com/stream',
{},
@ -263,7 +263,7 @@ describe('StreamingService', () => {
body: createMockStream(chunks)
});
const results: StreamChunk[] = [];
const results: IStreamChunk[] = [];
for await (const chunk of service.streamRequest(
'https://api.example.com/stream',
{},
@ -289,7 +289,7 @@ describe('StreamingService', () => {
body: createMockStream(chunks)
});
const results: StreamChunk[] = [];
const results: IStreamChunk[] = [];
for await (const chunk of service.streamRequest(
'https://api.example.com/stream',
{},
@ -317,7 +317,7 @@ describe('StreamingService', () => {
body: createMockStream(chunks)
});
const results: StreamChunk[] = [];
const results: IStreamChunk[] = [];
for await (const chunk of service.streamRequest(
'https://api.example.com/stream',
{},
@ -341,7 +341,7 @@ describe('StreamingService', () => {
body: createMockStream(chunks)
});
const results: StreamChunk[] = [];
const results: IStreamChunk[] = [];
for await (const chunk of service.streamRequest(
'https://api.example.com/stream',
{},
@ -367,7 +367,7 @@ describe('StreamingService', () => {
body: createMockStream(chunks)
});
const customParser = (chunk: string): StreamChunk => {
const customParser = (chunk: string): IStreamChunk => {
const trimmedChunk = chunk.trim();
return {
content: `Parsed: ${trimmedChunk}`,
@ -375,7 +375,7 @@ describe('StreamingService', () => {
};
};
const results: StreamChunk[] = [];
const results: IStreamChunk[] = [];
for await (const chunk of service.streamRequest(
'https://api.example.com/stream',
{},
@ -400,7 +400,7 @@ describe('StreamingService', () => {
body: createMockStream(chunks)
});
const parserWithFunctionCall = (chunk: string): StreamChunk => {
const parserWithFunctionCall = (chunk: string): IStreamChunk => {
const data = JSON.parse(chunk);
return {
content: data.content || '',
@ -409,7 +409,7 @@ describe('StreamingService', () => {
};
};
const results: StreamChunk[] = [];
const results: IStreamChunk[] = [];
for await (const chunk of service.streamRequest(
'https://api.example.com/stream',
{},
@ -453,7 +453,7 @@ describe('StreamingService', () => {
abortError.name = 'AbortError';
mockFetch.mockRejectedValue(abortError);
const results: StreamChunk[] = [];
const results: IStreamChunk[] = [];
for await (const chunk of service.streamRequest(
'https://api.example.com/stream',
{},
@ -480,7 +480,7 @@ describe('StreamingService', () => {
text: async () => 'Resource not found'
});
const results: StreamChunk[] = [];
const results: IStreamChunk[] = [];
for await (const chunk of service.streamRequest(
'https://api.example.com/stream',
{},
@ -501,7 +501,7 @@ describe('StreamingService', () => {
body: null
});
const results: StreamChunk[] = [];
const results: IStreamChunk[] = [];
for await (const chunk of service.streamRequest(
'https://api.example.com/stream',
{},
@ -518,7 +518,7 @@ describe('StreamingService', () => {
it('should handle network error', async () => {
mockFetch.mockRejectedValue(new Error('Network connection failed'));
const results: StreamChunk[] = [];
const results: IStreamChunk[] = [];
for await (const chunk of service.streamRequest(
'https://api.example.com/stream',
{},
@ -535,7 +535,7 @@ describe('StreamingService', () => {
it('should handle unknown error type', async () => {
mockFetch.mockRejectedValue('String error');
const results: StreamChunk[] = [];
const results: IStreamChunk[] = [];
for await (const chunk of service.streamRequest(
'https://api.example.com/stream',
{},
@ -557,7 +557,7 @@ describe('StreamingService', () => {
body: createMockStream([])
});
const results: StreamChunk[] = [];
const results: IStreamChunk[] = [];
for await (const chunk of service.streamRequest(
'https://api.example.com/stream',
{},
@ -576,7 +576,7 @@ describe('StreamingService', () => {
body: createMockStream([' \n\n \n'])
});
const results: StreamChunk[] = [];
const results: IStreamChunk[] = [];
for await (const chunk of service.streamRequest(
'https://api.example.com/stream',
{},
@ -595,7 +595,7 @@ describe('StreamingService', () => {
body: createMockStream(['data: {"content":"test","done":true}'])
});
const results: StreamChunk[] = [];
const results: IStreamChunk[] = [];
for await (const chunk of service.streamRequest(
'https://api.example.com/stream',
{},
@ -621,7 +621,7 @@ describe('StreamingService', () => {
body: createMockStream(chunks)
});
const results: StreamChunk[] = [];
const results: IStreamChunk[] = [];
for await (const chunk of service.streamRequest(
'https://api.example.com/stream',
{},

2
__tests__/Services/VaultCacheService.test.ts
View file

@ -1,7 +1,7 @@
import { describe, it, expect, vi, beforeEach, afterEach } from 'vitest';
import { VaultCacheService } from '../../Services/VaultCacheService';
import { VaultService } from '../../Services/VaultService';
import { TFile, TFolder, MetadataCache } from 'obsidian';
import { TFile, TFolder } from 'obsidian';
import { RegisterSingleton, DeregisterAllServices } from '../../Services/DependencyService';
import { Services } from '../../Services/Services';
import { FileEvent } from '../../Enums/FileEvent';

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

@ -32,7 +32,9 @@ const mockVault = {
const mockFileManager = {
renameFile: vi.fn()
} as unknown as FileManager;
} as unknown as FileManager & {
renameFile: ReturnType<typeof vi.fn>;
};
// Create a mutable plugin settings object that tests can modify
const mockPluginSettings = {