snapshot of https://github.com/RocketChat/Rocket.Chat.git (develop, 2026-09-16; blobs > 131072B omitted)

apps/meteor/server/lib/dataExport/processDataDownloads.spec.ts

@@ -0,0 +1,251 @@
+import fs from 'node:fs';
+
+import type { IExportOperation } from '@rocket.chat/core-typings';
+import { expect } from 'chai';
+import proxyquire from 'proxyquire';
+import Sinon from 'sinon';
+
+let exportOperation: IExportOperation | null = null;
+
+const modelsMock = {
+	ExportOperations: {
+		findLastOperationByUser: async (userId: string, fullExport = false) => {
+			if (exportOperation?.userId === userId && exportOperation?.fullExport === fullExport) {
+				return exportOperation;
+			}
+		},
+		countAllPendingBeforeMyRequest: async (requestDay: Date) => {
+			if (
+				exportOperation &&
+				exportOperation.createdAt < requestDay &&
+				exportOperation.status !== 'completed' &&
+				exportOperation.status !== 'skipped'
+			) {
+				return 1;
+			}
+			return 0;
+		},
+		create: async (data: any) => {
+			exportOperation = {
+				userNameTable: null, // need to keep this null for testing purposes
+				...data,
+				_id: 'exportOp1',
+				createdAt: new Date(),
+			};
+			return exportOperation?._id as IExportOperation['_id'];
+		},
+		updateOperation: async (data: IExportOperation) => {
+			if (exportOperation && exportOperation._id === data._id) {
+				exportOperation = { ...exportOperation, ...data };
+			}
+			return { modifiedCount: 1 };
+		},
+
+		findOnePending: async () => {
+			if (exportOperation && exportOperation.status !== 'completed' && exportOperation.status !== 'skipped') {
+				return exportOperation;
+			}
+			return null;
+		},
+	},
+	UserDataFiles: {
+		findOneById: async (fileId: string) => {
+			if (exportOperation?.fileId === fileId) {
+				return {
+					_id: fileId,
+				};
+			}
+		},
+		findLastFileByUser: async (userId: string) => {
+			if (exportOperation?.userId === userId && exportOperation.fileId) {
+				return {
+					_id: exportOperation.fileId,
+				};
+			}
+		},
+	},
+	Avatars: {
+		findOneByName: async (_name: string) => {
+			return null;
+		},
+	},
+	Subscriptions: {
+		findByUserId: (_userId: string) => {
+			return [
+				{
+					rid: 'general',
+				},
+			];
+		},
+	},
+	Messages: {
+		findPaginated: (_query: object, _options: object) => {
+			return {
+				cursor: {
+					toArray: async () => [
+						{
+							_id: 'msg1',
+							rid: 'general',
+							ts: new Date(),
+							msg: 'Hello World',
+							u: { _id: 'user1', username: 'userone' },
+						},
+						{
+							_id: 'msg2',
+							rid: 'general',
+							ts: new Date(),
+							msg: 'Second message',
+							u: { _id: 'user2', username: 'usertwo' },
+						},
+					],
+				},
+				totalCount: Promise.resolve(0),
+			};
+		},
+	},
+};
+
+const { exportRoomMessagesToFile } = proxyquire.noCallThru().load('./exportRoomMessagesToFile.ts', {
+	'@rocket.chat/models': modelsMock,
+	'../../settings': {
+		settings: {
+			get: (_key: string) => {
+				return undefined;
+			},
+		},
+	},
+	'../i18n': {
+		i18n: {
+			t: (key: string) => key,
+		},
+	},
+});
+
+const { requestDataDownload } = proxyquire.noCallThru().load('../../meteor-methods/platform/requestDataDownload.ts', {
+	'@rocket.chat/models': modelsMock,
+	'../../settings': {
+		settings: {
+			get: (_key: string) => {
+				return undefined;
+			},
+		},
+	},
+	'../../lib/dataExport': {
+		getPath: (fileId: string) => `/data-download/${fileId}`,
+	},
+	'meteor/meteor': {
+		Meteor: {
+			methods: Sinon.stub(),
+		},
+	},
+}) as {
+	requestDataDownload: (args: { userData: { _id: string }; fullExport?: boolean }) => Promise<{
+		requested: boolean;
+		exportOperation: IExportOperation;
+		url: string | null;
+		pendingOperationsBeforeMyRequest: number;
+	}>;
+};
+
+const { processDataDownloads } = proxyquire.noCallThru().load('./processDataDownloads.ts', {
+	'@rocket.chat/models': modelsMock,
+	'../media/file-upload': {
+		FileUpload: {
+			copy: async (fileId: string, _options: any) => {
+				return `copied-${fileId}`;
+			},
+		},
+	},
+	'../../settings': {
+		settings: {
+			get: (_key: string) => {
+				return undefined;
+			},
+		},
+	},
+	'../utils/getURL': {
+		getURL: (path: string) => `https://example.com${path}`,
+	},
+	'../i18n': {
+		i18n: {
+			t: (key: string) => key,
+		},
+	},
+	'./copyFileUpload': {
+		copyFileUpload: (_attachmentData: { _id: string; name: string }, _assetsPath: string) => {
+			return Promise.resolve();
+		},
+	},
+	'./exportRoomMessagesToFile': {
+		exportRoomMessagesToFile,
+	},
+	'./getRoomData': {
+		getRoomData: Sinon.stub().resolves({
+			roomId: 'GENERAL',
+			roomName: 'general',
+			type: 'c',
+			exportedCount: 0,
+			status: 'pending',
+			userId: 'user1',
+			targetFile: 'general.json',
+		}),
+	},
+	'./sendEmail': {
+		sendEmail: Sinon.stub().resolves(),
+	},
+	'./uploadZipFile': {
+		uploadZipFile: Sinon.stub().resolves({ _id: 'file1' }),
+	},
+}) as {
+	processDataDownloads: () => Promise<void>;
+};
+
+const userData = { _id: 'user1', username: 'userone' };
+
+describe('requestDataDownload', () => {
+	beforeEach(() => {
+		exportOperation = null;
+	});
+
+	it('should create a new export operation if none exists', async () => {
+		const result = await requestDataDownload({ userData, fullExport: false });
+
+		expect(result.requested).to.be.true;
+		expect(result.exportOperation).to.exist;
+		expect(result.exportOperation.userId).to.equal('user1');
+		expect(result.exportOperation.fullExport).to.be.false;
+		expect(result.url).to.be.null;
+		expect(result.pendingOperationsBeforeMyRequest).to.equal(0);
+		expect(result.exportOperation.status).to.equal('pending');
+	});
+});
+
+describe('export user data', async () => {
+	beforeEach(() => {
+		exportOperation = null;
+	});
+	it('should process data download for pending export operations', async () => {
+		await requestDataDownload({ userData, fullExport: true });
+
+		expect(exportOperation).to.not.be.null;
+		expect(exportOperation?.userId).to.equal('user1');
+		expect(exportOperation?.fullExport).to.be.true;
+		expect(exportOperation?.status).to.equal('pending');
+
+		await processDataDownloads();
+
+		expect(exportOperation?.status).to.equal('completed');
+		expect(exportOperation?.fileId).to.equal('file1');
+		expect(exportOperation?.generatedUserFile).to.be.true;
+		expect(exportOperation?.roomList).to.have.lengthOf(1);
+		expect(exportOperation?.roomList?.[0].roomId).to.equal('GENERAL');
+		expect(exportOperation?.roomList?.[0].exportedCount).to.equal(2);
+		expect(exportOperation?.exportPath).to.be.string;
+
+		expect(fs.readFileSync(`${exportOperation?.exportPath}/${exportOperation?.roomList?.[0].targetFile}`, 'utf-8')).to.contain(
+			'Hello World',
+		);
+		expect(exportOperation?.generatedFile).to.be.string;
+		expect(fs.existsSync(exportOperation?.generatedFile as string)).to.be.true;
+	});
+});