Create k6 scripts and http to test

Author: kevariable

.gitignore

@@ -20,4 +20,5 @@ yarn-error.log
 /.phpstorm.meta.php
 /_ide_helper.php
 /coverage
-/dist
+/dist
+/k6/node_modules/

docker-compose.yml

@@ -1,5 +1,3 @@
-version: '3.8'
-
 services:
   app:
     image: ts-restful-api:dev
@@ -11,10 +9,23 @@ services:
       - /app/node_modules
     command: ["npm", "run", "dev"]
     ports:
-      - 3000:3000
+      - "3000:3000"
     depends_on:
       - mysql
 
+  k6:
+    image: grafana/k6
+    working_dir: /app
+    environment:
+      K6_WEB_DASHBOARD: true
+    tty: true
+    command: >
+      run src/ping.js
+    ports:
+      - "5665:5665"
+    volumes:
+      - ./k6:/app
+
   mysql:
     image: mysql/mysql-server:8.0
     restart: always

k6/node_modules/@types/k6/README.md

@@ -0,0 +1,5 @@
+{
+  "devDependencies": {
+    "@types/k6": "^0.54.2"
+  }
+}

k6/package-lock.json

@@ -0,0 +1,42 @@
+import { vu } from 'k6/execution'
+import {loginUser} from "./supports/user.js";
+import {createContact} from "./supports/contact.js";
+
+export const options = {
+  vus: 10,
+  duration: '30s'
+}
+
+export function setup() {
+  const data = []
+
+  for (let i = 0; i < 10; i++) {
+    data.push({
+      first_name: 'Contact',
+      last_name: `No ${i}`,
+      email: `contact${i}@example.com`,
+      phone: `i`
+    })
+  }
+
+  return data
+}
+
+const getToken = () => {
+  const username = `kevin${vu.idInInstance}`
+  const loginRequest = JSON.stringify({
+    username,
+    password: 'password'
+  })
+
+
+  return loginUser(loginRequest)
+}
+
+export default function (data) {
+  const { token } = getToken()
+
+  for (const value of data) {
+    createContact(JSON.stringify(value), token)
+  }
+}

k6/package.json

@@ -0,0 +1,11 @@
+import http from 'k6/http';
+import { sleep } from 'k6';
+
+export const options = {
+  vus: 10,
+  duration: '30s',
+};
+
+export default function() {
+  http.get('http://app:3000/api/ping');
+}

k6/src/create-contact.js

@@ -0,0 +1,97 @@
+import http from 'k6/http';
+import {check, fail} from 'k6';
+import { uuidv4 } from 'https://jslib.k6.io/k6-utils/1.4.0/index.js';
+
+export const options = {
+  vus: 10,
+  duration: '10s',
+};
+
+const currentUserCallback = (data, { uniqueId }) => {
+  const currentUser = http.get('http://app:3000/api/users/current', {
+    headers: {
+      'Accept': 'application/json',
+      'Content-Type': 'application/json',
+      'X-API-TOKEN': data.token
+    }
+  });
+
+  const currentUserResponse = currentUser.json()
+
+  const currentUserCheck = check(currentUser, {
+    'Current user response status must 200': currentUser.status === 200,
+    'Current user response data must not null': currentUserResponse.data !== null ,
+  })
+
+  if (! currentUserCheck) {
+    fail(`Failed to get current user-${uniqueId}`)
+  }
+
+  return currentUserResponse
+}
+
+const loginCallback = (data, { uniqueId, password }) => {
+  const loginRequest = JSON.stringify({
+    username: data?.username || '',
+    password,
+  })
+
+  const login = http.post('http://app:3000/api/login', loginRequest, {
+    headers: {
+      'Accept': 'application/json',
+      'Content-Type': 'application/json'
+    }
+  });
+
+  const loginResponse = login.json()
+
+  const loginCheck = check(login, {
+    'Login response status must 200': login.status === 200,
+    'Login response token must not null': (loginResponse.data?.token || null) !== null ,
+  })
+
+  if (! loginCheck) {
+    fail(`Failed to get user-${uniqueId}`)
+  }
+
+  return loginResponse
+}
+
+const registerCallback = ({ uniqueId, password }) => {
+  const data = JSON.stringify({
+    username: `user-${uniqueId}`,
+    password,
+    name: `user-${uniqueId}`
+  })
+
+  const userRegistered = http.post('http://app:3000/api/users', data, {
+    headers: {
+      'Accept': 'application/json',
+      'Content-Type': 'application/json'
+    }
+  })
+
+  const userRegisteredResponse = userRegistered.json()
+
+  const userRegisteredCheck = check(userRegistered, {
+    'Register response status must 201': userRegistered.status === 201,
+    'Register response data must not null': userRegisteredResponse.data !== null ,
+  })
+
+  if (! userRegisteredCheck) {
+    fail(`Failed to registering user-${uniqueId}`)
+  }
+
+  return userRegisteredResponse
+}
+
+export default function() {
+  const uniqueId = uuidv4()
+  const password = 'password'
+
+  const registerResponse = registerCallback({ uniqueId, password })
+
+  const loginResponse = loginCallback(registerResponse.data, { uniqueId, password })
+
+  return currentUserCallback(loginResponse.data, { uniqueId })
+}

k6/src/ping.js

@@ -0,0 +1,22 @@
+import http from 'k6/http';
+
+export const options = {
+    stages: [
+        {
+            duration: '10s',
+            target: 20
+        },
+        {
+            duration: '10s',
+            target: 10
+        },
+        {
+            duration: '10s',
+            target: 0
+        },
+    ]
+};
+
+export default function() {
+    http.get('http://app:3000/api/ping');
+}

k6/src/register.js

@@ -0,0 +1,19 @@
+import http from "k6/http";
+import {check} from "k6";
+
+export const createContact = (createContactRequest, token) => {
+    const createContact = http.post('http://app:3000/api/contacts', createContactRequest, {
+        headers: {
+            'X-API-TOKEN': token,
+            'Accept': 'application/json',
+            'Content-Type': 'application/json'
+        }
+    })
+
+    const createContactResponse = createContact.json()
+
+    check(createContact, {
+        'response must be 200': createContact.status === 200,
+        'response contains id': createContactResponse.data.id !== undefined,
+    })
+}