it.geosolutions.geobatch.action.scripting.ScriptingTest.java Source code

Introduction

Here is the source code for it.geosolutions.geobatch.action.scripting.ScriptingTest.java

Source

/*
 *  GeoBatch - Open Source geospatial batch processing system
 *  http://geobatch.geo-solutions.it/
 *  Copyright (C) 2007-2012 GeoSolutions S.A.S.
 *  http://www.geo-solutions.it
 *
 *  GPLv3 + Classpath exception
 *
 *  This program 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, or
 *  (at your option) any later version.
 *
 *  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 General Public License for more details.
 *
 *  You should have received a copy of the GNU General Public License
 *  along with this program.  If not, see <http://www.gnu.org/licenses/>.
 */
package it.geosolutions.geobatch.action.scripting;

import java.io.File;
import java.io.FileReader;
import java.io.IOException;
import java.util.List;

import javax.script.ScriptEngine;
import javax.script.ScriptEngineFactory;
import javax.script.ScriptEngineManager;
import javax.script.ScriptException;

import junit.framework.Assert;

import org.junit.Test;
import org.springframework.core.io.ClassPathResource;

/**
 * 
 * @author etj
 * @author Carlo Cancellieri - carlo.cancellieri@geo-solutions.it
 */
public class ScriptingTest extends Assert {

    @Test
    public void testLoop() throws ScriptException {

        String engineName = "groovy";

        ScriptEngine foundEngine = null;
        // create a script engine manager
        ScriptEngineManager mgr = new ScriptEngineManager();
        List<ScriptEngineFactory> factories = mgr.getEngineFactories();
        System.out.println("FOUND " + factories.size() + " factories");
        for (ScriptEngineFactory sef : factories) {
            System.out.println("FACTORY: " + "'" + sef.getEngineName() + "' " + "'" + sef.getLanguageName() + "' "
                    + "'" + sef.getExtensions() + "' " + "'" + sef.getNames() + "' ");
            if (sef.getEngineName().contains(engineName)) {
                foundEngine = sef.getScriptEngine();
            }

        }

        assertNotNull("Can't find engine '" + engineName + "'", foundEngine);
        foundEngine.eval("print('Hello, World')");
    }

    @Test
    public void testGetEngineByExt() throws ScriptException {

        String engineExt = "js";

        ScriptEngineManager mgr = new ScriptEngineManager();
        // create a JavaScript engine
        ScriptEngine engine = mgr.getEngineByExtension(engineExt);
        assertNotNull("Can't find engine '" + engineExt + "'", engine);

        ScriptEngineFactory sef = engine.getFactory();
        System.out.println("FACTORY for " + engineExt + ": " + "'" + sef.getEngineName() + "' " + "'"
                + sef.getLanguageName() + "' " + "'" + sef.getExtensions() + "' " + "'" + sef.getNames() + "' ");

        // evaluate JavaScript code from String
        engine.eval("print('Hello, World')");

    }

    @Test
    public void testGroovy() throws ScriptException {

        String engineName = "groovy";

        ScriptEngineManager mgr = new ScriptEngineManager();
        // create a JavaScript engine
        ScriptEngine engine = mgr.getEngineByName(engineName);
        assertNotNull("Can't find engine '" + engineName + "'", engine);

        ScriptEngineFactory sef = engine.getFactory();
        System.out.println("FACTORY for " + engineName + ": " + "'" + sef.getEngineName() + "' " + "'"
                + sef.getLanguageName() + "' " + "'" + sef.getExtensions() + "' " + "'" + sef.getNames() + "' ");

        // evaluate code from String
        engine.eval("println \"hello, groovy\"");
    }

    @Test
    public void testGroovyFileAndParam() throws ScriptException, IOException {

        String engineName = "groovy";

        ScriptEngineManager mgr = new ScriptEngineManager();
        // create a JavaScript engine
        ScriptEngine engine = mgr.getEngineByName(engineName);
        assertNotNull("Can't find engine '" + engineName + "'", engine);

        File script = new ClassPathResource("test-data/test.groovy").getFile();

        assertNotNull("Can't find test script", script);

        engine.put("gbtest", "testok");

        // evaluate code from File
        engine.eval(new FileReader(script));
    }

}