feat(OUT-2045): added a script to seed tasks in the load testing workspace

Author: arpandhakal

src/cmd/load-testing/index.ts

@@ -10,8 +10,8 @@
       - 2000 tasks
 */
 
-import config from '@cmd/load-testing/load-testing.config.json'
 import LoadTester from '@cmd/load-testing/load-testing-v2.service'
+import { AssigneeType } from '@prisma/client'
 
 export const run = async () => {
   if (process.env.VERCEL_ENV === 'production') {
@@ -26,6 +26,8 @@ export const run = async () => {
   await loadTester.seedClients(200)
   // seed clients with a single company
   await loadTester.seedClients(200, true)
+  // seed tasks
+  await loadTester.seedTasks(100, AssigneeType.client)
 }
 
 run()

src/cmd/load-testing/load-testing-v2.service.ts

@@ -1,23 +1,27 @@
+import { MAX_FETCH_ASSIGNEE_COUNT } from '@/constants/users'
 import DBClient from '@/lib/db'
 import {
   ClientRequest,
   ClientRequestSchema,
+  ClientResponseSchema,
   CompanyCreateRequest,
   CompanyCreateRequestSchema,
-  InternalUsersResponse,
+  Token,
+  TokenSchema,
 } from '@/types/common'
+import { CreateTaskRequest, CreateTaskRequestSchema } from '@/types/dto/tasks.dto'
+import { WorkflowStateResponse } from '@/types/dto/workflowStates.dto'
 import { CopilotAPI } from '@/utils/CopilotAPI'
+import { faker } from '@faker-js/faker'
 import { AssigneeType, PrismaClient } from '@prisma/client'
 import Bottleneck from 'bottleneck'
-import { z } from 'zod'
-import { faker } from '@faker-js/faker'
 import fs from 'fs'
 import path from 'path'
-import { MAX_FETCH_ASSIGNEE_COUNT } from '@/constants/users'
+import { z } from 'zod'
 
 class LoadTester {
   protected db: PrismaClient = DBClient.getInstance()
-
+  private apiUrl = 'https://tasks-app-git-feature-m16-load-testing-copilot-platforms.vercel.app'
   private token = z.string().parse(process.env.LOAD_TESTING_COPILOT_TOKEN)
   private apiKey = z.string().parse(process.env.COPILOT_API_KEY)
   private copilot: CopilotAPI = new CopilotAPI(this.token, this.apiKey)
@@ -75,6 +79,55 @@ class LoadTester {
     return responses
   }
 
+  async seedTasks(noOfTasks: number, userType: AssigneeType) {
+    const [assigneeList, workflowStates, tokenPayload] = await Promise.all([
+      userType == AssigneeType.client
+        ? (await this.copilot.getClients({ limit: MAX_FETCH_ASSIGNEE_COUNT })).data
+        : (await this.copilot.getCompanies({ limit: MAX_FETCH_ASSIGNEE_COUNT })).data?.filter((data) => !data.isPlaceholder),
+      this.getAllWorkflowStates(this.token),
+      this.getTokenPayload(this.token),
+    ])
+
+    const seedPromises = []
+
+    for (let i = 0; i < noOfTasks; i++) {
+      if (assigneeList) {
+        const assignee = assigneeList[Math.floor(Math.random() * assigneeList.length)]
+        let clientId,
+          companyId,
+          internalUserId = undefined
+        if (userType == AssigneeType.client) {
+          clientId = assignee.id
+          companyId = ClientResponseSchema.parse(assignee).companyId
+        } else if (userType == AssigneeType.company) {
+          companyId = assignee.id
+        } else {
+          internalUserId = assignee.id
+        }
+        const workflowStateId = workflowStates[Math.floor(Math.random() * workflowStates.length)].id
+        const createdById = tokenPayload.internalUserId
+        const title = faker.music.songName()
+        const body = faker.lorem.paragraph()
+        const task = CreateTaskRequestSchema.parse({
+          internalUserId,
+          clientId,
+          companyId,
+          workflowStateId,
+          createdById,
+          title,
+          body,
+        })
+        const seedPromiseWithBottleneck = this.bottlenecks.copilot.schedule(() => {
+          console.info(`Seeding task ${task.title} `)
+          return this.handleCreate(this.token, task)
+        })
+        seedPromises.push(seedPromiseWithBottleneck)
+      }
+      const responses = await Promise.all(seedPromises)
+      return responses
+    }
+  }
+
   private exportToCSV(data: Record<string, any>[], type: AssigneeType) {
     if (data.length === 0) return
 
@@ -95,6 +148,30 @@ class LoadTester {
     fs.writeFileSync(outputPath, csvContent)
     console.log(`✅ CSV written to ${outputPath}`)
   }
+
+  private async getTokenPayload(token: string): Promise<Token> {
+    const payload = TokenSchema.parse(await this.copilot.getTokenPayload())
+    return payload as Token
+  }
+
+  private async getAllWorkflowStates(token: string): Promise<WorkflowStateResponse[]> {
+    const res = await fetch(`${this.apiUrl}/api/workflow-states?token=${token}`)
+
+    const data = await res.json()
+    return data.workflowStates
+  }
+
+  private async handleCreate(token: string, payload: CreateTaskRequest) {
+    try {
+      const response = await fetch(`${this.apiUrl}/api/tasks?token=${token}`, {
+        method: 'POST',
+        body: JSON.stringify(payload),
+      })
+      return await response.json()
+    } catch (e: unknown) {
+      console.error('Something went wrong while creating task!', e)
+    }
+  }
 }
 
 export default LoadTester