fix right stories

Kira-Pilot · Kira-Pilot · commit 8322b633ee07 · 2023-05-17T21:53:11.000Z
diff --git a/site/src/pages/CreateWorkspacePage/CreateWorkspacePageView.stories.tsx b/site/src/pages/CreateWorkspacePage/CreateWorkspacePageView.stories.tsx
@@ -7,41 +7,20 @@ import {
   MockTemplateVersionParameter1,
   MockTemplateVersionParameter2,
   MockTemplateVersionParameter3,
-  MockBuildInfo,
-  MockEntitlementsWithScheduling,
-  MockExperiments,
-  MockAppearance,
-} from "testHelpers/entities"
+} from "../../testHelpers/entities"
 import {
   CreateWorkspaceErrors,
   CreateWorkspacePageView,
   CreateWorkspacePageViewProps,
 } from "./CreateWorkspacePageView"
-import { DashboardProviderContext } from "components/Dashboard/DashboardProvider"
 
 export default {
   title: "pages/CreateWorkspacePageView",
   component: CreateWorkspacePageView,
 } as ComponentMeta<typeof CreateWorkspacePageView>
 
-const MockedAppearance = {
-  config: MockAppearance,
-  preview: false,
-  setPreview: () => null,
-  save: () => null,
-}
-
 const Template: Story<CreateWorkspacePageViewProps> = (args) => (
-  <DashboardProviderContext.Provider
-    value={{
-      buildInfo: MockBuildInfo,
-      entitlements: MockEntitlementsWithScheduling,
-      experiments: MockExperiments,
-      appearance: MockedAppearance,
-    }}
-  >
-    <CreateWorkspacePageView {...args} />
-  </DashboardProviderContext.Provider>
+  <CreateWorkspacePageView {...args} />
 )
 
 export const NoParameters = Template.bind({})
diff --git a/site/src/pages/WorkspacesPage/WorkspacesPageView.stories.tsx b/site/src/pages/WorkspacesPage/WorkspacesPageView.stories.tsx
@@ -6,13 +6,20 @@ import {
   Workspace,
   WorkspaceStatus,
   WorkspaceStatuses,
-} from "../../api/typesGenerated"
-import { MockWorkspace } from "../../testHelpers/entities"
-import { workspaceFilterQuery } from "../../utils/filters"
+} from "api/typesGenerated"
+import {
+  MockWorkspace,
+  MockAppearance,
+  MockBuildInfo,
+  MockEntitlementsWithScheduling,
+  MockExperiments,
+} from "testHelpers/entities"
+import { workspaceFilterQuery } from "utils/filters"
 import {
   WorkspacesPageView,
   WorkspacesPageViewProps,
 } from "./WorkspacesPageView"
+import { DashboardProviderContext } from "components/Dashboard/DashboardProvider"
 
 const createWorkspace = (
   status: WorkspaceStatus,
@@ -55,6 +62,13 @@ const allWorkspaces = [
   ...Object.values(additionalWorkspaces),
 ]
 
+const MockedAppearance = {
+  config: MockAppearance,
+  preview: false,
+  setPreview: () => null,
+  save: () => null,
+}
+
 export default {
   title: "pages/WorkspacesPageView",
   component: WorkspacesPageView,
@@ -64,7 +78,16 @@ export default {
 } as ComponentMeta<typeof WorkspacesPageView>
 
 const Template: Story<WorkspacesPageViewProps> = (args) => (
-  <WorkspacesPageView {...args} />
+  <DashboardProviderContext.Provider
+    value={{
+      buildInfo: MockBuildInfo,
+      entitlements: MockEntitlementsWithScheduling,
+      experiments: MockExperiments,
+      appearance: MockedAppearance,
+    }}
+  >
+    <WorkspacesPageView {...args} />
+  </DashboardProviderContext.Provider>
 )
 
 export const AllStates = Template.bind({})
