transferBalanceLoadTesting.js

import http from 'k6/http'
import { sleep } from 'k6'
import { Counter, Rate } from 'k6/metrics'

export const options = {
  scenarios: {
    example_scenario: {
      executor: 'constant-arrival-rate',
      rate: 8,
      timeUnit: '1s', // 1000 iterations per second, i.e. 1000 RPS
      duration: '30s',
      preAllocatedVUs: 400, // how large the initial pool of VUs would be
      maxVUs: 400, // if the preAllocatedVUs are not enough, we can initialize more
    },
  },
}

const count200 = new Counter('status_code_2xx')
const count300 = new Counter('status_code_3xx')
const count400 = new Counter('status_code_4xx')
const count500 = new Counter('status_code_5xx')

const rate200 = new Rate('rate_status_code_2xx')
const rate300 = new Rate('rate_status_code_3xx')
const rate400 = new Rate('rate_status_code_4xx')
const rate500 = new Rate('rate_status_code_5xx')

function recordRates(res) {
  if (res.status >= 200 && res.status < 300) {
    count200.add(1)
    rate200.add(1)
  } else if (res.status >= 300 && res.status < 400) {
    console.log(res.body)
    count300.add(1)
    rate300.add(1)
  } else if (res.status >= 400 && res.status < 500) {
    count400.add(1)
    rate400.add(1)
  } else if (res.status >= 500 && res.status < 600) {
    count500.add(1)
    rate500.add(1)
  }
}

export default function () {
  // console.log(`VU: ${__VU}  -  ITER: ${__ITER}`)

  // Sign up for an account
  // recordRates(http.get('http://localhost:3000/app/index'))
  // sleep(2);
  // recordRates(http.get('http://localhost:3000/app/signup'))
  // sleep(2);

  const headers = {
    'Content-Type': 'application/json',
  }
  const uniqueUser = `${__VU}-${__ITER}`
  recordRates(http.get('http://localhost:3000/app/index'))
  recordRates(http.get('http://localhost:3000/app/signup'))
  recordRates(
    http.post(
      'http://localhost:3000/auth/signup',
      `{"name":"${uniqueUser}","email":"${uniqueUser}@gmail.com","password":"12345678"}`,
      { headers: headers }
    )
  )

  // After signing up, you are redirected to login to the account automatically
  // sleep(2);
  recordRates(http.get('http://localhost:3000/app/login'))
  recordRates(
    http.post('http://localhost:3000/auth/login', `{"email":"${uniqueUser}@gmail.com","password":"12345678"}`, {
      headers: headers,
    })
  )

  // Login redirects to profile page automatically, pause to link an external chase account
  // recordRates(http.post('http://localhost:3000/getPlaidLinkToken')) this is automatically performed on entering /app/profile (which needs to be optimized eventually)
  // sleep(5);
  recordRates(http.get('http://localhost:3000/app/profile'))
  const res = http.post(
    'http://localhost:3000/createAccounts',
    '{"accounts":[{"account_id":"e3qz8qZNePty5K6bwnLwHXzjPmK9leFLlmp5J","balances":{"available":8113.27,"current":10000,"iso_currency_code":"CAD","limit":null,"unofficial_currency_code":null},"mask":"2163","name":"Chase Savings","official_name":"Chase College Savings","subtype":"savings","type":"depository"},{"account_id":"7mJeWJrzyaHZXWzNp51pUxjvMW76XEtgKxenj","balances":{"available":4939.59,"current":5000,"iso_currency_code":"CAD","limit":null,"unofficial_currency_code":null},"mask":"4409","name":"Chase Checking","official_name":"Chase College Savings","subtype":"checking","type":"depository"}]}',
    { headers: headers }
  )
  recordRates(res)

  const { newAccounts } = JSON.parse(res.body)
  recordRates(http.get('http://localhost:3000/app/transferBalance'))
  // console.log(newAccountIds[0], newAccountIds[1])
  recordRates(
    http.post(
      'http://localhost:3000/transferBalance',
      `{"fromAccountId":${newAccounts[0].insertId},"toAccountId":${newAccounts[1].insertId},"amount":5000}`,
      {
        headers: headers,
      }
    )
  )

  recordRates(
    http.post('http://localhost:3000/auth/logout', {
      headers: headers,
    })
  )

  // let query = `mutation {
  //   createUser(input:{
  //     userType:USER
  //     email:"test@gmail.com"
  //     name:"Test"
  //     password:"123123123"
  //   })
  // }`

  // recordRates(http.post('http://localhost:3000/graphql', JSON.stringify({ query }), { headers: headers }))

  // console.log(res.body)
}