pelias-model/test/serialize/test.js

Pelias data models

var Document = require('../../Document'),
    fixtures = require('./fixtures');

module.exports.tests = {};

// simulate the effect of sending the data to easticsearch.
// the result should map correctly to the schema.
function serializeDeserialize( model ){
  return JSON.parse( JSON.stringify( model ) );
}

// test the document serializes as expected
module.exports.tests.minimal = function(test) {
  test('minimal', function(t) {

    // create a new doc and serialize it
    var doc = new Document('mysource','mylayer','myid');
    var s = serializeDeserialize( doc );

    // document meta data
    t.equal(doc.getMeta('id'), 'myid', 'correct _id');

    // document body
    t.deepEqual(s, {
      'source': 'mysource',
      'layer': 'mylayer',
      'source_id': 'myid',
      'name': {},
      'parent': {},
      'address_parts': {},
      'category': [],
      'center_point': {},
      'addendum': {}
    }, 'valid document body');

    t.end();
  });
};

// test the document serializes as expected
module.exports.tests.complete = function(test) {
  test('complete', function(t) {

    // create a new doc and serialize it
    var doc = new Document( 'geoname', 'venue', 1003 )
      .setMeta( 'author', 'peter' )
      .setName( 'default', 'Hackney City Farm' )
      .setName( 'alt', 'Haggerston City Farm' )
      .addParent( 'country', 'Great Britain', '1001', 'GreatB' )
      .addParent( 'neighbourhood', 'Shoreditch', '2002' )
      .addParent( 'locality', 'London', '3003', 'LD', 'whosonfirst' )
      .setAddress( 'number', '10' )
      .setAddress( 'street', 'pelias place' )
      .addCategory( 'foo' )
      .addCategory( 'bar' )
      .removeCategory( 'foo' )
      .setPopulation(10)
      .setPopularity(3)
      .setAddendum('wikipedia', { slug: 'HackneyCityFarm' })
      .setAddendum('geonames', { foreignkey: 1 })
      .setCentroid({ lon: 0.5, lat: 50.1 })
      .setShape(fixtures.new_zealand)
      .setBoundingBox(fixtures.new_zealand_bbox);

    var s = serializeDeserialize( doc );

    // document meta data
    t.equal(doc.getMeta('id'), '1003', 'correct _id');

    // document body
    t.deepEqual(s, {

      // data partitioning
      'source': 'geoname',
      'layer': 'venue',

      // place name (ngram analysis)
      'name':{
        'default': 'Hackney City Farm',
        'alt': 'Haggerston City Farm'
      },

      // address data
      'address_parts':{
        'number': '10',
        'street': 'pelias place'
      },

      // WOF fields
      'parent': {
        'country': ['Great Britain'],
        'country_a': ['GreatB'],
        'country_id': ['1001'],
        'country_source': [null],
        'neighbourhood': ['Shoreditch'],
        'neighbourhood_a': [null],
        'neighbourhood_id': ['2002'],
        'neighbourhood_source': [null],
        'locality': ['London'],
        'locality_a': ['LD'],
        'locality_id': ['3003'],
        'locality_source': ['whosonfirst']
      },

      // geography
      'center_point':{
        'lon': 0.5,
        'lat': 50.1
      },

      // shape
      'shape':{
        'type': 'FeatureCollection',
        'features':[{
          'type': 'Feature',
          'properties': {},
          'geometry':{
            'type': 'Polygon',
            'coordinates': [[[163.82,-47.75],[163.82,-33.9],[179.42,-33.9],[179.42,-47.75],[163.82,-47.75]]]
          }
        }]
      },

      // bbox
      'bounding_box': '{"min_lat":-47.75,"max_lat":-33.9,"min_lon":163.82,"max_lon":179.42}',

      // meta info
      'source_id': '1003',
      'category':['bar'],
      'population': 10,
      'popularity': 3,

      // addendum
      'addendum': {
        'wikipedia': { 'slug': 'HackneyCityFarm' },
        'geonames': { 'foreignkey': 1 }
      }

    }, 'valid document body');

    t.end();
  });
};

module.exports.all = function (tape, common) {

  function test(name, testFunction) {
    return tape('serialize: ' + name, testFunction);
  }

  for( var testCase in module.exports.tests ){
    module.exports.tests[testCase](test, common);
  }
};