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()),
+ )