|
1 | 1 | package org.openmrs.performance.http;
|
2 | 2 |
|
3 | 3 | import com.google.gson.Gson;
|
| 4 | +import com.google.gson.GsonBuilder; |
4 | 5 | import io.gatling.javaapi.http.HttpRequestActionBuilder;
|
5 | 6 |
|
6 | 7 | import java.time.ZonedDateTime;
|
7 | 8 | import java.time.format.DateTimeFormatter;
|
| 9 | +import java.util.ArrayList; |
8 | 10 | import java.util.HashMap;
|
| 11 | +import java.util.List; |
9 | 12 | import java.util.Map;
|
10 | 13 |
|
11 | 14 | import static io.gatling.javaapi.core.CoreDsl.StringBody;
|
12 | 15 | import static io.gatling.javaapi.core.CoreDsl.bodyString;
|
| 16 | +import static io.gatling.javaapi.core.CoreDsl.exec; |
13 | 17 | import static io.gatling.javaapi.core.CoreDsl.jsonPath;
|
14 | 18 | import static io.gatling.javaapi.http.HttpDsl.http;
|
15 | 19 | import static org.openmrs.performance.Constants.CARE_SETTING_UUID;
|
| 20 | +import static org.openmrs.performance.Constants.CLINICIAN_ENCOUNTER_ROLE; |
16 | 21 | import static org.openmrs.performance.Constants.DAYS;
|
17 | 22 | import static org.openmrs.performance.Constants.DEFAULT_DOSING_TYPE;
|
18 | 23 | import static org.openmrs.performance.Constants.DRUG_ORDER;
|
|
21 | 26 | import static org.openmrs.performance.Constants.ORDER;
|
22 | 27 | import static org.openmrs.performance.Constants.OUTPATIENT_CLINIC_LOCATION_UUID;
|
23 | 28 | import static org.openmrs.performance.Constants.TABLET;
|
| 29 | +import static org.openmrs.performance.Constants.VISIT_NOTE_CONCEPT_UUID; |
| 30 | +import static org.openmrs.performance.Constants.VISIT_NOTE_ENCOUNTER_TYPE_UUID; |
| 31 | +import static org.openmrs.performance.Constants.VISIT_NOTE_FORM_UUID; |
24 | 32 |
|
25 | 33 | public class DoctorHttpService extends HttpService {
|
26 | 34 |
|
@@ -232,4 +240,65 @@ public HttpRequestActionBuilder saveOrder(String patientUuid, String visitUuid,
|
232 | 240 | .post("/openmrs/ws/rest/v1/encounter")
|
233 | 241 | .body(StringBody(body));
|
234 | 242 | }
|
| 243 | + |
| 244 | + public HttpRequestActionBuilder saveVisitNote(String patientUuid, String currentUser, String value) { |
| 245 | + Map<String, Object> visitNote = new HashMap<>(); |
| 246 | + visitNote.put("form", VISIT_NOTE_FORM_UUID); |
| 247 | + visitNote.put("patient", patientUuid); |
| 248 | + visitNote.put("location", OUTPATIENT_CLINIC_LOCATION_UUID); |
| 249 | + visitNote.put("encounterType", VISIT_NOTE_ENCOUNTER_TYPE_UUID); |
| 250 | + |
| 251 | + Map<String, Object> encounterProvider = new HashMap<>(); |
| 252 | + encounterProvider.put("encounterRole", CLINICIAN_ENCOUNTER_ROLE); |
| 253 | + encounterProvider.put("provider", currentUser); |
| 254 | + |
| 255 | + visitNote.put("encounterProviders", List.of(encounterProvider)); |
| 256 | + |
| 257 | + Map<String, Object> concept = new HashMap<>(); |
| 258 | + concept.put("uuid", VISIT_NOTE_CONCEPT_UUID); |
| 259 | + |
| 260 | + |
| 261 | + Map<String, Object> obs = new HashMap<>(); |
| 262 | + obs.put("concept", concept); |
| 263 | + obs.put("value", value); |
| 264 | + visitNote.put("obs", List.of(obs)); |
| 265 | + |
| 266 | + Gson gson = new Gson(); |
| 267 | + String body = gson.toJson(visitNote); |
| 268 | + |
| 269 | + exec(session -> { |
| 270 | + System.out.println(body); |
| 271 | + return session; |
| 272 | + }); |
| 273 | + |
| 274 | + return http("Save Visit Note").post("/openmrs/ws/rest/v1/encounter").body(StringBody(body)) |
| 275 | + .check(jsonPath("$.uuid").saveAs("encounter_uuid")); |
| 276 | + } |
| 277 | + |
| 278 | + public HttpRequestActionBuilder saveDiagnosis(String patientUuid, String encounterUuid, String diagnosisUuid, |
| 279 | + String certainty, int rank) { |
| 280 | + Map<String, Object> patientDiagnosis = new HashMap<>(); |
| 281 | + patientDiagnosis.put("patient", patientUuid); |
| 282 | + patientDiagnosis.put("encounter", encounterUuid); |
| 283 | + patientDiagnosis.put("certainty", certainty); |
| 284 | + patientDiagnosis.put("rank", rank); |
| 285 | + patientDiagnosis.put("condition", null); |
| 286 | + |
| 287 | + Map<String, Object> diagnosis = new HashMap<>(); |
| 288 | + diagnosis.put("coded", diagnosisUuid); |
| 289 | + patientDiagnosis.put("diagnosis", diagnosis); |
| 290 | + |
| 291 | + Gson gson = new GsonBuilder().serializeNulls().create(); |
| 292 | + String body = gson.toJson(patientDiagnosis); |
| 293 | + |
| 294 | + exec(seassion -> { |
| 295 | + System.out.println(body); |
| 296 | + return seassion; |
| 297 | + }); |
| 298 | + |
| 299 | + return http("Save Patient Diagnosis") |
| 300 | + .post("/openmrs/ws/rest/v1/patientdiagnoses") |
| 301 | + .body(StringBody(body)); |
| 302 | + } |
| 303 | + |
235 | 304 | }
|
0 commit comments