fix - use findByRole instead of getByRole

opengovsg · Nov 18, 2024 · 4c80133 · 4c80133
1 parent edb01c8
commit 4c80133
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/apps/studio/src/stories/Page/Search.stories.tsx b/apps/studio/src/stories/Page/Search.stories.tsx
@@ -48,7 +48,7 @@ export const Initial: Story = {
   play: async ({ canvasElement }) => {
     await waitFor(async () => {
       const screen = within(canvasElement)
-      const searchButton = screen.getByRole("button", {
+      const searchButton = await screen.findByRole("button", {
         name: "search-button",
       })
       await userEvent.click(searchButton)
@@ -65,7 +65,7 @@ export const Results: Story = {
   play: async ({ canvasElement }) => {
     await waitFor(async () => {
       const screen = within(canvasElement)
-      const searchButton = screen.getByRole("button", {
+      const searchButton = await screen.findByRole("button", {
         name: "search-button",
       })
       await userEvent.click(searchButton)
@@ -83,7 +83,7 @@ export const Loading: Story = {
   play: async ({ canvasElement }) => {
     await waitFor(async () => {
       const screen = within(canvasElement)
-      const searchButton = screen.getByRole("button", {
+      const searchButton = await screen.findByRole("button", {
         name: "search-button",
       })
       await userEvent.click(searchButton)
@@ -101,7 +101,7 @@ export const NoResults: Story = {
   play: async ({ canvasElement }) => {
     await waitFor(async () => {
       const screen = within(canvasElement)
-      const searchButton = screen.getByRole("button", {
+      const searchButton = await screen.findByRole("button", {
         name: "search-button",
       })
       await userEvent.click(searchButton)