Statistics
| Revision:

svn-gvsig-desktop / trunk / org.gvsig.desktop / org.gvsig.desktop.compat.cdc / org.gvsig.fmap.dal / org.gvsig.fmap.dal.impl / src / test / java / org / gvsig / fmap / dal / store / SQLBuilderTest.java @ 43129

History | View | Annotate | Download (19.1 KB)

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

    
3
import java.util.ArrayList;
4
import java.util.List;
5
import org.apache.commons.lang3.ArrayUtils;
6
import org.cresques.cts.IProjection;
7
import org.gvsig.fmap.crs.CRSFactory;
8
import org.gvsig.fmap.dal.ExpressionBuilder;
9
import org.gvsig.fmap.dal.ExpressionBuilder.Parameter;
10
import org.gvsig.fmap.dal.ExpressionBuilder.Variable;
11
import org.gvsig.fmap.dal.SQLBuilder;
12
import org.gvsig.fmap.dal.SQLBuilder.Privilege;
13
import org.gvsig.fmap.dal.feature.spi.SQLBuilderBase;
14
import org.gvsig.fmap.geom.DataTypes;
15
import org.gvsig.fmap.geom.Geometry;
16
import org.gvsig.fmap.geom.GeometryLocator;
17
import org.gvsig.fmap.geom.GeometryManager;
18
import org.gvsig.fmap.geom.primitive.Polygon;
19
import org.gvsig.tools.junit.AbstractLibraryAutoInitTestCase;
20

    
21
public class SQLBuilderTest extends AbstractLibraryAutoInitTestCase {
22
    
23
    @Override
24
    protected void doSetUp() throws Exception {
25

    
26
    }
27
    
28
    List<String> getVariableNames(ExpressionBuilder builder) {
29
        List<String> vars = new ArrayList<>();
30
        for (Variable var : builder.getVariables()) {
31
            vars.add(var.getName());
32
        }
33
        return vars;
34
    }
35
    
36
    List<String> getParameterNames(ExpressionBuilder builder) {
37
        List<String> params = new ArrayList<>();
38
        for (Parameter param : builder.getParameters()) {
39
            String s;
40
            switch(param.getType()) {
41
                case Constant:
42
                    Object value = param.getValue();
43
                    if( value==null ) {
44
                        s = "null";
45
                    } else if( value instanceof String ) {
46
                        s = "'" + (String)value + "'";
47
                    } else {
48
                        s = value.toString();
49
                    }    
50
                    break;
51
                case Geometry:
52
                case Variable:
53
                default:
54
                    s = "\"" + param.getName() + "\"";
55
            }
56
            params.add(s);
57
        }
58
        return params;
59
    }
60
    
61
 
62
    public void testCalulateEnvelope() throws Exception {
63
        GeometryManager geometryManager = GeometryLocator.getGeometryManager();
64
        IProjection proj = CRSFactory.getCRS("EPSG:4326");
65
        
66
        Polygon limit = geometryManager.createPolygon(Geometry.SUBTYPES.GEOM2D);
67
        limit.addVertex(0, 0);
68
        limit.addVertex(0, 100);
69
        limit.addVertex(100, 100);
70
        limit.addVertex(100, 0);
71
        limit.addVertex(0, 0);
72
        
73
        SQLBuilder builder = new SQLBuilderBase();
74
        
75
        builder.select().column().value(
76
            builder.getAsGeometry(
77
              builder.ST_ExtentAggregate(
78
                builder.column("the_geom")
79
              )
80
            )
81
        ).as("envelope");
82
        builder.select().from().table().database("master").schema("dbo").name("test1");
83
        builder.select().where().set(
84
            builder.ST_Intersects(
85
                builder.ST_Envelope(
86
                    builder.column("the_geom")
87
                ),
88
                builder.geometry(limit, proj)
89
            )
90
        );
91
        builder.select().where().and(
92
                builder.custom("x = 27")
93
        );
94
        
95
        // SELECT ST_AsBinary(ST_Extent("the_geom")) AS "envelope" FROM "master"."dbo"."test1" WHERE ST_Intersects((ST_Envelope("the_geom")), (ST_GeomFromText('POLYGON ((0 0, 0 100, 100 100, 100 0, 0 0))', (4326)))) AND x = 27
96
        
97
        System.out.println("# Test:: testCalulateEnvelope");
98
        System.out.println("# SQL:: " + builder.toString());
99
        System.out.println("# Variables:: " + ArrayUtils.toString(getVariableNames(builder)));
100
        System.out.println("# Parametros:: " + ArrayUtils.toString(getParameterNames(builder)));
101
        assertEquals(
102
                "SELECT ST_AsBinary(ST_Extent(\"the_geom\")) AS \"envelope\" FROM \"master\".\"dbo\".\"test1\" WHERE ST_Intersects((ST_Envelope(\"the_geom\")), (ST_GeomFromText('POLYGON ((0 0, 0 100, 100 100, 100 0, 0 0))', (4326)))) AND x = 27",
103
                builder.toString()
104
        );
105
        assertEquals(
106
                "[the_geom]",
107
                ArrayUtils.toString(getVariableNames(builder))
108
        );
109
        assertEquals(
110
                "[]",
111
                ArrayUtils.toString(getParameterNames(builder))
112
        );
113
    }
114

    
115
    public void testCount() throws Exception {
116
        SQLBuilder builder = new SQLBuilderBase();
117
        
118
        builder.select().column().value(builder.count().all());
119
        builder.select().from().table().database("master").schema("dbo").name("test1");
120
        builder.select().from().subquery(null);
121
        builder.select().where().set( builder.custom("pp = 200"));
122

    
123
        System.out.println("# Test:: testCount");
124
        System.out.println("# SQL:: " + builder.toString());
125
        System.out.println("# Variables:: " + ArrayUtils.toString(getVariableNames(builder)));
126
        System.out.println("# Parametros:: " + ArrayUtils.toString(getParameterNames(builder)));
127

    
128
        //# Test:: testCount
129
        //# SQL:: SELECT COUNT(*) FROM "master"."dbo"."test1" WHERE pp = 200
130
        //# Variables:: []
131
        //# Parametros:: []
132

    
133
        assertEquals(
134
                "SELECT COUNT(*) FROM \"master\".\"dbo\".\"test1\" WHERE pp = 200",
135
                builder.toString()
136
        );
137
        assertEquals(
138
                "[]",
139
                ArrayUtils.toString(getVariableNames(builder))
140
        );
141
        assertEquals(
142
                "[]",
143
                ArrayUtils.toString(getParameterNames(builder))
144
        );
145
    }
146
    
147
    public void testCreateTable() throws Exception {
148
        SQLBuilder builder = new SQLBuilderBase();
149

    
150
        builder.create_table().table().database("master").schema("dbo").name("test1");
151
        builder.create_table().add_column(
152
                "name",
153
                DataTypes.STRING,
154
                45,
155
                0,
156
                false,
157
                true,
158
                false,
159
                null
160
        );
161
        builder.create_table().add_column(
162
                "id",
163
                DataTypes.INT,
164
                0,
165
                0,
166
                true,
167
                false,
168
                true,
169
                0
170
        );
171
        builder.create_table().add_column(
172
                "geom",
173
                DataTypes.GEOMETRY,
174
                0,
175
                0,
176
                false,
177
                true,
178
                false,
179
                null
180
        );
181

    
182
        
183
        // CREATE TABLE "master"."dbo"."test1" ("name" VARCHAR(45) DEFAULT NULL NULL, "id" SERIAL DEFAULT '0' NOT NULL PRIMARY KEY, "geom" TEXT DEFAULT NULL NULL )
184
        System.out.println("# Test:: testCreateTable");
185
        System.out.println("# SQL:: " + builder.toString());
186
        System.out.println("# Variables:: " + ArrayUtils.toString(getVariableNames(builder)));
187
        System.out.println("# Parametros:: " + ArrayUtils.toString(getParameterNames(builder)));
188
        assertEquals(
189
                "CREATE TABLE \"master\".\"dbo\".\"test1\" (\"name\" VARCHAR(45) DEFAULT NULL NULL, \"id\" SERIAL DEFAULT '0' NOT NULL PRIMARY KEY, \"geom\" TEXT DEFAULT NULL NULL )",
190
                builder.toString()
191
        );
192
        assertEquals(
193
                "[]",
194
                ArrayUtils.toString(getVariableNames(builder))
195
        );
196
        assertEquals(
197
                "[]",
198
                ArrayUtils.toString(getParameterNames(builder))
199
        );
200
    }
201

    
202
    public void testDropTable() throws Exception {
203
        SQLBuilder builder = new SQLBuilderBase();
204
        
205
        builder.drop_table().table().database("master").schema("dbo").name("test1");
206

    
207
        // DROP TABLE "master"."dbo"."test1"; DELETE FROM GEOMETRY_COLUMNS WHERE f_table_schema = 'dbo' AND f_table_name = 'test1'
208
        
209
        System.out.println("# Test:: testDropTable");
210
        System.out.println("# SQL:: " + builder.toString());
211
        System.out.println("# Variables:: " + ArrayUtils.toString(getVariableNames(builder)));
212
        System.out.println("# Parametros:: " + ArrayUtils.toString(getParameterNames(builder)));
213
        assertEquals(
214
                "DROP TABLE \"master\".\"dbo\".\"test1\"; DELETE FROM GEOMETRY_COLUMNS WHERE f_table_schema = 'dbo' AND f_table_name = 'test1'",
215
                builder.toString()
216
        );
217
        assertEquals(
218
                "[]",
219
                ArrayUtils.toString(getVariableNames(builder))
220
        );
221
        assertEquals(
222
                "[]",
223
                ArrayUtils.toString(getParameterNames(builder))
224
        );
225
    }
226
    
227
    public void testFetchFeatureProviderByReference() throws Exception {
228
        SQLBuilder builder = new SQLBuilderBase();
229
        
230
        String value = "yoyo";
231
        builder.select().column().name("name");
232
        builder.select().column().name("id");
233
        builder.select().column().name("geom").as_geometry();
234
        builder.select().from().table().database("master").schema("dbo").name("test1");
235
        builder.select().where().set(
236
            builder.eq(
237
                builder.column("name"),
238
                builder.parameter(value).as_constant()
239
            )
240
        );
241
        builder.select().limit(1);
242

    
243
        // SELECT "name", "id", ST_AsBinary("geom") FROM "master"."dbo"."test1" WHERE ( ("name") = (?) ) LIMIT 1
244

    
245
        System.out.println("# Test:: testFetchFeatureProviderByReference");
246
        System.out.println("# SQL:: " + builder.toString());
247
        System.out.println("# Variables:: " + ArrayUtils.toString(getVariableNames(builder)));
248
        System.out.println("# Parametros:: " + ArrayUtils.toString(getParameterNames(builder)));
249
        assertEquals(
250
                "SELECT \"name\", \"id\", ST_AsBinary(\"geom\") FROM \"master\".\"dbo\".\"test1\" WHERE ( (\"name\") = (?) ) LIMIT 1",
251
                builder.toString()
252
        );
253
        assertEquals(
254
                "[geom, id, name]",
255
                ArrayUtils.toString(getVariableNames(builder))
256
        );
257
        assertEquals(
258
                "['yoyo']",
259
                ArrayUtils.toString(getParameterNames(builder))
260
        );
261
    }
262
    
263
    public void testFetchFeatureType() throws Exception {
264
        SQLBuilder builder = new SQLBuilderBase();
265

    
266
        builder.select().column().all();
267
        builder.select().from().table().database("master").schema("dbo").name("test1");
268
        builder.select().limit(1);
269

    
270
        System.out.println("# Test:: testFetchFeatureType");
271
        System.out.println("# SQL:: " + builder.toString());
272
        System.out.println("# Variables:: " + ArrayUtils.toString(getVariableNames(builder)));
273
        System.out.println("# Parametros:: " + ArrayUtils.toString(getParameterNames(builder)));
274
        
275
        //# Test:: testFetchFeatureType
276
        //# SQL:: SELECT * FROM [master].[dbo].[test1] LIMMIT 1
277
        //# Variables:: []
278
        //# Parametros:: []        
279
        
280
        assertEquals(
281
                "SELECT * FROM \"master\".\"dbo\".\"test1\" LIMIT 1",
282
                builder.toString()
283
        );
284
        assertEquals(
285
                "[]",
286
                ArrayUtils.toString(getVariableNames(builder))
287
        );
288
        assertEquals(
289
                "[]",
290
                ArrayUtils.toString(getParameterNames(builder))
291
        );
292
    }
293
        
294
    public void testPerformDeletes() throws Exception {
295
        SQLBuilder builder = new SQLBuilderBase();
296

    
297
        builder.delete().table().database("master").schema("dbo").name("test1");
298
        builder.delete().where().and(
299
            builder.eq( 
300
                builder.column("id1"),
301
                builder.parameter("id1").as_variable()
302
            )
303
        );
304
        builder.delete().where().and(
305
            builder.eq( 
306
                builder.column("id2"),
307
                builder.parameter("id2").as_variable()
308
            )
309
        );
310

    
311
        // DELETE FROM "master"."dbo"."test1" WHERE ( ("id1") = (?) ) AND ( ("id2") = (?) )
312

    
313
        System.out.println("# Test:: testPerformDeletes");
314
        System.out.println("# SQL:: " + builder.toString());
315
        System.out.println("# Variables:: " + ArrayUtils.toString(getVariableNames(builder)));
316
        System.out.println("# Parametros:: " + ArrayUtils.toString(getParameterNames(builder)));
317
        assertEquals(
318
                "DELETE FROM \"master\".\"dbo\".\"test1\" WHERE ( (\"id1\") = (?) ) AND ( (\"id2\") = (?) )",
319
                builder.toString()
320
        );
321
        assertEquals(
322
                "[id1, id2]",
323
                ArrayUtils.toString(getVariableNames(builder))
324
        );
325
        assertEquals(
326
                "[\"id1\", \"id2\"]",
327
                ArrayUtils.toString(getParameterNames(builder))
328
        );
329
    }
330

    
331
    public void testPerformInserts1() throws Exception {
332
        IProjection proj = CRSFactory.getCRS("EPSG:4326");
333

    
334
        SQLBuilder builder = new SQLBuilderBase();
335

    
336
        builder.insert().table().database("master").schema("dbo").name("test1");
337
        builder.insert().column().name("id").with_value(builder.parameter("id"));
338
        builder.insert().column().name("name").with_value(builder.parameter("name"));
339
        builder.insert().column().name("geom").with_value(builder.parameter("geom").as_geometry_variable().srs(proj));
340
        
341
        // INSERT INTO "master"."dbo"."test1" ( "id", "name", "geom" ) VALUES ( ?, ?, ST_GeomFromText(?, (4326)) )
342
        
343
        System.out.println("# Test:: testPerformInserts1");
344
        System.out.println("# SQL:: " + builder.toString());
345
        System.out.println("# Variables:: " + ArrayUtils.toString(getVariableNames(builder)));
346
        System.out.println("# Parametros:: " + ArrayUtils.toString(getParameterNames(builder)));
347
        assertEquals(
348
                "INSERT INTO \"master\".\"dbo\".\"test1\" ( \"id\", \"name\", \"geom\" ) VALUES ( ?, ?, ST_GeomFromText(?, (4326)) )",
349
                builder.toString()
350
        );
351
        assertEquals(
352
                "[geom, id, name]",
353
                ArrayUtils.toString(getVariableNames(builder))
354
        );
355
        assertEquals(
356
                "[\"id\", \"name\", \"geom\"]",
357
                ArrayUtils.toString(getParameterNames(builder))
358
        );
359
    }
360

    
361
    public void testPerformInserts2() throws Exception {
362
        IProjection proj = CRSFactory.getCRS("EPSG:4326");
363

    
364
        SQLBuilder builder = new SQLBuilderBase();
365

    
366
        builder.insert().table().database("master").schema("dbo").name("test1");
367
        builder.insert().column().name("id").with_value(builder.parameter("id"));
368
        builder.insert().column().name("name").with_value(builder.parameter("name"));
369
        builder.insert().column().name("geom").with_value(builder.parameter("geom").as_geometry_variable().srs(builder.parameter().value(proj)));
370
        
371
        System.out.println("# Test:: testPerformInserts2");
372
        System.out.println("# SQL:: " + builder.toString());
373
        System.out.println("# Variables:: " + ArrayUtils.toString(getVariableNames(builder)));
374
        System.out.println("# Parametros:: " + ArrayUtils.toString(getParameterNames(builder)));
375
        assertEquals(
376
                "INSERT INTO \"master\".\"dbo\".\"test1\" ( \"id\", \"name\", \"geom\" ) VALUES ( ?, ?, ST_GeomFromText(?, (?)) )",
377
                builder.toString()
378
        );
379
        assertEquals(
380
                "[geom, id, name]",
381
                ArrayUtils.toString(getVariableNames(builder))
382
        );
383
        assertEquals(
384
                "[\"id\", \"name\", \"geom\", 4326]",
385
                ArrayUtils.toString(getParameterNames(builder))
386
        );
387
    }
388

    
389
    public void testPerformUpdates() throws Exception {
390
        IProjection proj = CRSFactory.getCRS("EPSG:4326");
391

    
392
        SQLBuilder builder = new SQLBuilderBase();
393

    
394
        builder.update().table().database("master").schema("dbo").name("test1");
395
        builder.update().where().and(
396
            builder.eq(
397
                builder.column("id"), 
398
                builder.parameter("id").as_variable()
399
            )
400
        );
401
        builder.update().column().name("name").with_value(builder.parameter("name"));
402
        builder.update().column().name("geom").with_value(
403
                builder.parameter("geom").as_geometry_variable().srs(builder.parameter().value(proj)) 
404
        );
405

    
406
        // UPDATE "master"."dbo"."test1" SET "name" = ?, "geom" = ST_GeomFromText(?, (?)) WHERE ( ("id") = (?) )
407

    
408
        System.out.println("# Test:: testPerformUpdates");
409
        System.out.println("# SQL:: " + builder.toString());
410
        System.out.println("# Variables:: " + ArrayUtils.toString(getVariableNames(builder)));
411
        System.out.println("# Parametros:: " + ArrayUtils.toString(getParameterNames(builder)));
412
        assertEquals(
413
                "UPDATE \"master\".\"dbo\".\"test1\" SET \"name\" = ?, \"geom\" = ST_GeomFromText(?, (?)) WHERE ( (\"id\") = (?) )",
414
                builder.toString()
415
        );
416
        assertEquals(
417
                "[geom, id, name]",
418
                ArrayUtils.toString(getVariableNames(builder))
419
        );
420
        assertEquals(
421
                "[\"name\", \"geom\", 4326, \"id\"]",
422
                ArrayUtils.toString(getParameterNames(builder))
423
        );
424
    }
425

    
426
    public void testGrant1() throws Exception {
427

    
428
        SQLBuilder builder = new SQLBuilderBase();
429

    
430
        builder.grant().table().database("master").schema("dbo").name("test1");
431
        builder.grant().role("prueba").select().insert().update();
432
        builder.grant().role("gis").all();
433
                
434
        
435
        System.out.println("# Test:: testGrant1");
436
        System.out.println("# SQL:: " + builder.toString());
437
        System.out.println("# Variables:: " + ArrayUtils.toString(getVariableNames(builder)));
438
        System.out.println("# Parametros:: " + ArrayUtils.toString(getParameterNames(builder)));
439
        assertEquals(
440
                "",
441
                builder.toString()
442
        );
443
        assertEquals(
444
                "[]",
445
                ArrayUtils.toString(getVariableNames(builder))
446
        );
447
        assertEquals(
448
                "[]",
449
                ArrayUtils.toString(getParameterNames(builder))
450
        );
451
    }
452

    
453
    public void testGrant2() throws Exception {
454

    
455
        SQLBuilder builder = new SQLBuilderBase();
456

    
457
        builder.grant().table().database("master").schema("dbo").name("test1");
458
        builder.grant().role("prueba").privilege(Privilege.SELECT)
459
                .privilege(Privilege.INSERT)
460
                .privilege(Privilege.UPDATE);
461
        builder.grant().role("gis").privilege(Privilege.ALL);
462
                
463
        
464
        System.out.println("# Test:: testGrant2");
465
        System.out.println("# SQL:: " + builder.toString());
466
        System.out.println("# Variables:: " + ArrayUtils.toString(getVariableNames(builder)));
467
        System.out.println("# Parametros:: " + ArrayUtils.toString(getParameterNames(builder)));
468
        assertEquals(
469
                "",
470
                builder.toString()
471
        );
472
        assertEquals(
473
                "[]",
474
                ArrayUtils.toString(getVariableNames(builder))
475
        );
476
        assertEquals(
477
                "[]",
478
                ArrayUtils.toString(getParameterNames(builder))
479
        );
480
    }
481

    
482

    
483
}