feat(cli): add message revice test

apps/cli/src/commands/benchmark/connections.ts

@@ -3,6 +3,7 @@ import { io, Socket } from 'socket.io-client';
 import msgpackParser from 'socket.io-msgpack-parser';
 import fs from 'fs-extra';
 import ora from 'ora';
+import randomString from 'crypto-random-string';
 
 const CLIENT_CREATION_INTERVAL_IN_MS = 5;
 
@@ -10,40 +11,100 @@ export const benchmarkConnectionsCommand: CommandModule = {
   command: 'connections <url>',
   describe: 'Test Tailchat Connections',
   builder: (yargs) =>
-    yargs.demandOption('url', 'Backend Url').option('file', {
-      describe: 'Account Token Path',
-      demandOption: true,
-      type: 'string',
-      default: './accounts',
-    }),
+    yargs
+      .demandOption('url', 'Backend Url')
+      .option('file', {
+        describe: 'Account Token Path',
+        demandOption: true,
+        type: 'string',
+        default: './accounts',
+      })
+      .option('groupId', {
+        describe: 'Group Id which send Message',
+        type: 'string',
+      })
+      .option('converseId', {
+        describe: 'Converse Id which send Message',
+        type: 'string',
+      }),
   async handler(args) {
-    console.log('Reading account tokens from', args.file);
-    const account = await fs.readFile(args.file as string, {
+    const url = args.url as string;
+    const file = args.file as string;
+    const groupId = args.groupId as string;
+    const converseId = args.converseId as string;
+
+    console.log('Reading account tokens from', file);
+    const account = await fs.readFile(file as string, {
       encoding: 'utf8',
     });
-    createClients(
-      args.url as string,
+    const sockets = await createClients(
+      url as string,
       account.split('\n').map((s) => s.trim())
     );
+
+    if (groupId && converseId) {
+      // send message test
+      const randomMessage = randomString({ length: 16 });
+      const spinner = ora()
+        .info(`Start message receive test, message: ${randomMessage}`)
+        .start();
+      const start = Date.now();
+      let receiveCount = 0;
+      const len = sockets.length;
+
+      function receivedCallback() {
+        receiveCount += 1;
+        spinner.text = `Receive: ${receiveCount}/${len}`;
+
+        if (receiveCount === len) {
+          spinner.succeed(
+            `All client received, usage: ${Date.now() - start}ms`
+          );
+        }
+      }
+
+      sockets.forEach((socket) => {
+        socket.on('notify:chat.message.add', (message) => {
+          const content = message.content;
+
+          if (message.converseId === converseId && randomMessage === content) {
+            receivedCallback();
+          }
+        });
+      });
+
+      sockets[0].emit('chat.message.sendMessage', {
+        groupId,
+        converseId,
+        content: randomMessage,
+      });
+    }
   },
 };
 
-async function createClients(url: string, accountTokens: string[]) {
+async function createClients(
+  url: string,
+  accountTokens: string[]
+): Promise<Socket[]> {
   const maxCount = accountTokens.length;
   const spinner = ora().info(`Create Client Connection to ${url}`).start();
 
   let i = 0;
+  const sockets: Socket[] = [];
   for (const token of accountTokens) {
     await sleep(CLIENT_CREATION_INTERVAL_IN_MS);
     spinner.text = `Progress: ${++i}/${maxCount}`;
-    await createClient(url, token);
+    const socket = await createClient(url, token);
+    sockets.push(socket);
   }
 
   spinner.succeed(`${maxCount} clients has been create.`);
+
+  return sockets;
 }
 
 function createClient(url: string, token: string): Promise<Socket> {
-  return new Promise((resolve, reject) => {
+  return new Promise<Socket>((resolve, reject) => {
     const socket = io(url, {
       transports: ['websocket'],
       auth: {
@@ -63,6 +124,10 @@ function createClient(url: string, token: string): Promise<Socket> {
     socket.on('disconnect', (reason) => {
       console.log(`disconnect due to ${reason}`);
     });
+  }).then(async (socket) => {
+    await socket.emitWithAck('chat.converse.findAndJoinRoom', {});
+
+    return socket;
   });
 }
 

apps/cli/src/commands/benchmark/register.ts

@@ -24,9 +24,15 @@ export const benchmarkRegisterCommand: CommandModule = {
         demandOption: true,
         type: 'number',
         default: 100,
+      })
+      .option('invite', {
+        describe: 'Invite Code',
+        type: 'string',
       }),
   async handler(args) {
+    const url = args.url as string;
     const count = args.count as number;
+    const invite = args.invite as string | undefined;
     const tokens: string[] = [];
     const start = Date.now();
 
@@ -35,10 +41,11 @@ export const benchmarkRegisterCommand: CommandModule = {
     for (let i = 0; i < count; i++) {
       spinner.text = `Progress: ${i + 1}/${count}`;
 
-      const token = await registerTemporaryAccount(
-        args.url as string,
-        `benchUser-${i}`
-      );
+      const token = await registerTemporaryAccount(url, `benchUser-${i}`);
+      if (invite) {
+        // Apply group invite
+        applyGroupInviteCode(url, token, invite);
+      }
       tokens.push(token);
     }
 
@@ -64,3 +71,18 @@ async function registerTemporaryAccount(
 
   return res.data.token;
 }
+
+async function applyGroupInviteCode(
+  url: string,
+  token: string,
+  inviteCode: string
+) {
+  await got.post(`${url}/api/group/invite/applyInvite`, {
+    json: {
+      code: inviteCode,
+    },
+    headers: {
+      'X-Token': token,
+    },
+  });
+}