scripttest.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.action;

import java.util.HashMap;
import java.util.HashSet;
import java.util.Map;
import java.util.Set;

import junit.framework.TestCase;

import org.jbpm.context.def.ContextDefinition;
import org.jbpm.context.def.VariableAccess;
import org.jbpm.context.exe.ContextInstance;
import org.jbpm.graph.def.ProcessDefinition;
import org.jbpm.graph.exe.ExecutionContext;
import org.jbpm.graph.exe.ProcessInstance;
import org.jbpm.graph.exe.Token;

public class ScriptTest extends TestCase {

  public void testActionScript() {
    ProcessDefinition processDefinition = ProcessDefinition.parseXmlString( 
      "<process-definition>" +
      "  <start-state>" +
      "    <transition to='a' />" +
      "  </start-state>" +
      "  <state name='a'>" +
      "    <event type='node-enter'>" +
      "      <script>" +
      "        <variable name='a' access='write' />" +
      "        <expression>" +
      "          a = b + c;" +
      "        </expression>" +
      "      </script>" +
      "    </event>" +
      "  </state>" +
      "</process-definition>"
    );
    processDefinition.addDefinition(new ContextDefinition());

    ProcessInstance processInstance = new ProcessInstance(processDefinition);
    ContextInstance contextInstance = (ContextInstance) processInstance.getInstance(ContextInstance.class);
    contextInstance.setVariable("b", new Integer(3));
    contextInstance.setVariable("c", new Integer(9));
    processInstance.signal();

    assertEquals(new Integer(12), contextInstance.getVariable("a"));
  }

  public void testNodeScript() {
    ProcessDefinition processDefinition = ProcessDefinition.parseXmlString( 
      "<process-definition>" +
      "  <start-state>" +
      "    <transition to='a' />" +
      "  </start-state>" +
      "  <node name='a'>" +
      "    <script>" +
      "      executionContext.leaveNode(\"c\");" +
      "    </script>" +
      "    <transition name='b' to='b' />" +
      "    <transition name='c' to='c' />" +
      "  </node>" +
      "  <state name='b' />" +
      "  <state name='c' />" +
      "</process-definition>"
    );
    
    ProcessInstance processInstance = new ProcessInstance(processDefinition);
    processInstance.signal();
    assertSame(processDefinition.getNode("c"), processInstance.getRootToken().getNode());
  }

  public void testNullValues() {
    ProcessDefinition processDefinition = ProcessDefinition.parseXmlString( 
      "<process-definition>" +
      "  <start-state>" +
      "    <transition to='a'>" +
      "      <script>" +
      "        <expression>" +
      "        if (a==null) {" +
      "          b = null;" +
      "        }" +
      "        </expression>" +
      "        <variable name='a' access='write' />" +
      "        <variable name='b' access='read' />" +
      "      </script>" +
      "    </transition>" +
      "  </start-state>" +
      "  <state name='a' />" +
      "</process-definition>"
    );
    
    ProcessInstance processInstance = new ProcessInstance(processDefinition);
    processInstance.getContextInstance().setVariable("a", null);
    processInstance.signal();
    assertNull(processInstance.getContextInstance().getVariable("b"));
  }

  public void testScriptEvaluation() throws Exception {
    Map inputMap = new HashMap();
    inputMap.put("a", new Integer(1));
    inputMap.put("b", new Integer(1));
    
    Set outputNames = new HashSet();
    outputNames.add("c");
    outputNames.add("d");

    Script script = new Script();
    script.setExpression(
      "c = a + b;" +
      "d = a + b + c"
    );
    Map outputMap = script.eval(inputMap, outputNames);
    
    assertEquals(2, outputMap.size());
    assertEquals(new Integer(2), outputMap.get("c"));
    assertEquals(new Integer(4), outputMap.get("d"));
  }
  
  public void testExecute() throws Exception {
    ProcessDefinition processDefinition = ProcessDefinition.createNewProcessDefinition();
    ProcessInstance processInstance = new ProcessInstance(processDefinition);
    Token token = processInstance.getRootToken();
    ExecutionContext executionContext = new ExecutionContext(token);
    ContextInstance contextInstance = executionContext.getContextInstance();
    contextInstance.setVariable("a", new Integer(1));
    contextInstance.setVariable("b", new Integer(1));
    
    Script script = new Script();
    script.addVariableAccess(new VariableAccess("a", "read,write", null));
                                                // b is READ-ONLY ! 
    script.addVariableAccess(new VariableAccess("b", "read", null));
    script.addVariableAccess(new VariableAccess("c", "write", null));
    script.setExpression(
      "if (a!=1) throw new RuntimeException(\"a is not 1\");" +
      "if (b!=1) throw new RuntimeException(\"b is not 1\");" +
      "a = 2;" +
      "b = 3;" +
      "c = 4;"
    );
    script.execute(executionContext);
    
    assertEquals(new Integer(2), contextInstance.getVariable("a"));
    // b was READ-ONLY ! 
    assertEquals(new Integer(1), contextInstance.getVariable("b"));
    assertEquals(new Integer(4), contextInstance.getVariable("c"));
  }

  
  public void testCreateInputMapWithoutContext() {
    ProcessDefinition processDefinition = ProcessDefinition.createNewProcessDefinition();
    ProcessInstance processInstance = new ProcessInstance(processDefinition);
    Token token = processInstance.getRootToken();
    
    Script script = new Script();
    ExecutionContext executionContext = new ExecutionContext(token);
    Map inputMap = script.createInputMap(executionContext);
    assertSame( executionContext, inputMap.get("executionContext"));
    assertSame( token, inputMap.get("token"));
    assertNull( inputMap.get("node"));
    assertNull( inputMap.get("task"));
    assertNull( inputMap.get("taskInstance"));
  }

  public void testCreateInputMapWithoutVariableAccesses() {
    ProcessDefinition processDefinition = ProcessDefinition.createNewProcessDefinition();
    ProcessInstance processInstance = new ProcessInstance(processDefinition);
    Token token = processInstance.getRootToken();
    ContextInstance contextInstance = processInstance.getContextInstance();
    contextInstance.setVariable("a", new Integer(1) );
    contextInstance.setVariable("b", new Integer(1) );
    contextInstance.setVariable("c", new Integer(1) );
    
    Script script = new Script();
    Map inputMap = script.createInputMap(new ExecutionContext(token));
    assertEquals(new Integer(1), inputMap.get("a"));
    assertEquals(new Integer(1), inputMap.get("b"));
    assertEquals(new Integer(1), inputMap.get("c"));
  }

  public void testCreateInputMapWithOnlyWriteVariableAccesses() {
    ProcessDefinition processDefinition = ProcessDefinition.createNewProcessDefinition();
    ProcessInstance processInstance = new ProcessInstance(processDefinition);
    Token token = processInstance.getRootToken();
    ContextInstance contextInstance = processInstance.getContextInstance();
    contextInstance.setVariable("a", new Integer(1) );
    contextInstance.setVariable("b", new Integer(1) );
    contextInstance.setVariable("c", new Integer(1) );
    
    Script script = new Script();
    script.addVariableAccess(new VariableAccess("a", "write", null));
    script.addVariableAccess(new VariableAccess("b", "write-required", null));
    Map inputMap = script.createInputMap(new ExecutionContext(token));
    assertEquals(new Integer(1), inputMap.get("a"));
    assertEquals(new Integer(1), inputMap.get("b"));
    assertEquals(new Integer(1), inputMap.get("c"));
  }

  public void testCreateInputMapWithOnlyReadVariableAccesses() {
    ProcessDefinition processDefinition = ProcessDefinition.createNewProcessDefinition();
    ProcessInstance processInstance = new ProcessInstance(processDefinition);
    Token token = processInstance.getRootToken();
    ContextInstance contextInstance = processInstance.getContextInstance();
    contextInstance.setVariable("a", new Integer(1) );
    contextInstance.setVariable("b", new Integer(1) );
    contextInstance.setVariable("c", new Integer(1) );
    
    Script script = new Script();
    script.addVariableAccess(new VariableAccess("a", "read", null));
    Map inputMap = script.createInputMap(new ExecutionContext(token));
    assertEquals(new Integer(1), inputMap.get("a"));
    assertFalse(inputMap.containsKey("b"));
    assertFalse(inputMap.containsKey("c"));
  }

  public void testMappedNameTest() throws Exception {
    ProcessDefinition processDefinition = ProcessDefinition.createNewProcessDefinition();
    ProcessInstance processInstance = new ProcessInstance(processDefinition);
    Token token = processInstance.getRootToken();
    ContextInstance contextInstance = processInstance.getContextInstance();
    contextInstance.setVariable("a", new Integer(1) );
    
    Script script = new Script();
    script.setExpression("AAA++;");
    script.addVariableAccess(new VariableAccess("a", "read-write", "AAA"));
    script.execute(new ExecutionContext(token));

    assertEquals(new Integer(2), contextInstance.getVariable("a"));
  }
}