datahub/test/backend.elasticsearch.test.js

(function ($) {
module("Backend ElasticSearch");

test("ElasticSearch queryNormalize", function() { 
  var backend = new recline.Backend.ElasticSearch();
  var in_ = new recline.Model.Query();
  in_.set({q: ''});
  var out = backend._normalizeQuery(in_);
  equal(out.q, undefined);
  deepEqual(out.query.match_all, {});

  var backend = new recline.Backend.ElasticSearch();
  var in_ = new recline.Model.Query().toJSON();
  in_.q = '';
  var out = backend._normalizeQuery(in_);
  equal(out.q, undefined);
  deepEqual(out.query.match_all, {});

  var in_ = new recline.Model.Query().toJSON();
  in_.q = 'abc';
  var out = backend._normalizeQuery(in_);
  equal(out.query.query_string.query, 'abc');

  var in_ = new recline.Model.Query();
  in_.addTermFilter('xyz', 'XXX');
  in_ = in_.toJSON();
  var out = backend._normalizeQuery(in_);
  deepEqual(out.filter.and[0], {term: { xyz: 'XXX'}});
});

var mapping_data = {
  "note": {
    "properties": {
      "_created": {
        "format": "dateOptionalTime", 
        "type": "date"
      }, 
      "_last_modified": {
        "format": "dateOptionalTime", 
        "type": "date"
      }, 
      "end": {
        "type": "string"
      }, 
      "owner": {
        "type": "string"
      }, 
      "start": {
        "type": "string"
      }, 
      "title": {
        "type": "string"
      }
    }
  }
};

var sample_data = {
  "_shards": {
    "failed": 0, 
    "successful": 5, 
    "total": 5
  }, 
  "hits": {
    "hits": [
      {
        "_id": "u3rpLyuFS3yLNXrtxWkMwg", 
        "_index": "hypernotes", 
        "_score": 1.0, 
        "_source": {
          "_created": "2012-02-24T17:53:57.286Z", 
          "_last_modified": "2012-02-24T17:53:57.286Z", 
          "owner": "tester", 
          "title": "Note 1"
        }, 
        "_type": "note"
      }, 
      {
        "_id": "n7JMkFOHSASJCVTXgcpqkA", 
        "_index": "hypernotes", 
        "_score": 1.0, 
        "_source": {
          "_created": "2012-02-24T17:53:57.290Z", 
          "_last_modified": "2012-02-24T17:53:57.290Z", 
          "owner": "tester", 
          "title": "Note 3"
        }, 
        "_type": "note"
      }, 
      {
        "_id": "g7UMA55gTJijvsB3dFitzw", 
        "_index": "hypernotes", 
        "_score": 1.0, 
        "_source": {
          "_created": "2012-02-24T17:53:57.289Z", 
          "_last_modified": "2012-02-24T17:53:57.289Z", 
          "owner": "tester", 
          "title": "Note 2"
        }, 
        "_type": "note"
      }
    ], 
    "max_score": 1.0, 
    "total": 3
  }, 
  "timed_out": false, 
  "took": 2
};

test("ElasticSearch", function() { 
  var backend = new recline.Backend.ElasticSearch();
  var dataset = new recline.Model.Dataset({
      url: 'https://localhost:9200/my-es-db/my-es-type'
    },
    backend
  );

  var stub = sinon.stub($, 'ajax', function(options) {
    if (options.url.indexOf('_mapping') != -1) {
      return {
        done: function(callback) {
          callback(mapping_data);
          return this;
        },
        fail: function() {
          return this;
        }
      }
    } else {
      return {
        done: function(callback) {
          callback(sample_data);
        },
        fail: function() {
        }
      }
    }
  });

  dataset.fetch().then(function(dataset) {
    deepEqual(['_created', '_last_modified', 'end', 'owner', 'start', 'title'], _.pluck(dataset.fields.toJSON(), 'id'));
    dataset.query().then(function(docList) {
      equal(3, dataset.docCount);
      equal(3, docList.length);
      equal('Note 1', docList.models[0].get('title'));
      start();
    });
  });
  $.ajax.restore();
});

})(this.jQuery);