Add readable ZXYStream and createZXYStream method. Refs #42.

2014-10-29 17:11:59 -04:00
parent 5931bb739b
commit 59a6360035
3 changed files with 122 additions and 0 deletions
@@ -0,0 +1,75 @@
+var tape = require('tape');
+var MBTiles = require('../lib/mbtiles.js');
+var source;
+
+tape('zxystream setup', function(assert) {
+    new MBTiles(__dirname + '/fixtures/plain_2.mbtiles', function(err, s) {
+        assert.ifError(err);
+        source = s;
+        assert.end();
+    });
+});
+
+tape('zxystream default batch', function(assert) {
+    var stream = source.createZXYStream();
+    var output = '';
+    var called = 0;
+
+    assert.deepEqual(stream.source, source, 'sets stream.source');
+    assert.deepEqual(stream.batch, 10000, 'sets stream.batch = 10000');
+    assert.deepEqual(stream.offset, 0, 'sets stream.offset = 0');
+
+    stream.on('data', function(lines) {
+        output += lines;
+        called++;
+    });
+    stream.on('end', function() {
+        var queue = output.toString().split('\n');
+        assert.equal(queue.length, 270);
+        assert.equal(called, 1, 'emitted data x1 times');
+        checkTile(queue);
+        function checkTile(queue) {
+            if (!queue.length) return assert.end();
+            var zxy = queue.shift();
+            if (!zxy) return checkTile(queue);
+            zxy = zxy.split('/');
+            source.getTile(zxy[0], zxy[1], zxy[2], function(err, buffer, headers) {
+                assert.equal(!err && (buffer instanceof Buffer), true, zxy.join('/') + ' exists');
+                checkTile(queue);
+            });
+        }
+    });
+});
+
+
+tape('zxystream batch = 10', function(assert) {
+    var stream = source.createZXYStream({batch:10});
+    var output = '';
+    var called = 0;
+
+    assert.deepEqual(stream.source, source, 'sets stream.source');
+    assert.deepEqual(stream.batch, 10, 'sets stream.batch = 10');
+    assert.deepEqual(stream.offset, 0, 'sets stream.offset = 0');
+
+    stream.on('data', function(lines) {
+        output += lines;
+        called++;
+    });
+    stream.on('end', function() {
+        var queue = output.toString().split('\n');
+        assert.equal(queue.length, 270);
+        assert.equal(called, 27, 'emitted data x27 times');
+        checkTile(queue);
+        function checkTile(queue) {
+            if (!queue.length) return assert.end();
+            var zxy = queue.shift();
+            if (!zxy) return checkTile(queue);
+            zxy = zxy.split('/');
+            source.getTile(zxy[0], zxy[1], zxy[2], function(err, buffer, headers) {
+                assert.equal(!err && (buffer instanceof Buffer), true, zxy.join('/') + ' exists');
+                checkTile(queue);
+            });
+        }
+    });
+});
+