wehub-resource-sync
070959e133
landing-page-staging / Deploy landing page to staging (push) Has been skipped
landing-page-ci / Validate landing page (push) Failing after 4s
visual-baseline / Capture visual baselines (push) Has been cancelled
bake-plugin-previews / Bake plugin previews (push) Has been cancelled
78 lines
2.3 KiB
TypeScript
78 lines
2.3 KiB
TypeScript
import { expect, test } from 'vitest';
|
|
|
|
import { resolveSafePromptImagePaths, selectPromptImagePaths } from '../src/server.js';
|
|
|
|
test('selectPromptImagePaths uses staged AMR paths in prompt text', () => {
|
|
expect(
|
|
selectPromptImagePaths(
|
|
'amr',
|
|
['/tmp/od-uploads/original.png'],
|
|
['/project/.amr-attachments/staged.png'],
|
|
),
|
|
).toEqual(['/project/.amr-attachments/staged.png']);
|
|
});
|
|
|
|
test('selectPromptImagePaths keeps original paths for non-AMR agents', () => {
|
|
expect(
|
|
selectPromptImagePaths(
|
|
'opencode',
|
|
['/tmp/od-uploads/original.png'],
|
|
['/project/.amr-attachments/staged.png'],
|
|
),
|
|
).toEqual(['/tmp/od-uploads/original.png']);
|
|
});
|
|
|
|
test('resolveSafePromptImagePaths rejects images larger than 1 MB', () => {
|
|
const result = resolveSafePromptImagePaths(
|
|
['/tmp/od-uploads/too-large.png', '/tmp/od-uploads/ok.png'],
|
|
{
|
|
uploadDir: '/tmp/od-uploads',
|
|
existsSync: () => true,
|
|
statSync: (inputPath: string) => ({
|
|
isFile: () => true,
|
|
size: inputPath.endsWith('too-large.png') ? 1024 * 1024 + 1 : 1024,
|
|
}),
|
|
},
|
|
);
|
|
|
|
expect(result.safeImages).toEqual(['/tmp/od-uploads/ok.png']);
|
|
expect(result.oversizedImages).toEqual([
|
|
{ path: '/tmp/od-uploads/too-large.png', sizeBytes: 1024 * 1024 + 1 },
|
|
]);
|
|
});
|
|
|
|
test('resolveSafePromptImagePaths keeps images at or below 1 MB', () => {
|
|
const result = resolveSafePromptImagePaths(
|
|
['/tmp/od-uploads/exactly-1mb.png'],
|
|
{
|
|
uploadDir: '/tmp/od-uploads',
|
|
existsSync: () => true,
|
|
statSync: () => ({
|
|
isFile: () => true,
|
|
size: 1024 * 1024,
|
|
}),
|
|
},
|
|
);
|
|
|
|
expect(result.safeImages).toEqual(['/tmp/od-uploads/exactly-1mb.png']);
|
|
expect(result.oversizedImages).toEqual([]);
|
|
});
|
|
|
|
test('resolveSafePromptImagePaths surfaces stat failures instead of dropping the image', () => {
|
|
const result = resolveSafePromptImagePaths(['/tmp/od-uploads/unreadable.png'], {
|
|
uploadDir: '/tmp/od-uploads',
|
|
existsSync: () => true,
|
|
statSync: () => {
|
|
throw Object.assign(new Error('EACCES: permission denied'), {
|
|
code: 'EACCES',
|
|
});
|
|
},
|
|
});
|
|
|
|
expect(result.safeImages).toEqual([]);
|
|
expect(result.oversizedImages).toEqual([]);
|
|
expect(result.failedImages).toEqual([
|
|
{ path: '/tmp/od-uploads/unreadable.png', error: 'EACCES: permission denied' },
|
|
]);
|
|
});
|