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 / TestCreate.java @ 47611

History | View | Annotate | Download (19.7 KB)

1
package org.gvsig.fmap.dal.store.h2;
2

    
3
import java.util.ArrayList;
4
import java.util.Date;
5
import java.util.List;
6
import junit.framework.TestCase;
7
import static junit.framework.TestCase.assertEquals;
8
import org.gvsig.expressionevaluator.ExpressionBuilder;
9
import org.gvsig.expressionevaluator.ExpressionUtils;
10
import org.gvsig.fmap.dal.DALLocator;
11
import org.gvsig.fmap.dal.DataTypes;
12
import org.gvsig.fmap.dal.DataManager;
13
import org.gvsig.fmap.dal.DataStore;
14
import org.gvsig.fmap.dal.exception.ValidateDataParametersException;
15
import org.gvsig.fmap.dal.feature.EditableFeature;
16
import org.gvsig.fmap.dal.feature.EditableFeatureAttributeDescriptor;
17
import org.gvsig.fmap.dal.feature.EditableFeatureType;
18
import org.gvsig.fmap.dal.feature.Feature;
19
import org.gvsig.fmap.dal.feature.FeatureAttributeDescriptor;
20
import org.gvsig.fmap.dal.feature.FeatureQuery;
21
import org.gvsig.fmap.dal.feature.FeatureStore;
22
import org.gvsig.fmap.dal.feature.FeatureType;
23
import org.gvsig.fmap.dal.feature.NewFeatureStoreParameters;
24
import org.gvsig.fmap.dal.impl.expressionevaluator.DefaultFeatureAttributeEmulatorExpression;
25
import org.gvsig.fmap.dal.store.jdbc.JDBCStoreParameters;
26
import org.gvsig.fmap.dal.store.jdbc2.JDBCServerExplorer;
27
import org.gvsig.tools.dispose.DisposeUtils;
28
import org.gvsig.tools.library.impl.DefaultLibrariesInitializer;
29
import org.slf4j.Logger;
30
import org.slf4j.LoggerFactory;
31

    
32
public class TestCreate extends TestCase {
33
    private static final Logger LOGGER = LoggerFactory.getLogger(TestCreate.class);
34
    
35
    public static final String DBNAME = "testCreate";
36
    
37
    public TestCreate(String testName) {
38
        super(testName);
39
    }
40
    
41
    @Override
42
    protected void setUp() throws Exception {
43
        super.setUp();
44
        new DefaultLibrariesInitializer().fullInitialize();
45
    }
46
    
47
    @Override
48
    protected void tearDown() throws Exception {
49
        super.tearDown();
50
    }
51

    
52
    // TODO add test methods here. The name must begin with 'test'. For example:
53
    // public void testHello() {}
54
    
55
    protected String getProviderName() {
56
        return DataStore.H2SPATIAL_PROVIDER_NAME;
57
    }
58
    
59
    protected String getTargetName() {
60
        return "testCreateTarget1";
61
    }
62
    
63
    protected FeatureStore openTargetStore1(JDBCServerExplorer explorer) throws Exception {
64
        JDBCStoreParameters params = explorer.get(getTargetName());
65
        
66
        DataManager dataManager = DALLocator.getDataManager();
67
        FeatureStore store;
68
        try {
69
            store = (FeatureStore) dataManager.openStore(
70
                    getProviderName(), 
71
                    params
72
            );
73
        } catch(ValidateDataParametersException ex) {
74
            LOGGER.warn(ex.getLocalizedMessageStack());
75
            throw ex;
76
        }
77
        return store;
78
    }
79

    
80
    protected void createFrom(JDBCServerExplorer explorer, FeatureStore sourceStore) throws Exception {
81
        NewFeatureStoreParameters params = (NewFeatureStoreParameters) explorer.getAddParameters(
82
                getTargetName()
83
        );
84
        EditableFeatureType ft = params.getDefaultFeatureType();
85
        ft.addAll(sourceStore.getDefaultFeatureType());
86
        explorer.add(getProviderName(), params, true);
87
    }
88
    
89
    protected void checkTypes(JDBCServerExplorer explorer, FeatureType sourceFeatureType) throws Exception {
90
//        DataType STRING_TYPE = ToolsLocator.getDataTypesManager().get(DataTypes.STRING);
91
//        DataType INT_TYPE = ToolsLocator.getDataTypesManager().get(DataTypes.INT);
92
        
93
        FeatureStore targetStore = openTargetStore1(explorer);
94
        FeatureType targetFeatureType = targetStore.getDefaultFeatureType();
95

    
96
        assertEquals("Feature type size",sourceFeatureType.size(), targetFeatureType.size());
97
        for (int i = 0; i < sourceFeatureType.size(); i++) {
98
            FeatureAttributeDescriptor sourceAttr = sourceFeatureType.get(i);
99
            FeatureAttributeDescriptor targetAttr = targetFeatureType.get(i);
100
            switch(sourceAttr.getType()) {
101
                case DataTypes.BYTE:
102
                case DataTypes.INT:
103
                case DataTypes.LONG:
104
                case DataTypes.BOOLEAN:
105
                case DataTypes.DATE:
106
                case DataTypes.STRING:
107
                case DataTypes.TIME:
108
                case DataTypes.TIMESTAMP:
109
                case DataTypes.DOUBLE:
110
                case DataTypes.FLOAT:
111
                case DataTypes.DECIMAL:
112
                    assertEquals(
113
                            String.format("Field %s name mismatch", sourceAttr.getName()), 
114
                            sourceAttr.getName(), 
115
                            targetAttr.getName()
116
                    );
117
                    assertEquals(
118
                            String.format("Field %s type mismatch", sourceAttr.getName()), 
119
                            sourceAttr.getDataTypeName(), 
120
                            targetAttr.getDataTypeName()
121
                    );
122
                    assertEquals(
123
                            String.format("Field %s size mismatch", sourceAttr.getName()), 
124
                            sourceAttr.getSize(),
125
                            targetAttr.getSize()
126
                    );
127
                    assertEquals(
128
                            String.format("Field %s precision mismatch", sourceAttr.getName()), 
129
                            sourceAttr.getPrecision(),
130
                            targetAttr.getPrecision()
131
                    );
132
                    assertEquals(
133
                            String.format("Field %s scale mismatch", sourceAttr.getName()), 
134
                            sourceAttr.getScale(),
135
                            targetAttr.getScale()
136
                    );
137
                    break;
138
                case DataTypes.GEOMETRY:
139
                    assertEquals(
140
                            String.format("Field %s name mismatch", sourceAttr.getName()), 
141
                            sourceAttr.getName(), 
142
                            targetAttr.getName()
143
                    );
144
                    assertEquals(
145
                            String.format("Field %s type mismatch", sourceAttr.getName()), 
146
                            sourceAttr.getDataTypeName(), 
147
                            targetAttr.getDataTypeName()
148
                    );
149
                    assertEquals(
150
                            String.format("Field %s geometry type mismatch", sourceAttr.getName()), 
151
                            sourceAttr.getGeomType().getName(), 
152
                            targetAttr.getGeomType().getName()
153
                    );
154
                    assertEquals(
155
                            String.format("Field %s geometry SRS mismatch", sourceAttr.getName()), 
156
                            sourceAttr.getSRS().toString(), 
157
                            targetAttr.getSRS().toString()
158
                    );
159
                    assertEquals(
160
                            String.format("Field %s size mismatch", sourceAttr.getName()), 
161
                            sourceAttr.getSize(),
162
                            targetAttr.getSize()
163
                    );
164
                    assertEquals(
165
                            String.format("Field %s precision mismatch", sourceAttr.getName()), 
166
                            sourceAttr.getPrecision(),
167
                            targetAttr.getPrecision()
168
                    );
169
                    break;
170
                default:
171
                    fail(
172
                        String.format("Field %s type %d (%s) not supported.", 
173
                                targetAttr.getName(),
174
                                targetAttr.getType(),
175
                                targetAttr.getDataTypeName()
176
                        )
177
                    );
178
            }
179
        }
180
    }
181
    
182
    protected void copyFrom(JDBCServerExplorer explorer, FeatureStore sourceStore, int mode) throws Exception {
183
        FeatureStore targetStore = openTargetStore1(explorer);
184
        targetStore.edit(mode);
185
        try {
186
            for (Feature sourceFeature : sourceStore.getFeatureSet()) {
187
                EditableFeature targetFeature = targetStore.createNewFeature(sourceFeature);
188
                targetStore.insert(targetFeature);
189
            }
190
        } finally {
191
            targetStore.finishEditing();
192
        }
193
    }
194
    
195
    protected void checkData(JDBCServerExplorer explorer, FeatureStore sourceStore) throws Exception {
196
        FeatureStore targetStore = openTargetStore1(explorer);
197

    
198
        List<Feature> sourceFeatures = sourceStore.getFeatures();
199
        List<Feature> targetFeatures = targetStore.getFeatures();
200
        assertEquals("Count features", sourceFeatures.size(), targetFeatures.size());
201
        for (int i = 0; i < targetFeatures.size(); i++) {
202
            Feature sourceFeature = sourceFeatures.get(i);
203
            Feature targetFeature = targetFeatures.get(i);
204
            for (FeatureAttributeDescriptor sourceAttr : sourceStore.getDefaultFeatureType()) {
205
                switch(sourceAttr.getType()) {
206
                    case DataTypes.BYTE:
207
                        assertEquals(
208
                                String.format("Feature %03d attribute %s", i, sourceAttr.getName()),
209
                                sourceFeature.getInt(sourceAttr.getName()),
210
                                targetFeature.getInt(sourceAttr.getName())
211
                        );
212
                        break;
213
                    case DataTypes.TIMESTAMP:
214
                        Date sourceTimestamp = sourceFeature.getDate(sourceAttr.getName());
215
                        Date targetTimestamp = targetFeature.getDate(sourceAttr.getName());
216
                        assertEquals(
217
                                String.format("Feature %03d attribute %s", i, sourceAttr.getName()),
218
                                sourceTimestamp,
219
                                targetTimestamp
220
                        );
221
                        break;
222
                    case DataTypes.TIME:
223
                        assertEquals(
224
                                String.format("Feature %03d attribute %s", i, sourceAttr.getName()),
225
                                sourceFeature.getDate(sourceAttr.getName()),
226
                                targetFeature.getDate(sourceAttr.getName())
227
                        );
228
                        break;
229
                    case DataTypes.GEOMETRY:
230
                        assertEquals(
231
                                String.format("Feature %03d attribute %s", i, sourceAttr.getName()),
232
                                sourceFeature.get(sourceAttr.getName()),
233
                                targetFeature.get(sourceAttr.getName())
234
                        );
235
                        break;
236
                    case DataTypes.STRING:
237
                    case DataTypes.INT:
238
                    case DataTypes.LONG:
239
                    case DataTypes.FLOAT:
240
                    case DataTypes.DOUBLE:
241
                    default:
242
                        Object sourceValue = sourceFeature.get(sourceAttr.getName());
243
                        Object targetValue = targetFeature.get(sourceAttr.getName());
244
                        if( sourceValue == null ) {
245
                            LOGGER.info(String.format("Feature %03d attribute %s is null", i, sourceAttr.getName()));
246
                        }
247
                        assertEquals(
248
                                String.format("Feature %03d attribute %s", i, sourceAttr.getName()),
249
                                sourceValue,
250
                                targetValue
251
                        );
252
                }
253
            }
254
        }
255
    }
256
    
257
    
258
    public void testCreatePopulateAndCheckData() throws Exception {
259
        try {
260
            FeatureStore sourceStore = TestUtils.openSourceStore1();
261
            JDBCServerExplorer explorer = TestUtils.openServerExplorer(DBNAME);
262

    
263
            createFrom(explorer, sourceStore);        
264

    
265
            checkTypes(explorer, sourceStore.getDefaultFeatureType());
266
            copyFrom(explorer, sourceStore, FeatureStore.MODE_APPEND);
267
            checkData(explorer, sourceStore);
268

    
269
            JDBCStoreParameters params = explorer.get(this.getTargetName());
270
            explorer.remove(params);
271

    
272
            createFrom(explorer, sourceStore);
273
            copyFrom(explorer, sourceStore, FeatureStore.MODE_FULLEDIT);
274
            checkData(explorer, sourceStore);
275
        } catch (Exception ex) {
276
          LOGGER.warn("", ex);
277
          throw ex;
278
        } 
279

    
280
    }
281

    
282
    public void testFilterByDate() throws Exception {
283
        FeatureStore sourceStore = TestUtils.openSourceStore1();
284
        JDBCServerExplorer explorer = TestUtils.openServerExplorer(DBNAME);
285
        
286
        createFrom(explorer, sourceStore);        
287
        copyFrom(explorer, sourceStore, FeatureStore.MODE_APPEND);
288
      
289
        FeatureStore h2Store = openTargetStore1(explorer);
290
        ExpressionBuilder expbuilder = ExpressionUtils.createExpressionBuilder();
291

    
292
        String filter = expbuilder.or(
293
          expbuilder.and(
294
            expbuilder.gt(
295
                    expbuilder.column("Time"), 
296
                    expbuilder.time("01:02:03")
297
            ),
298
            expbuilder.lt(
299
                    expbuilder.column("Time"), 
300
                    expbuilder.time(new Date())
301
            )
302
          ),
303
          expbuilder.and(
304
            expbuilder.gt(
305
                    expbuilder.column("Date"), 
306
                    expbuilder.date("2019-02-17")
307
            ),
308
            expbuilder.lt(
309
                    expbuilder.column("Date"), 
310
                    expbuilder.date(new Date())
311
            )
312
          )
313
        ).toString();
314
        FeatureQuery query = sourceStore.createFeatureQuery();
315
        query.addFilter(filter);
316
        List<Feature> features = h2Store.getFeatures(query);
317
        int sz = features.size();
318
    }
319
    
320
    public void testComputed1() throws Exception {
321
        FeatureStore sourceStore = TestUtils.openSourceStore1();
322
        JDBCServerExplorer explorer = TestUtils.openServerExplorer(DBNAME);
323
        
324
        createFrom(explorer, sourceStore);        
325
        copyFrom(explorer, sourceStore, FeatureStore.MODE_APPEND);
326
      
327
        FeatureStore h2Store = openTargetStore1(explorer);
328
        h2Store.edit();
329
        FeatureType featureType = h2Store.getDefaultFeatureType();
330
        EditableFeatureType eFeatureType = featureType.getEditable();
331
        eFeatureType.add("Compu1", DataTypes.INTEGER, new DefaultFeatureAttributeEmulatorExpression(eFeatureType, ExpressionUtils.createExpression("ID*2")));
332
        eFeatureType.add("Compu2", DataTypes.INTEGER, new DefaultFeatureAttributeEmulatorExpression(eFeatureType, ExpressionUtils.createExpression("Long+10+Compu1")));
333
        h2Store.update(eFeatureType);
334
        h2Store.finishEditing();
335
        List<Feature> features = h2Store.getFeatures();
336
        for (int i = 0; i < features.size(); i++) {
337
            Feature feature = features.get(i);
338
            assertEquals("Compu1 "+i, feature.getInt("ID") * 2, feature.getInt("Compu1"));
339
            if(feature.get("Long")==null) {
340
                assertEquals("Compu2 "+i, null, feature.get("Compu2"));
341
            } else {
342
                assertEquals("Compu2 "+i, feature.getInt("Long") + 10 + feature.getInt("Compu1"), feature.getInt("Compu2"));
343
            }
344
        }
345
        DisposeUtils.dispose(h2Store);
346
    }
347
    
348
    public void testComputed2() throws Exception {
349
        FeatureStore sourceStore = TestUtils.openSourceStore1();
350
        JDBCServerExplorer explorer = TestUtils.openServerExplorer(DBNAME);
351
        
352
        createFrom(explorer, sourceStore);        
353
        copyFrom(explorer, sourceStore, FeatureStore.MODE_APPEND);
354
      
355
        FeatureStore h2Store = openTargetStore1(explorer);
356
        h2Store.edit();
357
        FeatureType featureType = h2Store.getDefaultFeatureType();
358
        EditableFeatureType eFeatureType = featureType.getEditable();
359
        FeatureQuery query = sourceStore.createFeatureQuery();
360
        eFeatureType.add("Compu1", DataTypes.INTEGER, new DefaultFeatureAttributeEmulatorExpression(eFeatureType, ExpressionUtils.createExpression("MOD(ID,10)")));
361
        eFeatureType.add("Compu2", DataTypes.INTEGER, new DefaultFeatureAttributeEmulatorExpression(eFeatureType, ExpressionUtils.createExpression("Long+300")));
362
        EditableFeatureAttributeDescriptor extraColumn1 = query.getExtraColumn().add("Extra1", DataTypes.INTEGER);
363
        EditableFeatureAttributeDescriptor extraColumn2 = query.getExtraColumn().add("Extra2", DataTypes.INTEGER);
364
//        EditableFeatureAttributeDescriptor extraColumn3 = query.getExtraColumn().add("LongMax", DataTypes.INTEGER);
365
        extraColumn1.setFeatureAttributeEmulator(new DefaultFeatureAttributeEmulatorExpression(eFeatureType, ExpressionUtils.createExpression("Compu1+1")));
366
        extraColumn2.setFeatureAttributeEmulator(new DefaultFeatureAttributeEmulatorExpression(eFeatureType, ExpressionUtils.createExpression("20+Byte")));
367
//        extraColumn3.setFeatureAttributeEmulator(new DefaultFeatureAttributeEmulatorExpression(eFeatureType, ExpressionUtils.createExpression("Long")));
368
        query.getGroupByColumns().add("Decimal");
369
        query.getGroupByColumns().add("Extra1");
370
//        query.getGroupByColumns().add("Compu1");
371
        query.getAggregateFunctions().put("ID", "MIN");
372
        query.getAggregateFunctions().put("Byte", "MIN");
373
//        query.getAggregateFunctions().put("Long", "MIN");
374
//        query.getAggregateFunctions().put("LongMax", "MAX");
375
        query.getAggregateFunctions().put("Compu1", "MIN");
376
        query.getAggregateFunctions().put("Double", "SUM");
377
        query.getAggregateFunctions().put("Extra2", "SUM");
378
        query.getAggregateFunctions().put("Compu2", "SUM");
379
        h2Store.update(eFeatureType);
380
        h2Store.finishEditing();
381
        
382
        
383
        List<Feature> features0 = h2Store.getFeatures(query);
384
        ArrayList<Feature> features = new ArrayList<>();
385
        
386
        for (int i = 0; i < features0.size(); i++) {
387
            Feature feature = features0.get(i);
388
            features.add(feature.getCopy());
389
        }
390
        features0 = null;
391
        Feature feature = features.get(1);
392
        feature.get("Compu1");
393
        feature.get("Compu2");
394
        assertEquals("Compu2", 3602, feature.getInt("Compu2"));
395
        assertEquals("Compu1", 1, feature.getInt("Compu1"));
396
//        assertEquals("Long0",null,features.get(0).get("Long"));
397
//        assertEquals("Long1",null,features.get(1).get("Long"));
398
//        assertEquals("Long2",1000L,features.get(2).getLong("Long"));
399
//        assertEquals("Long3",1001L,features.get(3).getLong("Long"));
400
//        assertEquals("Long4",2000L,features.get(4).getLong("Long"));
401
//        assertEquals("Long5",2002L,features.get(5).getLong("Long"));
402
//        assertEquals("Long6",3000L,features.get(6).getLong("Long"));
403
//        assertEquals("Long7",4000L,features.get(7).getLong("Long"));
404
//        assertEquals("Long8",4001L,features.get(8).getLong("Long"));
405
//        assertEquals("Long9",5000L,features.get(9).getLong("Long"));
406
//        assertEquals("Long10",5002L,features.get(10).getLong("Long"));
407
//        assertEquals("Long11",6000L,features.get(11).getLong("Long"));
408
//        assertEquals("Long12",6003L,features.get(12).getLong("Long"));
409
//        assertEquals("Long13",7000L,features.get(13).getLong("Long"));
410
//        assertEquals("Long14",7004L,features.get(14).getLong("Long"));
411
//        assertEquals("Long15",8000L,features.get(15).getLong("Long"));
412
//        assertEquals("Long16",8005L,features.get(16).getLong("Long"));
413
//        assertEquals("Long17",9000L,features.get(17).getLong("Long"));
414
//        assertEquals("Long18",9006L,features.get(18).getLong("Long"));
415
        
416
//        for (int i = 0; i < features.size(); i++) {
417
//            Feature feature = features.get(i);
418
//            assertEquals("Long", feature.getInt("ID") * 2, feature.getInt("Compu1"));
419
//            assertEquals("Compu2", feature.getInt("Long") + 10 + feature.getInt("Compu1"), feature.getInt("Compu2"));
420
//        }
421
        DisposeUtils.dispose(h2Store);
422
    }
423
    
424
    
425

    
426
}