137 lines
4.0 KiB
TypeScript
137 lines
4.0 KiB
TypeScript
import { Artifact } from "../src/Artifact"
|
|
import { GithubArtifactUploader } from "../src/ArtifactUploader"
|
|
import { Releases } from "../src/Releases";
|
|
|
|
const artifacts = [
|
|
new Artifact('a/art1'),
|
|
new Artifact('b/art2')
|
|
]
|
|
const fileContents = Buffer.from('artful facts', 'utf-8')
|
|
const contentLength = 42
|
|
const releaseId = 100
|
|
const url = 'http://api.example.com'
|
|
|
|
const deleteMock = jest.fn()
|
|
const listArtifactsMock = jest.fn()
|
|
const uploadMock = jest.fn()
|
|
|
|
jest.mock('fs', () => {
|
|
return {
|
|
readFileSync: () => fileContents,
|
|
statSync: () => { return { size: contentLength } }
|
|
};
|
|
})
|
|
|
|
describe('ArtifactUploader', () => {
|
|
beforeEach(() => {
|
|
deleteMock.mockClear()
|
|
listArtifactsMock.mockClear()
|
|
uploadMock.mockClear()
|
|
})
|
|
|
|
it('replaces all artifacts', async () => {
|
|
mockDeleteSuccess()
|
|
mockListWithAssets()
|
|
const uploader = createUploader(true)
|
|
|
|
await uploader.uploadArtifacts(artifacts, releaseId, url)
|
|
|
|
expect(uploadMock).toBeCalledTimes(2)
|
|
expect(uploadMock)
|
|
.toBeCalledWith(url, contentLength, 'raw', fileContents, 'art1')
|
|
expect(uploadMock)
|
|
.toBeCalledWith(url, contentLength, 'raw', fileContents, 'art2')
|
|
|
|
expect(deleteMock).toBeCalledTimes(2)
|
|
expect(deleteMock).toBeCalledWith(1)
|
|
expect(deleteMock).toBeCalledWith(2)
|
|
})
|
|
|
|
it('replaces no artifacts when previous asset list empty', async () => {
|
|
mockDeleteSuccess()
|
|
mockListWithoutAssets()
|
|
const uploader = createUploader(true)
|
|
|
|
await uploader.uploadArtifacts(artifacts, releaseId, url)
|
|
|
|
expect(uploadMock).toBeCalledTimes(2)
|
|
expect(uploadMock)
|
|
.toBeCalledWith(url, contentLength, 'raw', fileContents, 'art1')
|
|
expect(uploadMock)
|
|
.toBeCalledWith(url, contentLength, 'raw', fileContents, 'art2')
|
|
|
|
expect(deleteMock).toBeCalledTimes(0)
|
|
})
|
|
|
|
it('throws error from replace', async () => {
|
|
mockDeleteError()
|
|
mockListWithAssets()
|
|
const uploader = createUploader(true)
|
|
|
|
expect.hasAssertions()
|
|
try {
|
|
await uploader.uploadArtifacts(artifacts, releaseId, url)
|
|
} catch (error) {
|
|
expect(error).toEqual("error")
|
|
}
|
|
})
|
|
|
|
it('updates all artifacts, delete none', async () => {
|
|
mockDeleteError()
|
|
mockListWithAssets()
|
|
const uploader = createUploader(false)
|
|
|
|
await uploader.uploadArtifacts(artifacts, releaseId, url)
|
|
|
|
expect(uploadMock).toBeCalledTimes(2)
|
|
expect(uploadMock)
|
|
.toBeCalledWith(url, contentLength, 'raw', fileContents, 'art1')
|
|
expect(uploadMock)
|
|
.toBeCalledWith(url, contentLength, 'raw', fileContents, 'art2')
|
|
|
|
expect(deleteMock).toBeCalledTimes(0)
|
|
})
|
|
|
|
function createUploader(replaces: boolean): GithubArtifactUploader {
|
|
uploadMock.mockResolvedValue({})
|
|
const MockReleases = jest.fn<Releases, any>(() => {
|
|
return {
|
|
create: jest.fn(),
|
|
deleteArtifact: deleteMock,
|
|
getByTag: jest.fn(),
|
|
listArtifactsForRelease: listArtifactsMock,
|
|
listReleases: jest.fn(),
|
|
update: jest.fn(),
|
|
uploadArtifact: uploadMock
|
|
}
|
|
})
|
|
return new GithubArtifactUploader(new MockReleases(), replaces)
|
|
}
|
|
|
|
function mockDeleteError(): any {
|
|
deleteMock.mockRejectedValue("error")
|
|
}
|
|
|
|
function mockDeleteSuccess(): any {
|
|
deleteMock.mockResolvedValue({})
|
|
}
|
|
|
|
function mockListWithAssets() {
|
|
listArtifactsMock.mockResolvedValue({
|
|
data: [
|
|
{
|
|
name: "art1",
|
|
id: 1
|
|
},
|
|
{
|
|
name: "art2",
|
|
id: 2
|
|
}
|
|
]
|
|
})
|
|
}
|
|
|
|
function mockListWithoutAssets() {
|
|
listArtifactsMock.mockResolvedValue({ data: [] })
|
|
}
|
|
}); |