Tom Butcher 28c94159b4
Some checks failed
farmcontrol/farmcontrol-api/pipeline/head There was a failure building this commit
Added unit tests.
2025-12-29 00:39:13 +00:00

124 lines
3.1 KiB
JavaScript

import { jest } from '@jest/globals';
jest.unstable_mockModule('../../../database/database.js', () => ({
listObjects: jest.fn(),
getObject: jest.fn(),
editObject: jest.fn(),
newObject: jest.fn(),
deleteObject: jest.fn(),
listObjectsByProperties: jest.fn(),
flushFile: jest.fn(),
getModelStats: jest.fn(),
getModelHistory: jest.fn(),
}));
jest.unstable_mockModule('../../../database/schemas/management/file.schema.js', () => ({
fileModel: { modelName: 'File' },
}));
jest.unstable_mockModule('../../../database/ceph.js', () => ({
uploadFile: jest.fn(),
downloadFile: jest.fn(),
deleteFile: jest.fn(),
BUCKETS: { FILES: 'test-bucket' },
}));
jest.unstable_mockModule('../../../utils.js', () => ({
getFileMeta: jest.fn(),
}));
jest.unstable_mockModule('multer', () => {
const mockMemoryStorage = jest.fn();
const mockMulter = jest.fn(() => ({
single: jest.fn(),
}));
mockMulter.memoryStorage = mockMemoryStorage;
return {
default: mockMulter,
memoryStorage: mockMemoryStorage,
};
});
jest.unstable_mockModule('log4js', () => ({
default: {
getLogger: () => ({
level: 'info',
debug: jest.fn(),
error: jest.fn(),
warn: jest.fn(),
trace: jest.fn(),
}),
},
}));
const {
listFilesRouteHandler,
getFileRouteHandler,
editFileRouteHandler,
flushFileRouteHandler,
} = await import('../files.js');
const { listObjects, getObject, editObject, flushFile } = await import(
'../../../database/database.js'
);
const { fileModel } = await import('../../../database/schemas/management/file.schema.js');
describe('File Service Route Handlers', () => {
let req, res;
beforeEach(() => {
req = {
params: {},
query: {},
body: {},
user: { id: 'test-user-id' },
};
res = {
send: jest.fn(),
status: jest.fn().mockReturnThis(),
};
jest.clearAllMocks();
});
describe('listFilesRouteHandler', () => {
it('should list files', async () => {
const mockResult = [{ _id: '1', name: 'file.pdf' }];
listObjects.mockResolvedValue(mockResult);
await listFilesRouteHandler(req, res);
expect(listObjects).toHaveBeenCalledWith(expect.objectContaining({ model: fileModel }));
expect(res.send).toHaveBeenCalledWith(mockResult);
});
});
describe('getFileRouteHandler', () => {
it('should get a file by ID', async () => {
req.params.id = '123';
const mockFile = { _id: '123', name: 'test.pdf' };
getObject.mockResolvedValue(mockFile);
await getFileRouteHandler(req, res);
expect(getObject).toHaveBeenCalledWith(expect.objectContaining({ id: '123' }));
expect(res.send).toHaveBeenCalledWith(mockFile);
});
});
describe('flushFileRouteHandler', () => {
it('should flush/delete a file', async () => {
req.params.id = '507f1f77bcf86cd799439011';
const mockResult = { success: true };
flushFile.mockResolvedValue(mockResult);
await flushFileRouteHandler(req, res);
expect(flushFile).toHaveBeenCalledWith(
expect.objectContaining({ id: '507f1f77bcf86cd799439011' })
);
expect(res.send).toHaveBeenCalledWith(mockResult);
});
});
});