kibana-with-account/src/ui/public/courier/fetch/__tests__/fetch.js

kibana-with-account is develop based kibana project

import ngMock from 'ngMock';
import expect from 'expect.js';
import sinon from 'auto-release-sinon';

import FetchProvider from 'ui/courier/fetch';
import IndexPatternProvider from 'fixtures/stubbed_logstash_index_pattern';
import searchResp from 'fixtures/search_response';

describe('fetch service', function () {
  require('testUtils/noDigestPromises').activateForSuite();

  let es;
  let fetch;
  let Promise;
  let DocSource;
  let SearchSource;
  let indexPattern;

  beforeEach(ngMock.module('kibana'));
  beforeEach(ngMock.inject(function ($injector, Private) {
    es = $injector.get('es');
    Promise = $injector.get('Promise');
    fetch = Private(FetchProvider);
    indexPattern = Private(IndexPatternProvider);
    DocSource = Private(require('ui/courier/data_source/doc_source'));
    SearchSource = Private(require('ui/courier/data_source/search_source'));
  }));

  describe('#doc(docSource)', function () {
    it('fetches a single doc source', function () {
      const doc = {
        _index: 'test-index',
        _type: 'test-type',
        _id: 'test-id',
      };

      const source = new DocSource({
        index: doc._index,
        type: doc._type,
        id: doc._id
      });

      sinon.stub(es, 'mget').returns(Promise.resolve({
        docs: [doc]
      }));

      return fetch.doc(source).then(function (resp) {
        expect(resp).to.be(doc);
      });
    });
  });

  describe('#search(searchSource)', function () {
    it('fetches a single search source', function () {
      const resp = searchResp;
      const mresp = {
        responses: [resp]
      };

      const source = new SearchSource({
        index: indexPattern
      });

      sinon.stub(es, 'msearch').returns(Promise.resolve(mresp));

      return fetch
      .search(source)
      .then(function (courierResp) {
        expect(courierResp).to.be(resp);
      });
    });
  });
});