From 441d13e35efea2160835bb6b29d7c752546e8bd7 Mon Sep 17 00:00:00 2001
From: spalger <email@spalger.com>
Date: Wed, 4 Nov 2015 17:02:15 -0600
Subject: [PATCH] [courier/fetch] added tests

---
 .../public/courier/fetch/__tests__/fetch.js   | 73 +++++++++++++++++++
 1 file changed, 73 insertions(+)
 create mode 100644 src/ui/public/courier/fetch/__tests__/fetch.js

diff --git a/src/ui/public/courier/fetch/__tests__/fetch.js b/src/ui/public/courier/fetch/__tests__/fetch.js
new file mode 100644
index 000000000000..82ba398805c9
--- /dev/null
+++ b/src/ui/public/courier/fetch/__tests__/fetch.js
@@ -0,0 +1,73 @@
+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);
+      });
+    });
+  });
+});