From 353950894a28dc87e872df952d4ef9536c926043 Mon Sep 17 00:00:00 2001
From: Ayaan Zaidi <hi@obviy.us>
Date: Fri, 17 Apr 2026 08:37:11 +0530
Subject: [PATCH] test(plugins): address discovery review feedback

---
 src/plugins/discovery.test.ts | 1 +
 src/plugins/discovery.ts      | 3 ++-
 2 files changed, 3 insertions(+), 1 deletion(-)

diff --git a/src/plugins/discovery.test.ts b/src/plugins/discovery.test.ts
index 53552746bbc..58d1e372b9e 100644
--- a/src/plugins/discovery.test.ts
+++ b/src/plugins/discovery.test.ts
@@ -258,6 +258,7 @@ async function expectRejectedPackageExtensionEntry(params: {
 }
 
 afterEach(() => {
+  vi.restoreAllMocks();
   clearPluginDiscoveryCache();
   cleanupTrackedTempDirs(tempDirs);
 });
diff --git a/src/plugins/discovery.ts b/src/plugins/discovery.ts
index 17111b81858..3232cfbe7e0 100644
--- a/src/plugins/discovery.ts
+++ b/src/plugins/discovery.ts
@@ -102,8 +102,9 @@ function buildScopedDiscoveryCacheKey(params: {
     env: params.env,
   });
   const workspaceKey = roots.workspace ?? "";
+  const bundledRoot = roots.stock ?? "";
   const ownershipUid = params.ownershipUid ?? currentUid();
-  return `scoped::${workspaceKey}::${ownershipUid ?? "none"}::${JSON.stringify(loadPaths)}`;
+  return `scoped::${workspaceKey}::${bundledRoot}::${ownershipUid ?? "none"}::${JSON.stringify(loadPaths)}`;
 }
 
 function buildSharedDiscoveryCacheKey(params: {