Statistics
| Revision:

svn-gvsig-desktop / trunk / org.gvsig.desktop / org.gvsig.desktop.plugin / org.gvsig.sqlite / org.gvsig.sqlite.provider / src / test / java / org / gvsig / sqlite / dal / TestCreate.java @ 47542

History | View | Annotate | Download (19.5 KB)

1
package org.gvsig.sqlite.dal;
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(Throwable t) {
276
            LOGGER.warn("",t);
277
            throw t;
278
        }
279
    }
280

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

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

    
422
}