chore: test fixes

flanksource · Oct 29, 2024 · a00de91 · a00de91
1 parent 68646ad
commit a00de91
Showing 1 changed file with 4 additions and 28 deletions.
diff --git a/src/components/Connections/__tests__/ConnectionFormModal.unit.test.tsx b/src/components/Connections/__tests__/ConnectionFormModal.unit.test.tsx
@@ -1,4 +1,4 @@
-import { AuthContext } from "@flanksource-ui/context";
+import { AuthContext, FakeUser, Roles } from "@flanksource-ui/context";
 import { UserAccessStateContextProvider } from "@flanksource-ui/context/UserAccessContext/UserAccessContext";
 import { QueryClient } from "@tanstack/query-core";
 import { QueryClientProvider } from "@tanstack/react-query";
@@ -53,15 +53,7 @@ describe("ConnectionForm", () => {
   it("renders form when provided with initial value", async () => {
     render(
       <QueryClientProvider client={client}>
-        <AuthContext.Provider
-          value={{
-            user: {
-              id: "b149b5ee-db1c-4c0c-9711-98d06f1f1ce7",
-              email: "admin@local",
-              name: "John Doe"
-            }
-          }}
-        >
+        <AuthContext.Provider value={FakeUser([Roles.admin])}>
           <UserAccessStateContextProvider>
             <ConnectionFormModal
               formValue={formInitialValue}
@@ -105,15 +97,7 @@ describe("ConnectionForm", () => {
   it("renders list of connection types", async () => {
     render(
       <QueryClientProvider client={client}>
-        <AuthContext.Provider
-          value={{
-            user: {
-              id: "b149b5ee-db1c-4c0c-9711-98d06f1f1ce7",
-              email: "admin@local",
-              name: "John Doe"
-            }
-          }}
-        >
+        <AuthContext.Provider value={FakeUser([Roles.admin])}>
           <UserAccessStateContextProvider>
             <ConnectionFormModal
               formValue={undefined}
@@ -134,15 +118,7 @@ describe("ConnectionForm", () => {
   it("renders form when connection type is selected", async () => {
     render(
       <QueryClientProvider client={client}>
-        <AuthContext.Provider
-          value={{
-            user: {
-              id: "b149b5ee-db1c-4c0c-9711-98d06f1f1ce7",
-              email: "admin@local",
-              name: "John Doe"
-            }
-          }}
-        >
+        <AuthContext.Provider value={FakeUser([Roles.admin])}>
           <UserAccessStateContextProvider>
             <ConnectionFormModal
               formValue={undefined}