Examples with TrackedEntity org.hisp.dhis.tracker.domain.TrackedEntity used on opensource projects

Example 1 with TrackedEntity

use of org.hisp.dhis.tracker.domain.TrackedEntity in project dhis2-core by dhis2.

the class AssignValueImplementerTest method testAssignAttributeValueForEnrollmentsWhenAttributeIsAlreadyPresentInTei.

@Test
void testAssignAttributeValueForEnrollmentsWhenAttributeIsAlreadyPresentInTei() {
    List<Enrollment> enrollments = Lists.newArrayList(getEnrollmentWithAttributeNOTSet());
    List<TrackedEntity> trackedEntities = Lists.newArrayList(getTrackedEntitiesWithAttributeSet());
    bundle.setEnrollments(enrollments);
    bundle.setTrackedEntities(trackedEntities);
    bundle.setRuleEffects(getRuleEnrollmentEffects(enrollments));
    Map<String, List<ProgramRuleIssue>> enrollmentIssues = implementerToTest.validateEnrollments(bundle);
    Enrollment enrollment = bundle.getEnrollments().stream().filter(e -> e.getEnrollment().equals(SECOND_ENROLLMENT_ID)).findAny().get();
    TrackedEntity trackedEntity = bundle.getTrackedEntities().stream().filter(e -> e.getTrackedEntity().equals(TRACKED_ENTITY_ID)).findAny().get();
    Optional<Attribute> enrollmentAttribute = enrollment.getAttributes().stream().filter(at -> at.getAttribute().equals(ATTRIBUTE_ID)).findAny();
    Optional<Attribute> teiAttribute = trackedEntity.getAttributes().stream().filter(at -> at.getAttribute().equals(ATTRIBUTE_ID)).findAny();
    assertFalse(enrollmentAttribute.isPresent());
    assertTrue(teiAttribute.isPresent());
    assertEquals(TEI_ATTRIBUTE_OLD_VALUE, teiAttribute.get().getValue());
    assertEquals(1, enrollmentIssues.size());
    assertEquals(1, enrollmentIssues.get(SECOND_ENROLLMENT_ID).size());
    assertEquals(ERROR, enrollmentIssues.get(SECOND_ENROLLMENT_ID).get(0).getIssueType());
}

Example 2 with TrackedEntity

use of org.hisp.dhis.tracker.domain.TrackedEntity in project dhis2-core by dhis2.

the class AssignValueImplementerTest method getTrackedEntitiesWithAttributeNOTSet.

private TrackedEntity getTrackedEntitiesWithAttributeNOTSet() {
    TrackedEntity trackedEntity = new TrackedEntity();
    trackedEntity.setTrackedEntity(TRACKED_ENTITY_ID);
    return trackedEntity;
}

Example 3 with TrackedEntity

use of org.hisp.dhis.tracker.domain.TrackedEntity in project dhis2-core by dhis2.

the class AssignValueImplementerTest method testAssignAttributeValueForEnrollmentsWhenAttributeIsEmpty.

@Test
void testAssignAttributeValueForEnrollmentsWhenAttributeIsEmpty() {
    List<TrackedEntity> trackedEntities = Lists.newArrayList(getTrackedEntitiesWithAttributeNOTSet());
    List<Enrollment> enrollments = Lists.newArrayList(getEnrollmentWithAttributeNOTSet());
    bundle.setTrackedEntities(trackedEntities);
    bundle.setEnrollments(enrollments);
    bundle.setRuleEffects(getRuleEnrollmentEffects(enrollments));
    Map<String, List<ProgramRuleIssue>> enrollmentIssues = implementerToTest.validateEnrollments(bundle);
    Enrollment enrollment = bundle.getEnrollments().stream().filter(e -> e.getEnrollment().equals(SECOND_ENROLLMENT_ID)).findAny().get();
    Optional<Attribute> attribute = enrollment.getAttributes().stream().filter(at -> at.getAttribute().equals(ATTRIBUTE_ID)).findAny();
    assertTrue(attribute.isPresent());
    assertEquals(TEI_ATTRIBUTE_NEW_VALUE, attribute.get().getValue());
    assertEquals(1, enrollmentIssues.size());
    assertEquals(1, enrollmentIssues.get(SECOND_ENROLLMENT_ID).size());
    assertEquals(WARNING, enrollmentIssues.get(SECOND_ENROLLMENT_ID).get(0).getIssueType());
}

Example 4 with TrackedEntity

use of org.hisp.dhis.tracker.domain.TrackedEntity in project dhis2-core by dhis2.

the class AssignValueImplementerTest method getTrackedEntitiesWithAttributeSet.

private TrackedEntity getTrackedEntitiesWithAttributeSet() {
    TrackedEntity trackedEntity = new TrackedEntity();
    trackedEntity.setTrackedEntity(TRACKED_ENTITY_ID);
    trackedEntity.setAttributes(getAttributes());
    return trackedEntity;
}

Example 5 with TrackedEntity

use of org.hisp.dhis.tracker.domain.TrackedEntity in project dhis2-core by dhis2.

the class StrategyPreProcessorTest method setUp.

@BeforeEach
void setUp() {
    tei = new TrackedEntityInstance();
    tei.setUid(TEI_UID);
    trackedEntity = new TrackedEntity();
    trackedEntity.setTrackedEntity(TEI_UID);
    newTrackedEntity = new TrackedEntity();
    newTrackedEntity.setTrackedEntity(NEW_TEI_UID);
    pi = new ProgramInstance();
    pi.setUid(ENROLLMENT_UID);
    enrollment = new Enrollment();
    enrollment.setEnrollment(ENROLLMENT_UID);
    newEnrollment = new Enrollment();
    newEnrollment.setEnrollment(NEW_ENROLLMENT_UID);
    psi = new ProgramStageInstance();
    psi.setUid(EVENT_UID);
    event = new Event();
    event.setEvent(EVENT_UID);
    newEvent = new Event();
    newEvent.setEvent(NEW_EVENT_UID);
    relationship = new Relationship();
    relationship.setUid(RELATIONSHIP_UID);
    payloadRelationship = new org.hisp.dhis.tracker.domain.Relationship();
    payloadRelationship.setRelationship(RELATIONSHIP_UID);
    newPayloadRelationship = new org.hisp.dhis.tracker.domain.Relationship();
    newPayloadRelationship.setRelationship(NEW_RELATIONSHIP_UID);
    Mockito.when(preheat.getTrackedEntity(TrackerIdScheme.UID, TEI_UID)).thenReturn(tei);
    Mockito.when(preheat.getEnrollment(TrackerIdScheme.UID, ENROLLMENT_UID)).thenReturn(pi);
    Mockito.when(preheat.getEvent(TrackerIdScheme.UID, EVENT_UID)).thenReturn(psi);
    Mockito.when(preheat.getRelationship(TrackerIdScheme.UID, payloadRelationship)).thenReturn(relationship);
}