mirror of https://github.com/actions/toolkit
91 lines
2.3 KiB
TypeScript
91 lines
2.3 KiB
TypeScript
import {MockAgent, setGlobalDispatcher} from 'undici'
|
|
import {writeAttestation} from '../src/store'
|
|
|
|
describe('writeAttestation', () => {
|
|
const originalEnv = process.env
|
|
const attestation = {foo: 'bar '}
|
|
const token = 'token'
|
|
|
|
const mockAgent = new MockAgent()
|
|
setGlobalDispatcher(mockAgent)
|
|
|
|
beforeEach(() => {
|
|
process.env = {
|
|
...originalEnv,
|
|
GITHUB_REPOSITORY: 'foo/bar'
|
|
}
|
|
})
|
|
|
|
afterEach(() => {
|
|
process.env = originalEnv
|
|
})
|
|
|
|
describe('when the api call is successful', () => {
|
|
beforeEach(() => {
|
|
mockAgent
|
|
.get('https://api.github.com')
|
|
.intercept({
|
|
path: '/repos/foo/bar/attestations',
|
|
method: 'POST',
|
|
headers: {authorization: `token ${token}`},
|
|
body: JSON.stringify({bundle: attestation})
|
|
})
|
|
.reply(201, {id: '123'})
|
|
})
|
|
|
|
it('persists the attestation', async () => {
|
|
await expect(writeAttestation(attestation, token)).resolves.toEqual('123')
|
|
})
|
|
})
|
|
|
|
describe('when the api call fails', () => {
|
|
beforeEach(() => {
|
|
mockAgent
|
|
.get('https://api.github.com')
|
|
.intercept({
|
|
path: '/repos/foo/bar/attestations',
|
|
method: 'POST',
|
|
headers: {authorization: `token ${token}`},
|
|
body: JSON.stringify({bundle: attestation})
|
|
})
|
|
.reply(500, 'oops')
|
|
})
|
|
|
|
it('throws an error', async () => {
|
|
await expect(
|
|
writeAttestation(attestation, token, {retry: 0})
|
|
).rejects.toThrow(/oops/)
|
|
})
|
|
})
|
|
|
|
describe('when the api call fails but succeeds on retry', () => {
|
|
beforeEach(() => {
|
|
const pool = mockAgent.get('https://api.github.com')
|
|
|
|
pool
|
|
.intercept({
|
|
path: '/repos/foo/bar/attestations',
|
|
method: 'POST',
|
|
headers: {authorization: `token ${token}`},
|
|
body: JSON.stringify({bundle: attestation})
|
|
})
|
|
.reply(500, 'oops')
|
|
.times(1)
|
|
|
|
pool
|
|
.intercept({
|
|
path: '/repos/foo/bar/attestations',
|
|
method: 'POST',
|
|
headers: {authorization: `token ${token}`},
|
|
body: JSON.stringify({bundle: attestation})
|
|
})
|
|
.reply(201, {id: '123'})
|
|
.times(1)
|
|
})
|
|
|
|
it('persists the attestation', async () => {
|
|
await expect(writeAttestation(attestation, token)).resolves.toEqual('123')
|
|
})
|
|
})
|
|
})
|