Have two scenarios, one for allowed ODS codes, one for blocked ones

Author: wildjames

artillery/helper/odscodes.mjs

@@ -4,5 +4,5 @@ export const allowedOdsCodes = [
 ]
 
 export const blockedOdsCodes = [
-    // "B3J1Z"
-]
+    "B3J1Z"
+]

artillery/notify_entrypoint.mjs

@@ -7,28 +7,6 @@ export { getSharedAuthToken }
 
 const logger = pino()
 
-const LETTERS = "ABCDEFGHIJKLMNOPQRSTUVWXYZ";
-const DIGITS = "0123456789";
-
-const randomChar = (chars) => chars[Math.floor(Math.random() * chars.length)];
-
-/** Generate one two-letter, three-digit ODS code, e.g. "AB123" */
-const generateOdsCode = () =>
-  `${randomChar(LETTERS)}${randomChar(LETTERS)}${randomChar(DIGITS)}${randomChar(DIGITS)}${randomChar(DIGITS)}`;
-
-function buildFullOdsCodes(targetCount, seedCodes) {
-  const codes = new Set(seedCodes);
-
-  while (codes.size < targetCount) {
-    codes.add(generateOdsCode());
-  }
-
-  return Array.from(codes);
-}
-
-// The complete list of ODS codes
-const fullOdsCodes = allowedOdsCodes.concat(blockedOdsCodes)
-
 function computeCheckDigit(nhsNumber) {
     const factors = [10,9,8,7,6,5,4,3,2]
     let total = 0
@@ -62,23 +40,42 @@ function sampleNormal(mean = 0, sd = 1) {
   return z * sd + mean;
 }
 
-export function initUser(context, events, done) {
-    // Generate data for a patient
-    context.vars.odsCode = fullOdsCodes[Math.floor(Math.random()*fullOdsCodes.length)]
-    context.vars.nhsNumber = generateValidNhsNumber()
+function initUserContextVars(context) {
+  context.vars.nhsNumber = generateValidNhsNumber()
 
-    let prescriptionCount = Math.round(sampleNormal(3,1))
-    if (prescriptionCount < 1) prescriptionCount = 1 // just truncate at 1.
-    context.vars.prescriptionCount  = prescriptionCount
-    context.vars.loopcount = 0
-    
-    logger.info(`Patient ${context.vars.nhsNumber}, ODS ${context.vars.odsCode} has ${context.vars.prescriptionCount} prescriptions`)
-    
-    done()
+  let prescriptionCount = Math.round(sampleNormal(3,1))
+  if (prescriptionCount < 1) prescriptionCount = 1 // just truncate at 1.
+  context.vars.prescriptionCount  = prescriptionCount
+  context.vars.loopcount = 0
+}
+
+export function initUserAllowed(context, events, done) {
+  logger.info("Initializing user context variables for allowed ODS codes")
+  initUserContextVars(context)
+  
+  // Generate data for a patient with an allowed ODS code
+  context.vars.odsCode = allowedOdsCodes[Math.floor(Math.random() * allowedOdsCodes.length)]
+
+  logger.info(`[ALLOWED] Patient ${context.vars.nhsNumber}, ODS ${context.vars.odsCode} has ${context.vars.prescriptionCount} prescriptions`)
+  done()
+}
+
+export function initUserBlocked(context, events, done) {
+  logger.info("Initializing user context variables for allowed ODS codes")
+  initUserContextVars(context)
+
+  // Generate data for a patient with a blocked ODS code
+  context.vars.odsCode = blockedOdsCodes[Math.floor(Math.random() * blockedOdsCodes.length)]
+
+  logger.info(`[BLOCKED] Patient ${context.vars.nhsNumber}, ODS ${context.vars.odsCode} has ${context.vars.prescriptionCount} prescriptions`)
+  done()
 }
 
 export function generatePrescData(requestParams, context, ee, next) {
-  logger.debug(`Generating a prescription for patient ${context.vars.nhsNumber}`)
+  const isAllowed = allowedOdsCodes.includes(context.vars.odsCode);
+  const logPrefix = isAllowed ? "[ALLOWED]" : "[BLOCKED]";
+
+  logger.debug(`${logPrefix} Generating a prescription for patient ${context.vars.nhsNumber}`)
   const body = getBody(
     true,                   /* isValid */   
     "completed",            /* status */    
@@ -87,11 +84,11 @@ export function generatePrescData(requestParams, context, ee, next) {
     "ready to collect"      /* Item status */
   )
   // The body is fine - it works when I put it in postman
-  
+
   requestParams.json = body
   context.vars.x_request_id = uuidv4()
   context.vars.x_correlation_id = uuidv4()
-  
+
   context.vars.loopcount += 1
 
   // Wait this long between requests
@@ -104,8 +101,6 @@ export function generatePrescData(requestParams, context, ee, next) {
   }
 
   context.vars.nextDelay = delay
-  logger.debug(`Patient ${context.vars.nhsNumber} (on prescription update ${context.vars.loopcount}/${context.vars.prescriptionCount}) will think for ${context.vars.nextDelay} seconds`)
-  
+  logger.debug(`${logPrefix} Patient ${context.vars.nhsNumber} (on prescription update ${context.vars.loopcount}/${context.vars.prescriptionCount}) will think for ${context.vars.nextDelay} seconds`)
   next()
 }
-

artillery/notify_load_test.yml

@@ -28,12 +28,28 @@ before:
     - function: getSharedAuthToken
 
 scenarios:
-  - name: dynamic PSU calls for each patient
-  
+  - name: Allowed ODS code scenario
+    weight: 1
     flow:
-      - function: initUser
+      - function: initUserAllowed
+      - loop:
+          - function: getSharedAuthToken
+          - post:
+              url: "/prescription-status-update/"
+              beforeRequest: "generatePrescData"
+              headers:
+                Authorization: "Bearer {{ authToken }}"
+                x-request-id:     "{{ x_request_id }}"
+                x-correlation-id: "{{ x_correlation_id }}"
+              expect:
+                - statusCode: 201
+          - think: "{{ nextDelay }}seconds"
+        count: "{{ prescriptionCount }}"
 
-      # for each prescription (number of prescriptions is variable)
+  - name: Blocked ODS code scenario
+    weight: 3
+    flow:
+      - function: initUserBlocked
       - loop:
           - function: getSharedAuthToken
           - post: