Fix Audit story

presleyp · presleyp · commit e92cbbfa61e3 · 2022-10-25T18:04:45.000Z
diff --git a/site/src/pages/AuditPage/AuditPageView.stories.tsx b/site/src/pages/AuditPage/AuditPageView.stories.tsx
@@ -1,31 +1,31 @@
 import { ComponentMeta, Story } from "@storybook/react"
+import { createPaginationRef } from "components/PaginationWidget/utils"
 import { MockAuditLog, MockAuditLog2 } from "testHelpers/entities"
 import { AuditPageView, AuditPageViewProps } from "./AuditPageView"
 
 export default {
   title: "pages/AuditPageView",
   component: AuditPageView,
+  argTypes: {
+    auditLogs: {
+      defaultValue: [MockAuditLog, MockAuditLog2],
+    },
+    count: {
+      defaultValue: 1000,
+    },
+    paginationRef: {
+      defaultValue: createPaginationRef({ page: 1, limit: 25 })
+    }
+  }
 } as ComponentMeta<typeof AuditPageView>
 
 const Template: Story<AuditPageViewProps> = (args) => (
   <AuditPageView {...args} />
 )
 
 export const AuditPage = Template.bind({})
-AuditPage.args = {
-  auditLogs: [MockAuditLog, MockAuditLog2],
-  count: 1000,
-  page: 1,
-  limit: 25,
-}
 
 export const AuditPageSmallViewport = Template.bind({})
-AuditPageSmallViewport.args = {
-  auditLogs: [MockAuditLog, MockAuditLog2],
-  count: 1000,
-  page: 1,
-  limit: 25,
-}
 AuditPageSmallViewport.parameters = {
   chromatic: { viewports: [600] },
 }
