login-action/__tests__/retries_fail.test.ts

import { expect, jest, test, } from '@jest/globals';
import * as path from 'path';

// import * as dockerModule from '../src/docker';

import { login } from '../src/docker';
import { Docker } from '@docker/actions-toolkit/lib/docker/docker';

test('login retries function', async () => {
  let stderr_strings: string[] = []
  let call_count: number = -1

  // const execSpy = jest.spyOn(Docker, 'getExecOutput').mockImplementation(async () => {
  Docker.getExecOutput = jest.fn(async () => {
    call_count++
    console.log(`Mock: ${call_count}, ${stderr_strings}`)
    if (call_count >= stderr_strings.length) {
      return {
        exitCode: 0,
        stdout: 'Mock success',
        stderr: ''
      }
    }
    return {
      exitCode: 1,
      stdout: '',
      stderr: stderr_strings[(call_count) % stderr_strings.length]
    }
  })

  const username = 'dbowie';
  const password = 'groundcontrol';
  const registry = 'https://ghcr.io';

  stderr_strings = [
    'mock error, failed with status: 408 Request Timeout',
    'mock error, failed with status: 502 Request Timeout',
    'mock error, failed with status: 400 Request Timeout',
  ]
  call_count = -1
  await expect(async () => {
    await login(registry, username, password, 'false', ['408', '400'], 5, 0.1);
  })
    .rejects
    .toThrow("mock error, failed with status: 502 Request Timeout");
  expect(Docker.getExecOutput).toHaveBeenCalledTimes(2);

  stderr_strings = [
    'not matching error',
    'mock error, failed with status: 502 Request Timeout',
    'mock error, failed with status: 400 Request Timeout',
  ]
  call_count = -1
  await expect(async () => {
    await login(registry, username, password, 'false', ['408', '400'], 5, 0.1);
  })
    .rejects
    .toThrow('not matching error');
  expect(Docker.getExecOutput).toHaveBeenCalledTimes(2 + 1);

});