User auth and more tests

Author: rhysyngsun

load_testing/auth.ts

@@ -0,0 +1,13 @@
+import { SharedArray } from "k6/data"
+
+export function getAccessToken(): String {
+  return __ENV.AUTH_ACCESS_TOKEN
+}
+
+export const users = new SharedArray("users", function () {
+  if (!__ENV.USERS_JSON_FILE) {
+    return []
+  }
+
+  return JSON.parse(open(__ENV.USERS_JSON_FILE))
+})

load_testing/backend/client/client.ts

@@ -0,0 +1,26 @@
+import { Params } from "k6/http"
+import { Config } from "../../config.ts"
+import { MITLearnAPIClient as V0Client } from "./v0/api.ts"
+import { MITLearnAPIClient as V1Client } from "./v1/api.ts"
+import { getAccessToken } from "../../auth.ts"
+
+function getCommonRequestParameters(): Params {
+  return {
+    headers: {
+      Authorization: `Bearer ${getAccessToken()}`,
+    },
+  }
+}
+
+export function createV0Client(config: Config): V0Client {
+  return new V0Client({
+    baseUrl: config.apiBaseUrl,
+    commonRequestParameters: getCommonRequestParameters(),
+  })
+}
+export function createV1Client(config: Config): V1Client {
+  return new V1Client({
+    baseUrl: config.apiBaseUrl,
+    commonRequestParameters: getCommonRequestParameters(),
+  })
+}

load_testing/backend/test.ts

@@ -1,16 +1,54 @@
+import { check, group } from "k6"
+import exec from "k6/execution"
+import {
+  randomItem,
+  randomIntBetween,
+} from "https://jslib.k6.io/k6-utils/1.2.0/index.js"
+
 import { Config } from "../config.ts"
-import { check } from "k6"
-import { MITLearnAPIClient } from "./client/v0/api.ts"
+import { createV0Client, createV1Client } from "./client/client.ts"
 
 export function configure(config: Config) {
-  const v0Client = new MITLearnAPIClient({
-    baseUrl: config.apiBaseUrl,
-  })
   return function () {
-    const res = v0Client.videoShortsList({ limit: 50 })
-    check(res, {
-      "is status 200": (r) => r.status === 200,
-      "has results": (r) => r.json("results").length > 0,
+    group("api", function () {
+      group("v0", function () {
+        exec.vu.tags.apiVersion = "v0"
+
+        const client = createV0Client(config)
+
+        check(client.videoShortsList({ limit: 50 }), {
+          "is status 200": (r) => r.response.status === 200,
+          "has results": (r) => r.response.json("results").length > 0,
+        })
+
+        delete exec.vu.tags.apiVersion
+      })
+
+      group("v1", function () {
+        exec.vu.tags.apiVersion = "v1"
+
+        const client = createV1Client(config)
+
+        group("learning resources", function () {
+          const res = client.learningResourcesList({ limit: 50 })
+          const results = res.response.json("results")
+          check(res, {
+            "is status 200": (r) => r.response.status === 200,
+            "has results": (_) => results.length > 0,
+          })
+
+          for (let index = 0; index < randomIntBetween(3, 5); index++) {
+            const resource = randomItem(results)
+            check(client.learningResourcesRetrieve(resource.id), {
+              "is status 200": (r) => r.response.status === 200,
+              "matches resource id": (r) =>
+                r.response.json("id") == resource.id,
+            })
+          }
+        })
+
+        delete exec.vu.tags.apiVersion
+      })
     })
   }
 }