actions-cache/src/utils/testUtils.ts

import { Inputs } from "../constants";

// See: https://github.com/actions/toolkit/blob/master/packages/core/src/core.ts#L67
function getInputName(name: string): string {
    return `INPUT_${name.replace(/ /g, "_").toUpperCase()}`;
}

export function setInput(name: string, value: string): void {
    process.env[getInputName(name)] = value;
}

interface CacheInput {
    path: string;
    key: string;
    restoreKeys?: string[];
    strictRestore?: string;
    saveOnAnyFailure?: string;
}

export function setInputs(input: CacheInput): void {
    setInput(Inputs.Path, input.path);
    setInput(Inputs.Key, input.key);
    setInput(Inputs.SaveOnAnyFailure, "false");
    setInput(Inputs.FailOnCacheMiss, "false");
    input.restoreKeys &&
        setInput(Inputs.RestoreKeys, input.restoreKeys.join("\n"));
    input.strictRestore && setInput(Inputs.FailOnCacheMiss, input.strictRestore);
    input.saveOnAnyFailure &&
        setInput(Inputs.SaveOnAnyFailure, input.saveOnAnyFailure);
}

export function clearInputs(): void {
    delete process.env[getInputName(Inputs.Path)];
    delete process.env[getInputName(Inputs.Key)];
    delete process.env[getInputName(Inputs.RestoreKeys)];
    delete process.env[getInputName(Inputs.FailOnCacheMiss)];
    delete process.env[getInputName(Inputs.SaveOnAnyFailure)];
    delete process.env[getInputName(Inputs.UploadChunkSize)];
}