zitadel/load-test/src/use_cases/machine_jwt_profile_grant.ts

import { loginByUsernamePassword } from '../login_ui';
import { createOrg, removeOrg } from '../org';
import {createMachine, User, addMachineKey} from '../user';
import {JWTProfileRequest, token, userinfo} from '../oidc';
import { Config, MaxVUs } from '../config';
import encoding from 'k6/encoding';

const publicKey = encoding.b64encode(open('../.keys/key.pem.pub'));

export async function setup() {
  const tokens = loginByUsernamePassword(Config.admin as User);
  console.info('setup: admin signed in');
  
  const org = await createOrg(tokens.accessToken!);
  console.info(`setup: org (${org.organizationId}) created`);
  
  let machines = (
    await Promise.all(
      Array.from({ length: MaxVUs() }, (_, i) => {
        return createMachine(`zitachine-${i}`, org, tokens.accessToken!);
      }),
    )
  ).map((machine) => {
    return { userId: machine.userId, loginName: machine.loginNames[0] };
  });
  console.info(`setup: ${machines.length} machines created`);
  
  let keys = (
    await Promise.all(
      machines.map((machine) => {
        return addMachineKey(
          machine.userId, 
          org, 
          tokens.accessToken!,
          publicKey,        
        );
      }),
    )
  ).map((key, i) => {
    return { userId: machines[i].userId, keyId: key.keyId };
  });
  console.info(`setup: ${keys.length} keys added`);

  return { tokens, machines: keys, org };
}

export default function (data: any) {
  token(new JWTProfileRequest(data.machines[__VU - 1].userId, data.machines[__VU - 1].keyId))
}

export function teardown(data: any) {
  removeOrg(data.org, data.tokens.accessToken);
  console.info('teardown: org removed');
}