Application: gvSIG desktop

package org.gvsig.expresionevaluator.impl;

import static junit.framework.Assert.assertEquals;

import junit.framework.TestCase;

import org.gvsig.expressionevaluator.spi.AbstractCodeToStringConverter;

import org.gvsig.expressionevaluator.Code;

import org.gvsig.expressionevaluator.CodeToStringConverter;

import org.gvsig.expressionevaluator.Codes;

import org.gvsig.expressionevaluator.ExpressionEvaluatorLocator;

import org.gvsig.expressionevaluator.ExpressionEvaluatorManager;

import org.gvsig.expressionevaluator.Function;

import org.gvsig.expressionevaluator.Compiler;

import org.gvsig.expressionevaluator.LexicalAnalyzer;

import org.gvsig.expressionevaluator.SymbolTable;

import org.gvsig.expressionevaluator.impl.DefaultCompiler;

import org.gvsig.expressionevaluator.impl.SQLLexicalAnalyzer;

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

/**

 *

 * @author jjdelcerro

 */

public class TestCodeToStringConverter extends TestCase {

    public TestCodeToStringConverter(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() {

        SQLLexicalAnalyzer lexer = new SQLLexicalAnalyzer();

        return lexer;

    }

    protected Compiler createCompiler() {

        Compiler compiler = new DefaultCompiler();

        compiler.setLexicalAnalyzer(createLexicalAnalyzer());

        return compiler;

    }

    private static class SimpleCodeToStringConverter extends AbstractCodeToStringConverter {

        public SimpleCodeToStringConverter() {

            super();

            initFunctions();

        }

        private void initFunctions() {

            ExpressionEvaluatorManager manager = ExpressionEvaluatorLocator.getManager();

            SymbolTable symbolTable = manager.getSymbolTable("SQL");

            for (Function function : symbolTable.functions()) {

                this.addSupportedFunction(function.name());

            }

            symbolTable = manager.getSymbolTable("OGC");

            for (Function function : symbolTable.functions()) {

                this.addSupportedFunction(function.name());

            }

        }

        @Override

        protected String functionToString(String functionName, Codes args) {

            switch (functionName.toLowerCase()) {

                case "date":

                    return "DATE " + toString(args.get(0)) + " ";

                case "time":

                    return "TIME " + toString(args.get(0)) + " ";

                case "timestamp":

                    return "TIMESTAMP " + toString(args.get(0)) + " ";

            }

            return super.functionToString(functionName, args);

        }

    }

    public void testSimpleExpression() {

        String source = "(precio + 10) * 2 + 20 + 30 / 2";

        CodeToStringConverter converter = new SimpleCodeToStringConverter();

        Compiler compiler = createCompiler();

        Code code = compiler.compileExpression(source);

        assertEquals("(((\"precio\" + 10) * 2) + 20) + (30 / 2)", converter.toString(code));

    }    

    public void testFunctionExpression1() {

        String source = "(precio + 10) * abs(2 + 20) + 30 / 2";

        CodeToStringConverter converter = new SimpleCodeToStringConverter();

        Compiler compiler = createCompiler();

        Code code = compiler.compileExpression(source);

        assertEquals(true, converter.isValid(code));

        assertEquals("((\"precio\" + 10) * (abs(2 + 20))) + (30 / 2)", converter.toString(code));

    }    

}