feat: Return household JSON in household updater func and update test

Author: anth-volk

policyengine_api/routes/household_routes.py

@@ -133,7 +133,7 @@ def update_household(country_id: str, household_id: int) -> Response:
             raise NotFound(f"Household #{household_id} not found.")
 
         # Next, update the household
-        household_service.update_household(
+        updated_household: dict = household_service.update_household(
             country_id, household_id, household_json, label
         )
         return Response(
@@ -143,6 +143,7 @@ def update_household(country_id: str, household_id: int) -> Response:
                     "message": None,
                     "result": {
                         "household_id": household_id,
+                        "household_json": updated_household["household_json"],
                     },
                 }
             ),

policyengine_api/services/household_service.py

@@ -90,7 +90,7 @@ def update_household(
         household_id: str,
         household_json: dict,
         label: str,
-    ) -> None:
+    ) -> dict:
         """
         Update a household with the given data.
 
@@ -116,6 +116,12 @@ def update_household(
                     household_id,
                 ),
             )
+
+            # Fetch the updated JSON back from the table
+            updated_household: dict = self.get_household(
+                country_id, household_id
+            )
+            return updated_household
         except Exception as e:
             print(
                 f"Error updating household #{household_id}. Details: {str(e)}"

tests/python/test_household.py

@@ -132,6 +132,7 @@ def test_update_household_success(
         assert response.status_code == 200
         assert data["status"] == "ok"
         assert data["result"]["household_id"] == "1"
+        assert data["result"]["household_json"] == updated_data["data"]
 
     def test_update_nonexistent_household(self, rest_client, mock_database):
         """Test updating a non-existent household."""