From fa8c252a0bcd2c443efb1b37bc25892623c29719 Mon Sep 17 00:00:00 2001 From: Ella Kramer Date: Fri, 19 Jul 2024 17:16:25 -0400 Subject: [PATCH] Update tests to account for changes --- __tests__/restore.test.ts | 18 ++++++++++++------ __tests__/restoreImpl.test.ts | 3 ++- 2 files changed, 14 insertions(+), 7 deletions(-) diff --git a/__tests__/restore.test.ts b/__tests__/restore.test.ts index 250f7ef..f53d376 100644 --- a/__tests__/restore.test.ts +++ b/__tests__/restore.test.ts @@ -85,7 +85,8 @@ test("restore with no cache found", async () => { ); expect(stateMock).toHaveBeenCalledWith("CACHE_KEY", key); - expect(stateMock).toHaveBeenCalledTimes(1); + expect(stateMock).toHaveBeenCalledWith("CACHE_PATH", path); + expect(stateMock).toHaveBeenCalledTimes(2); expect(failedMock).toHaveBeenCalledTimes(0); @@ -128,7 +129,8 @@ test("restore with restore keys and no cache found", async () => { ); expect(stateMock).toHaveBeenCalledWith("CACHE_KEY", key); - expect(stateMock).toHaveBeenCalledTimes(1); + expect(stateMock).toHaveBeenCalledWith("CACHE_PATH", path); + expect(stateMock).toHaveBeenCalledTimes(2); expect(failedMock).toHaveBeenCalledTimes(0); @@ -171,7 +173,8 @@ test("restore with cache found for key", async () => { expect(stateMock).toHaveBeenCalledWith("CACHE_KEY", key); expect(stateMock).toHaveBeenCalledWith("CACHE_RESULT", key); - expect(stateMock).toHaveBeenCalledTimes(2); + expect(stateMock).toHaveBeenCalledWith("CACHE_PATH", path); + expect(stateMock).toHaveBeenCalledTimes(3); expect(setCacheHitOutputMock).toHaveBeenCalledTimes(1); expect(setCacheHitOutputMock).toHaveBeenCalledWith("cache-hit", "true"); @@ -216,7 +219,8 @@ test("restore with cache found for restore key", async () => { expect(stateMock).toHaveBeenCalledWith("CACHE_KEY", key); expect(stateMock).toHaveBeenCalledWith("CACHE_RESULT", restoreKey); - expect(stateMock).toHaveBeenCalledTimes(2); + expect(stateMock).toHaveBeenCalledWith("CACHE_PATH", path); + expect(stateMock).toHaveBeenCalledTimes(3); expect(setCacheHitOutputMock).toHaveBeenCalledTimes(1); expect(setCacheHitOutputMock).toHaveBeenCalledWith("cache-hit", "false"); @@ -304,7 +308,8 @@ test("restore when fail on cache miss is enabled and primary key doesn't match r expect(stateMock).toHaveBeenCalledWith("CACHE_KEY", key); expect(stateMock).toHaveBeenCalledWith("CACHE_RESULT", restoreKey); - expect(stateMock).toHaveBeenCalledTimes(2); + expect(stateMock).toHaveBeenCalledWith("CACHE_PATH", path); + expect(stateMock).toHaveBeenCalledTimes(3); expect(setCacheHitOutputMock).toHaveBeenCalledTimes(1); expect(setCacheHitOutputMock).toHaveBeenCalledWith("cache-hit", "false"); @@ -349,7 +354,8 @@ test("restore with fail on cache miss disabled and no cache found", async () => ); expect(stateMock).toHaveBeenCalledWith("CACHE_KEY", key); - expect(stateMock).toHaveBeenCalledTimes(1); + expect(stateMock).toHaveBeenCalledWith("CACHE_PATH", path); + expect(stateMock).toHaveBeenCalledTimes(2); expect(infoMock).toHaveBeenCalledWith( `Cache not found for input keys: ${key}, ${restoreKey}` diff --git a/__tests__/restoreImpl.test.ts b/__tests__/restoreImpl.test.ts index 16f5f72..9aad012 100644 --- a/__tests__/restoreImpl.test.ts +++ b/__tests__/restoreImpl.test.ts @@ -439,7 +439,8 @@ test("restore with lookup-only set", async () => { expect(stateMock).toHaveBeenCalledWith("CACHE_KEY", key); expect(stateMock).toHaveBeenCalledWith("CACHE_RESULT", key); - expect(stateMock).toHaveBeenCalledTimes(2); + expect(stateMock).toHaveBeenCalledWith("CACHE_PATH", path); + expect(stateMock).toHaveBeenCalledTimes(3); expect(setCacheHitOutputMock).toHaveBeenCalledTimes(1); expect(setCacheHitOutputMock).toHaveBeenCalledWith("cache-hit", "true");