adding wrapper to save-only and testcases

src/save-only.ts

@@ -0,0 +1,20 @@
+import * as core from "@actions/core";
+
+import { Inputs } from "./constants";
+import save from "./save";
+import * as utils from "./utils/actionUtils";
+
+async function runSaveAction(): Promise<void> {
+    if (!core.getInput(Inputs.Key)) {
+        utils.logWarning(`Error retrieving key from inputs.`);
+        return;
+    }
+    saveOnly = true;
+
+    await save();
+}
+
+runSaveAction();
+
+export default runSaveAction;
+export let saveOnly: boolean;

src/save.ts

@@ -2,6 +2,7 @@ import * as cache from "@actions/cache";
 import * as core from "@actions/core";
 
 import { Events, Inputs, State } from "./constants";
+import { saveOnly } from "./save-only";
 import * as utils from "./utils/actionUtils";
 
 // Catch and log any unhandled exceptions.  These exceptions can leak out of the uploadChunk method in
@@ -28,7 +29,9 @@ async function run(): Promise<void> {
 
         // Inputs are re-evaluted before the post action, so we want the original key used for restore
         const primaryKey =
-            core.getState(State.CachePrimaryKey) || core.getInput(Inputs.Key);
+            saveOnly === true
+                ? core.getInput(Inputs.Key)
+                : core.getState(State.CachePrimaryKey);
 
         if (!primaryKey) {
             utils.logWarning(`Error retrieving key from state.`);
@@ -58,6 +61,4 @@ async function run(): Promise<void> {
     }
 }
 
-run();
-
 export default run;