Expect redirect urls

storybookjs · May 15, 2024 · 9eee515 · 9eee515
1 parent ec7bdc9
commit 9eee515
Show file tree

Hide file tree

Showing 3 changed files with 9 additions and 5 deletions.
diff --git a/app/note/edit/[id]/page.stories.tsx b/app/note/edit/[id]/page.stories.tsx
@@ -67,7 +67,8 @@ export const SavingExistingNoteShouldUpdateDBAndRedirect: Story = {
     await userEvent.click(
       await canvas.findByRole('menuitem', { name: /done/i }),
     )
-    await expectRedirect()
+
+    await expectRedirect('/note/2')
 
     await expect(await db.note.findUnique({ where: { id: 2 } })).toEqual(
       expect.objectContaining({
@@ -90,7 +91,8 @@ export const DeleteNoteRemovesFromDBAndSidebar: Story = {
     await userEvent.click(
       await canvas.findByRole('menuitem', { name: /delete/i }),
     )
-    await expectRedirect()
+
+    await expectRedirect('/')
 
     await expect(
       await db.note.findMany({ where: { id: 2 } }),

diff --git a/app/note/edit/page.stories.tsx b/app/note/edit/page.stories.tsx
@@ -60,7 +60,7 @@ export const SaveNewNote: Story = {
       await canvas.findByRole('menuitem', { name: /done/i }),
     )
 
-    await expectRedirect()
+    await expectRedirect('/note/3')
 
     await expect(await db.note.findUnique({ where: { id: 3 } })).toEqual(
       expect.objectContaining({

diff --git a/lib/test-utils.ts b/lib/test-utils.ts
@@ -1,5 +1,7 @@
 import { expect, waitFor } from '@storybook/test'
 import { getRouter } from '@storybook/nextjs/navigation.mock'
 
-export const expectRedirect = () =>
-  waitFor(() => expect(getRouter().push).toHaveBeenCalled())
+export const expectRedirect = (url: string) =>
+  waitFor(() =>
+    expect(getRouter().push).toHaveBeenLastCalledWith(url, expect.anything()),
+  )