Examples with TrackerImportParams org.hisp.dhis.tracker.TrackerImportParams used on opensource projects

Example 21 with TrackerImportParams

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

the class FileResourceSupplierTest method verifySupplier.

@Test
void verifySupplier() {
    FileResource fileResource = createFileResource('A', "FileResource".getBytes());
    fileResource.setUid(FILE_RESOURCE_UID);
    FileResource anotherFileResource = createFileResource('B', "AnotherFileResource".getBytes());
    anotherFileResource.setUid(ANOTHER_FILE_RESOURCE_UID);
    when(fileResourceService.getFileResources(Lists.newArrayList(ANOTHER_FILE_RESOURCE_UID, FILE_RESOURCE_UID))).thenReturn(Lists.newArrayList(fileResource, anotherFileResource));
    final TrackerImportParams params = TrackerImportParams.builder().trackedEntities(Lists.newArrayList(getTrackedEntity())).enrollments(Lists.newArrayList(getEnrollment())).events(Lists.newArrayList(getEvent())).build();
    this.supplierToTest.preheatAdd(params, preheat);
    assertThat(preheat.getAll(FileResource.class), hasSize(2));
    assertThat(preheat.getAll(FileResource.class), containsInAnyOrder(fileResource, anotherFileResource));
}

Example 22 with TrackerImportParams

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

the class TrackerEntityInstanceStrategyTest method verifyStrategyIgnoresPersistedTei.

@Test
void verifyStrategyIgnoresPersistedTei() {
    // Create preheat params
    final List<TrackedEntity> trackedEntities = rnd.objects(TrackedEntity.class, 2).collect(Collectors.toList());
    final TrackerImportParams params = TrackerImportParams.builder().trackedEntities(trackedEntities).build();
    // Preheat
    User user = new User();
    TrackerPreheat preheat = new TrackerPreheat();
    preheat.setUser(user);
    final List<String> rootUids = trackedEntities.stream().map(TrackedEntity::getTrackedEntity).collect(Collectors.toList());
    // Add uid of non-root tei
    rootUids.add("noroottei");
    List<List<String>> uids = new ArrayList<>();
    uids.add(rootUids);
    when(trackedEntityInstanceStore.getIncludingDeleted(rootUids)).thenReturn(Lists.newArrayList(new TrackedEntityInstance() {

        {
            setUid(trackedEntities.get(0).getTrackedEntity());
        }
    }));
    // when
    strategy.add(params, uids, preheat);
    preheat.createReferenceTree();
    assertFalse(preheat.getReference(trackedEntities.get(0).getTrackedEntity()).isPresent());
    assertTrue(preheat.getReference(trackedEntities.get(1).getTrackedEntity()).isPresent());
    assertFalse(preheat.getReference("noroottei").isPresent());
}

Example 23 with TrackerImportParams

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

the class TrackerPreheatServiceIntegration method testPreheatWithDifferentIdSchemes.

@Test
@Disabled
void testPreheatWithDifferentIdSchemes() {
    TrackedEntity teA = TrackedEntity.builder().orgUnit("OUA").trackedEntityType(TET_UID).build();
    Enrollment enrollmentA = Enrollment.builder().orgUnit("OUA").program("PROGRAM1").trackedEntity(TE_UID).build();
    TrackerImportParams trackerPreheatParams = TrackerImportParams.builder().trackedEntities(Lists.newArrayList(teA)).enrollments(Lists.newArrayList(enrollmentA)).identifiers(TrackerIdentifierParams.builder().idScheme(TrackerIdentifier.UID).orgUnitIdScheme(TrackerIdentifier.CODE).programIdScheme(TrackerIdentifier.builder().idScheme(TrackerIdScheme.ATTRIBUTE).value(ATTRIBUTE_UID).build()).build()).build();
    TrackerPreheat preheat = trackerPreheatService.preheat(trackerPreheatParams);
    assertNotNull(preheat);
    assertNotNull(preheat.getMap());
}

Example 24 with TrackerImportParams

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

the class TrackerImportParamsBuilderTest method testFlushMode.

@Test
void testFlushMode() {
    Arrays.stream(FlushMode.values()).forEach(e -> {
        paramMap.put(FLUSH_MODE_KEY.getKey(), Collections.singletonList(e.name()));
        TrackerImportParams params = TrackerImportParamsBuilder.build(paramMap);
        assertThat(params.getFlushMode(), is(e));
    });
}

Example 25 with TrackerImportParams

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

the class TrackerImportParamsBuilderTest method testCategoryOptionIdentifier.

@Test
void testCategoryOptionIdentifier() {
    Arrays.stream(TrackerIdScheme.values()).forEach(e -> {
        paramMap.put("categoryOptionIdScheme", Collections.singletonList(e.name()));
        TrackerImportParams params = TrackerImportParamsBuilder.build(paramMap);
        assertThat(params.getIdentifiers().getCategoryOptionIdScheme().getIdScheme(), is(e));
    });
}