Update auth setup

Author: rhysyngsun

load_testing/auth.ts

@@ -1,13 +1,41 @@
 import { SharedArray } from "k6/data"
 
-export function getAccessToken(): String {
+export type AuthCredential = {
+  username: string
+  password: string
+}
+
+export function getAccessToken(): string | null {
   return __ENV.AUTH_ACCESS_TOKEN
 }
 
-export const users = new SharedArray("users", function () {
-  if (!__ENV.USERS_JSON_FILE) {
-    return []
+function _validate_credentials(credentials) {
+  if (typeof credentials !== "array") {
+    throw Error("Expected an array of credentials")
+  }
+
+  for (let index = 0; index < credentials.length; index++) {
+    const credential = credentials[index]
+    if (!Object.hasOwn(credential, "email")) {
+      throw Error(`User entry is missing 'email' at index ${index}`)
+    }
+    if (!Object.hasOwn(credential, "password")) {
+      throw Error(`User entry is missing 'password' at index ${index}`)
+    }
   }
+}
+
+export const credentials: AuthCredential[] = new SharedArray(
+  "credentials",
+  function () {
+    if (!__ENV.USERS_JSON_FILE) {
+      return []
+    }
+
+    const parsed = JSON.parse(open(__ENV.USERS_JSON_FILE))
+
+    _validate_credentials(parsed)
 
-  return JSON.parse(open(__ENV.USERS_JSON_FILE))
-})
+    return parsed
+  },
+)

load_testing/frontend/test.ts

@@ -1,7 +1,11 @@
 import { check } from "k6"
 import { browser, Page } from "k6/browser"
-import { randomIntBetween } from "https://jslib.k6.io/k6-utils/1.2.0/index.js"
+import {
+  randomIntBetween,
+  randomItem,
+} from "https://jslib.k6.io/k6-utils/1.2.0/index.js"
 import { BROWSER_CONTEXT_OPTIONS, FRONTEND_BASE_URL } from "../config.ts"
+import { AuthCredential, credentials } from "../auth.ts"
 
 async function home(page: Page) {
   await page.goto(FRONTEND_BASE_URL)
@@ -38,15 +42,47 @@ async function units(page: Page) {
   await page.goto(`${FRONTEND_BASE_URL}/units`)
 }
 
+async function login(page: Page) {
+  const credential: AuthCredential = randomItem(credentials)
+
+  if (!!credential) {
+    log.debug("Skipping login because no credentials provided")
+    return
+  }
+
+  const loginButton = page.getByTestId("login-button-desktop")
+  await loginButton.first().click()
+
+  await loginKeycloak(page)
+
+  await page.waitForURL(/.*\/dashboard.*/)
+}
+
+async function loginKeycloak(page: Page, credential: AuthCredential) {
+  await page.waitForURL(
+    /https:\/sso(\-qa)?\.ol\.mit\.edu\/realms\/olapps\/protocol\/openid-connect\/auth.*/,
+  )
+
+  const credentialnameInput = await page.locator("input#username")
+  await credentialnameInput.type(user.email)
+  await page.locator("button#kc-login").click()
+
+  const passwordInput = await page.locator("input#password")
+  await passwordInput.type(credential.password)
+  await page.locator("button#kc-login").click()
+}
+
 export async function testFrontend() {
   const page = await browser.newPage(BROWSER_CONTEXT_OPTIONS)
 
   try {
+    // always home page first
     await home(page)
     await search(page)
     await topics(page)
     await departments(page)
     await units(page)
+    await login(page)
   } finally {
     await page.close()
   }