Statistics
| Revision:

svn-gvsig-desktop / trunk / org.gvsig.desktop / org.gvsig.desktop.plugin / org.gvsig.h2spatial / org.gvsig.h2spatial.h2gis132 / org.gvsig.h2spatial.h2gis132.provider / src / test / java / org / gvsig / fmap / dal / store / h2 / operations / usecases / arena2 / TestMultipleExpansionOfCalculatedField.java @ 46110

History | View | Annotate | Download (4.72 KB)

1 46110 omartinez
package org.gvsig.fmap.dal.store.h2.operations.usecases.arena2;
2 45472 jjdelcerro
3
import java.util.List;
4
import junit.framework.TestCase;
5 46050 omartinez
import org.gvsig.expressionevaluator.ExpressionUtils;
6
import org.gvsig.fmap.dal.DataTypes;
7
import org.gvsig.fmap.dal.feature.EditableFeatureAttributeDescriptor;
8
import org.gvsig.fmap.dal.feature.EditableFeatureType;
9 45472 jjdelcerro
import org.gvsig.fmap.dal.feature.FeatureQuery;
10
import org.gvsig.fmap.dal.feature.FeatureStore;
11
import org.gvsig.fmap.dal.feature.FeatureType;
12 46050 omartinez
import org.gvsig.fmap.dal.impl.expressionevaluator.DefaultFeatureAttributeEmulatorExpression;
13 45472 jjdelcerro
import org.gvsig.fmap.dal.store.h2.TestUtils;
14
import org.gvsig.fmap.dal.store.jdbc2.JDBCHelper;
15
import org.gvsig.fmap.dal.store.jdbc2.OperationsFactory;
16
import org.gvsig.fmap.dal.store.jdbc2.OperationsFactory.TableReference;
17
import org.gvsig.fmap.dal.store.jdbc2.spi.JDBCSQLBuilderBase;
18 46104 omartinez
import org.gvsig.fmap.dal.store.jdbc2.spi.operations.CountOperation;
19 45472 jjdelcerro
import org.gvsig.fmap.dal.store.jdbc2.spi.operations.ResultSetForSetProviderOperation;
20
import org.gvsig.tools.library.impl.DefaultLibrariesInitializer;
21
import org.slf4j.Logger;
22
import org.slf4j.LoggerFactory;
23
24 46010 jjdelcerro
@SuppressWarnings("UseSpecificCatch")
25 46110 omartinez
public class TestMultipleExpansionOfCalculatedField extends TestCase {
26 45472 jjdelcerro
27 46110 omartinez
    private static final Logger LOGGER = LoggerFactory.getLogger(TestMultipleExpansionOfCalculatedField.class);
28 45472 jjdelcerro
29 46110 omartinez
    public TestMultipleExpansionOfCalculatedField(String testName) {
30 46105 omartinez
        super(testName);
31
    }
32 45472 jjdelcerro
33 46105 omartinez
    @Override
34
    protected void setUp() throws Exception {
35
        super.setUp();
36
        new DefaultLibrariesInitializer().fullInitialize();
37
    }
38 45472 jjdelcerro
39 46105 omartinez
    @Override
40
    protected void tearDown() throws Exception {
41
        super.tearDown();
42
    }
43 46050 omartinez
44 46105 omartinez
    protected List<String> getExpectedSQLs(String name) throws Exception {
45
        return TestUtils.getSQLs(name);
46
    }
47
48 46110 omartinez
    public void testMultipleExpansionOfCalculatedField() throws Exception {
49 46010 jjdelcerro
        try {
50
            JDBCHelper helper = TestUtils.createJDBCHelper();
51
            JDBCSQLBuilderBase sqlbuilder = helper.createSQLBuilder();
52
            OperationsFactory operations = helper.getOperations();
53 45472 jjdelcerro
54 46110 omartinez
            List<String> expectedSQLs = getExpectedSQLs("usecases/arena2/testMultipleExpansionOfCalculatedField.sql");
55 45472 jjdelcerro
56 46104 omartinez
            FeatureStore arena2_conductores = TestUtils.openStoreTest("usecases/arena2/ARENA2_CONDUCTORES.csv");
57
            FeatureStore arena2_accidentes = TestUtils.openStoreTest("usecases/arena2/ARENA2_ACCIDENTES.csv");
58
            FeatureStore arena2_vehiculos = TestUtils.openStoreTest("usecases/arena2/ARENA2_VEHICULOS.csv");
59 45472 jjdelcerro
60 46010 jjdelcerro
            TableReference table = operations.createTableReference(
61
                    "dbtest",
62 46104 omartinez
                    "public", //sqlbuilder.default_schema(),
63
                    "ARENA2_CONDUCTORES",
64 46010 jjdelcerro
                    null
65
            );
66 46104 omartinez
            FeatureType featureType = arena2_conductores.getDefaultFeatureType();
67 46050 omartinez
            EditableFeatureType eFeatureType = featureType.getEditable();
68 46104 omartinez
            FeatureQuery query = arena2_conductores.createFeatureQuery();
69 46110 omartinez
            addExtraColumn(eFeatureType, query, "AccKm0", DataTypes.DOUBLE, "FOREING_VALUE('ID_ACCIDENTE.KM')");
70
            addExtraColumn(eFeatureType, query, "AccKm1", DataTypes.DOUBLE, "AccKm0 * 100");
71
            addExtraColumn(eFeatureType, query, "AccKm2", DataTypes.DOUBLE, "AccKm0 * 222");
72 46105 omartinez
73
            CountOperation count = operations.createCount(eFeatureType, table, null, query);
74
            String sqlcount = count.getSQL();
75
            System.out.println("###### SQL:" + sqlcount);
76
            System.out.println("###### EXP:" + expectedSQLs.get(1));
77 46104 omartinez
            assertEquals("Count SQL", expectedSQLs.get(1), sqlcount);
78 46105 omartinez
79 46050 omartinez
            ResultSetForSetProviderOperation resultSetForSetProvider = operations.createResultSetForSetProvider(
80
                    table,
81
                    null,
82
                    null,
83
                    query,
84
                    eFeatureType,
85
                    eFeatureType,
86 46105 omartinez
                    15,
87 46050 omartinez
                    0,
88
                    0
89
            );
90
            String sql = resultSetForSetProvider.getSQL();
91
            System.out.println("###### SQL:" + sql);
92 46104 omartinez
            System.out.println("###### EXP:" + expectedSQLs.get(0));
93 46050 omartinez
94 46104 omartinez
            assertEquals("Select SQL", expectedSQLs.get(0), sql);
95 46105 omartinez
96
97 46050 omartinez
        } catch (Throwable th) {
98 46105 omartinez
            th.printStackTrace();
99 46050 omartinez
            throw th;
100
        }
101
    }
102 46110 omartinez
103 46105 omartinez
    private void addExtraColumn(EditableFeatureType eFeatureType, FeatureQuery query, String name, int type, String exp) {
104
        EditableFeatureAttributeDescriptor extraColumn = query.getExtraColumn().add(name, type);
105
        extraColumn.setFeatureAttributeEmulator(new DefaultFeatureAttributeEmulatorExpression(eFeatureType, ExpressionUtils.createExpression(exp)));
106
    }
107 45472 jjdelcerro
108
}