fix storybook

johnstcn · johnstcn · commit 3a1655773a28 · 2022-08-18T22:16:26.000Z
diff --git a/site/src/components/Workspace/Workspace.stories.tsx b/site/src/components/Workspace/Workspace.stories.tsx
@@ -1,5 +1,7 @@
 import { action } from "@storybook/addon-actions"
 import { Story } from "@storybook/react"
+import dayjs from "dayjs"
+import { canExtendDeadline, canReduceDeadline } from "util/schedule"
 import * as Mocks from "../../testHelpers/entities"
 import { Workspace, WorkspaceErrors, WorkspaceProps } from "./Workspace"
 
@@ -24,6 +26,16 @@ Started.args = {
     onDeadlinePlus: () => {
       // do nothing, this is just for storybook
     },
+    deadlineMinusEnabled: () => {
+      return canReduceDeadline(dayjs(Mocks.MockWorkspace.latest_build.deadline))
+    },
+    deadlinePlusEnabled: () => {
+      return canExtendDeadline(
+        dayjs(Mocks.MockWorkspace.latest_build.deadline),
+        Mocks.MockWorkspace,
+        Mocks.MockTemplate,
+      )
+    },
   },
   workspace: Mocks.MockWorkspace,
   handleStart: action("start"),
