/*
* Copyright 2013 Red Hat, Inc. and/or its affiliates.
*
* Licensed under the Apache License, Version 2.0 (the "License");
* you may not use this file except in compliance with the License.
* You may obtain a copy of the License at
*
* http://www.apache.org/licenses/LICENSE-2.0
*
* Unless required by applicable law or agreed to in writing, software
* distributed under the License is distributed on an "AS IS" BASIS,
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
* See the License for the specific language governing permissions and
* limitations under the License.
*/
package org.jbpm.test.functional.subprocess;
import java.util.Arrays;
import java.util.Collection;
import java.util.HashMap;
import java.util.List;
import java.util.Map;
import javax.naming.InitialContext;
import javax.transaction.UserTransaction;
import org.drools.persistence.jta.JtaTransactionManager;
import org.jbpm.process.instance.impl.demo.SystemOutWorkItemHandler;
import org.jbpm.services.task.admin.listener.TaskCleanUpProcessEventListener;
import org.jbpm.test.JbpmTestCase;
import org.junit.Test;
import org.junit.runner.RunWith;
import org.junit.runners.Parameterized;
import org.junit.runners.Parameterized.Parameters;
import org.kie.api.runtime.KieSession;
import org.kie.api.runtime.manager.RuntimeEngine;
import org.kie.api.runtime.manager.RuntimeManager;
import org.kie.api.runtime.process.ProcessInstance;
import org.kie.api.task.TaskService;
import org.kie.api.task.model.TaskSummary;
import org.kie.internal.runtime.manager.context.ProcessInstanceIdContext;
import org.kie.internal.task.api.InternalTaskService;
@RunWith(Parameterized.class)
public class CallActivitiesWithUserTasksProcessTest extends JbpmTestCase {
@Parameters(name="{1} | user managed={0}")
public static Collection<Object[]> parameters() {
Object[][] locking = new Object[][] {
{ true, Strategy.SINGLETON },
{ true, Strategy.PROCESS_INSTANCE },
{ false, Strategy.SINGLETON },
{ false, Strategy.PROCESS_INSTANCE },
};
return Arrays.asList(locking);
};
private final boolean userManagedTx;
private final Strategy strategy;
public CallActivitiesWithUserTasksProcessTest(boolean txManagedType, Strategy runtimeStrategy) {
super(true, true);
this.userManagedTx = txManagedType;
this.strategy = runtimeStrategy;
}
@Test
public void testCallActivitiesWithUserTasks() throws Exception {
TaskCleanUpProcessEventListener taskListener = new TaskCleanUpProcessEventListener(null);
addWorkItemHandler("Sysout", new SystemOutWorkItemHandler());
addProcessEventListener(taskListener);
InitialContext context = new InitialContext();
UserTransaction ut = (UserTransaction) context.lookup( JtaTransactionManager.DEFAULT_USER_TRANSACTION_NAME );
RuntimeManager manager = createRuntimeManager(strategy, (String) null, "org/jbpm/test/functional/subprocess/CallActivityWithTask-Main.bpmn2",
"org/jbpm/test/functional/subprocess/CallActivityWithTask-Sub.bpmn2");
RuntimeEngine runtimeEngine;
if( Strategy.SINGLETON.equals(strategy) ) {
runtimeEngine = getRuntimeEngine();
} else if( Strategy.PROCESS_INSTANCE.equals(strategy) ) {
runtimeEngine = getRuntimeEngine(ProcessInstanceIdContext.get());
} else {
throw new IllegalStateException("Not possible!");
}
KieSession ksession = runtimeEngine.getKieSession();
TaskService taskService = runtimeEngine.getTaskService();
// set created task service on listener
taskListener.setTaskService((InternalTaskService) taskService);
if( userManagedTx ) {
ut.begin();
}
Map<String, Object> params = new HashMap<String, Object>();
params.put("pActorId", "john");
ProcessInstance processInstance = ksession.startProcess("PolicyValueAnalysis", params);
if( userManagedTx ) {
ut.commit();
manager.disposeRuntimeEngine(runtimeEngine);
runtimeEngine = getRuntimeEngine(ProcessInstanceIdContext.get(processInstance.getId()));
ksession = runtimeEngine.getKieSession();
taskService = runtimeEngine.getTaskService();
taskListener.setTaskService((InternalTaskService) taskService);
ut.begin();
} else {
manager.disposeRuntimeEngine(runtimeEngine);
runtimeEngine = getRuntimeEngine(ProcessInstanceIdContext.get(processInstance.getId()));
ksession = runtimeEngine.getKieSession();
taskService = runtimeEngine.getTaskService();
taskListener.setTaskService((InternalTaskService) taskService);
}
List<TaskSummary> tasks = taskService.getTasksAssignedAsPotentialOwner("john", "en-UK");
assertNotNull(tasks);
assertEquals(1, tasks.size());
Long taskId = tasks.get(0).getId();
taskService.start(taskId, "john");
taskService.complete(taskId, "john", null);
if( userManagedTx ) {
ut.commit();
manager.disposeRuntimeEngine(runtimeEngine);
runtimeEngine = getRuntimeEngine(ProcessInstanceIdContext.get(processInstance.getId()));
ksession = runtimeEngine.getKieSession();
taskService = runtimeEngine.getTaskService();
taskListener.setTaskService((InternalTaskService) taskService);
ut.begin();
} else {
manager.disposeRuntimeEngine(runtimeEngine);
runtimeEngine = getRuntimeEngine(ProcessInstanceIdContext.get(processInstance.getId()));
ksession = runtimeEngine.getKieSession();
taskService = runtimeEngine.getTaskService();
taskListener.setTaskService((InternalTaskService) taskService);
}
tasks = taskService.getTasksAssignedAsPotentialOwner("john", "en-UK");
assertNotNull(tasks);
assertEquals(1, tasks.size());
taskId = tasks.get(0).getId();
taskService.start(taskId, "john");
taskService.complete(taskId, "john", null);
if( userManagedTx ) {
ut.commit();
ut.begin();
}
assertProcessInstanceCompleted(processInstance.getId());
if( userManagedTx ) {
ut.commit();
}
}
@Test
public void testCallActivitiesWith2ndUserTaskInSub() throws Exception {
TaskCleanUpProcessEventListener taskListener = new TaskCleanUpProcessEventListener(null);
addWorkItemHandler("Sysout", new SystemOutWorkItemHandler());
addProcessEventListener(taskListener);
InitialContext context = new InitialContext();
UserTransaction ut = (UserTransaction) context.lookup( JtaTransactionManager.DEFAULT_USER_TRANSACTION_NAME );
RuntimeManager manager = createRuntimeManager(strategy, (String) null, "org/jbpm/test/functional/subprocess/CallActivityWithTaskInSub-Main.bpmn2",
"org/jbpm/test/functional/subprocess/CallActivityWithTaskInSub-Sub.bpmn2");
RuntimeEngine runtimeEngine;
if( Strategy.SINGLETON.equals(strategy) ) {
runtimeEngine = getRuntimeEngine();
} else if( Strategy.PROCESS_INSTANCE.equals(strategy) ) {
runtimeEngine = getRuntimeEngine(ProcessInstanceIdContext.get());
} else {
throw new IllegalStateException("Not possible!");
}
KieSession ksession = runtimeEngine.getKieSession();
TaskService taskService = runtimeEngine.getTaskService();
// set created task service on listener
taskListener.setTaskService((InternalTaskService) taskService);
if( userManagedTx ) {
ut.begin();
}
Map<String, Object> params = new HashMap<String, Object>();
params.put("pActorId", "john");
ProcessInstance processInstance = ksession.startProcess("PolicyValueAnalysis", params);
if( userManagedTx ) {
ut.commit();
manager.disposeRuntimeEngine(runtimeEngine);
runtimeEngine = getRuntimeEngine(ProcessInstanceIdContext.get(processInstance.getId()));
ksession = runtimeEngine.getKieSession();
taskService = runtimeEngine.getTaskService();
taskListener.setTaskService((InternalTaskService) taskService);
ut.begin();
} else {
manager.disposeRuntimeEngine(runtimeEngine);
runtimeEngine = getRuntimeEngine(ProcessInstanceIdContext.get(processInstance.getId()));
ksession = runtimeEngine.getKieSession();
taskService = runtimeEngine.getTaskService();
taskListener.setTaskService((InternalTaskService) taskService);
}
List<TaskSummary> tasks = taskService.getTasksAssignedAsPotentialOwner("john", "en-UK");
assertNotNull(tasks);
assertEquals(1, tasks.size());
Long taskId = tasks.get(0).getId();
taskService.start(taskId, "john");
taskService.complete(taskId, "john", null);
if( userManagedTx ) {
ut.commit();
manager.disposeRuntimeEngine(runtimeEngine);
runtimeEngine = getRuntimeEngine(ProcessInstanceIdContext.get(processInstance.getId()));
ksession = runtimeEngine.getKieSession();
taskService = runtimeEngine.getTaskService();
taskListener.setTaskService((InternalTaskService) taskService);
ut.begin();
} else {
manager.disposeRuntimeEngine(runtimeEngine);
runtimeEngine = getRuntimeEngine(ProcessInstanceIdContext.get(processInstance.getId()));
ksession = runtimeEngine.getKieSession();
taskService = runtimeEngine.getTaskService();
taskListener.setTaskService((InternalTaskService) taskService);
}
// sub process task
tasks = taskService.getTasksAssignedAsPotentialOwner("john", "en-UK");
assertNotNull(tasks);
assertEquals(1, tasks.size());
taskId = tasks.get(0).getId();
taskService.start(taskId, "john");
taskService.complete(taskId, "john", null);
if( userManagedTx ) {
ut.commit();
manager.disposeRuntimeEngine(runtimeEngine);
runtimeEngine = getRuntimeEngine(ProcessInstanceIdContext.get(processInstance.getId()));
ksession = runtimeEngine.getKieSession();
taskService = runtimeEngine.getTaskService();
taskListener.setTaskService((InternalTaskService) taskService);
ut.begin();
} else {
manager.disposeRuntimeEngine(runtimeEngine);
runtimeEngine = getRuntimeEngine(ProcessInstanceIdContext.get(processInstance.getId()));
ksession = runtimeEngine.getKieSession();
taskService = runtimeEngine.getTaskService();
taskListener.setTaskService((InternalTaskService) taskService);
}
tasks = taskService.getTasksAssignedAsPotentialOwner("john", "en-UK");
assertNotNull(tasks);
assertEquals(1, tasks.size());
taskId = tasks.get(0).getId();
taskService.start(taskId, "john");
taskService.complete(taskId, "john", null);
if( userManagedTx ) {
ut.commit();
manager.disposeRuntimeEngine(runtimeEngine);
runtimeEngine = getRuntimeEngine(ProcessInstanceIdContext.get(processInstance.getId()));
ksession = runtimeEngine.getKieSession();
taskService = runtimeEngine.getTaskService();
taskListener.setTaskService((InternalTaskService) taskService);
ut.begin();
} else {
manager.disposeRuntimeEngine(runtimeEngine);
runtimeEngine = getRuntimeEngine(ProcessInstanceIdContext.get(processInstance.getId()));
ksession = runtimeEngine.getKieSession();
taskService = runtimeEngine.getTaskService();
taskListener.setTaskService((InternalTaskService) taskService);
}
// sub process task
tasks = taskService.getTasksAssignedAsPotentialOwner("john", "en-UK");
assertNotNull(tasks);
assertEquals(1, tasks.size());
taskId = tasks.get(0).getId();
taskService.start(taskId, "john");
taskService.complete(taskId, "john", null);
if( userManagedTx ) {
// since we are checking in same transaction it will already see it as completed
assertProcessInstanceCompleted(processInstance.getId());
ut.commit();
}
assertProcessInstanceCompleted(processInstance.getId());
}
}