test: Add checks for "LOADING" text in authorization tests

dictyBase · Aug 21, 2024 · 1b245cc · 1b245cc
1 parent b6d6663
commit 1b245cc
Showing 1 changed file with 6 additions and 4 deletions.
diff --git a/packages/auth/src/__tests__/useAuthorization.test.tsx b/packages/auth/src/__tests__/useAuthorization.test.tsx
@@ -1,6 +1,8 @@
 import { test, expect, vi } from "vitest"
 import { render, screen } from "@testing-library/react"
 import { useAuthorization } from "../useAuthorization"
+import { render, screen } from "@testing-library/react"
+import { useAuthorization } from "../useAuthorization"
 
 const mockUseLogto = vi.hoisted(() => vi.fn())
 
@@ -30,8 +32,8 @@ test("A user without a matching role is unauthorized", async () => {
     isAuthenticated: true,
   })
 
-  const { getByText } = render(<TestComponent authorizedRoles={["admin"]} />)
-  expect(getByText("LOADING")).toBeInTheDocument()
+  render(<TestComponent authorizedRoles={["admin"]} />)
+  expect(screen.getByText("LOADING")).toBeInTheDocument()
   expect(await screen.findByText("UNAUTHORIZED")).toBeInTheDocument()
 })
 
@@ -41,7 +43,7 @@ test("A user with a matching role is authorized", async () => {
     isAuthenticated: true,
   })
 
-  const { getByText } = render(<TestComponent authorizedRoles={["admin"]} />)
-  expect(getByText("LOADING")).toBeInTheDocument()
+  render(<TestComponent authorizedRoles={["admin"]} />)
+  expect(screen.getByText("LOADING")).toBeInTheDocument()
   expect(await screen.findByText("AUTHORIZED")).toBeInTheDocument()
 })