Application: gvSIG desktop

package org.gvsig.expresionevaluator.impl;

import java.net.URL;

import java.text.ParseException;

import junit.framework.TestCase;

import static junit.framework.TestCase.assertEquals;

import org.gvsig.expressionevaluator.LexicalAnalyzer;

import org.gvsig.expressionevaluator.Compiler;

import org.gvsig.expressionevaluator.SymbolTable;

import org.gvsig.expressionevaluator.Interpreter;

import org.gvsig.expressionevaluator.Code;

import org.gvsig.expressionevaluator.ExpressionEvaluatorLocator;

import org.gvsig.expressionevaluator.ExpressionEvaluatorManager;

import org.gvsig.expressionevaluator.MutableSymbolTable;

import org.gvsig.tools.library.impl.DefaultLibrariesInitializer;

import org.gvsig.tools.resourcesstorage.AbstractResourcesStorage;

import org.gvsig.tools.resourcesstorage.URLResource;

public class TestCreateFunction extends TestCase {

    private static class Test1ResourcesStorage extends AbstractResourcesStorage {

        public Test1ResourcesStorage() {

        }

        @Override

        public Resource getResource(String resourceName) {

            URL url = this.getClass().getResource("/org/gvsig/expressionevaluator/impl/"+resourceName+".cosa");

            if( url == null ) {

                return null;

            }

            Resource res = new URLResource(resourceName+".cosa", url);

            return res;

        }        

    }

    public TestCreateFunction(String testName) {

        super(testName);

    }

    @Override

    protected void setUp() throws Exception {

        super.setUp();

        new DefaultLibrariesInitializer().fullInitialize();

    }

    @Override

    protected void tearDown() throws Exception {

        super.tearDown();

    }

    // TODO add test methods here. The name must begin with 'test'. For example:

    // public void testHello() {}

    protected LexicalAnalyzer createLexicalAnalyzer() {

        ExpressionEvaluatorManager manager = ExpressionEvaluatorLocator.getManager();

        LexicalAnalyzer lexer = manager.createLexicalAnalyzer();

        return lexer;

    }

    protected org.gvsig.expressionevaluator.Compiler createCompiler() {

        ExpressionEvaluatorManager manager = ExpressionEvaluatorLocator.getManager();

        Compiler compiler = manager.createCompiler();

        compiler.setLexicalAnalyzer(createLexicalAnalyzer());

        return compiler;

    }

    protected SymbolTable createSymbolTable() {

        ExpressionEvaluatorManager manager = ExpressionEvaluatorLocator.getManager();

        MutableSymbolTable symbolTable = manager.createSymbolTable();

        symbolTable.setVar("precio", 200);

        symbolTable.setVar("1990", 0.168873933773767);

        return symbolTable;

    }

    protected Interpreter createInterpreter(SymbolTable symbolTable) {

        ExpressionEvaluatorManager manager = ExpressionEvaluatorLocator.getManager();

        manager.setScriptsResourcesStorage(new Test1ResourcesStorage());

        Interpreter interpreter = manager.createInterpreter();

        interpreter.setSymbolTable(symbolTable);

        return interpreter;

    }

    public void testStringUtilsCapitalize() {

        SymbolTable symbolTable = createSymbolTable();

        Compiler compiler = createCompiler();

        Interpreter interpreter = createInterpreter(symbolTable);

        String source = "BEGIN "

                + "CREATE FUNCTION CAPITALIZE(s) AS "

                + "'org.apache.commons.lang3.StringUtils', 'capitalize' "

                + "LANGUAGE 'java';"

                + "CAPITALIZE('hola');"

                + "END";

        Code code = compiler.compileExpression(source);

        assertEquals("BLOCK(CREATE_FUNCTION('CAPITALIZE', LIST('s'), NULL, 'org.apache.commons.lang3.StringUtils', 'capitalize', 'java', NULL), CAPITALIZE('hola'))", code.toString());

        Object v = interpreter.run(code);

        assertEquals("Hola", v.toString());

    }

    public void testSQRT() {

        SymbolTable symbolTable = createSymbolTable();

        Compiler compiler = createCompiler();

        Interpreter interpreter = createInterpreter(symbolTable);

        String source = "BEGIN "

                + "CREATE FUNCTION SQUARE(num) AS BEGIN RETURN num*num; END; "

                + "SQUARE(4);"

                + "END";

        Code code = compiler.compileExpression(source);

        assertEquals("BLOCK(CREATE_FUNCTION('SQUARE', LIST('num'), RETURN((\"num\" * \"num\")), NULL, NULL, NULL, NULL), SQUARE(4))", code.toString());

        Object v = interpreter.run(code);

        assertEquals(16, ((Number)v).intValue());

    }

    public void testCreateFunctionWithExcept1() {

        SymbolTable symbolTable = createSymbolTable();

        Compiler compiler = createCompiler();

        Interpreter interpreter = createInterpreter(symbolTable);

        String source = "BEGIN "

                + "CREATE FUNCTION SQUARE(num) AS BEGIN RETURN num*num; EXCEPT -1; END; "

                + "SQUARE(4);"

                + "END";

        Code code = compiler.compileExpression(source);

        assertEquals("BLOCK(CREATE_FUNCTION('SQUARE', LIST('num'), RETURN((\"num\" * \"num\")), NULL, NULL, NULL, -1), SQUARE(4))", code.toString());

        Object v = interpreter.run(code);

        assertEquals(16, ((Number)v).intValue());

    }

    public void testCreateFunctionWithExcept2() {

        SymbolTable symbolTable = createSymbolTable();

        Compiler compiler = createCompiler();

        Interpreter interpreter = createInterpreter(symbolTable);

        String source = "BEGIN "

                + "CREATE FUNCTION SQUARE(num) AS BEGIN RETURN 1/0; EXCEPT -1; END; "

                + "SQUARE(4);"

                + "END";

        Code code = compiler.compileExpression(source);

        assertEquals("BLOCK(CREATE_FUNCTION('SQUARE', LIST('num'), RETURN((1 / 0)), NULL, NULL, NULL, -1), SQUARE(4))", code.toString());

        Object v = interpreter.run(code);

        assertEquals(-1, ((Number)v).intValue());

    }

    public void testSQRTScriptEngine() {

        SymbolTable symbolTable = createSymbolTable();

        Compiler compiler = createCompiler();

        Interpreter interpreter = createInterpreter(symbolTable);

        String source = "BEGIN "

                + "CREATE FUNCTION SQUARE(num) AS 'testlib', 'SQUARE';"

                + "SQUARE(4);"

                + "END";

        Code code = compiler.compileExpression(source);

        assertEquals("BLOCK(CREATE_FUNCTION('SQUARE', LIST('num'), NULL, 'testlib', 'SQUARE', NULL, NULL), SQUARE(4))", code.toString());

        Object v = interpreter.run(code);

        assertEquals(16, ((Number)v).intValue());

    }

    public void testLambda1() throws ParseException {

        SymbolTable symbolTable = createSymbolTable();

        Compiler compiler = createCompiler();

        Interpreter interpreter = createInterpreter(symbolTable);

        String source = "begin set x to 10 ; set fn = lambda : x * 2; fn(); end";

        Code code = compiler.compileExpression(source);

        assertEquals("BLOCK(LET('x', 10), LET('fn', CREATE_FUNCTION(NULL, (\"x\" * 2))), fn())", code.toString());

        Object v = interpreter.run(code);

        assertEquals(20, ((Number)v).intValue());

    }

    public void testLambda1WithArgs() throws ParseException {

        SymbolTable symbolTable = createSymbolTable();

        Compiler compiler = createCompiler();

        Interpreter interpreter = createInterpreter(symbolTable);

        String source = "begin set x to 10 ; set fn = lambda n : (x * 2) + n; fn(3); end";

        Code code = compiler.compileExpression(source);

        assertEquals("BLOCK(LET('x', 10), LET('fn', CREATE_FUNCTION(LIST('n'), ((\"x\" * 2) + \"n\"))), fn(3))", code.toString());

        Object v = interpreter.run(code);

        assertEquals(23, ((Number)v).intValue());

    }

}