Add medicine filter to prescriptions (#2068)

* Log Prescriptions * pass medicine object instead of just id * refactor * add tests * update tests --------- Co-authored-by: Vignesh Hari <[email protected]>
ohcnetwork · May 8, 2024 · 0bf4d1d · 0bf4d1d
1 parent 8481175
commit 0bf4d1d
Show file tree

Hide file tree

Showing 2 changed files with 37 additions and 0 deletions.
diff --git a/care/facility/api/viewsets/prescription.py b/care/facility/api/viewsets/prescription.py
@@ -91,6 +91,7 @@ class ConsultationPrescriptionFilter(filters.FilterSet):
     dosage_type = MultiSelectFilter()
     prescription_type = CareChoiceFilter(choice_dict=inverse_prescription_type)
     discontinued = filters.BooleanFilter()
+    medicine = filters.UUIDFilter(field_name="medicine__external_id")
 
 
 class ConsultationPrescriptionViewSet(

diff --git a/care/facility/tests/test_prescriptions_api.py b/care/facility/tests/test_prescriptions_api.py
@@ -20,6 +20,7 @@ def setUp(self) -> None:
         super().setUp()
         self.consultation = self.create_consultation(self.patient, self.facility)
         self.medicine = MedibaseMedicine.objects.first()
+        self.medicine2 = MedibaseMedicine.objects.all()[1]
 
         self.normal_prescription_data = {
             "medicine": self.medicine.external_id,
@@ -29,6 +30,14 @@ def setUp(self) -> None:
             "dosage_type": "REGULAR",
         }
 
+        self.normal_prescription_data2 = {
+            "medicine": self.medicine2.external_id,
+            "prescription_type": "REGULAR",
+            "base_dosage": "1 mg",
+            "frequency": "OD",
+            "dosage_type": "REGULAR",
+        }
+
     def test_create_normal_prescription(self):
         response = self.client.post(
             f"/api/v1/consultation/{self.consultation.external_id}/prescriptions/",
@@ -113,3 +122,30 @@ def test_create_prn_prescription(self):
             prn_prescription_data,
         )
         self.assertEqual(response.status_code, status.HTTP_400_BAD_REQUEST)
+
+    def test_medicine_filter_for_prescription(self):
+        # post 2 prescriptions with different medicines
+        self.client.post(
+            f"/api/v1/consultation/{self.consultation.external_id}/prescriptions/",
+            self.normal_prescription_data,
+        )
+        self.client.post(
+            f"/api/v1/consultation/{self.consultation.external_id}/prescriptions/",
+            self.normal_prescription_data2,
+        )
+
+        # get all prescriptions without medicine filter
+        response = self.client.get(
+            f"/api/v1/consultation/{self.consultation.external_id}/prescriptions/",
+        )
+        self.assertEqual(response.data["count"], 2)
+
+        # get all prescriptions with medicine filter
+        response = self.client.get(
+            f"/api/v1/consultation/{self.consultation.external_id}/prescriptions/?medicine={self.medicine.external_id}",
+        )
+
+        for prescription in response.data["results"]:
+            self.assertEqual(
+                prescription["medicine_object"]["name"], self.medicine.name
+            )