adding test for report parameter

Evolveum · Aug 17, 2021 · 79b063a · 79b063a
1 parent f8e1e12
commit 79b063a
Showing 1 changed file with 57 additions and 7 deletions.
diff --git a/model/report-impl/src/test/java/com/evolveum/midpoint/report/TestCsvReportExportClassic.java b/model/report-impl/src/test/java/com/evolveum/midpoint/report/TestCsvReportExportClassic.java
@@ -8,18 +8,24 @@
 
 import java.io.File;
 
-import com.evolveum.midpoint.prism.PrismObject;
+import com.evolveum.midpoint.prism.*;
 
+import com.evolveum.midpoint.prism.path.ItemPath;
+import com.evolveum.midpoint.report.api.ReportConstants;
 import com.evolveum.midpoint.test.util.MidPointTestConstants;
 
+import com.evolveum.midpoint.util.exception.SchemaException;
+import com.evolveum.midpoint.xml.ns._public.common.common_3.*;
+
 import org.springframework.test.annotation.DirtiesContext;
 import org.springframework.test.context.ContextConfiguration;
 import org.testng.annotations.Test;
 
 import com.evolveum.midpoint.schema.result.OperationResult;
 import com.evolveum.midpoint.task.api.Task;
 import com.evolveum.midpoint.test.TestResource;
-import com.evolveum.midpoint.xml.ns._public.common.common_3.ReportType;
+
+import javax.xml.namespace.QName;
 
 import static org.testng.AssertJUnit.assertTrue;
 
@@ -149,10 +155,11 @@ public void test117ObjectCollectionReportWithFilterAndBasicCollectionWithoutView
         runTest(REPORT_OBJECT_COLLECTION_FILTER_BASIC_COLLECTION_WITHOUT_VIEW, 2, 1, null);
     }
 
-//    @Test //TODO uncomment after implementation of parameters in new task config
-//    public void test118ObjectCollectionWithParamReport() throws Exception {
-//        runTest(REPORT_OBJECT_COLLECTION_WITH_PARAM, 2, 2, null);
-//    }
+    @Test
+    public void test118ObjectCollectionWithParamReport() throws Exception {
+        ReportParameterType parameters = getParameters("givenName", String.class, "Will");
+        runTest(REPORT_OBJECT_COLLECTION_WITH_PARAM, 2, 2, null, parameters);
+    }
 
     @Test
     public void test119ObjectCollectionWithSubreportParamReport() throws Exception {
@@ -175,12 +182,27 @@ public void test121RunMidpointUsersScript() throws Exception {
         assertTrue("Target file is not there", targetFile.exists());
     }
 
-    private void runTest(TestResource<ReportType> reportResource, int expectedRows, int expectedColumns, CharSequence lastline) throws Exception {
+    private void runTest(TestResource<ReportType> reportResource, int expectedRows, int expectedColumns, CharSequence lastline,
+            ReportParameterType parameters) throws Exception {
         given();
 
         Task task = getTestTask();
         OperationResult result = task.getResult();
 
+        if (parameters != null) {
+            modifyObjectReplaceContainer(TaskType.class,
+                    TASK_EXPORT_CLASSIC.oid,
+                    ItemPath.create(TaskType.F_ACTIVITY,
+                            ActivityDefinitionType.F_WORK,
+                            WorkDefinitionsType.F_REPORT_EXPORT,
+                            ClassicReportImportWorkDefinitionType.F_REPORT_PARAM
+                    ),
+                    task,
+                    result,
+                    parameters
+            );
+        }
+
         runExportTask(reportResource, result);
 
         when();
@@ -196,4 +218,32 @@ private void runTest(TestResource<ReportType> reportResource, int expectedRows,
         PrismObject<ReportType> report = getObject(ReportType.class, reportResource.oid);
         basicCheckOutputFile(report, expectedRows, expectedColumns, lastline);
     }
+
+    private void runTest(TestResource<ReportType> reportResource, int expectedRows, int expectedColumns, CharSequence lastline)
+            throws Exception {
+        runTest(reportResource, expectedRows, expectedColumns, lastline, null);
+    }
+
+    private ReportParameterType getParameters(String name, Class<String> type, Object realValue) throws SchemaException {
+//        PrismContainerDefinition<ReportParameterType> paramContainerDef = prismContext.getSchemaRegistry().findContainerDefinitionByElementName(ReportConstants.REPORT_PARAMS_PROPERTY_NAME);
+//        PrismContainer<ReportParameterType> paramContainer;
+//        paramContainer = paramContainerDef.instantiate();
+        ReportParameterType reportParam = new ReportParameterType();
+        PrismContainerValue<ReportParameterType> reportParamValue = reportParam.asPrismContainerValue();
+//        reportParamValue.revive(prismContext);
+//        paramContainer.add(reportParamValue);
+
+        QName typeName = prismContext.getSchemaRegistry().determineTypeForClass(type);
+        MutablePrismPropertyDefinition<Object> def = prismContext.definitionFactory().createPropertyDefinition(
+                new QName(ReportConstants.NS_EXTENSION, name), typeName);
+        def.setDynamic(true);
+        def.setRuntimeSchema(true);
+        def.toMutable().setMaxOccurs(1);
+
+        PrismProperty prop = def.instantiate();
+        prop.addRealValue(realValue);
+        reportParamValue.add(prop);
+
+        return reportParam;
+    }
 }