public String execute() throws Exception {
// ---------------------------------------------------------------------
// Prepare values
// ---------------------------------------------------------------------
code = nullIfEmpty(code);
shortName = nullIfEmpty(shortName);
description = nullIfEmpty(description);
PeriodType periodType = PeriodType.getPeriodTypeByName(frequencySelect);
DataSet dataSet = new DataSet(name, shortName, code, periodType);
MapLegendSet legendSet = mappingService.getMapLegendSet(selectedLegendSetId);
dataSet.setExpiryDays(expiryDays);
dataSet.setTimelyDays(timelyDays);
dataSet.setSkipAggregation(skipAggregation);
for (String id : dataElementsSelectedList) {
dataSet.addDataElement(dataElementService.getDataElement(Integer.parseInt(id)));
}
Set<Indicator> indicators = new HashSet<Indicator>();
for (String id : indicatorsSelectedList) {
indicators.add(indicatorService.getIndicator(Integer.parseInt(id)));
}
if (categoryComboId != null) {
dataSet.setCategoryCombo(categoryService.getDataElementCategoryCombo(categoryComboId));
}
dataSet.setDescription(description);
dataSet.setVersion(1);
dataSet.setMobile(false);
dataSet.setIndicators(indicators);
dataSet.setNotificationRecipients(userGroupService.getUserGroup(notificationRecipients));
dataSet.setAllowFuturePeriods(allowFuturePeriods);
dataSet.setFieldCombinationRequired(fieldCombinationRequired);
dataSet.setValidCompleteOnly(validCompleteOnly);
dataSet.setNotifyCompletingUser(notifyCompletingUser);
dataSet.setApproveData(approveData);
dataSet.setSkipOffline(skipOffline);
dataSet.setDataElementDecoration(dataElementDecoration);
dataSet.setRenderAsTabs(renderAsTabs);
dataSet.setRenderHorizontally(renderHorizontally);
dataSet.setLegendSet(legendSet);
dataSetService.addDataSet(dataSet);
userService.assignDataSetToUserRole(dataSet);
return SUCCESS;
}
@Test
public void testAverageIntDataElement() {
DataElement dataElement = createDataElement('A', categoryCombo);
dataElement.setType(DataElement.VALUE_TYPE_INT);
dataElement.setAggregationOperator(DataElement.AGGREGATION_OPERATOR_AVERAGE_SUM);
dataElementIds.add(dataElementService.addDataElement(dataElement));
DataSet dataSet = createDataSet('A', new YearlyPeriodType());
dataSet.getDataElements().add(dataElement);
dataSetService.addDataSet(dataSet);
dataElement.getDataSets().add(dataSet);
dataElementService.updateDataElement(dataElement);
Period dataPeriod =
createPeriod(new YearlyPeriodType(), getDate(2000, 1, 1), getDate(2000, 12, 31));
OrganisationUnit unit = createOrganisationUnit('A');
organisationUnitIds.add(organisationUnitService.addOrganisationUnit(unit));
dataValueService.addDataValue(
createDataValue(dataElement, dataPeriod, unit, "100", categoryOptionCombo));
Period periodA =
createPeriod(new MonthlyPeriodType(), getDate(2000, 3, 1), getDate(2000, 3, 31));
Period periodB =
createPeriod(new MonthlyPeriodType(), getDate(2000, 4, 1), getDate(2000, 4, 30));
Period periodC =
createPeriod(new QuarterlyPeriodType(), getDate(2000, 3, 1), getDate(2000, 5, 31));
periodIds.add(periodService.addPeriod(periodA));
periodIds.add(periodService.addPeriod(periodB));
periodIds.add(periodService.addPeriod(periodC));
dataMartEngine.export(dataElementIds, indicatorIds, periodIds, organisationUnitIds);
assertEquals(
100.0,
aggregatedDataValueService.getAggregatedValue(
dataElement, categoryOptionCombo, periodA, unit),
DELTA);
assertEquals(
100.0,
aggregatedDataValueService.getAggregatedValue(
dataElement, categoryOptionCombo, periodB, unit),
DELTA);
assertEquals(
100.0,
aggregatedDataValueService.getAggregatedValue(
dataElement, categoryOptionCombo, periodC, unit),
DELTA);
}
@Override
public void setUpTest() {
mockDataValueBatchHandler = new MockBatchHandler<>();
mockBatchHandlerFactory = new MockBatchHandlerFactory();
mockBatchHandlerFactory.registerBatchHandler(
DataValueBatchHandler.class, mockDataValueBatchHandler);
setDependency(dataValueSetService, "batchHandlerFactory", mockBatchHandlerFactory);
categoryOptionA = createCategoryOption('A');
categoryOptionB = createCategoryOption('B');
categoryA = createDataElementCategory('A', categoryOptionA, categoryOptionB);
categoryComboA = createCategoryCombo('A', categoryA);
ocDef = categoryService.getDefaultDataElementCategoryOptionCombo();
ocA = createCategoryOptionCombo(categoryComboA, categoryOptionA);
ocB = createCategoryOptionCombo(categoryComboA, categoryOptionB);
deA = createDataElement('A');
deB = createDataElement('B');
deC = createDataElement('C');
deD = createDataElement('D');
dsA = createDataSet('A', new MonthlyPeriodType());
ouA = createOrganisationUnit('A');
ouB = createOrganisationUnit('B');
ouC = createOrganisationUnit('C');
peA =
createPeriod(
PeriodType.getByNameIgnoreCase(MonthlyPeriodType.NAME),
getDate(2012, 1, 1),
getDate(2012, 1, 31));
peB =
createPeriod(
PeriodType.getByNameIgnoreCase(MonthlyPeriodType.NAME),
getDate(2012, 2, 1),
getDate(2012, 2, 29));
ocA.setUid("kjuiHgy67hg");
ocB.setUid("Gad33qy67g5");
deA.setUid("f7n9E0hX8qk");
deB.setUid("Ix2HsbDMLea");
deC.setUid("eY5ehpbEsB7");
dsA.setUid("pBOMPrpg1QX");
ouA.setUid("DiszpKrYNg8");
ouB.setUid("BdfsJfj87js");
ouC.setUid("j7Hg26FpoIa");
ocA.setCode("OC_A");
ocB.setCode("OC_B");
deA.setCode("DE_A");
deB.setCode("DE_B");
deC.setCode("DE_C");
deD.setCode("DE_D");
dsA.setCode("DS_A");
ouA.setCode("OU_A");
ouB.setCode("OU_B");
ouC.setCode("OU_C");
categoryService.addDataElementCategoryOption(categoryOptionA);
categoryService.addDataElementCategoryOption(categoryOptionB);
categoryService.addDataElementCategory(categoryA);
categoryService.addDataElementCategoryCombo(categoryComboA);
categoryService.addDataElementCategoryOptionCombo(ocA);
categoryService.addDataElementCategoryOptionCombo(ocB);
dataElementService.addDataElement(deA);
dataElementService.addDataElement(deB);
dataElementService.addDataElement(deC);
dataElementService.addDataElement(deD);
dataSetService.addDataSet(dsA);
organisationUnitService.addOrganisationUnit(ouA);
organisationUnitService.addOrganisationUnit(ouB);
organisationUnitService.addOrganisationUnit(ouC);
periodService.addPeriod(peA);
periodService.addPeriod(peB);
user = createUser('A');
user.setOrganisationUnits(Sets.newHashSet(ouA, ouB));
CurrentUserService currentUserService = new MockCurrentUserService(user);
setDependency(dataValueSetService, "currentUserService", currentUserService);
}
