feat: adding file upload test

assets/file.json

@@ -0,0 +1,3 @@
+{
+  "app": "wow"
+}

src/BenchmarkRunner.ts

@@ -23,7 +23,8 @@ type EventKeys =
 	| 'subscribePresence'
 	| 'getRoutingConfig'
 	| 'getQueuedInquiries'
-	| 'takeInquiry';
+	| 'takeInquiry'
+	| 'uploadFile';
 
 const eventsPerSecond = (events: number): number => Math.ceil(1000 / events);
 

src/client/Client.ts

@@ -1,12 +1,16 @@
+import fs from 'fs';
+import path from 'path';
 import { URLSearchParams } from 'url';
 
 import RocketChatClient from '@rocket.chat/sdk/lib/clients/Rocketchat';
 import EJSON from 'ejson';
-import fetch from 'node-fetch';
+import FormData from 'form-data';
+import fetch, { BodyInit, RequestInit } from 'node-fetch';
 
 import { config } from '../config';
 import { Subscription, Department, Inquiry, Visitor } from '../definifitons';
 import { delay } from '../lib/delay';
+import { getRandomFileInFolder } from '../lib/file';
 import { username, email } from '../lib/ids';
 import * as prom from '../lib/prom';
 import { rand } from '../lib/rand';
@@ -275,6 +279,45 @@ export class Client {
 		await this.typing(rid, false);
 	}
 
+	async uploadFile({
+		rid,
+		authToken,
+		userId,
+	}: {
+		rid: string;
+		authToken: string;
+		userId: string;
+	}): Promise<void> {
+		await delay(8000);
+		const folderPath = path.join(__dirname, '..', '..', 'assets');
+
+		const { filePath } = getRandomFileInFolder(folderPath);
+
+		const endAction = prom.actions.startTimer({ action: 'uploadFile' });
+		const end = prom.messages.startTimer();
+		try {
+			const fileFormData = new FormData();
+
+			fileFormData.append('file', fs.createReadStream(filePath));
+
+			await this.httpPost(`/api/v1/rooms.upload/${rid}`, {
+				body: fileFormData as unknown as BodyInit,
+				headers: {
+					'X-Auth-Token': authToken,
+					'X-User-Id': userId,
+					'Content-Type': `multipart/form-data; boundary=${fileFormData.getBoundary()}`,
+				},
+			});
+
+			end({ status: 'success' });
+			endAction({ status: 'success' });
+		} catch (e) {
+			end({ status: 'error' });
+			endAction({ status: 'error' });
+			throw e;
+		}
+	}
+
 	async typing(rid: string, typing: boolean): Promise<void> {
 		this.client.methodCall(
 			'stream-notify-room',
@@ -531,4 +574,15 @@ export class Client {
 		});
 		return result.json();
 	}
+
+	protected async httpPost(
+		endpoint: string,
+		init?: RequestInit
+	): Promise<unknown> {
+		const result = await fetch(`${this.host}${endpoint}`, {
+			method: 'POST',
+			...init,
+		});
+		return result.json();
+	}
 }

src/client/WebClient.ts

@@ -70,6 +70,7 @@ export class WebClient extends Client {
 					'rooms-changed',
 					'webrtc',
 					'userData',
+					'uploadFile',
 				].map((event) =>
 					this.client.subscribe(
 						'stream-notify-user',

src/config.ts

@@ -48,6 +48,9 @@ const {
 	ROUTING_CONFIG_PER_SEC = '10',
 	QUEUED_INQUIRIES_PER_SEC = '0.5',
 	TAKE_INQUIRY_PER_SEC = '1',
+
+	FILES_PER_SECOND = '10',
+	FILE_UPLOADING_RATE = '0.00115428571', // ='0.001428571,
 } = process.env;
 
 export const config = {
@@ -107,4 +110,8 @@ export const config = {
 	ROUTING_CONFIG_PER_SEC: parseInt(ROUTING_CONFIG_PER_SEC),
 	QUEUED_INQUIRIES_PER_SEC: parseFloat(QUEUED_INQUIRIES_PER_SEC),
 	TAKE_INQUIRY_PER_SEC: parseInt(TAKE_INQUIRY_PER_SEC),
+
+	FILES_PER_SECOND: FILE_UPLOADING_RATE
+		? parseInt(HOW_MANY_USERS) * parseFloat(FILE_UPLOADING_RATE)
+		: parseInt(FILES_PER_SECOND),
 };

src/lib/file.ts

@@ -0,0 +1,25 @@
+import fs from 'fs';
+
+type FileInformation = {
+	fileName: string;
+	fileSize: number;
+	fileContent: Buffer;
+	filePath: string;
+};
+
+export const getRandomFileInFolder = (folderPath: string): FileInformation => {
+	const files = fs.readdirSync(folderPath);
+
+	const randomIndex = Math.floor(Math.random() * files.length);
+	const randomFile = files[randomIndex];
+
+	const filePath = `${folderPath}/${randomFile}`;
+	const fileContent = fs.readFileSync(filePath);
+
+	return {
+		fileName: randomFile,
+		fileSize: fileContent.length,
+		fileContent,
+		filePath,
+	};
+};

src/profile.ts

@@ -1,3 +1,4 @@
+import { ILoginResultAPI } from '@rocket.chat/sdk/interfaces';
 import { MongoClient } from 'mongodb';
 
 import { BenchmarkRunner } from './BenchmarkRunner';
@@ -82,6 +83,7 @@ export default (): void => {
 		openRoom: config.OPEN_ROOM_PER_SECOND,
 		setUserStatus: config.SET_STATUS_PER_SECOND,
 		subscribePresence: config.SUBSCRIBE_PRESENCE_PER_SECOND,
+		uploadFile: config.FILES_PER_SECOND,
 	});
 
 	b.on('ready', async () => {
@@ -148,5 +150,31 @@ export default (): void => {
 		client.listenPresence(userIds);
 	});
 
+	b.on('uploadFile', async () => {
+		try {
+			const client = rand(clients);
+			const subscription = client.getRandomSubscription();
+
+			if (!subscription) {
+				return;
+			}
+
+			const { authToken, userId } = client.client.currentLogin as {
+				username: string;
+				userId: string;
+				authToken: string;
+				result: ILoginResultAPI;
+			};
+
+			await client.uploadFile({
+				rid: subscription.rid,
+				authToken,
+				userId,
+			});
+		} catch (error) {
+			console.error('Error uploading file', error);
+		}
+	});
+
 	b.run();
 };