PropagateVariablesExec.java

/*
 * ProActive Parallel Suite(TM):
 * The Open Source library for parallel and distributed
 * Workflows & Scheduling, Orchestration, Cloud Automation
 * and Big Data Analysis on Enterprise Grids & Clouds.
 *
 * Copyright (c) 2007 - 2017 ActiveEon
 * Contact: contact@activeeon.com
 *
 * This library is free software: you can redistribute it and/or
 * modify it under the terms of the GNU Affero General Public License
 * as published by the Free Software Foundation: version 3 of
 * the License.
 *
 * This program 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 Affero General Public License for more details.
 *
 * You should have received a copy of the GNU Affero General Public License
 * along with this program. If not, see <http://www.gnu.org/licenses/>.
 *
 * If needed, contact us to obtain a release under GPL Version 2 or 3
 * or a different license than the AGPL.
 */
package functionaltests.executables;

import java.io.Serializable;
import java.util.Map;

import org.ow2.proactive.scheduler.common.task.TaskResult;
import org.ow2.proactive.scheduler.common.task.executable.JavaExecutable;


public class PropagateVariablesExec extends JavaExecutable {

    private Map<String, String> check;

    private Map<String, String> set;

    @Override
    public Serializable execute(TaskResult... results) throws Throwable {
        check();
        set();
        return "task_result";
    }

    private void check() throws Exception {
        if (check != null && !check.isEmpty()) {
            for (String key : check.keySet()) {
                String expected = check.get(key);
                if (expected != null) {
                    String actual = (String) getVariable(key);
                    if (actual == null || !actual.equals(expected)) {
                        throw new RuntimeException(String.format("Propagated variable error: %s, expected=%s, actual=%s",
                                                                 key,
                                                                 expected,
                                                                 actual));
                    }
                }

            }
        }
    }

    private void set() {
        if (set != null && !set.isEmpty()) {
            for (String key : set.keySet()) {
                setVariable(key, set.get(key));
            }
        }
    }

    private Serializable getVariable(String key) {
        return getVariables().get(key);
    }

    private void setVariable(String key, Serializable value) {
        getVariables().put(key, value);
    }

}