File size: 2,928 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 | /**
* @license
* Copyright 2025 Google LLC
* SPDX-License-Identifier: Apache-2.0
*/
import { describe, it, expect } from 'vitest';
import { flattenMemory } from './memory.js';
describe('memory', () => {
describe('flattenMemory', () => {
it('should return empty string for null or undefined', () => {
expect(flattenMemory(undefined)).toBe('');
expect(flattenMemory(null as unknown as undefined)).toBe('');
});
it('should return the string itself if a string is provided', () => {
expect(flattenMemory('raw string')).toBe('raw string');
});
it('should return empty string for an empty object', () => {
expect(flattenMemory({})).toBe('');
});
it('should return content with headers even if only global memory is present', () => {
expect(flattenMemory({ global: 'global content' })).toBe(
`--- Global ---
global content`,
);
});
it('should return content with headers even if only extension memory is present', () => {
expect(flattenMemory({ extension: 'extension content' })).toBe(
`--- Extension ---
extension content`,
);
});
it('should return content with headers even if only project memory is present', () => {
expect(flattenMemory({ project: 'project content' })).toBe(
`--- Project ---
project content`,
);
});
it('should include headers if multiple levels are present (global + project)', () => {
const result = flattenMemory({
global: 'global content',
project: 'project content',
});
expect(result).toContain('--- Global ---');
expect(result).toContain('global content');
expect(result).toContain('--- Project ---');
expect(result).toContain('project content');
expect(result).not.toContain('--- Extension ---');
});
it('should include headers if all levels are present', () => {
const result = flattenMemory({
global: 'global content',
extension: 'extension content',
project: 'project content',
});
expect(result).toContain('--- Global ---');
expect(result).toContain('--- Extension ---');
expect(result).toContain('--- Project ---');
expect(result).toBe(
`--- Global ---
global content
--- Extension ---
extension content
--- Project ---
project content`,
);
});
it('should trim content and ignore empty strings', () => {
const result = flattenMemory({
global: ' trimmed global ',
extension: ' ',
project: 'project\n',
});
expect(result).toBe(
`--- Global ---
trimmed global
--- Project ---
project`,
);
});
it('should return empty string if all levels are only whitespace', () => {
expect(
flattenMemory({
global: ' ',
extension: '\n',
project: ' ',
}),
).toBe('');
});
});
});
|