mirror of https://github.com/actions/toolkit
328 lines
12 KiB
TypeScript
328 lines
12 KiB
TypeScript
import * as core from '@actions/core'
|
|
import * as path from 'path'
|
|
import * as tar from '../src/internal/tar'
|
|
import * as config from '../src/internal/config'
|
|
import * as cacheUtils from '../src/internal/cacheUtils'
|
|
import * as cacheHttpClient from '../src/internal/cacheHttpClient'
|
|
import { restoreCache } from '../src/cache'
|
|
import { CacheFilename, CompressionMethod } from '../src/internal/constants'
|
|
import { ArtifactCacheEntry } from '../src/internal/contracts'
|
|
import { CacheServiceClientJSON } from '../src/generated/results/api/v1/cache.twirp'
|
|
|
|
jest.mock('../src/internal/cacheHttpClient')
|
|
jest.mock('../src/internal/cacheUtils')
|
|
jest.mock('../src/internal/config')
|
|
jest.mock('../src/internal/tar')
|
|
|
|
beforeAll(() => {
|
|
jest.spyOn(console, 'log').mockImplementation(() => { })
|
|
jest.spyOn(core, 'debug').mockImplementation(() => { })
|
|
jest.spyOn(core, 'info').mockImplementation(() => { })
|
|
jest.spyOn(core, 'warning').mockImplementation(() => { })
|
|
jest.spyOn(core, 'error').mockImplementation(() => { })
|
|
|
|
jest.spyOn(cacheUtils, 'getCacheFileName').mockImplementation(cm => {
|
|
const actualUtils = jest.requireActual('../src/internal/cacheUtils')
|
|
return actualUtils.getCacheFileName(cm)
|
|
})
|
|
|
|
// Ensure that we're using v2 for these tests
|
|
jest.spyOn(config, 'getCacheServiceVersion').mockReturnValue('v2')
|
|
})
|
|
|
|
test('restore with no path should fail', async () => {
|
|
const paths: string[] = []
|
|
const key = 'node-test'
|
|
await expect(restoreCache(paths, key)).rejects.toThrowError(
|
|
`Path Validation Error: At least one directory or file path is required`
|
|
)
|
|
})
|
|
|
|
test('restore with too many keys should fail', async () => {
|
|
const paths = ['node_modules']
|
|
const key = 'node-test'
|
|
const restoreKeys = [...Array(20).keys()].map(x => x.toString())
|
|
await expect(restoreCache(paths, key, restoreKeys)).rejects.toThrowError(
|
|
`Key Validation Error: Keys are limited to a maximum of 10.`
|
|
)
|
|
})
|
|
|
|
test('restore with large key should fail', async () => {
|
|
const paths = ['node_modules']
|
|
const key = 'foo'.repeat(512) // Over the 512 character limit
|
|
await expect(restoreCache(paths, key)).rejects.toThrowError(
|
|
`Key Validation Error: ${key} cannot be larger than 512 characters.`
|
|
)
|
|
})
|
|
|
|
test('restore with invalid key should fail', async () => {
|
|
const paths = ['node_modules']
|
|
const key = 'comma,comma'
|
|
await expect(restoreCache(paths, key)).rejects.toThrowError(
|
|
`Key Validation Error: ${key} cannot contain commas.`
|
|
)
|
|
})
|
|
|
|
test('restore with no cache found', async () => {
|
|
const paths = ['node_modules']
|
|
const key = 'node-test'
|
|
|
|
jest
|
|
.spyOn(CacheServiceClientJSON.prototype, 'GetCacheEntryDownloadURL')
|
|
.mockReturnValue(Promise.resolve({ ok: false, signedDownloadUrl: '' }))
|
|
|
|
const cacheKey = await restoreCache(paths, key)
|
|
|
|
expect(cacheKey).toBe(undefined)
|
|
})
|
|
|
|
test('restore with server error should fail', async () => {
|
|
const paths = ['node_modules']
|
|
const key = 'node-test'
|
|
const logWarningMock = jest.spyOn(core, 'warning')
|
|
|
|
jest
|
|
.spyOn(CacheServiceClientJSON.prototype, 'GetCacheEntryDownloadURL')
|
|
.mockImplementation(() => {
|
|
throw new Error('HTTP Error Occurred')
|
|
})
|
|
|
|
const cacheKey = await restoreCache(paths, key)
|
|
expect(cacheKey).toBe(undefined)
|
|
expect(logWarningMock).toHaveBeenCalledTimes(1)
|
|
expect(logWarningMock).toHaveBeenCalledWith(
|
|
'Failed to restore: HTTP Error Occurred'
|
|
)
|
|
})
|
|
|
|
// test('restore with restore keys and no cache found', async () => {
|
|
// const paths = ['node_modules']
|
|
// const key = 'node-test'
|
|
// const restoreKey = 'node-'
|
|
|
|
// jest
|
|
// .spyOn(CacheServiceClientJSON.prototype, 'GetCacheEntryDownloadURL')
|
|
// .mockImplementation(() => {
|
|
// return Promise.resolve(null)
|
|
// })
|
|
// jest.spyOn(cacheHttpClient, 'getCacheEntry').mockImplementation(async () => {
|
|
// return Promise.resolve(null)
|
|
// })
|
|
|
|
// const cacheKey = await restoreCache(paths, key, [restoreKey])
|
|
|
|
// expect(cacheKey).toBe(undefined)
|
|
// })
|
|
|
|
// test('restore with gzip compressed cache found', async () => {
|
|
// const paths = ['node_modules']
|
|
// const key = 'node-test'
|
|
|
|
// const cacheEntry: ArtifactCacheEntry = {
|
|
// cacheKey: key,
|
|
// scope: 'refs/heads/main',
|
|
// archiveLocation: 'www.actionscache.test/download'
|
|
// }
|
|
// const getCacheMock = jest.spyOn(cacheHttpClient, 'getCacheEntry')
|
|
// getCacheMock.mockImplementation(async () => {
|
|
// return Promise.resolve(cacheEntry)
|
|
// })
|
|
|
|
// const tempPath = '/foo/bar'
|
|
|
|
// const createTempDirectoryMock = jest.spyOn(cacheUtils, 'createTempDirectory')
|
|
// createTempDirectoryMock.mockImplementation(async () => {
|
|
// return Promise.resolve(tempPath)
|
|
// })
|
|
|
|
// const archivePath = path.join(tempPath, CacheFilename.Gzip)
|
|
// const downloadCacheMock = jest.spyOn(cacheHttpClient, 'downloadCache')
|
|
|
|
// const fileSize = 142
|
|
// const getArchiveFileSizeInBytesMock = jest
|
|
// .spyOn(cacheUtils, 'getArchiveFileSizeInBytes')
|
|
// .mockReturnValue(fileSize)
|
|
|
|
// const extractTarMock = jest.spyOn(tar, 'extractTar')
|
|
// const unlinkFileMock = jest.spyOn(cacheUtils, 'unlinkFile')
|
|
|
|
// const compression = CompressionMethod.Gzip
|
|
// const getCompressionMock = jest
|
|
// .spyOn(cacheUtils, 'getCompressionMethod')
|
|
// .mockReturnValue(Promise.resolve(compression))
|
|
|
|
// const cacheKey = await restoreCache(paths, key)
|
|
|
|
// expect(cacheKey).toBe(key)
|
|
// expect(getCacheMock).toHaveBeenCalledWith([key], paths, {
|
|
// compressionMethod: compression,
|
|
// enableCrossOsArchive: false
|
|
// })
|
|
// expect(createTempDirectoryMock).toHaveBeenCalledTimes(1)
|
|
// expect(downloadCacheMock).toHaveBeenCalledWith(
|
|
// cacheEntry.archiveLocation,
|
|
// archivePath,
|
|
// undefined
|
|
// )
|
|
// expect(getArchiveFileSizeInBytesMock).toHaveBeenCalledWith(archivePath)
|
|
|
|
// expect(extractTarMock).toHaveBeenCalledTimes(1)
|
|
// expect(extractTarMock).toHaveBeenCalledWith(archivePath, compression)
|
|
|
|
// expect(unlinkFileMock).toHaveBeenCalledTimes(1)
|
|
// expect(unlinkFileMock).toHaveBeenCalledWith(archivePath)
|
|
|
|
// expect(getCompressionMock).toHaveBeenCalledTimes(1)
|
|
// })
|
|
|
|
// test('restore with zstd compressed cache found', async () => {
|
|
// const paths = ['node_modules']
|
|
// const key = 'node-test'
|
|
|
|
// const infoMock = jest.spyOn(core, 'info')
|
|
|
|
// const cacheEntry: ArtifactCacheEntry = {
|
|
// cacheKey: key,
|
|
// scope: 'refs/heads/main',
|
|
// archiveLocation: 'www.actionscache.test/download'
|
|
// }
|
|
// const getCacheMock = jest.spyOn(cacheHttpClient, 'getCacheEntry')
|
|
// getCacheMock.mockImplementation(async () => {
|
|
// return Promise.resolve(cacheEntry)
|
|
// })
|
|
// const tempPath = '/foo/bar'
|
|
|
|
// const createTempDirectoryMock = jest.spyOn(cacheUtils, 'createTempDirectory')
|
|
// createTempDirectoryMock.mockImplementation(async () => {
|
|
// return Promise.resolve(tempPath)
|
|
// })
|
|
|
|
// const archivePath = path.join(tempPath, CacheFilename.Zstd)
|
|
// const downloadCacheMock = jest.spyOn(cacheHttpClient, 'downloadCache')
|
|
|
|
// const fileSize = 62915000
|
|
// const getArchiveFileSizeInBytesMock = jest
|
|
// .spyOn(cacheUtils, 'getArchiveFileSizeInBytes')
|
|
// .mockReturnValue(fileSize)
|
|
|
|
// const extractTarMock = jest.spyOn(tar, 'extractTar')
|
|
// const compression = CompressionMethod.Zstd
|
|
// const getCompressionMock = jest
|
|
// .spyOn(cacheUtils, 'getCompressionMethod')
|
|
// .mockReturnValue(Promise.resolve(compression))
|
|
|
|
// const cacheKey = await restoreCache(paths, key)
|
|
|
|
// expect(cacheKey).toBe(key)
|
|
// expect(getCacheMock).toHaveBeenCalledWith([key], paths, {
|
|
// compressionMethod: compression,
|
|
// enableCrossOsArchive: false
|
|
// })
|
|
// expect(createTempDirectoryMock).toHaveBeenCalledTimes(1)
|
|
// expect(downloadCacheMock).toHaveBeenCalledWith(
|
|
// cacheEntry.archiveLocation,
|
|
// archivePath,
|
|
// undefined
|
|
// )
|
|
// expect(getArchiveFileSizeInBytesMock).toHaveBeenCalledWith(archivePath)
|
|
// expect(infoMock).toHaveBeenCalledWith(`Cache Size: ~60 MB (62915000 B)`)
|
|
|
|
// expect(extractTarMock).toHaveBeenCalledTimes(1)
|
|
// expect(extractTarMock).toHaveBeenCalledWith(archivePath, compression)
|
|
// expect(getCompressionMock).toHaveBeenCalledTimes(1)
|
|
// })
|
|
|
|
// test('restore with cache found for restore key', async () => {
|
|
// const paths = ['node_modules']
|
|
// const key = 'node-test'
|
|
// const restoreKey = 'node-'
|
|
|
|
// const infoMock = jest.spyOn(core, 'info')
|
|
|
|
// const cacheEntry: ArtifactCacheEntry = {
|
|
// cacheKey: restoreKey,
|
|
// scope: 'refs/heads/main',
|
|
// archiveLocation: 'www.actionscache.test/download'
|
|
// }
|
|
// const getCacheMock = jest.spyOn(cacheHttpClient, 'getCacheEntry')
|
|
// getCacheMock.mockImplementation(async () => {
|
|
// return Promise.resolve(cacheEntry)
|
|
// })
|
|
// const tempPath = '/foo/bar'
|
|
|
|
// const createTempDirectoryMock = jest.spyOn(cacheUtils, 'createTempDirectory')
|
|
// createTempDirectoryMock.mockImplementation(async () => {
|
|
// return Promise.resolve(tempPath)
|
|
// })
|
|
|
|
// const archivePath = path.join(tempPath, CacheFilename.Zstd)
|
|
// const downloadCacheMock = jest.spyOn(cacheHttpClient, 'downloadCache')
|
|
|
|
// const fileSize = 142
|
|
// const getArchiveFileSizeInBytesMock = jest
|
|
// .spyOn(cacheUtils, 'getArchiveFileSizeInBytes')
|
|
// .mockReturnValue(fileSize)
|
|
|
|
// const extractTarMock = jest.spyOn(tar, 'extractTar')
|
|
// const compression = CompressionMethod.Zstd
|
|
// const getCompressionMock = jest
|
|
// .spyOn(cacheUtils, 'getCompressionMethod')
|
|
// .mockReturnValue(Promise.resolve(compression))
|
|
|
|
// const cacheKey = await restoreCache(paths, key, [restoreKey])
|
|
|
|
// expect(cacheKey).toBe(restoreKey)
|
|
// expect(getCacheMock).toHaveBeenCalledWith([key, restoreKey], paths, {
|
|
// compressionMethod: compression,
|
|
// enableCrossOsArchive: false
|
|
// })
|
|
// expect(createTempDirectoryMock).toHaveBeenCalledTimes(1)
|
|
// expect(downloadCacheMock).toHaveBeenCalledWith(
|
|
// cacheEntry.archiveLocation,
|
|
// archivePath,
|
|
// undefined
|
|
// )
|
|
// expect(getArchiveFileSizeInBytesMock).toHaveBeenCalledWith(archivePath)
|
|
// expect(infoMock).toHaveBeenCalledWith(`Cache Size: ~0 MB (142 B)`)
|
|
|
|
// expect(extractTarMock).toHaveBeenCalledTimes(1)
|
|
// expect(extractTarMock).toHaveBeenCalledWith(archivePath, compression)
|
|
// expect(getCompressionMock).toHaveBeenCalledTimes(1)
|
|
// })
|
|
|
|
// test('restore with dry run', async () => {
|
|
// const paths = ['node_modules']
|
|
// const key = 'node-test'
|
|
// const options = { lookupOnly: true }
|
|
|
|
// const cacheEntry: ArtifactCacheEntry = {
|
|
// cacheKey: key,
|
|
// scope: 'refs/heads/main',
|
|
// archiveLocation: 'www.actionscache.test/download'
|
|
// }
|
|
// const getCacheMock = jest.spyOn(cacheHttpClient, 'getCacheEntry')
|
|
// getCacheMock.mockImplementation(async () => {
|
|
// return Promise.resolve(cacheEntry)
|
|
// })
|
|
|
|
// const createTempDirectoryMock = jest.spyOn(cacheUtils, 'createTempDirectory')
|
|
// const downloadCacheMock = jest.spyOn(cacheHttpClient, 'downloadCache')
|
|
|
|
// const compression = CompressionMethod.Gzip
|
|
// const getCompressionMock = jest
|
|
// .spyOn(cacheUtils, 'getCompressionMethod')
|
|
// .mockReturnValue(Promise.resolve(compression))
|
|
|
|
// const cacheKey = await restoreCache(paths, key, undefined, options)
|
|
|
|
// expect(cacheKey).toBe(key)
|
|
// expect(getCompressionMock).toHaveBeenCalledTimes(1)
|
|
// expect(getCacheMock).toHaveBeenCalledWith([key], paths, {
|
|
// compressionMethod: compression,
|
|
// enableCrossOsArchive: false
|
|
// })
|
|
// // creating a tempDir and downloading the cache are skipped
|
|
// expect(createTempDirectoryMock).toHaveBeenCalledTimes(0)
|
|
// expect(downloadCacheMock).toHaveBeenCalledTimes(0)
|
|
// })
|