Filter programmes by team's programme_types from JWT

The Mavis API now includes programme_types in the JWT cis2_info
payload, allowing us to show only the programmes available to the
user's team rather than all programmes.

Author: mikejamesthompson

mavis/reporting/api_client/client.py

@@ -88,13 +88,21 @@ def get_variables(self) -> list[dict]:
         ]
 
     def get_programmes(self) -> list[dict]:
-        return [
+        all_programmes = [
             {"value": "flu", "text": "Flu"},
             {"value": "hpv", "text": "HPV"},
             {"value": "menacwy", "text": "MenACWY"},
             {"value": "td_ipv", "text": "Td/IPV"},
         ]
 
+        cis2_info = (self.session or {}).get("cis2_info", {})
+        programme_types = cis2_info.get("programme_types", [])
+
+        if not programme_types:
+            return all_programmes
+
+        return [p for p in all_programmes if p["value"] in programme_types]
+
     def get_year_groups(self) -> list[dict]:
         return [
             {"value": "0", "text": "Reception"},

tests/api_client/test_client.py

@@ -86,3 +86,26 @@ def test_td_ipv_returns_years_9_to_11(self, api_client):
     def test_unknown_programme_returns_empty_list(self, api_client):
         result = api_client.get_year_groups_for_programme("unknown")
         assert result == []
+
+
+class TestGetProgrammes:
+    def test_returns_all_when_no_programme_types(self, app):
+        client = MavisApiClient(app=app, session={"jwt": "x", "cis2_info": {}})
+        result = client.get_programmes()
+        values = [p["value"] for p in result]
+        assert values == ["flu", "hpv", "menacwy", "td_ipv"]
+
+    def test_filters_by_programme_types(self, app):
+        client = MavisApiClient(
+            app=app,
+            session={"jwt": "x", "cis2_info": {"programme_types": ["flu", "hpv"]}},
+        )
+        result = client.get_programmes()
+        values = [p["value"] for p in result]
+        assert values == ["flu", "hpv"]
+
+    def test_returns_all_when_cis2_info_missing(self, app):
+        client = MavisApiClient(app=app, session={"jwt": "x"})
+        result = client.get_programmes()
+        values = [p["value"] for p in result]
+        assert values == ["flu", "hpv", "menacwy", "td_ipv"]