/*
* 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.taskmgmt.exe;
import java.util.Iterator;
import java.util.List;
import org.jbpm.db.AbstractDbTestCase;
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.exe.Timer;
import org.jbpm.scheduler.impl.SchedulerThread;
public class TaskTimerExecutionDbTest extends AbstractDbTestCase {
static int counter = 0;
public static class PlusPlus implements ActionHandler {
private static final long serialVersionUID = 1L;
public void execute(ExecutionContext executionContext) throws Exception {
counter++;
}
}
public void setUp() throws Exception {
super.setUp();
counter = 0;
}
public void testTimerCreation() {
ProcessDefinition processDefinition = ProcessDefinition.parseXmlString(
"<process-definition>" +
" <start-state>" +
" <transition to='a' />" +
" </start-state>" +
" <task-node name='a'>" +
" <task name='clean ceiling'>" +
" <timer name='ceiling-timer' duedate='0 seconds'>" +
" <action class='org.jbpm.taskmgmt.exe.TaskTimerExecutionDbTest$PlusPlus' />" +
" </timer>" +
" </task>" +
" </task-node>" +
"</process-definition>"
);
processDefinition = saveAndReload(processDefinition);
ProcessInstance processInstance = new ProcessInstance(processDefinition);
processInstance.signal();
processInstance = saveAndReload(processInstance);
Iterator timersIter = schedulerSession.findTimersByDueDate();
assertTrue(timersIter.hasNext());
Timer timer = (Timer) timersIter.next();
assertEquals("ceiling-timer", timer.getName());
}
public void testTimerDeletion() {
ProcessDefinition processDefinition = ProcessDefinition.parseXmlString(
"<process-definition>" +
" <start-state>" +
" <transition to='a' />" +
" </start-state>" +
" <task-node name='a'>" +
" <task name='clean ceiling'>" +
" <timer name='ceiling-timer' duedate='0 seconds'>" +
" <action class='org.jbpm.taskmgmt.exe.TaskTimerExecutionDbTest$PlusPlus' />" +
" </timer>" +
" </task>" +
" <transition to='b' />" +
" </task-node>" +
" <state name='b' />" +
"</process-definition>"
);
processDefinition = saveAndReload(processDefinition);
ProcessInstance processInstance = new ProcessInstance(processDefinition);
processInstance.signal();
jbpmContext.save(processInstance);
newTransaction();
List taskInstances = taskMgmtSession.findTaskInstancesByToken(processInstance.getRootToken().getId());
assertEquals(1, taskInstances.size() );
TaskInstance taskInstance = (TaskInstance) taskInstances.get(0);
taskInstance.end();
jbpmContext.save(taskInstance.getTaskMgmtInstance().getProcessInstance());
newTransaction();
Iterator timersIter = schedulerSession.findTimersByDueDate();
assertFalse(timersIter.hasNext());
}
public void testTimerExecution() {
ProcessDefinition processDefinition = ProcessDefinition.parseXmlString(
"<process-definition>" +
" <start-state>" +
" <transition to='a' />" +
" </start-state>" +
" <task-node name='a'>" +
" <task name='clean ceiling'>" +
" <timer name='ceiling-timer' duedate='0 seconds'>" +
" <action class='org.jbpm.taskmgmt.exe.TaskTimerExecutionDbTest$PlusPlus' />" +
" </timer>" +
" </task>" +
" </task-node>" +
"</process-definition>"
);
processDefinition = saveAndReload(processDefinition);
ProcessInstance processInstance = new ProcessInstance(processDefinition);
processInstance.signal();
jbpmContext.save(processInstance);
newTransaction();
// the timer executor creates its own JbpmSession.
assertEquals(0, counter);
new SchedulerThread().executeTimers();
assertEquals(1, counter);
}
public void testTimerExecutionRepeat() {
ProcessDefinition processDefinition = ProcessDefinition.parseXmlString(
"<process-definition>" +
" <start-state>" +
" <transition to='a' />" +
" </start-state>" +
" <task-node name='a'>" +
" <task name='clean ceiling'>" +
" <timer name='ceiling-timer' duedate='0 seconds' repeat='10 second'>" +
" <action class='org.jbpm.taskmgmt.exe.TaskTimerExecutionDbTest$PlusPlus' />" +
" </timer>" +
" </task>" +
" </task-node>" +
"</process-definition>"
);
processDefinition = saveAndReload(processDefinition);
ProcessInstance processInstance = new ProcessInstance(processDefinition);
processInstance.signal();
jbpmContext.save(processInstance);
newTransaction();
// fetch the original duedate
Iterator timersIter = schedulerSession.findTimersByDueDate();
assertTrue(timersIter.hasNext());
Timer timer = (Timer) timersIter.next();
assertFalse(timersIter.hasNext());
long originalDueDate = timer.getDueDate().getTime();
commitAndCloseSession();
// the timer executor creates its own JbpmSession.
assertEquals(0, counter);
new SchedulerThread().executeTimers();
assertEquals(1, counter);
// begin another transaction
beginSessionTransaction();
// check if the timer has be re-scheduled because of the repeat.
timersIter = schedulerSession.findTimersByDueDate();
assertTrue(timersIter.hasNext());
timer = (Timer) timersIter.next();
assertFalse(timersIter.hasNext());
// check that the timer was rescheduled with a duedate 1 second after the original duedate.
assertEquals(originalDueDate+10000, timer.getDueDate().getTime());
}
}