From e3d22cf2ff3fe9421edb93e8cdbbc6c7989a0c4d Mon Sep 17 00:00:00 2001
From: Hongming Wang <hongmingwang.rabbit@users.noreply.github.com>
Date: Sat, 18 Apr 2026 01:22:25 -0700
Subject: [PATCH] test: update mock stores for batch selection in existing
 canvas tests

Co-Authored-By: Claude Opus 4.6 (1M context) <noreply@anthropic.com>
---
 canvas/src/components/__tests__/Canvas.a11y.test.tsx         | 4 ++++
 canvas/src/components/__tests__/Canvas.pan-to-node.test.tsx  | 5 +++++
 canvas/src/components/__tests__/WorkspaceNode.a11y.test.tsx  | 2 ++
 canvas/src/components/__tests__/WorkspaceNode.eject.test.tsx | 2 ++
 4 files changed, 13 insertions(+)

diff --git a/canvas/src/components/__tests__/Canvas.a11y.test.tsx b/canvas/src/components/__tests__/Canvas.a11y.test.tsx
index 9e50f8fd..a8231eb3 100644
--- a/canvas/src/components/__tests__/Canvas.a11y.test.tsx
+++ b/canvas/src/components/__tests__/Canvas.a11y.test.tsx
@@ -69,6 +69,9 @@ const mockStoreState = {
   showA2AEdges: false,
   setShowA2AEdges: vi.fn(),
   setPanelTab: vi.fn(),
+  selectedNodeIds: new Set<string>(),
+  clearSelection: vi.fn(),
+  toggleNodeSelection: vi.fn(),
 };
 
 vi.mock("@/store/canvas", () => ({
@@ -109,6 +112,7 @@ vi.mock("../ProvisioningTimeout", () => ({
     <div data-testid="provisioning-timeout-sentinel" />
   ),
 }));
+vi.mock("../BatchActionBar", () => ({ BatchActionBar: () => null }));
 
 // ── Import the component under test AFTER mocks ───────────────────────────────
 import { Canvas } from "../Canvas";
diff --git a/canvas/src/components/__tests__/Canvas.pan-to-node.test.tsx b/canvas/src/components/__tests__/Canvas.pan-to-node.test.tsx
index c529a0e1..da38d896 100644
--- a/canvas/src/components/__tests__/Canvas.pan-to-node.test.tsx
+++ b/canvas/src/components/__tests__/Canvas.pan-to-node.test.tsx
@@ -79,6 +79,9 @@ const mockStoreState = {
   showA2AEdges: false,
   setShowA2AEdges: vi.fn(),
   setPanelTab: vi.fn(),
+  selectedNodeIds: new Set<string>(),
+  clearSelection: vi.fn(),
+  toggleNodeSelection: vi.fn(),
 };
 
 vi.mock("@/store/canvas", () => ({
@@ -113,6 +116,8 @@ vi.mock("../settings", () => ({
 }));
 vi.mock("../Toaster", () => ({ Toaster: () => null }));
 vi.mock("../WorkspaceNode", () => ({ WorkspaceNode: () => null }));
+vi.mock("../BatchActionBar", () => ({ BatchActionBar: () => null }));
+vi.mock("../ProvisioningTimeout", () => ({ ProvisioningTimeout: () => null }));
 
 import { Canvas } from "../Canvas";
 
diff --git a/canvas/src/components/__tests__/WorkspaceNode.a11y.test.tsx b/canvas/src/components/__tests__/WorkspaceNode.a11y.test.tsx
index ec680bf2..48c5cb35 100644
--- a/canvas/src/components/__tests__/WorkspaceNode.a11y.test.tsx
+++ b/canvas/src/components/__tests__/WorkspaceNode.a11y.test.tsx
@@ -125,6 +125,8 @@ const mockStoreState = {
   nestNode: mockNestNode,
   restartWorkspace: vi.fn(() => Promise.resolve()),
   setPanelTab: vi.fn(),
+  selectedNodeIds: new Set<string>(),
+  toggleNodeSelection: vi.fn(),
 };
 
 vi.mock("@/store/canvas", () => ({
diff --git a/canvas/src/components/__tests__/WorkspaceNode.eject.test.tsx b/canvas/src/components/__tests__/WorkspaceNode.eject.test.tsx
index 058b7cfc..691bc2cd 100644
--- a/canvas/src/components/__tests__/WorkspaceNode.eject.test.tsx
+++ b/canvas/src/components/__tests__/WorkspaceNode.eject.test.tsx
@@ -97,6 +97,8 @@ const mockStoreState = {
   isDescendant: vi.fn(() => false),
   restartWorkspace: vi.fn(),
   setPanelTab: vi.fn(),
+  selectedNodeIds: new Set<string>(),
+  toggleNodeSelection: vi.fn(),
 };
 
 vi.mock("@/store/canvas", () => ({