/*
* JBoss, Home of Professional Open Source
* Copyright 2005, JBoss Inc., and individual contributors as indicated
* by the @authors tag. See the copyright.txt in the distribution for a
* full listing of individual contributors.
*
* This is free software; you can redistribute it and/or modify it
* under the terms of the GNU Lesser General Public License as
* published by the Free Software Foundation; either version 2.1 of
* the License, or (at your option) any later version.
*
* This software is distributed in the hope that it will be useful,
* but WITHOUT ANY WARRANTY; without even the implied warranty of
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
* Lesser General Public License for more details.
*
* You should have received a copy of the GNU Lesser General Public
* License along with this software; if not, write to the Free
* Software Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA
* 02110-1301 USA, or see the FSF site: http://www.fsf.org.
*/
package org.jbpm.scheduler.exe;
import java.text.DateFormat;
import java.text.SimpleDateFormat;
import java.util.Calendar;
import java.util.Date;
import java.util.GregorianCalendar;
import java.util.Iterator;
import java.util.List;
import org.jbpm.context.exe.ContextInstance;
import org.jbpm.db.AbstractDbTestCase;
import org.jbpm.graph.def.Action;
import org.jbpm.graph.def.ActionHandler;
import org.jbpm.graph.def.ProcessDefinition;
import org.jbpm.graph.exe.ExecutionContext;
import org.jbpm.graph.exe.ProcessInstance;
import org.jbpm.scheduler.impl.SchedulerThread;
import org.jbpm.taskmgmt.exe.TaskInstance;
public class TimerDbTest extends AbstractDbTestCase {
public void testSaveTimer() {
Date now = new Date();
Timer timer = new Timer();
timer.setName("timer-name");
timer.setDueDate(now);
timer.setTransitionName("transition-name");
timer.setRepeat("repeat-duration");
timer = saveAndReloadTimer(timer);
assertEquals("timer-name", timer.getName());
// we test for the same date in a simple format
DateFormat df = SimpleDateFormat.getDateInstance();
assertEquals(df.format(now), df.format(timer.getDueDate()));
// we test for each part of the date to see where we fail per database
// to help with debugging.
Calendar ncal = new GregorianCalendar();
ncal.setTime(now);
Calendar tcal = new GregorianCalendar();
tcal.setTime(timer.getDueDate());
assertEquals(ncal.get(Calendar.YEAR), tcal.get(Calendar.YEAR));
assertEquals(ncal.get(Calendar.MONTH), tcal.get(Calendar.MONTH));
assertEquals(ncal.get(Calendar.DAY_OF_MONTH), tcal.get(Calendar.DAY_OF_MONTH));
assertEquals(ncal.get(Calendar.HOUR_OF_DAY), tcal.get(Calendar.HOUR_OF_DAY));
assertEquals(ncal.get(Calendar.MINUTE), tcal.get(Calendar.MINUTE));
assertEquals(ncal.get(Calendar.SECOND), tcal.get(Calendar.SECOND));
assertEquals(ncal.get(Calendar.MILLISECOND), tcal.get(Calendar.MILLISECOND));
// we test for the actual date
assertEquals(now, timer.getDueDate());
assertEquals("transition-name", timer.getTransitionName());
assertEquals("repeat-duration", timer.getRepeat());
}
public void testTimerCreation() {
ProcessDefinition processDefinition = ProcessDefinition.parseXmlString(
"<process-definition>" +
" <start-state>" +
" <transition to='catch crooks' />" +
" </start-state>" +
" <state name='catch crooks'>" +
" <timer name='reminder' duedate='5 seconds' />" +
" <transition to='end'/>" +
" </state>" +
" <end-state name='end'/>" +
"</process-definition>"
);
graphSession.saveProcessDefinition(processDefinition);
ProcessInstance processInstance = new ProcessInstance(processDefinition);
jbpmContext.save(processInstance);
long before = System.currentTimeMillis();
processInstance.signal();
long after = System.currentTimeMillis();
jbpmContext.save(processInstance);
newTransaction();
Timer timer = (Timer) schedulerSession.findTimersByDueDate().next();
assertEquals("reminder", timer.getName());
assertTrue( (before+5000) <= timer.getDueDate().getTime() );
assertTrue( timer.getDueDate().getTime() <= (after+5000) );
assertEquals("catch crooks", timer.getGraphElement().getName());
}
public void testTimerCancellation() {
ProcessDefinition processDefinition = ProcessDefinition.parseXmlString(
"<process-definition>" +
" <start-state>" +
" <transition to='catch crooks' />" +
" </start-state>" +
" <state name='catch crooks'>" +
" <timer name='reminder' duedate='5 seconds' />" +
" <transition to='end'/>" +
" </state>" +
" <end-state name='end'/>" +
"</process-definition>"
);
graphSession.saveProcessDefinition(processDefinition);
ProcessInstance processInstance = new ProcessInstance(processDefinition);
jbpmContext.save(processInstance);
processInstance.signal();
processInstance.signal();
jbpmContext.save(processInstance);
newTransaction();
assertFalse(schedulerSession.findTimersByDueDate().hasNext());
}
public void testTimerAction() {
ProcessDefinition processDefinition = ProcessDefinition.parseXmlString(
"<process-definition name='process'>" +
" <start-state>" +
" <transition to='sometask' />" +
" </start-state>" +
" <task-node name='sometask'>" +
" <timer name='reminder'" +
" duedate='1 business minutes'" +
" repeat='1 business minutes'" +
" transition='time-out-transition' >" +
" <action class='my-action-handler-class-name' />" +
" </timer>" +
" <task name='do something'/>" +
" <transition name='time-out-transition' to='sometask' />" +
" </task-node>" +
"</process-definition>"
);
graphSession.saveProcessDefinition(processDefinition);
ProcessInstance processInstance = new ProcessInstance(processDefinition);
jbpmContext.save(processInstance);
newTransaction();
processInstance = graphSession.loadProcessInstance(processInstance.getId());
processInstance.signal();
processDefinition = processInstance.getProcessDefinition();
Iterator iter = processDefinition.getNode("sometask").getEvent("node-enter").getActions().iterator();
while (iter.hasNext()) {
Action action = (Action) iter.next();
action.getId();
}
jbpmContext.save(processInstance);
newTransaction();
Timer timer = (Timer) schedulerSession.findTimersByDueDate().next();
assertNotNull(timer);
}
public Timer saveAndReloadTimer(Timer timer) {
schedulerSession.saveTimer(timer);
newTransaction();
timer = (Timer) session.load(Timer.class, new Long(timer.getId()));
return timer;
}
public void testTaskTimerExecution() {
ProcessDefinition processDefinition = ProcessDefinition.parseXmlString(
"<process-definition>" +
" <start-state>" +
" <transition to='timed task' />" +
" </start-state>" +
" <task-node name='timed task'>" +
" <task>" +
" <timer duedate='23 business seconds'>" +
" <action class='geftem-eu-shuppe-oender-ze-konte'/>" +
" </timer>" +
" </task>" +
" </task-node>" +
"</process-definition>"
);
processDefinition = saveAndReload(processDefinition);
ProcessInstance processInstance = new ProcessInstance(processDefinition);
processInstance.signal();
processInstance = saveAndReload(processInstance);
assertTrue(schedulerSession.findTimersByDueDate().hasNext());
}
public void testTimerCancellationAtProcessEnd() {
ProcessDefinition processDefinition = ProcessDefinition.parseXmlString(
"<process-definition>" +
" <start-state>" +
" <transition to='s' />" +
" </start-state>" +
" <state name='s'>" +
" <event type='node-enter'>" +
" <create-timer duedate='26 business seconds'>" +
" <action class='claim.you.are.Innocent' />" +
" </create-timer>" +
" </event>" +
" <transition to='end' />" +
" </state>" +
" <end-state name='end' />" +
"</process-definition>"
);
processDefinition = saveAndReload(processDefinition);
ProcessInstance processInstance = new ProcessInstance(processDefinition);
processInstance.signal();
processInstance = saveAndReload(processInstance);
assertTrue(schedulerSession.findTimersByDueDate().hasNext());
processInstance.signal();
processInstance = saveAndReload(processInstance);
assertFalse(schedulerSession.findTimersByDueDate().hasNext());
}
public void testFindTimersByName() {
ProcessDefinition processDefinition = ProcessDefinition.parseXmlString(
"<process-definition>" +
" <start-state>" +
" <transition to='timed task' />" +
" </start-state>" +
" <task-node name='timed task'>" +
" <task name='find the hole in the market'>" +
" <timer name='reminder' duedate='23 business seconds'>" +
" <action class='geftem-eu-shuppe-oender-ze-konte'/>" +
" </timer>" +
" </task>" +
" </task-node>" +
"</process-definition>"
);
processDefinition = saveAndReload(processDefinition);
ProcessInstance processInstance = new ProcessInstance(processDefinition);
processInstance.signal();
processInstance = saveAndReload(processInstance);
List timersByName = schedulerSession.findTimersByName("reminder", processInstance.getRootToken());
assertNotNull(timersByName);
assertEquals(1, timersByName.size());
Timer timer = (Timer) timersByName.get(0);
assertEquals("geftem-eu-shuppe-oender-ze-konte", timer.getAction().getActionDelegation().getClassName());
}
public static class NoOp implements ActionHandler {
private static final long serialVersionUID = 1L;
public void execute(ExecutionContext executionContext) throws Exception {
}
}
public void testTimerRepeat() {
ProcessDefinition processDefinition = ProcessDefinition.parseXmlString(
"<process-definition>" +
" <start-state>" +
" <transition to='a' />" +
" </start-state>" +
" <state name='a'>" +
" <timer name='reminder' " +
" duedate='0 seconds'" +
" repeat='5 seconds' >" +
" <action class='org.jbpm.scheduler.exe.TimerDbTest$NoOp' />" +
" </timer>" +
" <transition to='b'/>" +
" <transition name='back' to='a'/>" +
" </state>" +
" <state name='b'/>" +
"</process-definition>"
);
processDefinition = saveAndReload(processDefinition);
ProcessInstance processInstance = new ProcessInstance(processDefinition);
long before = System.currentTimeMillis();
processInstance.signal();
long after = System.currentTimeMillis();
jbpmContext.save(processInstance);
newTransaction();
Iterator iter = schedulerSession.findTimersByDueDate();
assertTrue(iter.hasNext());
Timer timer = (Timer) iter.next();
assertNotNull(timer);
Date date = timer.getDueDate();
assertNotNull(date);
assertTrue(before <= date.getTime());
assertTrue(date.getTime() <= after);
long origDueDate = date.getTime();
assertFalse(iter.hasNext());
commitAndCloseSession();
SchedulerThread schedulerThread = new SchedulerThread(jbpmConfiguration);
schedulerThread.executeTimers();
beginSessionTransaction();
iter = schedulerSession.findTimersByDueDate();
assertTrue(iter.hasNext());
timer = (Timer) iter.next();
assertFalse(iter.hasNext());
assertEquals(origDueDate+5000, timer.getDueDate().getTime());
processInstance = jbpmContext.loadProcessInstance(processInstance.getId());
before = System.currentTimeMillis();
processInstance.signal("back");
after = System.currentTimeMillis();
jbpmContext.save(processInstance);
newTransaction();
iter = schedulerSession.findTimersByDueDate();
assertTrue(iter.hasNext());
timer = (Timer) iter.next();
assertNotNull(timer);
date = timer.getDueDate();
assertNotNull(date);
assertTrue(before <= date.getTime());
assertTrue(date.getTime() <= after);
assertFalse(iter.hasNext());
newTransaction();
processInstance = jbpmContext.loadProcessInstance(processInstance.getId());
processInstance.signal();
jbpmContext.save(processInstance);
newTransaction();
iter = schedulerSession.findTimersByDueDate();
assertFalse(iter.hasNext());
}
public static class UpdateVariables implements ActionHandler {
private static final long serialVersionUID = 1L;
public void execute(ExecutionContext executionContext) throws Exception {
executionContext.setVariable("a", "value a updated");
executionContext.setVariable("b", "value b updated");
}
}
public void testTimerUpdatingProcessVariables() {
ProcessDefinition processDefinition = ProcessDefinition.parseXmlString(
"<process-definition>" +
" <start-state>" +
" <transition to='a' />" +
" </start-state>" +
" <task-node name='a'>" +
" <task name='wait for var updates'>" +
" <controller>" +
// variable a will be a task instance local variable
// variable b will be a process instance variable
" <variable name='a' />" +
" </controller>" +
" <timer name='update variables' " +
" duedate='0 seconds'" +
" repeat='5 seconds' >" +
" <action class='org.jbpm.scheduler.exe.TimerDbTest$UpdateVariables' />" +
" </timer>" +
" </task>" +
" </task-node>" +
"</process-definition>"
);
processDefinition = saveAndReload(processDefinition);
ProcessInstance processInstance = new ProcessInstance(processDefinition);
ContextInstance contextInstance = processInstance.getContextInstance();
contextInstance.setVariable("a", "value a");
contextInstance.setVariable("b", "value b");
processInstance.signal();
jbpmContext.save(processInstance);
commitAndCloseSession();
SchedulerThread schedulerThread = new SchedulerThread(jbpmConfiguration);
schedulerThread.executeTimers();
beginSessionTransaction();
processInstance = jbpmContext.loadProcessInstance(processInstance.getId());
contextInstance = processInstance.getContextInstance();
assertEquals("value a", contextInstance.getVariable("a") );
assertEquals("value b updated", contextInstance.getVariable("b") );
TaskInstance taskInstance = (TaskInstance) processInstance.getTaskMgmtInstance().getTaskInstances().iterator().next();
assertEquals("value a updated", taskInstance.getVariable("a") );
assertEquals("value b updated", taskInstance.getVariable("b") );
}
public static class ConcurrentUpdateAction implements ActionHandler {
private static final long serialVersionUID = 1L;
public void execute(ExecutionContext executionContext) throws Exception {
executionContext.setVariable("a", "value a timer actioned updated");
}
}
}