mirror of
https://github.com/danny-avila/LibreChat.git
synced 2026-06-29 02:41:26 +00:00
2c8d54e18c
Some checks are pending
Docker Dev Branch Images Build / build (Dockerfile, lc-dev, node) (push) Waiting to run
Docker Dev Branch Images Build / build (Dockerfile.multi, lc-dev-api, api-build) (push) Waiting to run
GitNexus Index / index (push) Waiting to run
GitNexus Index / post-index (push) Blocked by required conditions
* feat: Add deployment skill directory * chore: Address deployment skill review feedback * fix: Include deployment skill file metadata * test: Add deployment skills e2e smoke test
107 lines
3.4 KiB
JavaScript
107 lines
3.4 KiB
JavaScript
const mockSaveBuffer = jest.fn();
|
|
const mockDeleteFile = jest.fn();
|
|
const mockGetStrategyFunctions = jest.fn();
|
|
const mockGetFileStrategy = jest.fn();
|
|
const mockGetStorageMetadata = jest.fn();
|
|
const mockResolveRequestTenantId = jest.fn();
|
|
const mockCreateDeploymentSkillMethods = jest.fn((methods) => methods);
|
|
|
|
jest.mock('~/server/services/Files/strategies', () => ({
|
|
getStrategyFunctions: (...args) => mockGetStrategyFunctions(...args),
|
|
}));
|
|
|
|
jest.mock('~/server/services/Files/Code/crud', () => ({
|
|
batchUploadCodeEnvFiles: jest.fn(),
|
|
}));
|
|
|
|
jest.mock('~/server/services/Files/Code/process', () => ({
|
|
getSessionInfo: jest.fn(),
|
|
checkIfActive: jest.fn(),
|
|
readSandboxFile: jest.fn(),
|
|
writeSandboxFile: jest.fn(),
|
|
}));
|
|
|
|
jest.mock('@librechat/api', () => ({
|
|
checkAccess: jest.fn(),
|
|
createDeploymentSkillMethods: (...args) => mockCreateDeploymentSkillMethods(...args),
|
|
enrichWithSkillConfigurable: jest.fn(),
|
|
getDeploymentSkillDownloadStream: jest.fn(),
|
|
getStorageMetadata: (...args) => mockGetStorageMetadata(...args),
|
|
isDeploymentSkillFileSource: jest.fn(() => false),
|
|
mergeDeploymentSkillIds: jest.fn((ids = []) => ids),
|
|
resolveRequestTenantId: (...args) => mockResolveRequestTenantId(...args),
|
|
}));
|
|
|
|
jest.mock('librechat-data-provider', () => ({
|
|
AccessRoleIds: { SKILL_OWNER: 'SKILL_OWNER' },
|
|
FileContext: { skill_file: 'skill_file' },
|
|
PermissionBits: { EDIT: 2 },
|
|
Permissions: { USE: 'USE', CREATE: 'CREATE' },
|
|
PermissionTypes: { SKILLS: 'SKILLS' },
|
|
PrincipalType: { USER: 'USER' },
|
|
ResourceType: { SKILL: 'SKILL' },
|
|
isEphemeralAgentId: jest.fn(() => false),
|
|
}));
|
|
|
|
jest.mock('~/server/services/PermissionService', () => ({
|
|
checkPermission: jest.fn(),
|
|
grantPermission: jest.fn(),
|
|
}));
|
|
|
|
jest.mock('~/server/utils/getFileStrategy', () => ({
|
|
getFileStrategy: (...args) => mockGetFileStrategy(...args),
|
|
}));
|
|
|
|
const mockDb = {
|
|
getSkillFileByPath: jest.fn(),
|
|
upsertSkillFile: jest.fn(),
|
|
};
|
|
|
|
jest.mock('~/models', () => mockDb);
|
|
|
|
const { getSkillToolDeps } = require('./skillDeps');
|
|
|
|
describe('skillDeps saveSkillFileContent', () => {
|
|
beforeEach(() => {
|
|
jest.clearAllMocks();
|
|
mockGetFileStrategy.mockReturnValue('s3');
|
|
mockGetStrategyFunctions.mockReturnValue({
|
|
saveBuffer: mockSaveBuffer,
|
|
deleteFile: mockDeleteFile,
|
|
});
|
|
mockSaveBuffer.mockResolvedValue('https://files.example.test/uploads/file.txt');
|
|
mockDeleteFile.mockResolvedValue(undefined);
|
|
mockGetStorageMetadata.mockReturnValue({
|
|
storageKey: 'uploads/file.txt',
|
|
storageRegion: 'us-east-2',
|
|
});
|
|
mockResolveRequestTenantId.mockReturnValue('tenant-1');
|
|
mockDb.getSkillFileByPath.mockResolvedValue(null);
|
|
});
|
|
|
|
it('cleans up the uploaded object when metadata upsert returns no row', async () => {
|
|
mockDb.upsertSkillFile.mockResolvedValue(null);
|
|
|
|
await expect(
|
|
getSkillToolDeps().saveSkillFileContent({
|
|
req: {
|
|
user: { id: 'user-1', _id: 'user-1' },
|
|
config: {},
|
|
},
|
|
skillId: 'skill-1',
|
|
relativePath: 'references/template.html',
|
|
content: '<html></html>',
|
|
mimeType: 'text/html',
|
|
}),
|
|
).rejects.toMatchObject({ code: 'SKILL_FILE_UPSERT_NOT_FOUND' });
|
|
|
|
expect(mockDeleteFile).toHaveBeenCalledWith(
|
|
expect.objectContaining({ user: expect.objectContaining({ id: 'user-1' }) }),
|
|
{
|
|
filepath: 'https://files.example.test/uploads/file.txt',
|
|
user: 'user-1',
|
|
tenantId: 'tenant-1',
|
|
},
|
|
);
|
|
});
|
|
});
|