add tests for retries

Signed-off-by: Fedor Dikarev <fedor.dikarev@gmail.com>
This commit is contained in:
Fedor Dikarev 2025-01-23 14:24:04 +01:00
parent fd11be1ef0
commit c1f8f4fdcd
2 changed files with 119 additions and 0 deletions

View file

@ -0,0 +1,61 @@
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);
});

View file

@ -0,0 +1,58 @@
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 success function', async () => {
// let stderr_strings: string[] = []
let stderr_strings: string[] = [
'mock error, failed with status: 408 Request Timeout',
'mock error, failed with status: 502 Request Timeout',
'mock error, failed with status: 400 Request Timeout',
]
let call_count: number = -1
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 = []
call_count = -1
expect(async () => {
await login(registry, username, password, 'false', ['408', '502', '400'], 5, 0.1);
})
.resolves;
expect(Docker.getExecOutput).toHaveBeenCalledTimes(1);
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
expect(async () => {
await login(registry, username, password, 'false', ['408', '502', '400'], 5, 0.1);
})
.resolves;
expect(Docker.getExecOutput).toHaveBeenCalledTimes(0);
});