fix: final test

Author: rohitbytecode

tests/billing_flow.test.js

@@ -68,7 +68,11 @@ describe('HMS Billing Flow Integration', () => {
       });
 
     expect([200, 201]).toContain(res.statusCode);
-    billId = res.body._id;
+
+    const bill = res.body.data || res.body;
+
+    billId = bill._id;
+
     expect(billId).toBeDefined();
   });
 
@@ -91,7 +95,7 @@ describe('HMS Billing Flow Integration', () => {
       });
 
     expect(res.statusCode).toBe(200);
-    const updatedBill = res.body;
+    const updatedBill = res.body.data || res.body;
     expect(updatedBill.status).toBe('paid');
   });
 });

tests/patient.test.js

@@ -60,7 +60,10 @@ describe('Patient Integration Tests', () => {
         });
 
       expect(res.statusCode).toBe(201);
-      expect(res.body._id).toBeDefined();
+
+      const patient = res.body.data || res.body;
+
+      expect(patient._id).toBeDefined();
     });
 
     it('should fail if missing required patient details', async () => {

tests/patient_flow.test.js

@@ -54,8 +54,11 @@ describe('HMS Patient Flow Integration', () => {
       });
 
     expect(res.statusCode).toBe(201);
-    expect(res.body._id).toBeDefined();
-    patientId = res.body._id;
+
+    const patient = res.body.data || res.body;
+
+    expect(patient._id).toBeDefined();
+    patientId = patient._id;
   });
 
   it('2. Should fetch patient by ID (Verifying the 404 fix)', async () => {