attempting to fix workspace stories

Kira-Pilot · Kira-Pilot · commit 819983c28210 · 2023-05-17T21:03:17.000Z
diff --git a/site/src/components/Workspace/Workspace.stories.tsx b/site/src/components/Workspace/Workspace.stories.tsx
@@ -2,12 +2,12 @@ import { action } from "@storybook/addon-actions"
 import { Story } from "@storybook/react"
 import { WatchAgentMetadataContext } from "components/Resources/AgentMetadata"
 import { ProvisionerJobLog } from "api/typesGenerated"
-import * as Mocks from "../../testHelpers/entities"
+import * as Mocks from "testHelpers/entities"
 import { Workspace, WorkspaceErrors, WorkspaceProps } from "./Workspace"
 import { withReactContext } from "storybook-react-context"
 import EventSource from "eventsourcemock"
 import { ProxyContext, getPreferredProxy } from "contexts/ProxyContext"
-import { MockProxyLatencies } from "../../testHelpers/entities"
+import { DashboardProviderContext } from "components/Dashboard/DashboardProvider"
 
 export default {
   title: "components/Workspace",
@@ -24,21 +24,37 @@ export default {
   ],
 }
 
+const MockedAppearance = {
+  config: Mocks.MockAppearance,
+  preview: false,
+  setPreview: () => {},
+  save: () => {},
+}
+
 const Template: Story<WorkspaceProps> = (args) => (
-  <ProxyContext.Provider
+  <DashboardProviderContext.Provider
     value={{
-      proxyLatencies: MockProxyLatencies,
-      proxy: getPreferredProxy([], undefined),
-      proxies: [],
-      isLoading: false,
-      isFetched: true,
-      setProxy: () => {
-        return
-      },
+      buildInfo: Mocks.MockBuildInfo,
+      entitlements: Mocks.MockEntitlementsWithScheduling,
+      experiments: Mocks.MockExperiments,
+      appearance: MockedAppearance,
     }}
   >
-    <Workspace {...args} />
-  </ProxyContext.Provider>
+    <ProxyContext.Provider
+      value={{
+        proxyLatencies: Mocks.MockProxyLatencies,
+        proxy: getPreferredProxy([], undefined),
+        proxies: [],
+        isLoading: false,
+        isFetched: true,
+        setProxy: () => {
+          return
+        },
+      }}
+    >
+      <Workspace {...args} />
+    </ProxyContext.Provider>
+  </DashboardProviderContext.Provider>
 )
 
 export const Running = Template.bind({})
