runner-container-hooks/packages/k8s/tests/cleanup-job-test.ts

import * as k8s from '@kubernetes/client-node'
import { cleanupJob, prepareJob } from '../src/hooks'
import { RunnerInstanceLabel } from '../src/hooks/constants'
import { namespace } from '../src/k8s'
import { TestHelper } from './test-setup'
import { PrepareJobArgs } from 'hooklib'

let testHelper: TestHelper

describe('Cleanup Job', () => {
  beforeEach(async () => {
    testHelper = new TestHelper()
    await testHelper.initialize()
    let prepareJobData = testHelper.getPrepareJobDefinition()
    const prepareJobOutputFilePath = testHelper.createFile(
      'prepare-job-output.json'
    )
    await prepareJob(
      prepareJobData.args as PrepareJobArgs,
      prepareJobOutputFilePath
    )
  })

  afterEach(async () => {
    await testHelper.cleanup()
  })

  it('should not throw', async () => {
    await expect(cleanupJob()).resolves.not.toThrow()
  })

  it('should have no runner linked pods running', async () => {
    await cleanupJob()
    const kc = new k8s.KubeConfig()

    kc.loadFromDefault()
    const k8sApi = kc.makeApiClient(k8s.CoreV1Api)

    const podList = await k8sApi.listNamespacedPod({
      namespace: namespace(),
      labelSelector: new RunnerInstanceLabel().toString()
    })

    expect(podList.items.length).toBe(0)
  })

  it('should have no runner linked secrets', async () => {
    await cleanupJob()
    const kc = new k8s.KubeConfig()

    kc.loadFromDefault()
    const k8sApi = kc.makeApiClient(k8s.CoreV1Api)

    const secretList = await k8sApi.listNamespacedSecret({
      namespace: namespace(),
      labelSelector: new RunnerInstanceLabel().toString()
    })

    expect(secretList.items.length).toBe(0)
  })
})