Tie test cases to sessions

Author: wrightmartin

lib/create-data.js

@@ -21,6 +21,7 @@ import {
   RegistrationOutcome,
   SchoolPhase,
   ScreenOutcome,
+  SessionPresetName,
   SessionPresets,
   SessionType,
   UploadType,
@@ -64,6 +65,7 @@ import {
   formatDate,
   removeDays,
   today,
+  getAcademicYear,
   getCurrentAcademicYear
 } from '../app/utils/date.js'
 import { range } from '../app/utils/number.js'
@@ -720,26 +722,80 @@ const seededMmrDob = new Date('2020-09-01')
 function buildSeededMmrVaccination({
   uuid,
   patient_uuid,
+  patientSession_uuid,
   ageMonths,
   ageDays = 0,
   given,
-  service = false
+  service = false,
+  notGivenOutcome
 }) {
   const createdAt = addDays(addMonths(seededMmrDob, ageMonths), ageDays)
   return new Vaccination({
     uuid,
     createdAt,
+    createdBy_uid: nurse.uid,
     patient_uuid,
+    patientSession_uuid,
     programme_id: 'mmr',
     outcome: given
-      ? (service ? VaccinationOutcome.Vaccinated : VaccinationOutcome.AlreadyVaccinated)
-      : VaccinationOutcome.Unwell,
+      ? VaccinationOutcome.Vaccinated
+      : (notGivenOutcome || VaccinationOutcome.Unwell),
     source: service || !given
       ? VaccinationSource.Service
       : VaccinationSource.NhsImmunisationsApi
   })
 }
 
+// Seeded MMR sessions at Seva Primary. Each session date is derived from
+// seededMmrDob + ageMonths so that service doses land on a real session.
+// sessionKey groups doses across patients into the same session.
+const seededMmrSessions = {
+  session_50m: {
+    id: 'mmr-seed-s1',
+    date: addMonths(seededMmrDob, 50),
+    openAt: addDays(addMonths(seededMmrDob, 50), -42)
+  },
+  session_54m: {
+    id: 'mmr-seed-s2',
+    date: addMonths(seededMmrDob, 54),
+    openAt: addDays(addMonths(seededMmrDob, 54), -42)
+  },
+  session_57m: {
+    id: 'mmr-seed-s3',
+    date: addMonths(seededMmrDob, 57),
+    openAt: addDays(addMonths(seededMmrDob, 57), -42)
+  },
+  session_61m: {
+    id: 'mmr-seed-s4',
+    date: addMonths(seededMmrDob, 61),
+    openAt: addDays(addMonths(seededMmrDob, 61), -42)
+  },
+  session_63m: {
+    id: 'mmr-seed-s5',
+    date: addDays(addMonths(seededMmrDob, 63), 10),
+    openAt: addMonths(seededMmrDob, 63)
+  }
+}
+
+for (const [, cfg] of Object.entries(seededMmrSessions)) {
+  const session = new Session(
+    {
+      id: cfg.id,
+      createdAt: cfg.openAt,
+      createdBy_uid: nurse.uid,
+      date: cfg.date,
+      openAt: cfg.openAt,
+      academicYear: getAcademicYear(cfg.date),
+      type: SessionType.School,
+      school_id: seededMmrSchoolId,
+      presetNames: [SessionPresetName.MMR],
+      registration: true
+    },
+    context
+  )
+  context.sessions[session.id] = session
+}
+
 const seededMmrPatients = [
   {
     uuid: 'mmr00001-0000-4000-8000-000000000001',
@@ -749,7 +805,7 @@ const seededMmrPatients = [
     doses: [
       // Case 1 — two valid doses
       { uuid: 'mmr00001-v001-4000-8000-000000000001', ageMonths: 12, ageDays: 14, given: true },
-      { uuid: 'mmr00001-v002-4000-8000-000000000002', ageMonths: 15, ageDays: 7, given: true, service: true }
+      { uuid: 'mmr00001-v002-4000-8000-000000000002', ageMonths: 40, ageDays: 10, given: true }
     ]
   },
   {
@@ -758,10 +814,13 @@ const seededMmrPatients = [
     firstName: 'Bilal',
     lastName: 'Begum',
     doses: [
-      // Case 2 — two valid doses plus a failed attempt between them
+      // Case 2 — two valid doses with repeated failed attempts between them
       { uuid: 'mmr00002-v001-4000-8000-000000000001', ageMonths: 12, ageDays: 14, given: true },
-      { uuid: 'mmr00002-v002-4000-8000-000000000002', ageMonths: 14, ageDays: 0, given: false },
-      { uuid: 'mmr00002-v003-4000-8000-000000000003', ageMonths: 15, ageDays: 10, given: true }
+      { uuid: 'mmr00002-v002-4000-8000-000000000002', ageMonths: 50, ageDays: 0, given: false, notGivenOutcome: VaccinationOutcome.Absent, sessionKey: 'session_50m' },
+      { uuid: 'mmr00002-v003-4000-8000-000000000003', ageMonths: 54, ageDays: 0, given: false, notGivenOutcome: VaccinationOutcome.ConsentRefused, sessionKey: 'session_54m' },
+      { uuid: 'mmr00002-v004-4000-8000-000000000004', ageMonths: 57, ageDays: 0, given: false, notGivenOutcome: VaccinationOutcome.Absent, sessionKey: 'session_57m' },
+      { uuid: 'mmr00002-v005-4000-8000-000000000005', ageMonths: 61, ageDays: 0, given: false, notGivenOutcome: VaccinationOutcome.Refused, sessionKey: 'session_61m' },
+      { uuid: 'mmr00002-v006-4000-8000-000000000006', ageMonths: 63, ageDays: 10, given: true, service: true, sessionKey: 'session_63m' }
     ]
   },
   {
@@ -770,10 +829,10 @@ const seededMmrPatients = [
     firstName: 'Chiamaka',
     lastName: 'Chen',
     doses: [
-      // Case 3 — dose at 11m (ignored), then two valid doses
+      // Case 3 — dose at 11m (ignored), then two valid doses given at school age by SAIS
       { uuid: 'mmr00003-v001-4000-8000-000000000001', ageMonths: 11, ageDays: 0, given: true },
-      { uuid: 'mmr00003-v002-4000-8000-000000000002', ageMonths: 13, ageDays: 0, given: true },
-      { uuid: 'mmr00003-v003-4000-8000-000000000003', ageMonths: 15, ageDays: 7, given: true }
+      { uuid: 'mmr00003-v002-4000-8000-000000000002', ageMonths: 50, ageDays: 0, given: true, service: true, sessionKey: 'session_50m' },
+      { uuid: 'mmr00003-v003-4000-8000-000000000003', ageMonths: 54, ageDays: 0, given: true, service: true, sessionKey: 'session_54m' }
     ]
   },
   {
@@ -791,10 +850,30 @@ const seededMmrPatients = [
 
 for (const seed of seededMmrPatients) {
   const vaccination_uuids = []
+  const patientSession_uuids = []
+
   for (const dose of seed.doses) {
+    let patientSession_uuid
+    if (dose.sessionKey) {
+      const cfg = seededMmrSessions[dose.sessionKey]
+      const patientSession = new PatientSession(
+        {
+          createdAt: cfg.openAt,
+          patient_uuid: seed.uuid,
+          programme_id: 'mmr',
+          session_id: cfg.id
+        },
+        context
+      )
+      context.patientSessions[patientSession.uuid] = patientSession
+      patientSession_uuid = patientSession.uuid
+      patientSession_uuids.push(patientSession.uuid)
+    }
+
     const vaccination = buildSeededMmrVaccination({
       ...dose,
-      patient_uuid: seed.uuid
+      patient_uuid: seed.uuid,
+      patientSession_uuid
     })
     context.vaccinations[vaccination.uuid] = vaccination
     vaccination_uuids.push(vaccination.uuid)
@@ -812,6 +891,7 @@ for (const seed of seededMmrPatients) {
       addressLevel1: 'Coventry',
       postalCode: 'CV1 1AA'
     },
+    patientSession_uuids,
     vaccination_uuids
   })
   context.patients[patient.uuid] = patient