chore: import upstream snapshot with attribution
CI / Shell Format Check (push) Has been cancelled
CI / Check Ruby (3.4) (push) Has been cancelled
CI / CI Config (push) Has been cancelled
CI / Test on Node ${{ matrix.node }} and ${{ matrix.os }}${{ matrix.shard && format(' (shard {0}/3)', matrix.shard) || '' }} (push) Has been cancelled
CI / Build on Node ${{ matrix.node }} (push) Has been cancelled
CI / Style Check (push) Has been cancelled
CI / Generate Assets (push) Has been cancelled
CI / Check Python (3.14) (push) Has been cancelled
CI / Check Python (3.9) (push) Has been cancelled
CI / Build Docs (push) Has been cancelled
CI / Code Scan Action (push) Has been cancelled
CI / Site tests (push) Has been cancelled
CI / webui tests (push) Has been cancelled
CI / Run Integration Tests (push) Has been cancelled
CI / Run Smoke Tests (push) Has been cancelled
CI / Go Tests (push) Has been cancelled
CI / Share Test (push) Has been cancelled
CI / Redteam (Production API) (push) Has been cancelled
CI / Redteam (Staging API) (push) Has been cancelled
CI / GitHub Actions Lint (push) Has been cancelled
CI / Check Ruby (3.0) (push) Has been cancelled
release-please / release-please (push) Has been cancelled
release-please / build (push) Has been cancelled
release-please / publish-npm (push) Has been cancelled
release-please / publish-npm-backfill (push) Has been cancelled
release-please / docker (push) Has been cancelled
release-please / publish-code-scan-action (push) Has been cancelled
release-please / attest-code-scan-action (push) Has been cancelled
Deploy local.promptfoo.app / Deploy to Cloudflare Pages (push) Has been cancelled
Test and Publish Multi-arch Docker Image / test (push) Has been cancelled
Test and Publish Multi-arch Docker Image / build-docker-and-push-digests (map[digest-suffix:linux-amd64 platform:linux/amd64 runner:ubuntu-latest]) (push) Has been cancelled
Test and Publish Multi-arch Docker Image / build-docker-and-push-digests (map[digest-suffix:linux-arm64 platform:linux/arm64 runner:ubuntu-24.04-arm]) (push) Has been cancelled
Test and Publish Multi-arch Docker Image / merge-docker-digests (push) Has been cancelled
Test and Publish Multi-arch Docker Image / Attest Multi-arch Image (push) Has been cancelled
Validate Renovate Config / Validate Renovate Configuration (push) Has been cancelled
CI / Shell Format Check (push) Has been cancelled
CI / Check Ruby (3.4) (push) Has been cancelled
CI / CI Config (push) Has been cancelled
CI / Test on Node ${{ matrix.node }} and ${{ matrix.os }}${{ matrix.shard && format(' (shard {0}/3)', matrix.shard) || '' }} (push) Has been cancelled
CI / Build on Node ${{ matrix.node }} (push) Has been cancelled
CI / Style Check (push) Has been cancelled
CI / Generate Assets (push) Has been cancelled
CI / Check Python (3.14) (push) Has been cancelled
CI / Check Python (3.9) (push) Has been cancelled
CI / Build Docs (push) Has been cancelled
CI / Code Scan Action (push) Has been cancelled
CI / Site tests (push) Has been cancelled
CI / webui tests (push) Has been cancelled
CI / Run Integration Tests (push) Has been cancelled
CI / Run Smoke Tests (push) Has been cancelled
CI / Go Tests (push) Has been cancelled
CI / Share Test (push) Has been cancelled
CI / Redteam (Production API) (push) Has been cancelled
CI / Redteam (Staging API) (push) Has been cancelled
CI / GitHub Actions Lint (push) Has been cancelled
CI / Check Ruby (3.0) (push) Has been cancelled
release-please / release-please (push) Has been cancelled
release-please / build (push) Has been cancelled
release-please / publish-npm (push) Has been cancelled
release-please / publish-npm-backfill (push) Has been cancelled
release-please / docker (push) Has been cancelled
release-please / publish-code-scan-action (push) Has been cancelled
release-please / attest-code-scan-action (push) Has been cancelled
Deploy local.promptfoo.app / Deploy to Cloudflare Pages (push) Has been cancelled
Test and Publish Multi-arch Docker Image / test (push) Has been cancelled
Test and Publish Multi-arch Docker Image / build-docker-and-push-digests (map[digest-suffix:linux-amd64 platform:linux/amd64 runner:ubuntu-latest]) (push) Has been cancelled
Test and Publish Multi-arch Docker Image / build-docker-and-push-digests (map[digest-suffix:linux-arm64 platform:linux/arm64 runner:ubuntu-24.04-arm]) (push) Has been cancelled
Test and Publish Multi-arch Docker Image / merge-docker-digests (push) Has been cancelled
Test and Publish Multi-arch Docker Image / Attest Multi-arch Image (push) Has been cancelled
Validate Renovate Config / Validate Renovate Configuration (push) Has been cancelled
This commit is contained in:
@@ -0,0 +1,155 @@
|
||||
import * as fs from 'fs';
|
||||
|
||||
import { beforeEach, describe, expect, it, vi } from 'vitest';
|
||||
import { processJsonFile } from '../../../src/prompts/processors/json';
|
||||
import * as fileModule from '../../../src/util/file';
|
||||
|
||||
vi.mock('fs');
|
||||
vi.mock('../../../src/util/file');
|
||||
|
||||
describe('processJsonFile', () => {
|
||||
const mockReadFileSync = vi.mocked(fs.readFileSync);
|
||||
const mockMaybeLoadConfigFromExternalFile = vi.mocked(fileModule.maybeLoadConfigFromExternalFile);
|
||||
|
||||
beforeEach(() => {
|
||||
vi.clearAllMocks();
|
||||
// By default, maybeLoadConfigFromExternalFile returns its input unchanged
|
||||
mockMaybeLoadConfigFromExternalFile.mockImplementation((input) => input);
|
||||
});
|
||||
|
||||
it('should process a valid JSON file without a label', () => {
|
||||
const filePath = 'file.json';
|
||||
const fileContent = JSON.stringify({ key: 'value' });
|
||||
mockReadFileSync.mockReturnValue(fileContent);
|
||||
expect(processJsonFile(filePath, {})).toEqual([
|
||||
{
|
||||
raw: fileContent,
|
||||
label: `${filePath}: ${fileContent}`,
|
||||
},
|
||||
]);
|
||||
expect(mockReadFileSync).toHaveBeenCalledWith(filePath, 'utf8');
|
||||
});
|
||||
|
||||
it('should process a valid JSON file with a label', () => {
|
||||
const filePath = 'file.json';
|
||||
const fileContent = JSON.stringify({ key: 'value' });
|
||||
mockReadFileSync.mockReturnValue(fileContent);
|
||||
expect(processJsonFile(filePath, { label: 'Label' })).toEqual([
|
||||
{
|
||||
raw: fileContent,
|
||||
label: `Label`,
|
||||
},
|
||||
]);
|
||||
expect(mockReadFileSync).toHaveBeenCalledWith(filePath, 'utf8');
|
||||
});
|
||||
|
||||
it('should throw an error if the file cannot be read', () => {
|
||||
const filePath = 'nonexistent.json';
|
||||
mockReadFileSync.mockImplementation(() => {
|
||||
throw new Error('File not found');
|
||||
});
|
||||
|
||||
expect(() => processJsonFile(filePath, {})).toThrow('File not found');
|
||||
expect(mockReadFileSync).toHaveBeenCalledWith(filePath, 'utf8');
|
||||
});
|
||||
|
||||
describe('recursive file:// resolution', () => {
|
||||
it('should recursively resolve file:// references in JSON content', () => {
|
||||
const filePath = 'conversation.json';
|
||||
const fileContent = JSON.stringify([
|
||||
{ role: 'system', content: 'file://system.md' },
|
||||
{ role: 'user', content: 'file://user.md' },
|
||||
]);
|
||||
|
||||
const parsedJson = [
|
||||
{ role: 'system', content: 'file://system.md' },
|
||||
{ role: 'user', content: 'file://user.md' },
|
||||
];
|
||||
|
||||
const resolvedContent = [
|
||||
{ role: 'system', content: 'You are a helpful assistant.' },
|
||||
{ role: 'user', content: 'What is 2 + 2?' },
|
||||
];
|
||||
|
||||
mockReadFileSync.mockReturnValue(fileContent);
|
||||
mockMaybeLoadConfigFromExternalFile.mockReturnValue(resolvedContent);
|
||||
|
||||
const result = processJsonFile(filePath, {});
|
||||
|
||||
expect(mockMaybeLoadConfigFromExternalFile).toHaveBeenCalledWith(parsedJson);
|
||||
expect(result[0].raw).toBe(JSON.stringify(resolvedContent));
|
||||
});
|
||||
|
||||
it('should handle nested file:// references in complex JSON structures', () => {
|
||||
const filePath = 'config.json';
|
||||
const fileContent = JSON.stringify({
|
||||
provider: {
|
||||
id: 'openai:gpt-4',
|
||||
config: {
|
||||
temperature: 'file://temperature.json',
|
||||
tools: 'file://tools.json',
|
||||
},
|
||||
},
|
||||
messages: [{ role: 'system', content: 'file://system.txt' }],
|
||||
});
|
||||
|
||||
const parsedJson = {
|
||||
provider: {
|
||||
id: 'openai:gpt-4',
|
||||
config: {
|
||||
temperature: 'file://temperature.json',
|
||||
tools: 'file://tools.json',
|
||||
},
|
||||
},
|
||||
messages: [{ role: 'system', content: 'file://system.txt' }],
|
||||
};
|
||||
|
||||
const resolvedContent = {
|
||||
provider: {
|
||||
id: 'openai:gpt-4',
|
||||
config: {
|
||||
temperature: 0.7,
|
||||
tools: [{ name: 'search', description: 'Search the web' }],
|
||||
},
|
||||
},
|
||||
messages: [{ role: 'system', content: 'You are a helpful assistant.' }],
|
||||
};
|
||||
|
||||
mockReadFileSync.mockReturnValue(fileContent);
|
||||
mockMaybeLoadConfigFromExternalFile.mockReturnValue(resolvedContent);
|
||||
|
||||
const result = processJsonFile(filePath, {});
|
||||
|
||||
expect(mockMaybeLoadConfigFromExternalFile).toHaveBeenCalledWith(parsedJson);
|
||||
expect(result[0].raw).toBe(JSON.stringify(resolvedContent));
|
||||
});
|
||||
|
||||
it('should handle invalid JSON and return original content', () => {
|
||||
const filePath = 'invalid.json';
|
||||
const fileContent = 'This is not valid JSON {]';
|
||||
|
||||
mockReadFileSync.mockReturnValue(fileContent);
|
||||
|
||||
const result = processJsonFile(filePath, {});
|
||||
|
||||
// When JSON parsing fails, it should return the original content
|
||||
expect(mockMaybeLoadConfigFromExternalFile).not.toHaveBeenCalled();
|
||||
expect(result[0].raw).toBe(fileContent);
|
||||
});
|
||||
|
||||
it('should handle when maybeLoadConfigFromExternalFile returns unchanged content', () => {
|
||||
const filePath = 'no-refs.json';
|
||||
const fileContent = JSON.stringify({ key1: 'value1', key2: 'value2' });
|
||||
|
||||
const parsedJson = { key1: 'value1', key2: 'value2' };
|
||||
|
||||
mockReadFileSync.mockReturnValue(fileContent);
|
||||
mockMaybeLoadConfigFromExternalFile.mockReturnValue(parsedJson);
|
||||
|
||||
const result = processJsonFile(filePath, {});
|
||||
|
||||
expect(mockMaybeLoadConfigFromExternalFile).toHaveBeenCalledWith(parsedJson);
|
||||
expect(result[0].raw).toBe(JSON.stringify(parsedJson));
|
||||
});
|
||||
});
|
||||
});
|
||||
Reference in New Issue
Block a user