File size: 4,059 Bytes
7a1ad33 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 | /**
* @license
* Copyright 2026 Google LLC
* SPDX-License-Identifier: Apache-2.0
*/
import { describe, it, expect, vi, beforeEach } from 'vitest';
import { discoverJitContext, appendJitContext } from './jit-context.js';
import type { Config } from '../config/config.js';
import type { MemoryContextManager } from '../context/memoryContextManager.js';
describe('jit-context', () => {
describe('discoverJitContext', () => {
let mockConfig: Config;
let mockMemoryContextManager: MemoryContextManager;
beforeEach(() => {
mockMemoryContextManager = {
discoverContext: vi.fn().mockResolvedValue(''),
} as unknown as MemoryContextManager;
mockConfig = {
getMemoryContextManager: vi
.fn()
.mockReturnValue(mockMemoryContextManager),
getWorkspaceContext: vi.fn().mockReturnValue({
getDirectories: vi.fn().mockReturnValue(['/app']),
}),
} as unknown as Config;
});
it('should return empty string when memoryContextManager is undefined', async () => {
vi.mocked(mockConfig.getMemoryContextManager).mockReturnValue(undefined);
const result = await discoverJitContext(mockConfig, '/app/src/file.ts');
expect(result).toBe('');
});
it('should call memoryContextManager.discoverContext with correct args', async () => {
vi.mocked(mockMemoryContextManager.discoverContext).mockResolvedValue(
'Subdirectory context content',
);
const result = await discoverJitContext(mockConfig, '/app/src/file.ts');
expect(mockMemoryContextManager.discoverContext).toHaveBeenCalledWith(
'/app/src/file.ts',
['/app'],
);
expect(result).toBe('Subdirectory context content');
});
it('should pass all workspace directories as trusted roots', async () => {
vi.mocked(mockConfig.getWorkspaceContext).mockReturnValue({
getDirectories: vi.fn().mockReturnValue(['/app', '/lib']),
} as unknown as ReturnType<Config['getWorkspaceContext']>);
vi.mocked(mockMemoryContextManager.discoverContext).mockResolvedValue('');
await discoverJitContext(mockConfig, '/app/src/file.ts');
expect(mockMemoryContextManager.discoverContext).toHaveBeenCalledWith(
'/app/src/file.ts',
['/app', '/lib'],
);
});
it('should return empty string when no new context is found', async () => {
vi.mocked(mockMemoryContextManager.discoverContext).mockResolvedValue('');
const result = await discoverJitContext(mockConfig, '/app/src/file.ts');
expect(result).toBe('');
});
it('should return empty string when discoverContext throws', async () => {
vi.mocked(mockMemoryContextManager.discoverContext).mockRejectedValue(
new Error('Permission denied'),
);
const result = await discoverJitContext(mockConfig, '/app/src/file.ts');
expect(result).toBe('');
});
});
describe('appendJitContext', () => {
it('should return original content when jitContext is empty', () => {
const content = 'file contents here';
const result = appendJitContext(content, '');
expect(result).toBe(content);
});
it('should append delimited context when jitContext is non-empty', () => {
const content = 'file contents here';
const jitContext = 'Use the useAuth hook.';
const result = appendJitContext(content, jitContext);
expect(result).toContain(content);
expect(result).toContain('--- Newly Discovered Project Context ---');
expect(result).toContain(jitContext);
expect(result).toContain('--- End Project Context ---');
});
it('should place context after the original content', () => {
const content = 'original output';
const jitContext = 'context rules';
const result = appendJitContext(content, jitContext);
const contentIndex = result.indexOf(content);
const contextIndex = result.indexOf(jitContext);
expect(contentIndex).toBeLessThan(contextIndex);
});
});
});
|