diff --git a/src/ui/public/visualize/__tests__/visualization_editor.js b/src/ui/public/visualize/__tests__/visualization_editor.js index faf5280b3ba3f..b7981b4261505 100644 --- a/src/ui/public/visualize/__tests__/visualization_editor.js +++ b/src/ui/public/visualize/__tests__/visualization_editor.js @@ -5,6 +5,7 @@ import { VisProvider } from 'ui/vis'; import FixturesStubbedLogstashIndexPatternProvider from 'fixtures/stubbed_logstash_index_pattern'; import FixturesStubbedSearchSourceProvider from 'fixtures/stubbed_search_source'; import MockState from 'fixtures/mock_state'; +import { AggResponseIndexProvider } from 'ui/agg_response'; describe('visualization_editor directive', function () { let $rootScope; @@ -18,6 +19,7 @@ describe('visualization_editor directive', function () { let appState; let $timeout; let vis; + let aggResponse; beforeEach(ngMock.module('kibana', 'kibana/table_vis')); beforeEach(ngMock.inject(function (Private, $injector) { @@ -30,6 +32,7 @@ describe('visualization_editor directive', function () { appState.toJSON = () => { return {}; }; indexPattern = Private(FixturesStubbedLogstashIndexPatternProvider); searchSource = Private(FixturesStubbedSearchSourceProvider); + aggResponse = Private(AggResponseIndexProvider); const requiresSearch = false; vis = new CreateVis(null, requiresSearch); @@ -43,7 +46,7 @@ describe('visualization_editor directive', function () { vis.aggs.forEach(function (agg, i) { agg.id = 'agg_' + (i + 1); }); $rootScope.vis = vis; - $rootScope.visData = esResponse; + $rootScope.visData = aggResponse.tabify(vis, esResponse); $rootScope.uiState = require('fixtures/mock_ui_state'); $rootScope.searchSource = searchSource; $el = $('<visualization-editor vis="vis" vis-data="visData" ui-state="uiState" search-source="searchSource">');