throws CoreException {
try {
IFolder folder = file.getProject().getFolder("src/main/java");
IPackageFragmentRoot packageFragmentRoot = javaProject
.getPackageFragmentRoot(folder);
IPackageFragment packageFragment = packageFragmentRoot
.createPackageFragment(pName, true, monitor);
String processName = process.getName();
processName = processName.replaceAll("\\s", "_");
if (processName == null || processName.trim().length() == 0) {
processName = "Process";
}
String fileName = processName + "JUnitTest";
String output =
"package " + pName + ";\n" +
"\n" +
"import java.util.ArrayList;\n" +
"import java.util.HashMap;\n" +
"import java.util.List;\n" +
"import java.util.Map;\n" +
"\n" +
"import org.jbpm.process.instance.impl.demo.SystemOutWorkItemHandler;\n" +
"import org.jbpm.test.JbpmJUnitTestCase;\n" +
"\n" +
"import org.junit.Test;\n" +
"\n" +
"import org.kie.api.runtime.KieSession;\n" +
"import org.kie.api.runtime.process.ProcessInstance;\n" +
"import org.kie.api.task.TaskService;\n" +
"import org.kie.api.task.model.TaskSummary;\n" +
"\n" +
"public class " + fileName + " extends JbpmJUnitTestCase {\n" +
"\n";
boolean containsHumanTasks = containsHumanTasks(process);
if (containsHumanTasks) {
output +=
" public " + fileName + "() {\n" +
" super(true);\n" +
" }\n" +
"\n";
}
Map<String, String> cases = new HashMap<String, String>();
Map<String, String> ongoingCases = new HashMap<String, String>();
boolean done = processNodes("", process.getStart(null), "", cases, ongoingCases);
if (!done) {
if (ongoingCases.size() == 1) {
cases.put("Implicit", ongoingCases.values().iterator().next());
} else {
throw new IllegalArgumentException("Could not create implicit case: " + ongoingCases.size());
}
}
for (Map.Entry<String, String> entry: cases.entrySet()) {
output +=
" @Test\n" +
" public void test" + entry.getKey() + "() {\n" +
" KieSession ksession = createKnowledgeSession(\"" + file.getName() + "\");\n";
Set<String> serviceTasks = new HashSet<String>();
containsServiceTasks(serviceTasks, process);
for (String service: serviceTasks) {
output += " ksession.getWorkItemManager().registerWorkItemHandler(\"" + service + "\", new SystemOutWorkItemHandler());\n";
}
if (containsHumanTasks) {
output += " TaskService taskService = getTaskService();\n";
}
List<Variable> variables = process.getVariableScope().getVariables();
if (variables != null && variables.size() > 0) {
output +=
" Map<String, Object> params = new HashMap<String, Object>();\n" +
" // initialize variables here if necessary\n";
for (Variable v: variables) {
output +=
" // params.put(\"" + v.getName() + "\", value); // type " + v.getType().getStringType() + "\n";
}
output +=
" ProcessInstance processInstance = ksession.startProcess(\"" + process.getId() + "\", params);\n";
} else {
output +=
" ProcessInstance processInstance = ksession.startProcess(\"" + process.getId() + "\");\n";
}
output +=
entry.getValue() +
" // do your checks here\n" +
" // for example, assertProcessInstanceCompleted(processInstance.getId(), ksession);\n" +
" }\n" +
"\n";
}
output +="}";
packageFragment.createCompilationUnit(fileName + ".java", output, true, monitor);
} catch (Exception e) {
e.printStackTrace();
MessageDialog.openError(null, "Error", e.getMessage());
}
}