@@ -16,7 +16,7 @@ import { useToastContext } from '@src/generic/toast-context';
1616import { ParentIds } from '@src/generic/types' ;
1717import {
1818 QueryClient ,
19- skipToken , useQuery , useQueryClient ,
19+ skipToken , useMutation , useQuery , useQueryClient ,
2020} from '@tanstack/react-query' ;
2121import { useDispatch } from 'react-redux' ;
2222import {
@@ -253,8 +253,8 @@ export const useConfigureSubsection = () => {
253253
254254export const useConfigureUnit = ( ) => {
255255 const queryClient = useQueryClient ( ) ;
256- const { showToast } = useToastContext ( ) ;
257- return useMutationWithProcessingNotification ( {
256+ const { showToast, closeToast } = useToastContext ( ) ;
257+ return useMutation ( {
258258 mutationFn : ( variables : ConfigureUnitData & ParentIds ) => configureCourseUnit ( variables ) ,
259259 onMutate : ( variables ) => {
260260 const msg = getNotificationMessage ( variables . type , variables . isVisibleToStaffOnly , true ) ;
@@ -264,6 +264,7 @@ export const useConfigureUnit = () => {
264264 onSettled : ( _data , _err , variables ) => {
265265 queryClient . invalidateQueries ( { queryKey : courseOutlineQueryKeys . courseDetails ( getCourseKey ( variables . unitId ) ) } ) ;
266266 invalidateParentQueries ( queryClient , variables ) . catch ( ( e ) => handleResponseErrors ( e ) ) ;
267+ closeToast ( ) ;
267268 } ,
268269 } ) ;
269270} ;
0 commit comments