fix(allergies): fixes allergies tab not refreshing data after allergy…

… has been added
HospitalRun · Aug 20, 2020 · 0e29639 · 0e29639
1 parent bc9709a
commit 0e29639
Show file tree

Hide file tree

Showing 3 changed files with 5 additions and 5 deletions.
diff --git a/src/__tests__/patients/hooks/useAddAllergy.test.tsx b/src/__tests__/patients/hooks/useAddAllergy.test.tsx
@@ -42,6 +42,6 @@ describe('use add allergy', () => {
     expect(PatientRepository.find).toHaveBeenCalledTimes(1)
     expect(PatientRepository.saveOrUpdate).toHaveBeenCalledTimes(1)
     expect(PatientRepository.saveOrUpdate).toHaveBeenCalledWith(expectedPatient)
-    expect(result).toEqual(expectedAllergy)
+    expect(result).toEqual([expectedAllergy])
   })
 })
diff --git a/src/__tests__/patients/hooks/useAllergy.tsx → ...ests__/patients/hooks/useAllergy.test.tsx b/src/__tests__/patients/hooks/useAllergy.tsx → ...ests__/patients/hooks/useAllergy.test.tsx
diff --git a/src/patients/hooks/useAddAllergy.tsx b/src/patients/hooks/useAddAllergy.tsx
@@ -11,7 +11,7 @@ interface AddAllergyRequest {
   allergy: Allergy
 }
 
-async function addAllergy(request: AddAllergyRequest): Promise<Allergy> {
+async function addAllergy(request: AddAllergyRequest): Promise<Allergy[]> {
   const error = validateAllergy(request.allergy)
 
   if (isEmpty(error)) {
@@ -28,16 +28,16 @@ async function addAllergy(request: AddAllergyRequest): Promise<Allergy> {
       allergies,
     })
 
-    return newAllergy
+    return allergies
   }
 
   throw error
 }
 
 export default function useAddAllergy() {
   return useMutation(addAllergy, {
-    onSuccess: async (_, variables) => {
-      await queryCache.invalidateQueries(['allergies', variables.patientId])
+    onSuccess: async (data, variables) => {
+      await queryCache.setQueryData(['allergies', variables.patientId], data)
     },
     throwOnError: true,
   })