package org.jbpm.persistence.session;
import static org.jbpm.persistence.util.PersistenceUtil.JBPM_PERSISTENCE_UNIT_NAME;
import static org.jbpm.persistence.util.PersistenceUtil.cleanUp;
import static org.jbpm.persistence.util.PersistenceUtil.setupWithPoolingDataSource;
import static org.junit.Assert.assertEquals;
import static org.junit.Assert.assertNotNull;
import static org.junit.Assert.assertNull;
import static org.junit.Assert.fail;
import java.util.ArrayList;
import java.util.HashMap;
import java.util.List;
import java.util.Map;
import javax.naming.NamingException;
import javax.transaction.HeuristicMixedException;
import javax.transaction.HeuristicRollbackException;
import javax.transaction.NotSupportedException;
import javax.transaction.RollbackException;
import javax.transaction.Status;
import javax.transaction.SystemException;
import javax.transaction.UserTransaction;
import junit.framework.Assert;
import org.drools.core.impl.InternalKnowledgeBase;
import org.drools.core.impl.KnowledgeBaseImpl;
import org.drools.core.io.impl.ClassPathResource;
import org.drools.core.marshalling.impl.ClassObjectMarshallingStrategyAcceptor;
import org.drools.core.marshalling.impl.SerializablePlaceholderResolverStrategy;
import org.drools.core.process.core.Work;
import org.drools.core.process.core.datatype.impl.type.ObjectDataType;
import org.drools.core.process.core.impl.WorkImpl;
import org.drools.persistence.infinispan.marshaller.InfinispanPlaceholderResolverStrategy;
import org.infinispan.Cache;
import org.infinispan.manager.DefaultCacheManager;
import org.jbpm.persistence.session.objects.MyEntity;
import org.jbpm.persistence.session.objects.MyEntityMethods;
import org.jbpm.persistence.session.objects.MyEntityOnlyFields;
import org.jbpm.persistence.session.objects.MySubEntity;
import org.jbpm.persistence.session.objects.MySubEntityMethods;
import org.jbpm.persistence.session.objects.MyVariableExtendingSerializable;
import org.jbpm.persistence.session.objects.MyVariableSerializable;
import org.jbpm.persistence.session.objects.TestWorkItemHandler;
import org.jbpm.persistence.util.PersistenceUtil;
import org.jbpm.process.core.context.variable.Variable;
import org.jbpm.process.instance.impl.Action;
import org.jbpm.ruleflow.core.RuleFlowProcess;
import org.jbpm.workflow.core.DroolsAction;
import org.jbpm.workflow.core.Node;
import org.jbpm.workflow.core.impl.ConnectionImpl;
import org.jbpm.workflow.core.impl.DroolsConsequenceAction;
import org.jbpm.workflow.core.node.ActionNode;
import org.jbpm.workflow.core.node.EndNode;
import org.jbpm.workflow.core.node.StartNode;
import org.jbpm.workflow.core.node.WorkItemNode;
import org.junit.After;
import org.junit.Before;
import org.junit.Test;
import org.kie.api.io.ResourceType;
import org.kie.api.marshalling.ObjectMarshallingStrategy;
import org.kie.api.runtime.Environment;
import org.kie.api.runtime.EnvironmentName;
import org.kie.api.runtime.process.ProcessContext;
import org.kie.api.runtime.process.ProcessInstance;
import org.kie.api.runtime.process.WorkItem;
import org.kie.api.runtime.process.WorkflowProcessInstance;
import org.kie.internal.KnowledgeBase;
import org.kie.internal.KnowledgeBaseFactory;
import org.kie.internal.builder.KnowledgeBuilder;
import org.kie.internal.builder.KnowledgeBuilderConfiguration;
import org.kie.internal.builder.KnowledgeBuilderError;
import org.kie.internal.builder.KnowledgeBuilderFactory;
import org.kie.internal.persistence.infinispan.InfinispanKnowledgeService;
import org.kie.internal.runtime.StatefulKnowledgeSession;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
public class VariablePersistenceStrategyTest {
private static Logger logger = LoggerFactory.getLogger( VariablePersistenceStrategyTest.class );
private HashMap<String, Object> context;
private DefaultCacheManager cm;
@Before
public void setUp() throws Exception {
context = setupWithPoolingDataSource(JBPM_PERSISTENCE_UNIT_NAME);
cm = (DefaultCacheManager) context.get(EnvironmentName.ENTITY_MANAGER_FACTORY);
}
@After
public void tearDown() throws Exception {
cleanUp(context);
}
@Test
public void testExtendingInterfaceVariablePersistence() throws Exception {
// Setup
Environment env = createEnvironment();
String processId = "extendingInterfaceVariablePersistence";
String variableText = "my extending serializable variable text";
KnowledgeBase kbase = getKnowledgeBaseForExtendingInterfaceVariablePersistence(processId,
variableText);
StatefulKnowledgeSession ksession = createSession( kbase , env );
Map<String, Object> initialParams = new HashMap<String, Object>();
initialParams.put( "x", new MyVariableExtendingSerializable( variableText ) );
// Start process and execute workItem
long processInstanceId = ksession.startProcess( processId, initialParams ).getId();
ksession = reloadSession( ksession, kbase, env );
long workItemId = TestWorkItemHandler.getInstance().getWorkItem().getId();
ksession.getWorkItemManager().completeWorkItem( workItemId, null );
// Test
Assert.assertNull( ksession.getProcessInstance( processInstanceId ) );
}
private KnowledgeBase getKnowledgeBaseForExtendingInterfaceVariablePersistence(String processId, final String variableText) {
RuleFlowProcess process = new RuleFlowProcess();
process.setId( processId );
List<Variable> variables = new ArrayList<Variable>();
Variable variable = new Variable();
variable.setName("x");
ObjectDataType extendingSerializableDataType = new ObjectDataType();
extendingSerializableDataType.setClassName(MyVariableExtendingSerializable.class.getName());
variable.setType(extendingSerializableDataType);
variables.add(variable);
process.getVariableScope().setVariables(variables);
StartNode startNode = new StartNode();
startNode.setName( "Start" );
startNode.setId(1);
WorkItemNode workItemNode = new WorkItemNode();
workItemNode.setName( "workItemNode" );
workItemNode.setId( 2 );
Work work = new WorkImpl();
work.setName( "MyWork" );
workItemNode.setWork( work );
ActionNode actionNode = new ActionNode();
actionNode.setName( "Print" );
DroolsAction action = new DroolsConsequenceAction( "java" , null);
action.setMetaData( "Action" , new Action() {
public void execute(ProcessContext context) throws Exception {
Assert.assertEquals( variableText , ((MyVariableExtendingSerializable) context.getVariable( "x" )).getText()); ;
}
});
actionNode.setAction(action);
actionNode.setId( 3 );
EndNode endNode = new EndNode();
endNode.setName("EndNode");
endNode.setId(4);
connect( startNode, workItemNode );
connect( workItemNode, actionNode );
connect( actionNode, endNode );
process.addNode( startNode );
process.addNode( workItemNode );
process.addNode( actionNode );
process.addNode( endNode );
KnowledgeBase kbase = KnowledgeBaseFactory.newKnowledgeBase();
((KnowledgeBaseImpl) kbase).addProcess(process);
return kbase;
}
@Test
public void testPersistenceVariables() throws NamingException, NotSupportedException, SystemException, RollbackException, HeuristicMixedException, HeuristicRollbackException {
Cache<String, Object> cache = cm.getCache("jbpm-configured-cache");
UserTransaction utx = (UserTransaction) cache.getAdvancedCache().getTransactionManager();
if( utx.getStatus() == Status.STATUS_NO_TRANSACTION ) {
utx.begin();
}
int origNumMyEntities = getEntitiesFromCache(cache, "myEntity", MyEntity.class).size();
int origNumMyEntityMethods = getEntitiesFromCache(cache, "myEntityMethods", MyEntityMethods.class).size();
int origNumMyEntityOnlyFields = getEntitiesFromCache(cache, "myEntityOnlyFields", MyEntityOnlyFields.class).size();
if( utx.getStatus() == Status.STATUS_ACTIVE ) {
utx.commit();
}
// Setup entities
MyEntity myEntity = new MyEntity("This is a test Entity with annotation in fields");
MyEntityMethods myEntityMethods = new MyEntityMethods("This is a test Entity with annotations in methods");
MyEntityOnlyFields myEntityOnlyFields = new MyEntityOnlyFields("This is a test Entity with annotations in fields and without accesors methods");
MyVariableSerializable myVariableSerializable = new MyVariableSerializable("This is a test SerializableObject");
// persist entities
utx = (UserTransaction) cache.getAdvancedCache().getTransactionManager();
utx.begin();
cache.put(generateId(cache, myEntity), myEntity);
cache.put(generateId(cache, myEntityMethods), myEntityMethods);
cache.put(generateId(cache, myEntityOnlyFields), myEntityOnlyFields);
utx.commit();
// More setup
Environment env = createEnvironment();
KnowledgeBase kbase = createKnowledgeBase( "VariablePersistenceStrategyProcess.rf" );
StatefulKnowledgeSession ksession = createSession( kbase, env );
logger.debug("### Starting process ###");
Map<String, Object> parameters = new HashMap<String, Object>();
parameters.put("x", "SomeString");
parameters.put("y", myEntity);
parameters.put("m", myEntityMethods);
parameters.put("f", myEntityOnlyFields);
parameters.put("z", myVariableSerializable);
// Start process
long processInstanceId = ksession.startProcess( "com.sample.ruleflow", parameters ).getId();
TestWorkItemHandler handler = TestWorkItemHandler.getInstance();
WorkItem workItem = handler.getWorkItem();
assertNotNull( workItem );
// Test results
List<?> result = getEntitiesFromCache(cache, "myEntity", MyEntity.class);
assertEquals(origNumMyEntities + 1, result.size());
result = getEntitiesFromCache(cache, "myEntityMethods", MyEntityMethods.class);
assertEquals(origNumMyEntityMethods + 1, result.size());
result = getEntitiesFromCache(cache, "myEntityOnlyFields", MyEntityOnlyFields.class);
assertEquals(origNumMyEntityOnlyFields + 1, result.size());
logger.debug("### Retrieving process instance ###");
ksession = reloadSession( ksession, kbase, env );
WorkflowProcessInstance processInstance = (WorkflowProcessInstance)
ksession.getProcessInstance( processInstanceId );
assertNotNull( processInstance );
assertEquals("SomeString", processInstance.getVariable("x"));
assertEquals("This is a test Entity with annotation in fields", ((MyEntity) processInstance.getVariable("y")).getTest());
assertEquals("This is a test Entity with annotations in methods", ((MyEntityMethods) processInstance.getVariable("m")).getTest());
assertEquals("This is a test Entity with annotations in fields and without accesors methods", ((MyEntityOnlyFields) processInstance.getVariable("f")).test);
assertEquals("This is a test SerializableObject", ((MyVariableSerializable) processInstance.getVariable("z")).getText());
assertNull(processInstance.getVariable("a"));
assertNull(processInstance.getVariable("b"));
assertNull(processInstance.getVariable("c"));
logger.debug("### Completing first work item ###");
ksession.getWorkItemManager().completeWorkItem( workItem.getId(), null );
workItem = handler.getWorkItem();
assertNotNull( workItem );
ksession.getWorkItemManager().completeWorkItem(workItem.getId(), null);
logger.debug("### Retrieving process instance ###");
ksession = reloadSession( ksession, kbase, env);
processInstance = (WorkflowProcessInstance)
ksession.getProcessInstance(processInstanceId);
assertNotNull(processInstance);
assertEquals("SomeString", processInstance.getVariable("x"));
assertEquals("This is a test Entity with annotation in fields", ((MyEntity) processInstance.getVariable("y")).getTest());
assertEquals("This is a test Entity with annotations in methods", ((MyEntityMethods) processInstance.getVariable("m")).getTest());
assertEquals("This is a test Entity with annotations in fields and without accesors methods", ((MyEntityOnlyFields) processInstance.getVariable("f")).test);
assertEquals("This is a test SerializableObject", ((MyVariableSerializable) processInstance.getVariable("z")).getText());
assertEquals("Some changed String", processInstance.getVariable("a"));
assertEquals("This is a changed test Entity", ((MyEntity) processInstance.getVariable("b")).getTest());
assertEquals("This is a changed test SerializableObject", ((MyVariableSerializable) processInstance.getVariable("c")).getText());
logger.debug("### Completing third work item ###");
ksession.getWorkItemManager().completeWorkItem(workItem.getId(), null);
workItem = handler.getWorkItem();
assertNotNull(workItem);
logger.debug("### Retrieving process instance ###");
ksession = reloadSession( ksession, kbase, env);
processInstance = (WorkflowProcessInstance)
ksession.getProcessInstance(processInstanceId);
assertNotNull(processInstance);
assertEquals("SomeString", processInstance.getVariable("x"));
assertEquals("This is a test Entity with annotation in fields", ((MyEntity) processInstance.getVariable("y")).getTest());
assertEquals("This is a test Entity with annotations in methods", ((MyEntityMethods) processInstance.getVariable("m")).getTest());
assertEquals("This is a test Entity with annotations in fields and without accesors methods", ((MyEntityOnlyFields) processInstance.getVariable("f")).test);
assertEquals("This is a test SerializableObject", ((MyVariableSerializable) processInstance.getVariable("z")).getText());
assertEquals("Some changed String", processInstance.getVariable("a"));
assertEquals("This is a changed test Entity", ((MyEntity) processInstance.getVariable("b")).getTest());
assertEquals("This is a changed test SerializableObject", ((MyVariableSerializable) processInstance.getVariable("c")).getText());
logger.debug("### Completing third work item ###");
ksession.getWorkItemManager().completeWorkItem(workItem.getId(), null);
workItem = handler.getWorkItem();
assertNull(workItem);
ksession = reloadSession( ksession, kbase, env );
processInstance = (WorkflowProcessInstance)
ksession.getProcessInstance(processInstanceId);
assertNull(processInstance);
}
private List<?> getEntitiesFromCache(Cache<String, Object> cache, String keyPrefix, Class<?> clazz) {
List<Object> retval = new ArrayList<Object>();
for (Map.Entry<String, Object> entry : cache.entrySet()) {
if (entry.getKey().startsWith(keyPrefix) && clazz.isInstance(entry.getValue())) {
retval.add(cache.get(entry.getValue()));
}
}
return retval;
}
private static long MYENTITYMETHODS_KEY = 1;
private static long MYENTITYONLYFIELDS_KEY = 1;
private static long MYENTITY_KEY = 1;
private static final Object syncObject = new Object();
private String generateId(Cache<String, Object> cache, MyEntityOnlyFields myEntityOnlyFields) {
synchronized (syncObject) {
while (cache.containsKey("myEntityOnlyFields" + MYENTITYONLYFIELDS_KEY)) {
MYENTITYONLYFIELDS_KEY++;
}
myEntityOnlyFields.id = MYENTITYONLYFIELDS_KEY;
}
return "myEntityOnlyFields" + MYENTITYONLYFIELDS_KEY;
}
private String generateId(Cache<String, Object> cache, MyEntityMethods myEntityMethods) {
synchronized (syncObject) {
while (cache.containsKey("myEntityMethods" + MYENTITYMETHODS_KEY)) {
MYENTITYMETHODS_KEY++;
}
myEntityMethods.setId(MYENTITYMETHODS_KEY);
}
return "myEntityMethods" + MYENTITYMETHODS_KEY;
}
private String generateId(Cache<String, Object> cache, MyEntity myEntity) {
synchronized (syncObject) {
while (cache.containsKey("myEntity" + MYENTITY_KEY)) {
MYENTITY_KEY++;
}
myEntity.setId(MYENTITY_KEY);
}
return "myEntity" + MYENTITY_KEY;
}
@Test
public void testPersistenceVariablesWithTypeChange() throws NamingException, NotSupportedException, SystemException, RollbackException, HeuristicMixedException, HeuristicRollbackException {
MyEntity myEntity = new MyEntity("This is a test Entity with annotation in fields");
MyEntityMethods myEntityMethods = new MyEntityMethods("This is a test Entity with annotations in methods");
MyEntityOnlyFields myEntityOnlyFields = new MyEntityOnlyFields("This is a test Entity with annotations in fields and without accesors methods");
MyVariableSerializable myVariableSerializable = new MyVariableSerializable("This is a test SerializableObject");
Cache<String, Object> cache = cm.getCache("jbpm-configured-cache");
UserTransaction utx = (UserTransaction) cache.getAdvancedCache().getTransactionManager();
int s = utx.getStatus();
if( utx.getStatus() == Status.STATUS_NO_TRANSACTION ) {
utx.begin();
}
cache.put(generateId(cache, myEntity), myEntity);
cache.put(generateId(cache, myEntityMethods), myEntityMethods);
cache.put(generateId(cache, myEntityOnlyFields), myEntityOnlyFields);
if( utx.getStatus() == Status.STATUS_ACTIVE ) {
utx.commit();
}
Environment env = createEnvironment();
KnowledgeBase kbase = createKnowledgeBase( "VariablePersistenceStrategyProcessTypeChange.rf" );
StatefulKnowledgeSession ksession = createSession( kbase, env );
Map<String, Object> parameters = new HashMap<String, Object>();
parameters.put("x", "SomeString");
parameters.put("y", myEntity);
parameters.put("m", myEntityMethods );
parameters.put("f", myEntityOnlyFields);
parameters.put("z", myVariableSerializable);
long processInstanceId = ksession.startProcess( "com.sample.ruleflow", parameters ).getId();
TestWorkItemHandler handler = TestWorkItemHandler.getInstance();
WorkItem workItem = handler.getWorkItem();
assertNotNull( workItem );
ksession = reloadSession( ksession, kbase, env );
ProcessInstance processInstance = ksession.getProcessInstance( processInstanceId );
assertNotNull( processInstance );
ksession.getWorkItemManager().completeWorkItem( workItem.getId(), null );
workItem = handler.getWorkItem();
assertNotNull( workItem );
ksession = reloadSession( ksession, kbase, env );
processInstance = ksession.getProcessInstance( processInstanceId );
assertNotNull( processInstance );
ksession.getWorkItemManager().completeWorkItem( workItem.getId(), null );
workItem = handler.getWorkItem();
assertNull( workItem );
ksession = reloadSession( ksession, kbase, env );
processInstance = ksession.getProcessInstance( processInstanceId );
assertNull( processInstance );
}
@Test
public void testPersistenceVariablesSubProcess() throws NamingException, NotSupportedException, SystemException, RollbackException, HeuristicMixedException, HeuristicRollbackException {
MyEntity myEntity = new MyEntity("This is a test Entity with annotation in fields");
MyEntityMethods myEntityMethods = new MyEntityMethods("This is a test Entity with annotations in methods");
MyEntityOnlyFields myEntityOnlyFields = new MyEntityOnlyFields("This is a test Entity with annotations in fields and without accesors methods");
MyVariableSerializable myVariableSerializable = new MyVariableSerializable("This is a test SerializableObject");
Cache<String, Object> cache = cm.getCache("jbpm-configured-cache");
UserTransaction utx = (UserTransaction) cache.getAdvancedCache().getTransactionManager();
utx.begin();
cache.put(generateId(cache, myEntity), myEntity);
cache.put(generateId(cache, myEntityMethods), myEntityMethods);
cache.put(generateId(cache, myEntityOnlyFields), myEntityOnlyFields);
utx.commit();
Environment env = createEnvironment();
KnowledgeBase kbase = createKnowledgeBase( "VariablePersistenceStrategySubProcess.rf" );
StatefulKnowledgeSession ksession = createSession( kbase, env );
Map<String, Object> parameters = new HashMap<String, Object>();
parameters.put("x", "SomeString");
parameters.put("y", myEntity);
parameters.put("m", myEntityMethods);
parameters.put("f", myEntityOnlyFields);
parameters.put("z", myVariableSerializable);
long processInstanceId = ksession.startProcess( "com.sample.ruleflow", parameters ).getId();
TestWorkItemHandler handler = TestWorkItemHandler.getInstance();
WorkItem workItem = handler.getWorkItem();
assertNotNull( workItem );
ksession = reloadSession( ksession, kbase, env );
ProcessInstance processInstance = ksession.getProcessInstance( processInstanceId );
assertNotNull( processInstance );
ksession.getWorkItemManager().completeWorkItem( workItem.getId(), null );
workItem = handler.getWorkItem();
assertNotNull( workItem );
ksession = reloadSession( ksession, kbase, env );
processInstance = ksession.getProcessInstance( processInstanceId );
assertNotNull( processInstance );
ksession.getWorkItemManager().completeWorkItem( workItem.getId(), null );
workItem = handler.getWorkItem();
assertNotNull( workItem );
ksession = reloadSession( ksession, kbase, env );
processInstance = ksession.getProcessInstance( processInstanceId );
assertNotNull( processInstance );
ksession.getWorkItemManager().completeWorkItem( workItem.getId(), null );
workItem = handler.getWorkItem();
assertNull( workItem );
ksession = reloadSession( ksession, kbase, env );
processInstance = ksession.getProcessInstance( processInstanceId );
assertNull( processInstance );
}
@Test
public void testWorkItemWithVariablePersistence() throws Exception{
MyEntity myEntity = new MyEntity("This is a test Entity");
MyVariableSerializable myVariableSerializable = new MyVariableSerializable("This is a test SerializableObject");
Cache<String, Object> cache = cm.getCache("jbpm-configured-cache");
UserTransaction utx = (UserTransaction) cache.getAdvancedCache().getTransactionManager();
utx.begin();
cache.put(generateId(cache, myEntity), myEntity);
utx.commit();
Environment env = createEnvironment();
KnowledgeBase kbase = createKnowledgeBase( "VPSProcessWithWorkItems.rf" );
StatefulKnowledgeSession ksession = createSession( kbase , env);
logger.debug("### Starting process ###");
Map<String, Object> parameters = new HashMap<String, Object>();
parameters.put("x", "SomeString");
parameters.put("y", myEntity);
parameters.put("z", myVariableSerializable);
long processInstanceId = ksession.startProcess( "com.sample.ruleflow", parameters ).getId();
TestWorkItemHandler handler = TestWorkItemHandler.getInstance();
WorkItem workItem = handler.getWorkItem();
assertNotNull( workItem );
logger.debug("### Retrieving process instance ###");
ksession = reloadSession( ksession, kbase , env);
WorkflowProcessInstance processInstance = (WorkflowProcessInstance)
ksession.getProcessInstance( processInstanceId );
assertNotNull( processInstance );
assertEquals("SomeString", processInstance.getVariable("x"));
assertEquals("This is a test Entity", ((MyEntity) processInstance.getVariable("y")).getTest());
assertEquals("This is a test SerializableObject", ((MyVariableSerializable) processInstance.getVariable("z")).getText());
assertNull(processInstance.getVariable("a"));
assertNull(processInstance.getVariable("b"));
assertNull(processInstance.getVariable("c"));
logger.debug("### Completing first work item ###");
Map<String, Object> results = new HashMap<String, Object>();
results.put("zeta", processInstance.getVariable("z"));
results.put("equis", processInstance.getVariable("x")+"->modifiedResult");
ksession.getWorkItemManager().completeWorkItem( workItem.getId(), results );
workItem = handler.getWorkItem();
assertNotNull( workItem );
logger.debug("### Retrieving process instance ###");
ksession = reloadSession( ksession, kbase, env );
processInstance = (WorkflowProcessInstance)
ksession.getProcessInstance(processInstanceId);
assertNotNull(processInstance);
logger.debug("######## Getting the already Persisted Variables #########");
assertEquals("SomeString->modifiedResult", processInstance.getVariable("x"));
assertEquals("This is a test Entity", ((MyEntity) processInstance.getVariable("y")).getTest());
assertEquals("This is a test SerializableObject", ((MyVariableSerializable) processInstance.getVariable("z")).getText());
assertEquals("Some new String", processInstance.getVariable("a"));
assertEquals("This is a new test Entity", ((MyEntity) processInstance.getVariable("b")).getTest());
assertEquals("This is a new test SerializableObject", ((MyVariableSerializable) processInstance.getVariable("c")).getText());
logger.debug("### Completing second work item ###");
results = new HashMap<String, Object>();
results.put("zeta", processInstance.getVariable("z"));
results.put("equis", processInstance.getVariable("x"));
ksession.getWorkItemManager().completeWorkItem( workItem.getId(), results );
workItem = handler.getWorkItem();
assertNotNull(workItem);
logger.debug("### Retrieving process instance ###");
ksession = reloadSession( ksession, kbase, env );
processInstance = (WorkflowProcessInstance)
ksession.getProcessInstance(processInstanceId);
assertNotNull(processInstance);
assertEquals("SomeString->modifiedResult", processInstance.getVariable("x"));
assertEquals("This is a test Entity", ((MyEntity) processInstance.getVariable("y")).getTest());
assertEquals("This is a test SerializableObject", ((MyVariableSerializable) processInstance.getVariable("z")).getText());
assertEquals("Some changed String", processInstance.getVariable("a"));
assertEquals("This is a changed test Entity", ((MyEntity) processInstance.getVariable("b")).getTest());
assertEquals("This is a changed test SerializableObject", ((MyVariableSerializable) processInstance.getVariable("c")).getText());
logger.debug("### Completing third work item ###");
results = new HashMap<String, Object>();
results.put("zeta", processInstance.getVariable("z"));
results.put("equis", processInstance.getVariable("x"));
ksession.getWorkItemManager().completeWorkItem( workItem.getId(), results );
workItem = handler.getWorkItem();
assertNull(workItem);
ksession = reloadSession( ksession, kbase, env );
processInstance = (WorkflowProcessInstance)
ksession.getProcessInstance(processInstanceId);
assertNull(processInstance);
}
@Test
public void testEntityWithSuperClassAnnotationField() throws Exception {
MySubEntity subEntity = new MySubEntity();
subEntity.setId(3L);
assertEquals(3L, InfinispanPlaceholderResolverStrategy.getClassIdValue(subEntity));
}
@Test
public void testEntityWithSuperClassAnnotationMethod() throws Exception {
MySubEntityMethods subEntity = new MySubEntityMethods();
subEntity.setId(3L);
assertEquals(3L, InfinispanPlaceholderResolverStrategy.getClassIdValue(subEntity));
}
@Test
public void testAbortWorkItemWithVariablePersistence() throws Exception{
MyEntity myEntity = new MyEntity("This is a test Entity");
MyVariableSerializable myVariableSerializable = new MyVariableSerializable("This is a test SerializableObject");
Cache<String, Object> cache = cm.getCache("jbpm-configured-cache");
UserTransaction utx = (UserTransaction) cache.getAdvancedCache().getTransactionManager();
utx.begin();
cache.put(generateId(cache, myEntity), myEntity);
utx.commit();
Environment env = createEnvironment();
KnowledgeBase kbase = createKnowledgeBase( "VPSProcessWithWorkItems.rf" );
StatefulKnowledgeSession ksession = createSession( kbase , env);
logger.debug("### Starting process ###");
Map<String, Object> parameters = new HashMap<String, Object>();
parameters.put("x", "SomeString");
parameters.put("y", myEntity);
parameters.put("z", myVariableSerializable);
long processInstanceId = ksession.startProcess( "com.sample.ruleflow", parameters ).getId();
TestWorkItemHandler handler = TestWorkItemHandler.getInstance();
WorkItem workItem = handler.getWorkItem();
assertNotNull( workItem );
logger.debug("### Retrieving process instance ###");
ksession = reloadSession( ksession, kbase , env);
WorkflowProcessInstance processInstance = (WorkflowProcessInstance)
ksession.getProcessInstance( processInstanceId );
assertNotNull( processInstance );
assertEquals("SomeString", processInstance.getVariable("x"));
assertEquals("This is a test Entity", ((MyEntity) processInstance.getVariable("y")).getTest());
assertEquals("This is a test SerializableObject", ((MyVariableSerializable) processInstance.getVariable("z")).getText());
assertNull(processInstance.getVariable("a"));
assertNull(processInstance.getVariable("b"));
assertNull(processInstance.getVariable("c"));
logger.debug("### Completing first work item ###");
Map<String, Object> results = new HashMap<String, Object>();
results.put("zeta", processInstance.getVariable("z"));
results.put("equis", processInstance.getVariable("x")+"->modifiedResult");
ksession.getWorkItemManager().completeWorkItem(workItem.getId(), results);
workItem = handler.getWorkItem();
assertNotNull( workItem );
// we simulate a failure here, aborting the work item
ksession.getWorkItemManager().abortWorkItem( workItem.getId() );
workItem = handler.getWorkItem();
assertNotNull( workItem );
logger.debug("### Retrieving process instance ###");
ksession = reloadSession( ksession, kbase, env );
processInstance = (WorkflowProcessInstance)
ksession.getProcessInstance(processInstanceId);
assertNotNull(processInstance);
logger.debug("######## Getting the already Persisted Variables #########");
// we expect the variables to be unmodifed
assertEquals("SomeString->modifiedResult", processInstance.getVariable("x"));
assertEquals("This is a test Entity", ((MyEntity) processInstance.getVariable("y")).getTest());
assertEquals("This is a test SerializableObject", ((MyVariableSerializable) processInstance.getVariable("z")).getText());
assertEquals("Some changed String", processInstance.getVariable("a"));
assertEquals("This is a changed test Entity", ((MyEntity) processInstance.getVariable("b")).getTest());
assertEquals("This is a changed test SerializableObject", ((MyVariableSerializable) processInstance.getVariable("c")).getText());
}
private StatefulKnowledgeSession createSession(KnowledgeBase kbase, Environment env){
return InfinispanKnowledgeService.newStatefulKnowledgeSession( kbase, null, env );
}
private StatefulKnowledgeSession reloadSession(StatefulKnowledgeSession ksession, KnowledgeBase kbase, Environment env){
int sessionId = ksession.getId();
ksession.dispose();
return InfinispanKnowledgeService.loadStatefulKnowledgeSession( sessionId, kbase, null, env);
}
private KnowledgeBase createKnowledgeBase(String flowFile) {
KnowledgeBuilderConfiguration conf = KnowledgeBuilderFactory.newKnowledgeBuilderConfiguration();
conf.setProperty("drools.dialect.java.compiler", "JANINO");
KnowledgeBuilder kbuilder = KnowledgeBuilderFactory.newKnowledgeBuilder(conf);
kbuilder.add( new ClassPathResource( flowFile ), ResourceType.DRF );
if(kbuilder.hasErrors()){
StringBuilder errorMessage = new StringBuilder();
for (KnowledgeBuilderError error: kbuilder.getErrors()) {
errorMessage.append( error.getMessage() );
errorMessage.append( System.getProperty( "line.separator" ) );
}
fail( errorMessage.toString());
}
KnowledgeBase kbase = KnowledgeBaseFactory.newKnowledgeBase();
kbase.addKnowledgePackages( kbuilder.getKnowledgePackages() );
return kbase;
}
private Environment createEnvironment() {
Environment env = PersistenceUtil.createEnvironment(context);
env.set(EnvironmentName.OBJECT_MARSHALLING_STRATEGIES, new ObjectMarshallingStrategy[]{
new InfinispanPlaceholderResolverStrategy(env),
new SerializablePlaceholderResolverStrategy( ClassObjectMarshallingStrategyAcceptor.DEFAULT )
});
return env;
}
private void connect(Node sourceNode,
Node targetNode) {
new ConnectionImpl (sourceNode, Node.CONNECTION_DEFAULT_TYPE,
targetNode, Node.CONNECTION_DEFAULT_TYPE);
}
}