feat(DashboardPage): add loading story

opengovsg · Jul 19, 2024 · d3bb1a3 · d3bb1a3
1 parent c7bd38b
commit d3bb1a3
Show file tree

Hide file tree

Showing 2 changed files with 17 additions and 2 deletions.
diff --git a/apps/studio/src/stories/Page/DashboardPage.stories.tsx b/apps/studio/src/stories/Page/DashboardPage.stories.tsx
@@ -22,3 +22,11 @@ type Story = StoryObj<typeof DashboardPage>
 export const Default: Story = {
   name: "Dashboard Page",
 }
+
+export const Loading: Story = {
+  parameters: {
+    msw: {
+      handlers: [meHandlers.me(), sitesHandlers.list.loading()],
+    },
+  },
+}
diff --git a/apps/studio/tests/msw/handlers/sites.ts b/apps/studio/tests/msw/handlers/sites.ts
@@ -1,7 +1,13 @@
+import type { DelayMode } from "msw"
+import { delay } from "msw"
+
 import { trpcMsw } from "../mockTrpc"
 
-const siteListQuery = () => {
-  return trpcMsw.site.list.query(() => {
+const siteListQuery = (wait?: DelayMode | number) => {
+  return trpcMsw.site.list.query(async () => {
+    if (wait !== undefined) {
+      await delay(wait)
+    }
     return [
       {
         id: 1,
@@ -14,5 +20,6 @@ const siteListQuery = () => {
 export const sitesHandlers = {
   list: {
     default: siteListQuery,
+    loading: () => siteListQuery("infinite"),
   },
 }