test: improve Storybook tests

AbhineetJain · AbhineetJain · commit 317782166c65 · 2022-06-01T20:22:57.000Z
diff --git a/site/src/components/UserDropdown/UserDropdown.stories.tsx b/site/src/components/UserDropdown/UserDropdown.stories.tsx
@@ -24,3 +24,12 @@ ExampleNoRoles.args = {
     return Promise.resolve()
   },
 }
+
+export const Open = Template.bind({})
+Open.args = {
+  isOpen: true,
+  user: MockUser,
+  onSignOut: () => {
+    return Promise.resolve()
+  },
+}
diff --git a/site/src/components/UserDropdown/UsersDropdown.tsx b/site/src/components/UserDropdown/UsersDropdown.tsx
@@ -22,11 +22,12 @@ export const Language = {
   signOutLabel: "Sign Out",
 }
 export interface UserDropdownProps {
+  isOpen?: boolean // *DO NOT USE* Only used for testing via Storybook
   user: TypesGen.User
   onSignOut: () => void
 }
 
-export const UserDropdown: React.FC<UserDropdownProps> = ({ user, onSignOut }: UserDropdownProps) => {
+export const UserDropdown: React.FC<UserDropdownProps> = ({ isOpen, user, onSignOut }: UserDropdownProps) => {
   const styles = useStyles()
   const [anchorEl, setAnchorEl] = useState<HTMLElement | undefined>()
 
@@ -51,7 +52,7 @@ export const UserDropdown: React.FC<UserDropdownProps> = ({ user, onSignOut }: U
       <BorderedMenu
         anchorEl={anchorEl}
         getContentAnchorEl={null}
-        open={!!anchorEl}
+        open={!!anchorEl || !!isOpen}
         anchorOrigin={{
           vertical: "bottom",
           horizontal: "right",
diff --git a/site/src/components/UserProfileCard/UserProfileCard.stories.tsx b/site/src/components/UserProfileCard/UserProfileCard.stories.tsx
@@ -4,11 +4,9 @@ import { MockUser } from "../../testHelpers/entities"
 import { UserProfileCard, UserProfileCardProps } from "./UserProfileCard"
 
 export default {
-  title: "components/UserDropdown",
+  title: "components/UserProfileCard",
   component: UserProfileCard,
-  argTypes: {
-    onSignOut: { action: "Sign Out" },
-  },
+  argTypes: {},
 }
 
 const Template: Story<UserProfileCardProps> = (args: UserProfileCardProps) => <UserProfileCard {...args} />
