Application: gvSIG desktop

package org.gvsig.expressionevaluator;

import junit.framework.TestCase;

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

/**

 *

 * @author jjdelcerro

 */

public class TestGrammarCompiler extends TestCase {

    public TestGrammarCompiler(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();

        Interpreter interpreter = manager.createInterpreter();

        interpreter.setSymbolTable(symbolTable);

        return interpreter;

    }

    private void link(Code code) {

      code.link(createSymbolTable());

    }

    public void testSelect() {

        StringBuilder source = new StringBuilder();

        source.append("SELECT * FROM countries;");

        Compiler compiler = createCompiler();

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

        link(code);

        assertEquals("SELECT(TABLE:'countries')", code.toString());

    }

    public void testSelect2() {

        StringBuilder source = new StringBuilder();

        source.append("BEGIN ");

        source.append("  SET X = 0; ");

        source.append("  FOR row in (SELECT * FROM countries;) ");

        source.append("    BEGIN ");

        source.append("      IF row.LASTCENSUS > 0 THEN ");

        source.append("        SET X = X + row.LASTCENSUS ");

        source.append("      END IF ");

        source.append("    END FOR ");

        source.append("END");

        Compiler compiler = createCompiler();

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

        link(code);

        assertEquals("BLOCK(LET('X', 0), FOREACH('row', SELECT(TABLE:'countries'), IFF((\"row\".\"LASTCENSUS\" > 0), LET('X', (\"X\" + \"row\".\"LASTCENSUS\")))))", code.toString());

    }

    public void testSelectWhere() {

        StringBuilder source = new StringBuilder();

        source.append("SELECT * FROM countries  ");

        source.append("  WHERE LASTCENSUS > 0;  ");

        Compiler compiler = createCompiler();

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

        link(code);

        assertEquals("SELECT(TABLE:'countries', WHERE:(\"LASTCENSUS\" > 0))", code.toString());

    }

    public void testSelectCountWhere() {

        StringBuilder source = new StringBuilder();

        source.append("SELECT COUNT(*) FROM countries  ");

        source.append("  WHERE LASTCENSUS > 0 ; ");

        Compiler compiler = createCompiler();

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

        link(code);

        assertEquals("SELECT_COUNT(TABLE:'countries', WHERE:(\"LASTCENSUS\" > 0))", code.toString());

    }

    public void testSelectWhere2() {

        StringBuilder source = new StringBuilder();

        source.append("SELECT * FROM countries  ");

        source.append("  WHERE countries.LASTCENSUS > 0 ; ");

        Compiler compiler = createCompiler();

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

        link(code);

        assertEquals("SELECT(TABLE:'countries', WHERE:(\"countries\".\"LASTCENSUS\" > 0))", code.toString());

    }

    public void testSelectOrder() {

        StringBuilder source = new StringBuilder();

        source.append("SELECT * FROM countries  ");

        source.append("  ORDER BY CONTINENT ASC, LASTCENSUS DESC; ");

        Compiler compiler = createCompiler();

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

        link(code);

        assertEquals("SELECT(TABLE:'countries', ORDER1:\"CONTINENT\", ORDER2:\"LASTCENSUS\", ORDER_MODE1:TRUE, ORDER_MODE2:FALSE)", code.toString());

    }

    public void testSelectWhereOrder() {

        StringBuilder source = new StringBuilder();

        source.append("SELECT * FROM countries  ");

        source.append("  WHERE LASTCENSUS > 0  ");

        source.append("  ORDER BY ID ; ");

        Compiler compiler = createCompiler();

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

        link(code);

        assertEquals("SELECT(TABLE:'countries', WHERE:(\"LASTCENSUS\" > 0), ORDER1:\"ID\", ORDER_MODE1:TRUE)", code.toString());

    }

    public void testSelectWhereOrderLimit() {

        StringBuilder source = new StringBuilder();

        source.append("SELECT * FROM countries  ");

        source.append("  WHERE LASTCENSUS > 0  ");

        source.append("  ORDER BY LASTCENSUS DESC");

        source.append("  LIMIT 3 ;");

        Compiler compiler = createCompiler();

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

        link(code);

        assertEquals("SELECT(TABLE:'countries', WHERE:(\"LASTCENSUS\" > 0), ORDER1:\"LASTCENSUS\", ORDER_MODE1:FALSE, LIMIT:3)", code.toString());

    }

    public void testSelectLimit() {

        StringBuilder source = new StringBuilder();

        source.append("SELECT * FROM countries  ");

        source.append("  LIMIT 3; ");

        Compiler compiler = createCompiler();

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

        link(code);

        assertEquals("SELECT(TABLE:'countries', LIMIT:3)", code.toString());

    }

    public void testExists() {

        StringBuilder source = new StringBuilder();

        source.append("EXISTS(NULL)");

        Compiler compiler = createCompiler();

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

        link(code);

        String id = "????????????????????????????????";

        String s = code.toString();

        s = s.substring(0, s.length()-id.length()-2)+id+s.substring(s.length()-2);

        assertEquals("EXISTS(NULL, 'EXISTS????????????????????????????????')", s);

    }

    public void testExistsSelect1() {

        StringBuilder source = new StringBuilder();

        source.append("EXISTS(");

        source.append(" SELECT 1 FROM countries");

        source.append("   WHERE countries.LASTCENSUS > 0 ; ");

        source.append(")");

        Compiler compiler = createCompiler();

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

        link(code);

        String id = "????????????????????????????????";

        String s = code.toString();

        s = s.substring(0, s.length()-id.length()-2)+id+s.substring(s.length()-2);

        assertEquals("EXISTS(SELECT(TABLE:'countries', WHERE:(\"countries\".\"LASTCENSUS\" > 0)), 'EXISTS????????????????????????????????')", s);

    }

    public void testExistsSelectLimit1() {

        StringBuilder source = new StringBuilder();

        source.append("EXISTS(");

        source.append(" SELECT 1 FROM countries  ");

        source.append("   WHERE countries.LASTCENSUS > 0  ");

        source.append("   LIMIT 1; ");

        source.append(")");

        Compiler compiler = createCompiler();

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

        link(code);

        String id = "????????????????????????????????";

        String s = code.toString();

        s = s.substring(0, s.length()-id.length()-2)+id+s.substring(s.length()-2);

        assertEquals("EXISTS(SELECT(TABLE:'countries', WHERE:(\"countries\".\"LASTCENSUS\" > 0), LIMIT:1), 'EXISTS????????????????????????????????')", s);

    }

    public void testExistsSelectLimit2() {

        // Sobre la tabla de continentes, para obtener la lista de continentes 

        // que tienen paises sin censo.

        // Filtramos continentes por un campo del pais.

        StringBuilder source = new StringBuilder();

        source.append("EXISTS(");

        source.append(" SELECT 1 FROM countries");

        source.append("   WHERE ");

        source.append("     continents.NAME = countries.CONTINENT AND ");

        source.append("     countries.LASTCENSUS < 0 ");

        source.append("   LIMIT 1; ");

        source.append(")");

        Compiler compiler = createCompiler();

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

        link(code);

        String id = "????????????????????????????????";

        String s = code.toString();

        s = s.substring(0, s.length()-id.length()-2)+id+s.substring(s.length()-2);

        assertEquals("EXISTS(SELECT(TABLE:'countries', WHERE:((\"continents\".\"NAME\" = \"countries\".\"CONTINENT\") AND (\"countries\".\"LASTCENSUS\" < 0)), LIMIT:1), 'EXISTS????????????????????????????????')", s);

    }

    public void testExistsSelectLimit3() {

        // Sobre la tabla de continentes, para obtener la lista de continentes 

        // que tienen paises sin censo.

        // Filtramos continentes por un campo del pais.

        StringBuilder source = new StringBuilder();

        source.append("EXISTS(");

        source.append(" SELECT 1 FROM countries");

        source.append("   WHERE ");

        source.append("     continents.NAME = countries.CONTINENT AND ");

        source.append("     countries.LASTCENSUS < 0 ");

        source.append("   LIMIT 1; ");

        source.append(" ,");

        source.append(" 'patata'");

        source.append(")");

        Compiler compiler = createCompiler();

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

        link(code);

        assertEquals("EXISTS(SELECT(TABLE:'countries', WHERE:((\"continents\".\"NAME\" = \"countries\".\"CONTINENT\") AND (\"countries\".\"LASTCENSUS\" < 0)), LIMIT:1), 'patata')", code.toString());

    }

    public void testExistsSelectLimit4() {

        StringBuilder source = new StringBuilder();

        source.append("EXISTS(");

        source.append(" SELECT(");

        source.append("   TABLE:'countries', ");

        source.append("   WHERE:(countries.LASTCENSUS > 0), ");

        source.append("   LIMIT:1 ");

        source.append("  ) ");

        source.append(" , ");

        source.append(" 'EXISTS1234567890ABCDEFGHHIJKLMNOPQRSTU' ");

        source.append(") ");

        Compiler compiler = createCompiler();

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

        link(code);

        String id = "????????????????????????????????";

        String s = code.toString();

        s = s.substring(0, s.length()-id.length()-2)+id+s.substring(s.length()-2);

        assertEquals("EXISTS(SELECT(TABLE:'countries', WHERE:(\"countries\".\"LASTCENSUS\" > 0), LIMIT:1), 'EXISTS????????????????????????????????')", s);

    }

}