Adds person_data_reader and tests

Author: shweta-nhs

src/eligibility_signposting_api/services/person_data_reader.py

@@ -0,0 +1,21 @@
+from __future__ import annotations
+
+from collections.abc import Collection, Mapping
+from typing import Any
+
+from wireup import service
+
+Row = Collection[Mapping[str, Any]]
+
+
+@service
+class PersonDataReader:
+    """Handles extracting and interpreting person data."""
+
+    def get_person_cohorts(self, person_data: Row) -> set[str]:
+        cohorts_row: Mapping[str, dict[str, dict[str, dict[str, Any]]]] = next(
+            (row for row in person_data if row.get("ATTRIBUTE_TYPE") == "COHORTS"),
+            {},
+        )
+        return set(cohorts_row.get("COHORT_MAP", {}).get("cohorts", {}).get("M", {}).keys())
+

tests/unit/services/test_person_data_reader.py

@@ -0,0 +1,85 @@
+import pytest
+from hamcrest import assert_that, is_
+
+from eligibility_signposting_api.services.person_data_reader import PersonDataReader
+
+
+@pytest.fixture
+def person_data_reader():
+    return PersonDataReader()
+
+
+def test_get_person_cohorts_empty_data(person_data_reader):
+    result = person_data_reader.get_person_cohorts([])
+    assert_that(result, is_(set()))
+
+
+def test_get_person_cohorts_no_cohorts_attribute_type(person_data_reader):
+    no_cohorts_type = [
+        {"ATTRIBUTE_TYPE": "NAME", "VALUE": "John Doe"},
+        {"ATTRIBUTE_TYPE": "AGE", "VALUE": 30},
+    ]
+    result = person_data_reader.get_person_cohorts(no_cohorts_type)
+    assert_that(result, is_(set()))
+
+
+def test_get_person_cohorts_no_cohort_map_key(person_data_reader):
+    no_cohorts_map = [
+        {"ATTRIBUTE_TYPE": "COHORTS", "OTHER_FIELD": "value"},
+    ]
+    result = person_data_reader.get_person_cohorts(no_cohorts_map)
+    assert_that(result, is_(set()))
+
+
+def test_get_person_cohorts_no_cohorts_list_key(person_data_reader):
+    no_cohorts_list = [
+        {"ATTRIBUTE_TYPE": "COHORTS", "COHORT_MAP": {"another_key": {}}},
+    ]
+    result = person_data_reader.get_person_cohorts(no_cohorts_list)
+    assert_that(result, is_(set()))
+
+
+def test_get_person_cohorts_no_m_key(person_data_reader):
+    no_m_key = [
+        {"ATTRIBUTE_TYPE": "COHORTS", "COHORT_MAP": {"cohorts": {"X": {}}}},
+    ]
+    result = person_data_reader.get_person_cohorts(no_m_key)
+    assert_that(result, is_(set()))
+
+
+def test_get_person_cohorts_single_cohort(person_data_reader):
+    single_cohorts = [
+        {"ATTRIBUTE_TYPE": "COHORTS", "COHORT_MAP": {"cohorts": {"M": {"COHORT_A": {}}}}},
+        {"ATTRIBUTE_TYPE": "NAME", "VALUE": "Jane Smith"},
+    ]
+    result = person_data_reader.get_person_cohorts(single_cohorts)
+    assert_that(result, is_({"COHORT_A"}))
+
+
+def test_get_person_cohorts_multiple_cohorts(person_data_reader):
+    multiple_cohorts = [
+        {"ATTRIBUTE_TYPE": "COHORTS", "COHORT_MAP": {"cohorts": {"M": {"COHORT_B": {}, "COHORT_C": {}}}}},
+        {"ATTRIBUTE_TYPE": "AGE", "VALUE": 45},
+    ]
+    result = person_data_reader.get_person_cohorts(multiple_cohorts)
+    assert_that(result, is_({"COHORT_B", "COHORT_C"}))
+
+
+def test_get_person_cohorts_mixed_data(person_data_reader):
+    mixed_data = [
+        {"ATTRIBUTE_TYPE": "NAME", "VALUE": "Alice"},
+        {"ATTRIBUTE_TYPE": "COHORTS", "COHORT_MAP": {"cohorts": {"M": {"COHORT_D": {}, "COHORT_E": {}}}}},
+        {"ATTRIBUTE_TYPE": "ADDRESS", "VALUE": "123 Main St"},
+    ]
+    result = person_data_reader.get_person_cohorts(mixed_data)
+    assert_that(result, is_({"COHORT_D", "COHORT_E"}))
+
+
+def test_get_person_cohorts_with_other_attribute_types_present(person_data_reader):
+    data = [
+        {"ATTRIBUTE_TYPE": "NAME", "VALUE": "Charlie"},
+        {"ATTRIBUTE_TYPE": "COHORTS", "COHORT_MAP": {"cohorts": {"M": {"COHORT_F": {}}}}},
+        {"ATTRIBUTE_TYPE": "AGE", "VALUE": 25},
+    ]
+    result = person_data_reader.get_person_cohorts(data)
+    assert_that(result, is_({"COHORT_F"}))