com.devoteam.srit.xmlloader.core.operations.functions.Function.java Source code

Java tutorial

Introduction

Here is the source code for com.devoteam.srit.xmlloader.core.operations.functions.Function.java

Source

/* 
 * Copyright 2012 Devoteam http://www.devoteam.com
 * DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS FILE HEADER.
 * 
 * 
 * This file is part of Multi-Protocol Test Suite (MTS).
 * 
 * Multi-Protocol Test Suite (MTS) is free software: you can redistribute
 * it and/or modify it under the terms of the GNU General Public License
 * as published by the Free Software Foundation, either version 3 of the
 * License.
 * 
 * Multi-Protocol Test Suite (MTS) 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 General Public License for more details.
 * 
 * You should have received a copy of the GNU General Public License
 * along with Multi-Protocol Test Suite (MTS).
 * If not, see <http://www.gnu.org/licenses/>.
 * 
 */

package com.devoteam.srit.xmlloader.core.operations.functions;

import com.devoteam.srit.xmlloader.core.Parameter;
import com.devoteam.srit.xmlloader.core.Runner;
import com.devoteam.srit.xmlloader.core.Scenario;
import com.devoteam.srit.xmlloader.core.ScenarioRunner;
import com.devoteam.srit.xmlloader.core.exception.ExecutionException;
import com.devoteam.srit.xmlloader.core.operations.basic.OperationSequence;
import com.devoteam.srit.xmlloader.core.protocol.Msg;
import com.devoteam.srit.xmlloader.core.utils.Utils;
import gp.utils.arrays.Array;
import java.util.HashMap;
import java.util.Map.Entry;
import java.util.concurrent.Semaphore;
import org.dom4j.Element;

/**
 *
 * @author gpasquiers
 */
public class Function {

    private String _name;
    private long _version;
    private OperationSequence _do;
    private Element _root;

    public Function(Element root, Scenario scenario, long version) throws Exception {
        _name = root.attributeValue("name");
        _version = version;
        _root = root;

        // TODO : handle stopCount better (at executions and not parsing ?)
        _do = new OperationSequence(root.element("do"), scenario);
    }

    public long getVersion() {
        return _version;
    }

    public String getName() {
        return _name;
    }

    public HashMap<String, Parameter> execute(Runner runner, HashMap<String, Parameter> inputs) throws Exception {
        HashMap<String, Parameter> outputs = new HashMap();
        try {
            // check args (type, mandatory, create with default value if necessary ...)
            // also extract the parameters from the input hashmap and put them into the pool
            ((ScenarioRunner) runner).stackFunctionParameterPool();

            for (Object object : _root.selectNodes("./input/parameter")) {
                Element element = (Element) object;
                String inputName = element.attributeValue("name");
                String inputDefault = element.attributeValue("default");
                String inputType = element.attributeValue("type");

                if (!Parameter.matchesParameter(inputName)) {
                    throw new ExecutionException(
                            "invalid input argument name format in function call\r\n" + element.asXML(), null);
                }

                if (inputName.contains("function:")) {
                    inputName = Utils.replaceNoRegex(inputName, "function:", "");
                }

                Parameter inputParameter = inputs.get(inputName);
                if (null == inputParameter && inputDefault != null) {
                    inputParameter = new Parameter();
                    if (!"EMPTY_LIST".equals(inputDefault)) {
                        inputParameter.add(inputDefault);
                    }
                } else if (null == inputParameter && inputDefault == null) {
                    throw new ExecutionException(
                            "mandatory input argument (no default value) not passed from the function call\r\n"
                                    + element.asXML(),
                            null);
                }

                // check the type
                checkType(inputParameter, inputType);

                runner.getParameterPool().set(inputName, inputParameter);
            }

            // put the input args them into the pool
            for (Entry<String, Parameter> entry : inputs.entrySet()) {
                runner.getParameterPool().set(entry.getKey(), entry.getValue());
            }

            // execute the function
            _do.execute(runner);

            // extract the parameters from the pool and put them into the output hashmap
            for (Object object : _root.selectNodes("./output/parameter")) {
                Element element = (Element) object;
                String outputName = element.attributeValue("name");

                if (outputName.contains("function:")) {
                    outputName = Utils.replaceNoRegex(outputName, "function:", "");
                }

                if (!runner.getParameterPool().existsLocal(outputName)) {
                    throw new ExecutionException(
                            "invalid ouput arg value (does not exists) in function def\r\n" + element.asXML(),
                            null);
                }

                outputs.put(outputName, runner.getParameterPool().get(outputName));
            }
        } finally {
            ((ScenarioRunner) runner).unstackFunctionParameterPool();
        }

        return outputs;
    }

    /* checks (asserts) the type of a parameter
     * throw an exception if it is not the expected type
     */
    private void checkType(Parameter parameter, String type) throws ExecutionException {
        for (Object o : parameter.getArray()) {
            if ("semaphore".equalsIgnoreCase(type)) {
                if (!(o instanceof Semaphore)) {
                    throw new ExecutionException("parameter content is not a semaphore\r\n" + parameter);
                }
            } else if ("message".equalsIgnoreCase(type)) {
                if (!(o instanceof Msg)) {
                    throw new ExecutionException("parameter content is not a message\r\n" + parameter);
                }
            } else {
                if ("number".equalsIgnoreCase(type)) {
                    try {
                        Double.parseDouble((String) o);
                    } catch (Exception e) {
                        throw new ExecutionException("parameter content is not a number\r\n" + parameter, e);
                    }
                } else if ("boolean".equalsIgnoreCase(type)) {
                    try {
                        Boolean.parseBoolean((String) o);
                    } catch (Exception e) {
                        throw new ExecutionException("parameter content is not a boolean\r\n" + parameter, e);
                    }
                } else if ("binary".equalsIgnoreCase(type)) {
                    try {
                        // TODO: improve; it's simple but expensive as it is
                        Array.fromHexString((String) o);
                    } catch (Exception e) {
                        throw new ExecutionException("parameter content is not a binary\r\n" + parameter, e);
                    }
                } else {
                    // do nothing, don't check the type
                }
            }
        }
    }
}