Add more stories for deployment dropdown

code-asher · code-asher · commit 72bf3453d513 · 2024-08-01T14:33:48.000-08:00
diff --git a/site/src/modules/dashboard/Navbar/NavbarView.stories.tsx b/site/src/modules/dashboard/Navbar/NavbarView.stories.tsx
@@ -1,4 +1,5 @@
 import type { Meta, StoryObj } from "@storybook/react";
+import { within, userEvent } from "@storybook/test";
 import { chromaticWithTablet } from "testHelpers/chromatic";
 import { MockUser, MockUser2 } from "testHelpers/entities";
 import { withDashboardProvider } from "testHelpers/storybook";
@@ -10,26 +11,63 @@ const meta: Meta<typeof NavbarView> = {
   component: NavbarView,
   args: {
     user: MockUser,
+    canViewAllUsers: true,
     canViewAuditLog: true,
     canViewDeployment: true,
-    canViewAllUsers: true,
     canViewHealth: true,
+    canViewOrganizations: true,
   },
   decorators: [withDashboardProvider],
 };
 
 export default meta;
 type Story = StoryObj<typeof NavbarView>;
 
-export const ForAdmin: Story = {};
+export const ForAdmin: Story = {
+  play: async ({ canvasElement }) => {
+    const canvas = within(canvasElement);
+    await userEvent.click(canvas.getByText("Deployment"));
+  },
+};
+
+export const ForAuditor: Story = {
+  args: {
+    user: MockUser2,
+    canViewAllUsers: false,
+    canViewAuditLog: true,
+    canViewDeployment: false,
+    canViewHealth: false,
+    canViewOrganizations: false,
+  },
+  play: async ({ canvasElement }) => {
+    const canvas = within(canvasElement);
+    await userEvent.click(canvas.getByText("Deployment"));
+  },
+};
+
+export const ForOrgAdmin: Story = {
+  args: {
+    user: MockUser2,
+    canViewAllUsers: false,
+    canViewAuditLog: true,
+    canViewDeployment: false,
+    canViewHealth: false,
+    canViewOrganizations: true,
+  },
+  play: async ({ canvasElement }) => {
+    const canvas = within(canvasElement);
+    await userEvent.click(canvas.getByText("Deployment"));
+  },
+};
 
 export const ForMember: Story = {
   args: {
     user: MockUser2,
+    canViewAllUsers: false,
     canViewAuditLog: false,
     canViewDeployment: false,
-    canViewAllUsers: false,
     canViewHealth: false,
+    canViewOrganizations: false,
   },
 };
 
