test: fix stories

DanielleMaywood · DanielleMaywood · commit 97432c9e2488 · 2025-08-06T17:10:05.000Z
diff --git a/site/src/modules/resources/AgentRow.stories.tsx b/site/src/modules/resources/AgentRow.stories.tsx
@@ -286,11 +286,17 @@ export const GroupApp: Story = {
 };
 
 export const Devcontainer: Story = {
-	beforeEach: () => {
-		spyOn(API, "getAgentContainers").mockResolvedValue({
-			devcontainers: [M.MockWorkspaceAgentDevcontainer],
-			containers: [M.MockWorkspaceAgentContainer],
-		});
+	parameters: {
+		queries: [
+			{
+				key: ["agents", M.MockWorkspaceAgent.id, "containers"],
+				data: {
+					devcontainers: [M.MockWorkspaceAgentDevcontainer],
+					containers: [M.MockWorkspaceAgentContainer],
+				},
+			},
+		],
+		webSocket: [],
 	},
 };
 
@@ -300,17 +306,23 @@ export const FoundDevcontainer: Story = {
 			...M.MockWorkspaceAgentReady,
 		},
 	},
-	beforeEach: () => {
-		spyOn(API, "getAgentContainers").mockResolvedValue({
-			devcontainers: [
-				{
-					...M.MockWorkspaceAgentDevcontainer,
-					status: "stopped",
-					container: undefined,
-					agent: undefined,
+	parameters: {
+		queries: [
+			{
+				key: ["agents", M.MockWorkspaceAgentReady.id, "containers"],
+				data: {
+					devcontainers: [
+						{
+							...M.MockWorkspaceAgentDevcontainer,
+							status: "stopped",
+							container: undefined,
+							agent: undefined,
+						},
+					],
+					containers: [],
 				},
-			],
-			containers: [],
-		});
+			},
+		],
+		webSocket: [],
 	},
 };
