O3-4476: Add Diagnosis Data to Save Visit Note Scenario (#68)

Author: Bawanthathilan

src/test/java/org/openmrs/performance/Constants.java

@@ -43,9 +43,14 @@ public class Constants {
 	public static final String CODED_ALLERGEN_UUID = "71617AAAAAAAAAAAAAAAAAAAAAAAAAAAAAAA";
 	public static final String SEVERITY_UUID = "1498AAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAA";
 
-	//Visit Notes
+	// Visit Notes
 	public static final String VISIT_NOTE_FORM_UUID = "c75f120a-04ec-11e3-8780-2b40bef9a44b";
 	public static final String CLINICIAN_ENCOUNTER_ROLE = "240b26f9-dd88-4172-823d-4a8bfeb7841f";
 	public static final String VISIT_NOTE_ENCOUNTER_TYPE_UUID = "d7151f82-c1f3-4152-a605-2f9ea7414a79";
 	public static final String VISIT_NOTE_CONCEPT_UUID = "162169AAAAAAAAAAAAAAAAAAAAAAAAAAAAAA";
+
+	// Diagnosis 
+	public static final String DIABETIC_KETOSIS_CONCEPT = "123107AAAAAAAAAAAAAAAAAAAAAAAAAAAAAA";
+	public static final String DIABETIC_FOOT_ULCER_CONCEPT = "149069AAAAAAAAAAAAAAAAAAAAAAAAAAAAAA";
+
 }

src/test/java/org/openmrs/performance/http/DoctorHttpService.java

@@ -1,5 +1,6 @@
 package org.openmrs.performance.http;
 
+import com.fasterxml.jackson.annotation.JsonInclude;
 import com.fasterxml.jackson.core.JsonProcessingException;
 import com.fasterxml.jackson.databind.ObjectMapper;
 import io.gatling.javaapi.http.HttpRequestActionBuilder;
@@ -321,4 +322,29 @@ public HttpRequestActionBuilder saveVisitNote(String patientUuid, String current
 			throw new RuntimeException("Error converting visitNote to JSON", e);
 		}
 	}
+
+	public HttpRequestActionBuilder saveDiagnosis(String patientUuid, String encounterUuid, String diagnosisUuid,
+	        String certainty, int rank) {
+		Map<String, Object> patientDiagnosis = new HashMap<>();
+		patientDiagnosis.put("patient", patientUuid);
+		patientDiagnosis.put("encounter", encounterUuid);
+		patientDiagnosis.put("certainty", certainty);
+		patientDiagnosis.put("rank", rank);
+		patientDiagnosis.put("condition", null);
+		
+		Map<String, Object> diagnosis = new HashMap<>();
+		diagnosis.put("coded", diagnosisUuid);
+		patientDiagnosis.put("diagnosis", diagnosis);
+		
+		try {
+			String body = new ObjectMapper().setSerializationInclusion(JsonInclude.Include.ALWAYS)
+			        .writeValueAsString(patientDiagnosis);
+			
+			return http("Save Patient Diagnosis").post("/openmrs/ws/rest/v1/patientdiagnoses").body(StringBody(body));
+		}
+		catch (JsonProcessingException e) {
+			throw new RuntimeException("Error converting patientDiagnosis to JSON", e);
+		}
+	}
+
 }

src/test/java/org/openmrs/performance/registries/DoctorRegistry.java

@@ -9,6 +9,8 @@
 import static io.gatling.javaapi.core.CoreDsl.exec;
 import static org.openmrs.performance.Constants.ALLERGY_REACTION_UUID;
 import static org.openmrs.performance.Constants.ARTERIAL_BLOOD_OXYGEN_SATURATION;
+import static org.openmrs.performance.Constants.DIABETIC_FOOT_ULCER_CONCEPT;
+import static org.openmrs.performance.Constants.DIABETIC_KETOSIS_CONCEPT;
 import static org.openmrs.performance.Constants.DIASTOLIC_BLOOD_PRESSURE;
 import static org.openmrs.performance.Constants.DRUG_ALLERGEN_UUID;
 import static org.openmrs.performance.Constants.ENVIRONMENTAL_ALLERGEN_UUID;
@@ -147,6 +149,12 @@ public ChainBuilder addDrugOrder(String patientUuid, String visitUuid, String cu
 
 	public ChainBuilder addVisitNote(String patientUuid, String currentUserUuid) {
 		String visitNoteText = "Patient visit note";
-		return exec(httpService.saveVisitNote(patientUuid, currentUserUuid, visitNoteText));
+		String certainty = "PROVISIONAL";
+		String encounterUuid = "#{encounterUuid}";
+		
+		return exec(
+		    httpService.saveVisitNote(patientUuid, currentUserUuid, visitNoteText),
+		    httpService.saveDiagnosis(patientUuid, encounterUuid, DIABETIC_KETOSIS_CONCEPT, certainty, 1),
+		    httpService.saveDiagnosis(patientUuid, encounterUuid, DIABETIC_FOOT_ULCER_CONCEPT, certainty, 2));
 	}
 }