testsource.js

var tilelive = require('@mapbox/tilelive');
var zlib = require('zlib');
var fs = require('fs');
var path = require('path');

module.exports = Testsource;

// Load fixture data.
var infos = {
    a: {
        minzoom:0,
        maxzoom:1,
        vector_layers: [
            {
                "id": "coastline",
                "description": "",
                "minzoom": 0,
                "maxzoom": 22,
                "fields": {
                    "FeatureCla": "String",
                    "Note": "String",
                    "ScaleRank": "Number"
                }
            }
        ]
    },
    b: {
        minzoom:0,
        maxzoom:2,
        fillzoom:1,
        vector_layers: [
            {
                "id": "coastline",
                "description": "",
                "minzoom": 0,
                "maxzoom": 22,
                "fields": {
                    "FeatureCla": "String",
                    "Note": "String",
                    "ScaleRank": "Number"
                }
            }
        ]
    },
    i: {
        minzoom:0,
        maxzoom:1,
        vector_layers: []
    },
    'invalid-novector': {
        minzoom:0,
        maxzoom:1
    },
    gz: {
        minzoom:0,
        maxzoom:0,
        vector_layers: [
            {
                "id": "coastline",
                "description": "",
                "minzoom": 0,
                "maxzoom": 22,
                "fields": {
                    "FeatureCla": "String",
                    "Note": "String",
                    "ScaleRank": "Number"
                }
            }
        ]
    },
    expires: {
        minzoom:0,
        maxzoom:1,
        vector_layers: [
            {
                "id": "coastline",
                "description": "",
                "minzoom": 0,
                "maxzoom": 22,
                "fields": {
                    "FeatureCla": "String",
                    "Note": "String",
                    "ScaleRank": "Number"
                }
            }
        ]
    },
    invalid: {
        minzoom:0,
        maxzoom:2,
        vector_layers: [
            {
                "id": "coastline",
                "description": "",
                "minzoom": 0,
                "maxzoom": 22,
                "fields": {
                    "FeatureCla": "String",
                    "Note": "String",
                    "ScaleRank": "Number"
                }
            }
        ]
    }
};

Testsource.tiles = {
    a: fs.readdirSync(path.resolve(path.join(__dirname, 'fixtures','a'))).reduce(function(memo, basename) {
        var key = basename.split('.').slice(0,3).join('.');
        memo[key] = fs.readFileSync(path.resolve(path.join(__dirname, 'fixtures', 'a', basename)));
        return memo;
    }, {}),
    b: fs.readdirSync(path.resolve(path.join(__dirname, 'fixtures','b'))).reduce(function(memo, basename) {
        var key = basename.split('.').slice(0,3).join('.');
        memo[key] = fs.readFileSync(path.resolve(path.join(__dirname, 'fixtures', 'b', basename)));
        return memo;
    }, {}),
    i: fs.readdirSync(path.resolve(path.join(__dirname, 'fixtures','i'))).reduce(function(memo, basename) {
        var key = basename.split('.').slice(0,3).join('.');
        memo[key] = fs.readFileSync(path.resolve(path.join(__dirname, 'fixtures', 'i', basename)));
        return memo;
    }, {}),
    gz: fs.readdirSync(path.resolve(path.join(__dirname, 'fixtures','gz'))).reduce(function(memo, basename) {
        var key = basename.split('.').slice(0,3).join('.');
        memo[key] = fs.readFileSync(path.resolve(path.join(__dirname, 'fixtures', 'gz', basename)));
        return memo;
    }, {}),
    expires: fs.readdirSync(path.resolve(path.join(__dirname, 'fixtures','a'))).reduce(function(memo, basename) {
        var key = basename.split('.').slice(0,3).join('.');
        memo[key] = fs.readFileSync(path.resolve(path.join(__dirname, 'fixtures', 'a', basename)));
        return memo;
    }, {}),
    invalid: {}
};

// Additional error tile fixtures.
zlib.deflate(new Buffer(0), function(err, deflated) {
    if (err) throw err;
    Testsource.tiles.invalid['0.0.0'] = deflated;
});

Testsource.now = new Date;

function Testsource(uri, callback) {
    if (uri && uri.pathname) uri = uri.pathname.slice(1);

    this.uri = uri;
    if (uri) this.data = {
        minzoom: infos[uri].minzoom,
        maxzoom: infos[uri].maxzoom,
        fillzoom: infos[uri].fillzoom,
        vector_layers: infos[uri].vector_layers
    };
    this.stats = {};
    return callback && callback(null, this);
};

Testsource.prototype.getTile = function(z,x,y,callback) {
    var key = [z,x,y].join('.');
    if (callback.scale == undefined) {
        return callback(new Error("Expected the callback to carry through scale option"));
    }

    if (callback.legacy == undefined) {
        return callback(new Error("Expected the callback to carry through legacy option"));
    }

    if (callback.upgrade == undefined) {
        return callback(new Error("Expected the callback to carry through upgrade option"));
    }

    // Count number of times each key is requested for tests.
    this.stats[key] = this.stats[key] || 0;
    this.stats[key]++;

    // Headers.
    var headers = {
        'Last-Modified': Testsource.now.toUTCString(),
        'ETag':'73f12a518adef759138c142865287a18',
        'Content-Type':'application/x-protobuf'
    };

    // Additional headers.
    if (this.uri === 'expires') headers['expires'] = new Date('2020-01-01').toUTCString();

    if (!Testsource.tiles[this.uri][key]) {
        return callback(new Error('Tile does not exist'));
    } else {
        return callback(null, Testsource.tiles[this.uri][key], headers);
    }
};

Testsource.prototype.getInfo = function(callback) {
    return callback(null, this.data);
};