From 41f23846bba086f04a5c5fe2eb9da7d9e9c5758d Mon Sep 17 00:00:00 2001
From: Daniel Kennedy <danwkennedy@github.com>
Date: Thu, 29 Jan 2026 14:12:58 -0500
Subject: [PATCH] Add retries to restore-cache.mjs

---
 packages/cache/__tests__/restore-cache.mjs | 33 +++++++++++++++-------
 1 file changed, 23 insertions(+), 10 deletions(-)

diff --git a/packages/cache/__tests__/restore-cache.mjs b/packages/cache/__tests__/restore-cache.mjs
index ad42c44b..7be76930 100644
--- a/packages/cache/__tests__/restore-cache.mjs
+++ b/packages/cache/__tests__/restore-cache.mjs
@@ -17,16 +17,29 @@ console.log(`Restoring cache with key: ${key}`)
 console.log(`Paths: ${paths.join(', ')}`)
 console.log(`Using Azure SDK: ${options.useAzureSdk}`)
 
-try {
-  const restoredKey = await cache.restoreCache(paths, key, [], options)
+const maxRetries = 3
+const retryDelayMs = 5000
 
-  if (restoredKey) {
-    console.log(`Cache restored with key: ${restoredKey}`)
-  } else {
-    console.error('Cache not found')
-    process.exit(1)
+for (let attempt = 1; attempt <= maxRetries; attempt++) {
+  try {
+    console.log(`Attempt ${attempt} of ${maxRetries}`)
+    const restoredKey = await cache.restoreCache(paths, key, [], options)
+
+    if (restoredKey) {
+      console.log(`Cache restored with key: ${restoredKey}`)
+      process.exit(0)
+    } else {
+      console.log('Cache not found on this attempt')
+    }
+  } catch (error) {
+    console.error(`Error on attempt ${attempt}:`, error.message)
+  }
+
+  if (attempt < maxRetries) {
+    console.log(`Waiting ${retryDelayMs / 1000}s before retry...`)
+    await new Promise(resolve => setTimeout(resolve, retryDelayMs))
   }
-} catch (error) {
-  console.error('Error restoring cache:', error)
-  process.exit(1)
 }
+
+console.error(`Failed to restore cache after ${maxRetries} attempts`)
+process.exit(1)