NRL-1876 Include dedicated ODS codes for 15m & 55m in permissions generated

Author: anjalitrace2-nhs

tests/performance/producer/generate_permissions.py

@@ -3,7 +3,11 @@
 
 import fire
 
-from tests.performance.seed_data_constants import DEFAULT_CUSTODIAN_DISTRIBUTIONS
+from tests.performance.seed_data_constants import (
+    DEFAULT_CUSTODIAN_DISTRIBUTIONS,
+    VOL_15M_POINTERS_CUSTODIAN_DISTRIBUTIONS,
+    VOL_55M_POINTERS_CUSTODIAN_DISTRIBUTIONS,
+)
 
 
 def main(output_dir="../../dist/nrlf_permissions/K6PerformanceTest"):
@@ -15,6 +19,12 @@ def main(output_dir="../../dist/nrlf_permissions/K6PerformanceTest"):
     for pointer_type, custodians in DEFAULT_CUSTODIAN_DISTRIBUTIONS.items():
         for custodian, _ in custodians.items():
             custodian_permissions.setdefault(custodian, []).append(pointer_type)
+    for pointer_type, custodians in VOL_15M_POINTERS_CUSTODIAN_DISTRIBUTIONS.items():
+        for custodian, _ in custodians.items():
+            custodian_permissions.setdefault(custodian, []).append(pointer_type)
+    for pointer_type, custodians in VOL_55M_POINTERS_CUSTODIAN_DISTRIBUTIONS.items():
+        for custodian, _ in custodians.items():
+            custodian_permissions.setdefault(custodian, []).append(pointer_type)
 
     for custodian, pointer_types in custodian_permissions.items():
         permissions = [f"http://snomed.info/sct|{pt}" for pt in pointer_types]