Fix stories

Emyrk · Emyrk · commit 27ef4a928589 · 2023-04-27T14:09:41.000-05:00
diff --git a/site/src/components/AppLink/AppLink.stories.tsx b/site/src/components/AppLink/AppLink.stories.tsx
@@ -1,17 +1,33 @@
 import { Story } from "@storybook/react"
 import {
+  MockPrimaryRegion,
+  MockRegions,
   MockWorkspace,
   MockWorkspaceAgent,
   MockWorkspaceApp,
 } from "testHelpers/entities"
 import { AppLink, AppLinkProps } from "./AppLink"
+import { ProxyContext } from "contexts/ProxyContext"
+import { getURLs } from "contexts/ProxyContext"
 
 export default {
   title: "components/AppLink",
   component: AppLink,
 }
 
-const Template: Story<AppLinkProps> = (args) => <AppLink {...args} />
+const Template: Story<AppLinkProps> = (args) => (
+  <ProxyContext.Provider
+    value={{
+      proxy: getURLs(MockRegions, MockPrimaryRegion),
+      isLoading: false,
+      setProxy: () => {
+        return
+      },
+    }}
+  >
+    <AppLink {...args} />
+  </ProxyContext.Provider>
+)
 
 export const WithIcon = Template.bind({})
 WithIcon.args = {
diff --git a/site/src/components/Resources/AgentRow.stories.tsx b/site/src/components/Resources/AgentRow.stories.tsx
@@ -1,5 +1,7 @@
 import { Story } from "@storybook/react"
 import {
+  MockPrimaryRegion,
+  MockRegions,
   MockWorkspace,
   MockWorkspaceAgent,
   MockWorkspaceAgentConnecting,
@@ -16,6 +18,7 @@ import {
   MockWorkspaceApp,
 } from "testHelpers/entities"
 import { AgentRow, AgentRowProps } from "./AgentRow"
+import { ProxyContext, getURLs } from "contexts/ProxyContext"
 
 export default {
   title: "components/AgentRow",
@@ -36,7 +39,19 @@ export default {
   },
 }
 
-const Template: Story<AgentRowProps> = (args) => <AgentRow {...args} />
+const Template: Story<AgentRowProps> = (args) => (
+  <ProxyContext.Provider
+    value={{
+      proxy: getURLs(MockRegions, MockPrimaryRegion),
+      isLoading: false,
+      setProxy: () => {
+        return
+      },
+    }}
+  >
+    <AgentRow {...args} />
+  </ProxyContext.Provider>
+)
 
 const defaultAgentMetadata = [
   {
diff --git a/site/src/components/Workspace/Workspace.stories.tsx b/site/src/components/Workspace/Workspace.stories.tsx
@@ -6,6 +6,7 @@ import * as Mocks from "../../testHelpers/entities"
 import { Workspace, WorkspaceErrors, WorkspaceProps } from "./Workspace"
 import { withReactContext } from "storybook-react-context"
 import EventSource from "eventsourcemock"
+import { ProxyContext, getURLs } from "contexts/ProxyContext"
 
 export default {
   title: "components/Workspace",
@@ -22,7 +23,19 @@ export default {
   ],
 }
 
-const Template: Story<WorkspaceProps> = (args) => <Workspace {...args} />
+const Template: Story<WorkspaceProps> = (args) => (
+  <ProxyContext.Provider
+    value={{
+      proxy: getURLs(Mocks.MockRegions, Mocks.MockPrimaryRegion),
+      isLoading: false,
+      setProxy: () => {
+        return
+      },
+    }}
+  >
+    <Workspace {...args} />
+  </ProxyContext.Provider>
+)
 
 export const Running = Template.bind({})
 Running.args = {
diff --git a/site/src/pages/UserSettingsPage/WorkspaceProxyPage/WorspaceProxyView.stories.tsx b/site/src/pages/UserSettingsPage/WorkspaceProxyPage/WorspaceProxyView.stories.tsx
@@ -1,5 +1,10 @@
 import { Story } from "@storybook/react"
-import { makeMockApiError, MockRegions, MockPrimaryRegion, MockHealthyWildRegion } from "testHelpers/entities"
+import {
+  makeMockApiError,
+  MockRegions,
+  MockPrimaryRegion,
+  MockHealthyWildRegion,
+} from "testHelpers/entities"
 import {
   WorkspaceProxyPageView,
   WorkspaceProxyPageViewProps,
