use of org.hisp.dhis.trackedentity.TrackedEntityType in project dhis2-core by dhis2.
the class TrackerEnrollmentsExportControllerTest method testGetEnrollmentByIdContainsFollowUp.
@Test
void testGetEnrollmentByIdContainsFollowUp() {
OrganisationUnit orgUnit = createOrganisationUnit('A');
manager.save(orgUnit);
Program program = createProgram('A');
manager.save(program);
TrackedEntityType trackedEntityType = createTrackedEntityType('A');
manager.save(trackedEntityType);
TrackedEntityInstance tei = createTrackedEntityInstance(orgUnit);
tei.setTrackedEntityType(trackedEntityType);
manager.save(tei);
ProgramInstance programInstance = new ProgramInstance(program, tei, orgUnit);
programInstance.setAutoFields();
programInstance.setEnrollmentDate(new Date());
programInstance.setIncidentDate(new Date());
programInstance.setStatus(ProgramStatus.COMPLETED);
programInstance.setFollowup(true);
manager.save(programInstance);
JsonObject enrollment = GET("/tracker/enrollments/{id}", programInstance.getUid()).content(HttpStatus.OK);
assertTrue(enrollment.isObject());
assertFalse(enrollment.isEmpty());
assertEquals(programInstance.getUid(), enrollment.getString("enrollment").string());
assertEquals(tei.getUid(), enrollment.getString("trackedEntity").string());
assertEquals(program.getUid(), enrollment.getString("program").string());
assertEquals("COMPLETED", enrollment.getString("status").string());
assertEquals(orgUnit.getUid(), enrollment.getString("orgUnit").string());
assertEquals(orgUnit.getName(), enrollment.getString("orgUnitName").string());
assertTrue(enrollment.has("enrolledAt", "occurredAt"));
assertTrue(enrollment.getBoolean("followUp").booleanValue());
assertFalse(enrollment.getBoolean("deleted").booleanValue());
assertTrue(enrollment.getArray("events").isEmpty());
assertTrue(enrollment.getArray("relationships").isEmpty());
assertTrue(enrollment.getArray("attributes").isEmpty());
assertTrue(enrollment.getArray("notes").isEmpty());
}
use of org.hisp.dhis.trackedentity.TrackedEntityType in project dhis2-core by dhis2.
the class TrackedEntityCriteriaMapper method validateTrackedEntityType.
private TrackedEntityType validateTrackedEntityType(TrackedEntityInstanceCriteria criteria) {
Function<String, TrackedEntityType> getTeiType = uid -> {
if (isNotEmpty(uid)) {
return trackedEntityTypeService.getTrackedEntityType(uid);
}
return null;
};
final TrackedEntityType trackedEntityType = getTeiType.apply(criteria.getTrackedEntityType());
if (isNotEmpty(criteria.getTrackedEntityType()) && trackedEntityType == null) {
throw new IllegalQueryException("Tracked entity type does not exist: " + criteria.getTrackedEntityType());
}
return trackedEntityType;
}
Aggregations