Add test for token replacement in postgis queries

Author: Rafa de la Torre

test/data/postgis-create-db-and-tables.sql

@@ -0,0 +1,9 @@
+CREATE TABLE test(gid serial PRIMARY KEY, geom geometry, colbigint bigint, col_text text, "col-char" char, "col+bool" boolean, "colnumeric" numeric, "colsmallint" smallint, "colfloat4" real, "colfloat8" double precision, "colcharacter" character);
+INSERT INTO test VALUES (DEFAULT, GeomFromEWKT('SRID=4326;POINT(0 0)'), -9223372036854775808, 'I am a point', 'A', TRUE, 1234567809990001, 0, 0.0, 0.0, 'A');
+INSERT INTO test VALUES (DEFAULT, GeomFromEWKT('SRID=4326;POINT(-2 2)'), 9223372036854775807, 'I, too, am a point!', 'B', FALSE, -123456780999001, 0, 0.0, 0.0, 'A');
+INSERT INTO test VALUES (DEFAULT, GeomFromEWKT('SRID=4326;MULTIPOINT(2 1,1 2)'), -1, 'I`m even a MULTI Point', 'Z', FALSE, 12345678099901, 0, 0.0, 0.0, 'A');
+INSERT INTO test VALUES (DEFAULT, GeomFromEWKT('SRID=4326;LINESTRING(0 0,1 1,1 2)'), 0, 'This is a line string', 'ß', FALSE, -9, 0, 0.0, 0.0, 'A');
+INSERT INTO test VALUES (DEFAULT, GeomFromEWKT('SRID=4326;MULTILINESTRING((1 0,0 1,3 2),(3 2,5 4))'), 1, 'multi line string', 'Ü', TRUE, 0.00001, 0, 0.0, 0.0, 'A');
+INSERT INTO test VALUES (DEFAULT, GeomFromEWKT('SRID=4326;POLYGON((0 0,4 0,4 4,0 4,0 0),(1 1, 2 1, 2 2, 1 2,1 1))'), 1, 'polygon', 'Ü', TRUE, 0.00001, 0, 0.0, 0.0, 'A');
+INSERT INTO test VALUES (DEFAULT, GeomFromEWKT('SRID=4326;MULTIPOLYGON(((1 1,3 1,3 3,1 3,1 1),(1 1,2 1,2 2,1 2,1 1)), ((-1 -1,-1 -2,-2 -2,-2 -1,-1 -1)))'), 5432, 'multi ploygon', 'X', TRUE, 999, 0, 0.0, 0.0, 'A');
+INSERT INTO test VALUES (DEFAULT, GeomFromEWKT('SRID=4326;GEOMETRYCOLLECTION(POLYGON((1 1, 2 1, 2 2, 1 2,1 1)),POINT(2 3),LINESTRING(2 3,3 4))'), 8080, 'GEOMETRYCOLLECTION', 'm', TRUE, 9999, 0, 0.0, 0.0, 'A');

test/data/postgis_datasource_tokens_query.xml

@@ -0,0 +1,17 @@
+<?xml version="1.0" encoding="utf-8"?>
+<Map srs="+init=epsg:3857">
+  <Layer name="field_shapes" status="on" srs="+init=epsg:4326">
+    <Datasource>
+      <Parameter name="type">postgis</Parameter>
+      <Parameter name="host">localhost</Parameter>
+      <Parameter name="dbname">node-mapnik-tmp-postgis-test-db</Parameter>
+      <Parameter name="srid">4326</Parameter>
+      <Parameter name="geometry_field">geom</Parameter>
+      <Parameter name="table">
+  (SELECT gid, geom FROM test WHERE gid = @fieldid) as field_shapes
+      </Parameter>
+      <Parameter name="estimate_extent">false</Parameter>
+      <Parameter name="extent">-150,-85,150,85</Parameter>
+    </Datasource>
+  </Layer>
+</Map>

test/vector-tile.postgis.test.js

@@ -0,0 +1,89 @@
+"use strict";
+
+var mapnik = require('../');
+var assert = require('assert');
+var fs = require('fs');
+var path = require('path');
+var spawn = require('child_process').spawn;
+
+var dbname = 'node-mapnik-tmp-postgis-test-db';
+
+var postgis_registered = false;
+var hasPostgisAvailable = false;
+
+describe('mapnik.VectorTile postgis.input', function() {
+
+    before(function(done) {
+        // Check that the postgis plugin is available
+        mapnik.register_datasource(path.join(mapnik.settings.paths.input_plugins,'postgis.input'));
+        postgis_registered = mapnik.datasources().indexOf('postgis') > -1;
+
+        if (postgis_registered) {
+            // Check if postgres is available
+            spawn('psql', ['--version'])
+                .on('error', function(code, signal) {
+                    console.warn('psql --version could not be executed.');
+                    return done();
+                })
+                .on('exit', function(code, signal) {
+                    if (code !== 0) {
+                        console.warn('psql --version returned ' + code);
+                        return done();
+                    }
+                    // Drop the test database if it exists
+                    spawn('dropdb', ['--if-exists', dbname])
+                        .on('exit', function(code, signal) {
+                            if (code !== 0) {
+                                console.warn('dropdb --if-exists ' + dbname + ' returned ' + code);
+                                return done();
+                            }
+                            // Create the test database
+                            spawn('createdb', ['-T', 'template_postgis', dbname])
+                                .on('exit', function(code, signal) {
+                                    if (code !== 0) {
+                                        console.warn('createdb -T template_postgis ' + dbname + 'returned ' + code);
+                                        return done();
+                                    }
+                                    hasPostgisAvailable = true;
+                                    return done();
+                                });
+                        })
+                });
+        } else {
+            console.warn('postgis input datasource not registered.');
+            return done();
+        }
+    });
+
+    it('passes variables to replace tokens in query', function(done) {
+        if (!hasPostgisAvailable) {
+            this.skip('postgis not availabel');
+        }
+
+        spawn('psql', ['-q', '-f', './test/data/postgis-create-db-and-tables.sql', dbname])
+            .on('exit', function(code, signal) {
+                assert.equal(code, 0, 'could not load data in postgis');
+                        var map = new mapnik.Map(256, 256);
+                map.loadSync('./test/data/postgis_datasource_tokens_query.xml');
+
+                var opts = {};
+                opts.variables = { "fieldid": 2 };
+                map.render(new mapnik.VectorTile(0, 0, 0), opts, function(err, vtile) {
+                    if (err) throw err;
+                    //fs.writeFileSync('/tmp/tile.vector.pbf',vtile.getData());
+                    assert(!vtile.empty());
+                    var out = JSON.parse(vtile.toGeoJSON(0));
+                    assert.equal(out.type,'FeatureCollection');
+                    assert.equal(out.features.length,1);
+                    assert.equal(out.features[0].properties.gid, 2);
+                    var coords = out.features[0].geometry.coordinates;
+                    var expected_coords = [-2.0, 2.0]; // From DB, GeomFromEWKT('SRID=4326;POINT(-2 2)')
+                    assert.ok(Math.abs(coords[0] - expected_coords[0]) < 0.3);
+                    assert.ok(Math.abs(coords[1] - expected_coords[1]) < 0.3);
+                    done();
+                });
+            });
+
+        // TODO: develop reasonable assertion
+    });
+});