commentdbtest.java

jbpm demo 是一款非常不错的开源工作流,简单易用,适合扩张开发!

/*
 * 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.graph.exe;

import java.util.ArrayList;
import java.util.Collection;
import java.util.List;

import org.jbpm.db.AbstractDbTestCase;
import org.jbpm.graph.def.ProcessDefinition;
import org.jbpm.taskmgmt.exe.TaskInstance;
import org.jbpm.taskmgmt.exe.TaskMgmtInstance;

public class CommentDbTest extends AbstractDbTestCase {
  
  public void testComments() {
    ProcessInstance processInstance = null;
    
    jbpmContext.setActorId("miketyson");
    try {
      ProcessDefinition processDefinition = new ProcessDefinition();
      graphSession.saveProcessDefinition(processDefinition);
      
      processInstance = new ProcessInstance(processDefinition);
      Token token =  processInstance.getRootToken();
      token.addComment("first");
      token.addComment("second");
      token.addComment("third");
      
      processInstance = saveAndReload(processInstance);
      
    } finally {
      jbpmContext.setActorId(null);
    }
    
    Token token = processInstance.getRootToken();
    List comments = token.getComments();
    
    assertNotNull(comments);
    assertEquals(3, comments.size());

    assertEquals("miketyson", ((Comment)comments.get(0)).getActorId());
    assertNotNull(((Comment)comments.get(0)).getTime());
    assertEquals("first", ((Comment)comments.get(0)).getMessage());
    
    assertEquals("miketyson", ((Comment)comments.get(1)).getActorId());
    assertNotNull(((Comment)comments.get(1)).getTime());
    assertEquals("second", ((Comment)comments.get(1)).getMessage());

    assertEquals("miketyson", ((Comment)comments.get(2)).getActorId());
    assertNotNull(((Comment)comments.get(2)).getTime());
    assertEquals("third", ((Comment)comments.get(2)).getMessage());
  }
  
  public void testCommentsOnDifferentTokens() {
    Token token = new Token();
    token.addComment("one");
    token.addComment("two");
    token.addComment("three");
    session.save(token);
    long firstTokenId = token.getId();
    
    token = new Token();
    token.addComment("first");
    token.addComment("second");
    token.addComment("third");
    session.save(token);
    long secondTokenId = token.getId();
    
    newTransaction();
    
    token = (Token) session.load(Token.class, new Long(firstTokenId));
    List comments = token.getComments();
    assertEquals(3, comments.size());
    assertEquals("one", ((Comment)comments.get(0)).getMessage());
    assertEquals("two", ((Comment)comments.get(1)).getMessage());
    assertEquals("three", ((Comment)comments.get(2)).getMessage());

    token = (Token) session.load(Token.class, new Long(secondTokenId));
    comments = token.getComments();
    assertEquals(3, comments.size());
    assertEquals("first", ((Comment)comments.get(0)).getMessage());
    assertEquals("second", ((Comment)comments.get(1)).getMessage());
    assertEquals("third", ((Comment)comments.get(2)).getMessage());
  }
  
  public void testTaskInstanceComment() {
    TaskInstance taskInstance = new TaskInstance();
    taskInstance.addComment("one");
    taskInstance.addComment("two");
    taskInstance.addComment("three");
    session.save(taskInstance);
    
    newTransaction();
    
    taskInstance = (TaskInstance) session.load(TaskInstance.class, new Long(taskInstance.getId()));
    List comments = taskInstance.getComments();
    assertEquals(3, comments.size());
    
    Comment comment = (Comment)comments.get(0);
    assertEquals("one", comment.getMessage());
    assertSame(taskInstance, comment.getTaskInstance());
    
    assertEquals("two", ((Comment)comments.get(1)).getMessage());
    assertEquals("three", ((Comment)comments.get(2)).getMessage());
  }

  public void testCommentToTokenAndTaskInstance() {
    ProcessDefinition processDefinition = ProcessDefinition.parseXmlString(
      "<process-definition>" +
      "  <start-state>" +
      "    <transition to='a' />" +
      "  </start-state>" +
      "  <task-node name='a'>" +
      "    <task name='clean ceiling' />" +
      "  </task-node>" +
      "</process-definition>"
    );
    graphSession.saveProcessDefinition(processDefinition);
    
    ProcessInstance processInstance = new ProcessInstance(processDefinition);
    processInstance.signal();
    
    processInstance = saveAndReload(processInstance);
    
    TaskMgmtInstance tmi = processInstance.getTaskMgmtInstance();
    TaskInstance taskInstance = (TaskInstance) tmi.getTaskInstances().iterator().next();
    taskInstance.addComment("one");
    taskInstance.addComment("two");
    taskInstance.addComment("three");
    
    processInstance = saveAndReload(processInstance);
    Token rootToken = processInstance.getRootToken();
    
    taskInstance = (TaskInstance) processInstance.getTaskMgmtInstance().getTaskInstances().iterator().next();
    assertEquals(3, taskInstance.getComments().size());
    assertEquals(3, rootToken.getComments().size());
    
    ArrayList tokenComments = new ArrayList(rootToken.getComments());
    ArrayList taskComments = new ArrayList(taskInstance.getComments());
    assertEquals(tokenComments, taskComments);
  }
  
  public void testTaskCommentAndLoadProcessInstance() {
    ProcessDefinition processDefinition = ProcessDefinition.parseXmlString(
      "<process-definition>" +
      "  <start-state>" +
      "    <transition to='a' />" +
      "  </start-state>" +
      "  <task-node name='a'>" +
      "    <task name='clean ceiling' />" +
      "    <transition to='end' />" +
      "  </task-node>" +
      "  <end-state name='end' />" +
      "</process-definition>"
    );

    processDefinition = saveAndReload(processDefinition);
    
    ProcessInstance processInstance = new ProcessInstance(processDefinition);
    processInstance.signal();
    Collection unfinishedTasks = processInstance.getTaskMgmtInstance().getUnfinishedTasks(processInstance.getRootToken());
    TaskInstance taskInstance = (TaskInstance) unfinishedTasks.iterator().next();
    taskInstance.addComment("please hurry!");
    
    processInstance = saveAndReload(processInstance);

    taskMgmtSession.loadTaskInstance(taskInstance.getId());
    graphSession.deleteProcessInstance(processInstance.getId());
  }
  
}