chore(site): fix storybook for agent row with port forward button (#12247)

BrunoQuaresma · web-flow · commit 0398e3c531f8 · 2024-02-21T14:32:39.000Z
diff --git a/site/src/modules/resources/AgentRow.stories.tsx b/site/src/modules/resources/AgentRow.stories.tsx
@@ -24,6 +24,7 @@ import {
 } from "testHelpers/entities";
 import { AgentRow, LineWithID } from "./AgentRow";
 import { ProxyContext, getPreferredProxy } from "contexts/ProxyContext";
+import { withDashboardProvider } from "testHelpers/storybook";
 
 const defaultAgentMetadata = [
   {
@@ -113,6 +114,7 @@ const meta: Meta<typeof AgentRow> = {
       },
     ],
   },
+
   component: AgentRow,
   args: {
     storybookLogs,
@@ -147,6 +149,7 @@ const meta: Meta<typeof AgentRow> = {
         <Story />
       </ProxyContext.Provider>
     ),
+    withDashboardProvider,
   ],
 };
 
