Application: gvSIG desktop

package org.gvsig.sqlite.dal.operations.usecases.arena2;

import junit.framework.TestCase;

import org.gvsig.expressionevaluator.Code;

import org.gvsig.expressionevaluator.ExpressionEvaluatorLocator;

import org.gvsig.expressionevaluator.ExpressionEvaluatorManager;

import org.gvsig.expressionevaluator.Interpreter;

import org.gvsig.expressionevaluator.LexicalAnalyzer;

import org.gvsig.expressionevaluator.MutableSymbolTable;

import org.gvsig.expressionevaluator.SymbolTable;

import org.gvsig.fmap.dal.DALLocator;

import org.gvsig.fmap.dal.DataManager;

import org.gvsig.fmap.dal.feature.FeatureStore;

import org.gvsig.fmap.dal.store.jdbc2.AbstractTestUtils;

import org.gvsig.fmap.dal.store.jdbc2.JDBCServerExplorer;

import org.gvsig.sqlite.dal.TestUtilsSQLite;

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

import org.slf4j.Logger;

import org.slf4j.LoggerFactory;

@SuppressWarnings("UseSpecificCatch")

public class TestSelectWithLocalVariables extends TestCase {

    private static final Logger LOGGER = LoggerFactory.getLogger(TestSelectWithLocalVariables.class);

    public TestSelectWithLocalVariables(String testName) {

        super(testName);

    }

    @Override

    protected void setUp() throws Exception {

        super.setUp();

        new DefaultLibrariesInitializer().fullInitialize();

    }

    @Override

    protected void tearDown() throws Exception {

        super.tearDown();

    }

    protected AbstractTestUtils utils;

    public AbstractTestUtils utils() {

        if (this.utils == null) {

            this.utils = this.createUtils();

        }

        return this.utils;

    }

    protected AbstractTestUtils createUtils() {

        return new TestUtilsSQLite();

    }

    protected LexicalAnalyzer createLexicalAnalyzer() {

        ExpressionEvaluatorManager manager = ExpressionEvaluatorLocator.getManager();

        LexicalAnalyzer lexer = manager.createLexicalAnalyzer();

        return lexer;

    }

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

        ExpressionEvaluatorManager manager = ExpressionEvaluatorLocator.getManager();

        org.gvsig.expressionevaluator.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();

        Interpreter interpreter = manager.createInterpreter();

        interpreter.setSymbolTable(symbolTable);

        return interpreter;

    }

    private void link(Code code) {

      code.link(createSymbolTable());

    }

    public void testSelectWithLocalVariables() throws Exception {

        try {

            FeatureStore sourceStore = utils().openCountriesStore();

            JDBCServerExplorer explorer = utils().openServerExplorer("testCountXXXXX");

            utils().info_jdbc(explorer);

            utils().drop_tables(explorer, sourceStore.getName());

            utils().create_table_from(explorer, sourceStore.getName(), sourceStore);

            utils().insert_into_from(explorer, sourceStore.getName(), sourceStore, FeatureStore.MODE_APPEND);

            DataManager dataManager = DALLocator.getDataManager();

            dataManager.getStoresRepository().add(sourceStore.getName(), explorer.get(sourceStore.getName()));

            StringBuilder source = new StringBuilder();

            source.append("BEGIN ");

            source.append("CREATE PROCEDURE fn1() AS");

            source.append("  BEGIN ");

            source.append("    X := 10; ");

            source.append("    SELECT * INTO ROW FROM countries ORDER BY (AREA + X) LIMIT 1;"); 

            source.append("  END; ");

            source.append("fn1();");

            source.append("END ");

            MutableSymbolTable symbolTable = (MutableSymbolTable) createSymbolTable();

            org.gvsig.expressionevaluator.Compiler compiler = createCompiler();

            Interpreter interpreter = createInterpreter(symbolTable);

            Code code = compiler.compileExpression(source.toString());

            link(code);            

            Object v = interpreter.run(code);

        } catch (Throwable th) {

            LOGGER.warn("", th);

            throw th;

        }

    }

}