From 27dcb0608389e41d223c615124ff19c3a4ca4e84 Mon Sep 17 00:00:00 2001
From: Daniel Hougaard <62331820+DanielHougaard@users.noreply.github.com>
Date: Tue, 12 Mar 2024 14:27:02 +0100
Subject: [PATCH] Fix: Invalidate after selecting organization

---
 frontend/src/hooks/api/auth/queries.tsx | 7 ++++++-
 1 file changed, 6 insertions(+), 1 deletion(-)

diff --git a/frontend/src/hooks/api/auth/queries.tsx b/frontend/src/hooks/api/auth/queries.tsx
index 9bc7bb034..5936408e1 100644
--- a/frontend/src/hooks/api/auth/queries.tsx
+++ b/frontend/src/hooks/api/auth/queries.tsx
@@ -1,9 +1,10 @@
-import { useMutation, useQuery } from "@tanstack/react-query";
+import { useMutation, useQuery, useQueryClient } from "@tanstack/react-query";
 
 import SecurityClient from "@app/components/utilities/SecurityClient";
 import { apiRequest } from "@app/config/request";
 import { setAuthToken } from "@app/reactQuery";
 
+import { organizationKeys } from "../organization/queries";
 import {
   ChangePasswordDTO,
   CompleteAccountDTO,
@@ -66,6 +67,7 @@ export const selectOrganization = async (data: { organizationId: string }) => {
 };
 
 export const useSelectOrganization = () => {
+  const queryClient = useQueryClient();
   return useMutation({
     mutationFn: async (details: { organizationId: string }) => {
       const data = await selectOrganization(details);
@@ -74,6 +76,9 @@ export const useSelectOrganization = () => {
       SecurityClient.setProviderAuthToken("");
 
       return data;
+    },
+    onSuccess: () => {
+      queryClient.invalidateQueries(organizationKeys.getUserOrganizations);
     }
   });
 };