actions/toolkit
1
0
Fork 0
mirror of https://git.mirrors.martin98.com/https://github.com/actions/toolkit synced 2026-03-19 23:32:36 +08:00
Code Issues Packages Projects Releases Wiki Activity

Replace downloadCacheFile with downloadCacheStorageSDK

Browse Source
This commit is contained in:
Bassem Dghaidi
2024-11-27 04:51:21 -08:00
committed by GitHub
parent b050504b2d
commit 27e5cf2514
3 changed files with 59 additions and 50 deletions
Download Patch File Download Diff File Expand all files Collapse all files

75
packages/cache/__tests__/restoreCacheV2.test.ts vendored
View File

@@ -3,11 +3,11 @@ 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 downloadCacheModule from '../src/internal/blob/download-cache'
import * as downloadUtils from '../src/internal/downloadUtils'
import {restoreCache} from '../src/cache'
import {CacheFilename, CompressionMethod} from '../src/internal/constants'
import {CacheServiceClientJSON} from '../src/generated/results/api/v1/cache.twirp'
import {BlobDownloadResponseParsed} from '@azure/storage-blob'
import {DownloadOptions} from '../src/options'
jest.mock('../src/internal/cacheHttpClient')
jest.mock('../src/internal/cacheUtils')
@@ -142,6 +142,7 @@ test('restore with gzip compressed cache found', async () => {
const signedDownloadUrl = 'https://blob-storage.local?signed=true'
const cacheVersion =
'd90f107aaeb22920dba0c637a23c37b5bc497b4dfa3b07fe3f79bf88a273c11b'
const options: DownloadOptions = {timeoutInMs: 30000}
const getCacheVersionMock = jest.spyOn(cacheUtils, 'getCacheVersion')
getCacheVersionMock.mockReturnValue(cacheVersion)
@@ -169,17 +170,11 @@ test('restore with gzip compressed cache found', async () => {
})
const archivePath = path.join(tempPath, CacheFilename.Gzip)
const downloadCacheFileMock = jest.spyOn(
downloadCacheModule,
'downloadCacheFile'
)
downloadCacheFileMock.mockReturnValue(
Promise.resolve({
_response: {
status: 200
}
} as BlobDownloadResponseParsed)
const downloadCacheStorageSDKMock = jest.spyOn(
downloadUtils,
'downloadCacheStorageSDK'
)
downloadCacheStorageSDKMock.mockReturnValue(Promise.resolve())
const fileSize = 142
const getArchiveFileSizeInBytesMock = jest
@@ -203,9 +198,10 @@ test('restore with gzip compressed cache found', async () => {
version: cacheVersion
})
expect(createTempDirectoryMock).toHaveBeenCalledTimes(1)
expect(downloadCacheFileMock).toHaveBeenCalledWith(
expect(downloadCacheStorageSDKMock).toHaveBeenCalledWith(
signedDownloadUrl,
archivePath
archivePath,
options
)
expect(getArchiveFileSizeInBytesMock).toHaveBeenCalledWith(archivePath)
expect(logInfoMock).toHaveBeenCalledWith(`Cache Size: ~0 MB (142 B)`)
@@ -226,6 +222,7 @@ test('restore with zstd compressed cache found', async () => {
const signedDownloadUrl = 'https://blob-storage.local?signed=true'
const cacheVersion =
'8e2e96a184cb0cd6b48285b176c06a418f3d7fce14c29d9886fd1bb4f05c513d'
const options: DownloadOptions = {timeoutInMs: 30000}
const getCacheVersionMock = jest.spyOn(cacheUtils, 'getCacheVersion')
getCacheVersionMock.mockReturnValue(cacheVersion)
@@ -253,17 +250,11 @@ test('restore with zstd compressed cache found', async () => {
})
const archivePath = path.join(tempPath, CacheFilename.Zstd)
const downloadCacheFileMock = jest.spyOn(
downloadCacheModule,
'downloadCacheFile'
)
downloadCacheFileMock.mockReturnValue(
Promise.resolve({
_response: {
status: 200
}
} as BlobDownloadResponseParsed)
const downloadCacheStorageSDKMock = jest.spyOn(
downloadUtils,
'downloadCacheStorageSDK'
)
downloadCacheStorageSDKMock.mockReturnValue(Promise.resolve())
const fileSize = 62915000
const getArchiveFileSizeInBytesMock = jest
@@ -287,9 +278,10 @@ test('restore with zstd compressed cache found', async () => {
version: cacheVersion
})
expect(createTempDirectoryMock).toHaveBeenCalledTimes(1)
expect(downloadCacheFileMock).toHaveBeenCalledWith(
expect(downloadCacheStorageSDKMock).toHaveBeenCalledWith(
signedDownloadUrl,
archivePath
archivePath,
options
)
expect(getArchiveFileSizeInBytesMock).toHaveBeenCalledWith(archivePath)
expect(logInfoMock).toHaveBeenCalledWith(`Cache Size: ~60 MB (62915000 B)`)
@@ -311,6 +303,7 @@ test('restore with cache found for restore key', async () => {
const signedDownloadUrl = 'https://blob-storage.local?signed=true'
const cacheVersion =
'b8b58e9bd7b1e8f83d9f05c7e06ea865ba44a0330e07a14db74ac74386677bed'
const options: DownloadOptions = {timeoutInMs: 30000}
const getCacheVersionMock = jest.spyOn(cacheUtils, 'getCacheVersion')
getCacheVersionMock.mockReturnValue(cacheVersion)
@@ -338,17 +331,11 @@ test('restore with cache found for restore key', async () => {
})
const archivePath = path.join(tempPath, CacheFilename.Gzip)
const downloadCacheFileMock = jest.spyOn(
downloadCacheModule,
'downloadCacheFile'
)
downloadCacheFileMock.mockReturnValue(
Promise.resolve({
_response: {
status: 200
}
} as BlobDownloadResponseParsed)
const downloadCacheStorageSDKMock = jest.spyOn(
downloadUtils,
'downloadCacheStorageSDK'
)
downloadCacheStorageSDKMock.mockReturnValue(Promise.resolve())
const fileSize = 142
const getArchiveFileSizeInBytesMock = jest
@@ -372,9 +359,10 @@ test('restore with cache found for restore key', async () => {
version: cacheVersion
})
expect(createTempDirectoryMock).toHaveBeenCalledTimes(1)
expect(downloadCacheFileMock).toHaveBeenCalledWith(
expect(downloadCacheStorageSDKMock).toHaveBeenCalledWith(
signedDownloadUrl,
archivePath
archivePath,
options
)
expect(getArchiveFileSizeInBytesMock).toHaveBeenCalledWith(archivePath)
expect(logInfoMock).toHaveBeenCalledWith(`Cache Size: ~0 MB (142 B)`)
@@ -391,11 +379,11 @@ test('restore with cache found for restore key', async () => {
test('restore with dry run', async () => {
const paths = ['node_modules']
const key = 'node-test'
const options = {lookupOnly: true}
const compressionMethod = CompressionMethod.Gzip
const signedDownloadUrl = 'https://blob-storage.local?signed=true'
const cacheVersion =
'd90f107aaeb22920dba0c637a23c37b5bc497b4dfa3b07fe3f79bf88a273c11b'
const options: DownloadOptions = {lookupOnly: true, timeoutInMs: 30000}
const getCacheVersionMock = jest.spyOn(cacheUtils, 'getCacheVersion')
getCacheVersionMock.mockReturnValue(cacheVersion)
@@ -416,10 +404,11 @@ test('restore with dry run', async () => {
)
const createTempDirectoryMock = jest.spyOn(cacheUtils, 'createTempDirectory')
const downloadCacheFileMock = jest.spyOn(
downloadCacheModule,
'downloadCacheFile'
const downloadCacheStorageSDKMock = jest.spyOn(
downloadUtils,
'downloadCacheStorageSDK'
)
downloadCacheStorageSDKMock.mockReturnValue(Promise.resolve())
const cacheKey = await restoreCache(paths, key, undefined, options)
@@ -438,5 +427,5 @@ test('restore with dry run', async () => {
// creating a tempDir and downloading the cache are skipped
expect(createTempDirectoryMock).toHaveBeenCalledTimes(0)
expect(downloadCacheFileMock).toHaveBeenCalledTimes(0)
expect(downloadCacheStorageSDKMock).toHaveBeenCalledTimes(0)
})