Running the flow, but getting 401 errors

Author: wildjames

artillery/helper/allowed_odscodes.mjs

@@ -0,0 +1,8 @@
+// These ODS codes should match the values in AWS parameter store. They're assumed to be enabled in the whitelist
+export const allowedOdsCodes = [
+    "FA565"
+]
+
+export const blockedOdsCodes = [
+    "B3J1Z"
+]

artillery/helper/odscodes.mjs

@@ -1,11 +1,11 @@
 import {v4 as uuidv4}   from "uuid"
 import pino from "pino"
 import {getSharedAuthToken, getBody} from "./helper/psu.mjs"
-import {allowedOdsCodes} from "./helper/allowed_odscodes.mjs"
+import {allowedOdsCodes, blockedOdsCodes} from "./helper/odscodes.mjs"
 
-const logger = pino()
+export { getSharedAuthToken }
 
-const NUM_ODS_CODES = 1000
+const logger = pino()
 
 const LETTERS = "ABCDEFGHIJKLMNOPQRSTUVWXYZ";
 const DIGITS = "0123456789";
@@ -26,8 +26,8 @@ function buildFullOdsCodes(targetCount, seedCodes) {
   return Array.from(codes);
 }
 
-/** The complete list of 10k ODS codes */
-const fullOdsCodes = buildFullOdsCodes(NUM_ODS_CODES, allowedOdsCodes);
+// 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]
@@ -71,11 +71,13 @@ export function initUser(context, events, done) {
     if (prescriptionCount < 1) prescriptionCount = 1 // just truncate at 1.
     context.vars.prescriptionCount  = prescriptionCount
 
+    logger.info(`Patient ${context.vars.nhsNumber}, ODS ${context.vars.odsCode} has ${context.vars.prescriptionCount} prescriptions`)
+    
     done()
 }
 
-// beforeEach request
 export function generatePrescData(requestParams, context, ee, next) {
+  logger.info(`Generating a prescription for patient ${context.vars.nhsNumber}`)
   const body = getBody(
     true,                   /* isValid */   
     "ready to collect",     /* status */    
@@ -91,8 +93,8 @@ export function generatePrescData(requestParams, context, ee, next) {
   let delay = sampleNormal(150, 60)
   if (delay < 0) delay = 0
   context.vars.nextDelay = delay
+  logger.info(`Patient ${context.vars.nhsNumber} will think for ${context.vars.nextDelay} seconds`)
 
   next()
 }
 
-export { getSharedAuthToken }

artillery/notify_entrypoint.mjs

@@ -12,13 +12,13 @@ config:
       maxVusers: "{{ $env.maxVusers }}"
   environments:
     dev:
-      target: https://internal-dev.api.service.nhs.uk
+      target: https://internal-dev.api.service.nhs.uk/
     ref:
-      target: https://ref.api.service.nhs.uk
+      target: https://ref.api.service.nhs.uk/
     int:
-      target: https://int.api.service.nhs.uk
+      target: https://int.api.service.nhs.uk/
     pr:
-      target: https://psu-pr-{{ prNumber }}.dev.eps.national.nhs.uk
+      target: https://psu-pr-{{ prNumber }}.dev.eps.national.nhs.uk/
   defaults:
     headers:
       Authorization: "Bearer {{ authToken }}"
@@ -29,21 +29,19 @@ scenarios:
     # grab token once per VU
     before:
       - function: getSharedAuthToken
-
+      
     flow:
       - function: initUser
 
       # for each prescription (number of prescriptions is variable)
       - loop:
-          - count: "{{ prescriptionCount }}"
-            steps:
-              - beforeRequest: generatePrescData
-              - post:
-                  url: "/prescription-status-update/"
-                  headers:
-                    x-request-id:     "{{ x_request_id }}"
-                    x-correlation-id: "{{ x_correlation_id }}"
-                  expect:
-                    - statusCode: 201
-              # wait before next prescription
-              - think: "{{ nextDelay }}"
+          - post:
+              url: "/prescription-status-update/"
+              beforeRequest: "generatePrescData"
+              headers:
+                x-request-id:     "{{ x_request_id }}"
+                x-correlation-id: "{{ x_correlation_id }}"
+              expect:
+                - statusCode: 201
+          - think: "{{ nextDelay }}"
+        count: "{{ prescriptionCount }}"

artillery/notify_load_test.yml

@@ -55,6 +55,7 @@ EOF
 
 echo "Running Artillery test locally..."
 echo ""
+echo "Environment: ${environment}"
 echo "Max Virtual Users: ${maxVusers}"
 echo "Phase Duration: ${duration}"
 echo "Arrival Rate: ${arrivalRate}"