O3-4280: implement Save a note in Visit Patient scenario

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

@@ -42,4 +42,10 @@ public class Constants {
 	public static final String ALLERGY_REACTION_UUID = "162553AAAAAAAAAAAAAAAAAAAAAAAAAAAAAA";
 	public static final String CODED_ALLERGEN_UUID = "71617AAAAAAAAAAAAAAAAAAAAAAAAAAAAAAA";
 	public static final String SEVERITY_UUID = "1498AAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAA";
+
+	//Save Visits
+	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";
 }

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

@@ -13,10 +13,12 @@
 
 import static io.gatling.javaapi.core.CoreDsl.StringBody;
 import static io.gatling.javaapi.core.CoreDsl.bodyString;
+import static io.gatling.javaapi.core.CoreDsl.exec;
 import static io.gatling.javaapi.core.CoreDsl.jsonPath;
 import static io.gatling.javaapi.http.HttpDsl.http;
 import static org.openmrs.performance.Constants.ALLERGY_REACTION_UUID;
 import static org.openmrs.performance.Constants.CARE_SETTING_UUID;
+import static org.openmrs.performance.Constants.CLINICIAN_ENCOUNTER_ROLE;
 import static org.openmrs.performance.Constants.CODED_ALLERGEN_UUID;
 import static org.openmrs.performance.Constants.DAYS;
 import static org.openmrs.performance.Constants.DEFAULT_DOSING_TYPE;
@@ -27,6 +29,9 @@
 import static org.openmrs.performance.Constants.OUTPATIENT_CLINIC_LOCATION_UUID;
 import static org.openmrs.performance.Constants.SEVERITY_UUID;
 import static org.openmrs.performance.Constants.TABLET;
+import static org.openmrs.performance.Constants.VISIT_NOTE_CONCEPT_UUID;
+import static org.openmrs.performance.Constants.VISIT_NOTE_ENCOUNTER_TYPE_UUID;
+import static org.openmrs.performance.Constants.VISIT_NOTE_FORM_UUID;
 
 public class DoctorHttpService extends HttpService {
 
@@ -287,4 +292,65 @@ public HttpRequestActionBuilder saveOrder(String patientUuid, String visitUuid,
             throw new RuntimeException(e);
         }
     }
+
+	public HttpRequestActionBuilder saveVisitNote(String patientUuid, String currentUser, String value) {
+		Map<String, Object> visitNote = new HashMap<>();
+		visitNote.put("form", VISIT_NOTE_FORM_UUID);
+		visitNote.put("patient", patientUuid);
+		visitNote.put("location", OUTPATIENT_CLINIC_LOCATION_UUID);
+		visitNote.put("encounterType", VISIT_NOTE_ENCOUNTER_TYPE_UUID);
+
+		Map<String, Object> encounterProvider = new HashMap<>();
+		encounterProvider.put("encounterRole", CLINICIAN_ENCOUNTER_ROLE);
+		encounterProvider.put("provider", currentUser);
+
+		Map<String, Object> obs = new HashMap<>();
+		obs.put("concept", Map.of("uuid", VISIT_NOTE_CONCEPT_UUID));
+		obs.put("value", value);
+
+		visitNote.put("encounterProviders", List.of(encounterProvider));
+		visitNote.put("obs", List.of(obs));
+
+		try {
+			String body = new ObjectMapper().writeValueAsString(visitNote); // Convert Map to JSON
+
+			return http("Save Visit Note")
+				.post("/openmrs/ws/rest/v1/encounter")
+				.body(StringBody(body))
+				.check(jsonPath("$.uuid").saveAs("encounterUuid")); // Store encounter UUID
+		} catch (JsonProcessingException e) {
+			throw new RuntimeException("Error converting visitNote to JSON", e);
+		}
+	}
+
+
+	public HttpRequestActionBuilder saveDiagnosis(String patientUuid, String encounterUuid, String diagnosisUuid,String certainty, int rank) {
+		try {
+			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);
+
+			ObjectMapper objectMapper = new ObjectMapper();
+			String body = objectMapper.writeValueAsString(patientDiagnosis);
+
+			exec(seassion -> {
+				System.out.println(body);
+				return seassion;
+			});
+
+			return http("Save Patient Diagnosis")
+					.post("/openmrs/ws/rest/v1/patientdiagnoses")
+					.body(StringBody(body));
+
+		} catch (Exception e) {
+			throw new RuntimeException("Error while serializing diagnosis data", e);
+		}
+}
 }

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

@@ -144,4 +144,11 @@ 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)
+		);
+	}
 }

src/test/java/org/openmrs/performance/scenarios/VisitPatientScenario.java

@@ -35,6 +35,7 @@ public ScenarioBuilder getScenarioBuilder() {
 				.exec(registry.openImmunizationsTab("#{patient_uuid}"))
 				.exec(registry.openAttachmentsTab("#{patient_uuid}"))
 				.exec(registry.addDrugOrder("#{patient_uuid}", "#{visitUuid}", "#{currentUserUuid}"))
+				.exec(registry.addVisitNote("#{patient_uuid}", "#{currentUserUuid}"))
 				.exec(registry.endVisit("#{patient_uuid}"));
 	}
 }