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,17 @@
+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 FormData from 'form-data';
+import type { BodyInit, RequestInit } from 'node-fetch';
 import fetch from 'node-fetch';
 
 import { config } from '../config';
 import type { Subscription, Department, Inquiry, Visitor } from '../definifitons';
 import { delay } from '../lib/delay';
+import { getRandomFileFromFolder } from '../lib/file';
 import { username, email } from '../lib/ids';
 import * as prom from '../lib/prom';
 import { rand } from '../lib/rand';
@@ -264,6 +269,39 @@ export class Client {
 		await this.typing(rid, false);
 	}
 
+	async uploadFile(rid: string): Promise<void> {
+		if (!this.client.currentLogin) {
+			return;
+		}
+
+		const folderPath = path.join(__dirname, '..', '..', 'assets');
+
+		const { fullPath: filePath } = getRandomFileFromFolder(folderPath);
+
+		const { authToken, userId } = this.client.currentLogin;
+
+		const endAction = prom.actions.startTimer({ action: 'uploadFile' });
+		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()}`,
+				},
+			});
+
+			endAction({ status: 'success' });
+		} catch (e) {
+			endAction({ status: 'error' });
+			throw e;
+		}
+	}
+
 	async typing(rid: string, typing: boolean): Promise<void> {
 		if (!this.loggedIn) {
 			await this.login();
@@ -510,4 +548,12 @@ 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

@@ -54,6 +54,7 @@ export class WebClient extends Client {
 						'roles-change',
 						'voip.statuschanged',
 						'permissions-changed',
+						'uploadFile',
 					].map((event) => this.client.subscribe('stream-notify-logged', event, false)),
 				);
 

src/config.ts

@@ -45,6 +45,9 @@ const {
 	QUEUED_INQUIRIES_PER_SEC = '0.5',
 	TAKE_INQUIRY_PER_SEC = '1',
 
+	FILES_PER_SECOND = '10',
+	FILE_UPLOADING_RATE = '0,0001', // ~8 files per day per user
+
 	DYNAMIC_LOGIN = 'false',
 } = process.env;
 
@@ -96,5 +99,7 @@ export const config = {
 	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),
+
 	DYNAMIC_LOGIN: ['true', 'yes'].includes(DYNAMIC_LOGIN),
 };

src/lib/file.ts

@@ -0,0 +1,11 @@
+import fs from 'fs';
+import path from 'path';
+
+export const getRandomFileFromFolder = (folderPath: string): { fileName: string; fullPath: string } => {
+	const files = fs.readdirSync(folderPath);
+
+	const randomIndex = Math.floor(Math.random() * files.length);
+	const randomFile = files[randomIndex];
+
+	return { fileName: randomFile, fullPath: path.join(folderPath, randomFile) };
+};

src/profile.ts

@@ -82,6 +82,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 () => {
@@ -149,6 +150,21 @@ export default (): void => {
 		await client.listenPresence(userIds);
 	});
 
+	b.on('uploadFile', async () => {
+		try {
+			const client = await getLoggedInClient();
+			const subscription = client.getRandomSubscription();
+
+			if (!subscription) {
+				return;
+			}
+
+			await client.uploadFile(subscription.rid);
+		} catch (error) {
+			console.error('Error uploading file', error);
+		}
+	});
+
 	b.run().catch((e) => {
 		console.error('Error during run', e);
 	});