dartboard/k6/crds/crd_utils.js

import { check, fail, sleep } from 'k6';
import http from 'k6/http'
import { retryUntilExpected } from "../rancher/rancher_utils.js";
import * as YAML from '../lib/js-yaml-4.1.0.mjs'


export const baseCRDPath = "v1/apiextensions.k8s.io.customresourcedefinitions"
export const crdTag = { url: `/v1/apiextensions.k8s.io.customresourcedefinitions/<CRD ID>` }
export const crdsTag = { url: `/v1/apiextensions.k8s.io.customresourcedefinitions` }
export const putCRDTag = { url: `/v1/apiextensions.k8s.io.customresourcedefinitions/<CRD Name>` }

export const crdRefreshDelaySeconds = 2
export const crdRefreshDelayMs = crdRefreshDelaySeconds * 1000
export const backgroundRefreshSeconds = 10
export const backgroundRefreshMs = backgroundRefreshSeconds * 1000


export function cleanupMatchingCRDs(baseUrl, cookies, namePrefix) {
  let res = http.get(`${baseUrl}/${baseCRDPath}`, { cookies: cookies })
  check(res, {
    '/v1/apiextensions.k8s.io.customresourcedefinitions returns status 200': (r) => r.status === 200,
  })
  JSON.parse(res.body)["data"].filter(r => r["metadata"]["name"].startsWith(namePrefix)).forEach(r => {
    res = http.del(`${baseUrl}/${baseCRDPath}/${r["id"]}`, { cookies: cookies })
    check(res, {
      'DELETE /v1/apiextensions.k8s.io.customresourcedefinitions returns status 204': (r) => r.status === 204,
    })
  })
}

export function getCRD(baseUrl, cookies, id) {
  let res = http.get(`${baseUrl}/${baseCRDPath}/${id}`, { cookies: cookies, tag: crdTag })
  let criteria = []
  criteria[`GET /${baseCRDPath}/<CRD ID> returns status 200`] = (r) => r.status === 200
  check(res, criteria)
  // console.log(`GET CRD status: ${res.status}`)
  return res
}

export function getCRDs(baseUrl, cookies) {
  let res = http.get(`${baseUrl}/${baseCRDPath}`, { cookies: cookies, tags: crdsTag })
  check(res, {
    'GET /v1/apiextensions.k8s.io.customresourcedefinitions returns status 200': (r) => r.status === 200,
  })
  return res
}

export function verifyCRDs(baseUrl, cookies, namePrefix, expectedLength, timeoutMs) {
  const timeWas = new Date();
  let timeSpent = null
  let res = null
  let criteria = []
  let currentLength = -1
  // Poll customresourcedefinitions until receiving a 200
  while (new Date() - timeWas < timeoutMs) {
    res = retryUntilExpected(200, 9, () => { return getCRDs(baseUrl, cookies) })
    timeSpent = new Date() - timeWas
    if (res.status === 200) {
      let data = JSON.parse(res.body)["data"]
      data = data.filter(r => r["metadata"]["name"].startsWith(namePrefix))
      currentLength = data.length
      if (currentLength == expectedLength) {
        console.log("Polling conditions met after ", timeSpent, "ms");
        break;
      }
    } else {
      console.log("Polling CRDs failed to receive 200 status")
    }
  }
  criteria[`detected the expected # of CRDs "${expectedLength}" matches the received # of CRDs "${currentLength}"`] = (r) => currentLength == expectedLength
  check(res, criteria)
  return { res: res, timeSpent: timeSpent }
}

export function createCRD(baseUrl, cookies, suffix) {
  const namePattern = `-test-${suffix}`
  const res = http.post(
    `${baseUrl}/${baseCRDPath}`,
    `apiVersion: apiextensions.k8s.io/v1\nkind: CustomResourceDefinition\nmetadata:\n  name: crontabs${namePattern}.stable.example.com\nspec:\n  group: stable.example.com\n  versions:\n    - name: v1\n      served: true\n      storage: false\n      schema:\n        openAPIV3Schema:\n          type: object\n          properties:\n            spec:\n              type: object\n              properties:\n                cronSpec:\n                  type: string\n                image:\n                  type: string\n                replicas:\n                  type: integer\n    - name: v2\n      served: true\n      storage: true\n      schema:\n        openAPIV3Schema:\n          type: object\n          properties:\n            spec:\n              type: object\n              properties:\n                cronSpec:\n                  type: string\n                newField:\n                  type: string\n                image:\n                  type: string\n                replicas:\n                  type: integer\n  scope: Namespaced\n  names:\n    plural: crontabs${namePattern}\n    singular: crontab${namePattern}\n    kind: CronTab${namePattern}\n    shortNames:\n      - ct${namePattern}\n`,
    {
      headers: {
        "content-type": "application/yaml",
      },
      cookies: cookies,
      tags: crdsTag,
    }
  )
  check(res, {
    'CRD post returns 201 (created)': (r) => r.status === 201,
  })
  return res
}

export function deleteCRD(baseUrl, cookies, id) {
  let res = http.del(`${baseUrl}/${baseCRDPath}/${id}`, null, { cookies: cookies, tags: crdTag })

  check(res, {
    'DELETE /v1/apiextensions.k8s.io.customresourcedefinitions returns status 200': (r) => r.status === 200 || r.status === 204,
  })
  return res
}

export function getCRDsMatchingName(baseUrl, cookies, namePrefix) {
  let res = getCRDs(baseUrl, cookies)
  if (!Object.hasOwn(res, 'body') || !Object.hasOwn(JSON.parse(res.body), 'data')) {
    console.log("Response doesn't have body");
    return { res: res, crdArray: [] };
  }
  let crdArray = []
  crdArray = JSON.parse(res.body)["data"]
  crdArray = crdArray.filter(r => r["metadata"]["name"].startsWith(namePrefix))
  return { res: res, crdArray: crdArray }
}

export function getCRDsMatchingNameVersions(baseUrl, cookies, namePrefix, numVersions) {
  let res = getCRDs(baseUrl, cookies)
  if (!Object.hasOwn(res, 'body') || !Object.hasOwn(JSON.parse(res.body), 'data')) {
    console.log("Response doesn't have body");
    return { res: res, crdArray: [] };
  }
  let crdArray = JSON.parse(res.body)["data"]
  crdArray = crdArray.filter(r => r["metadata"]["name"].startsWith(namePrefix) && r["spec"]["versions"].length == numVersions)
  return crdArray
}

export function teardown(data) {
  cleanup(data.cookies)
}

export function updateCRD(baseUrl, cookies, crd) {
  let body = YAML.dump(crd)
  let res = http.put(
    `${baseUrl}/${baseCRDPath}/${crd.metadata.name}`,
    body,
    {
      headers: {
        accept: 'application/json',
        'content-type': 'application/yaml',
      },
      cookies: cookies,
      tags: putCRDTag,
    }
  )
  check(res, {
    'PUT /v1/apiextensions.k8s.io.customresourcedefinitions returns status 200': (r) => r.status === 200,
  })
  return res
}