From c0c805306c6ffa5622dc44d8fee4eeb7ca9b751c Mon Sep 17 00:00:00 2001
From: Luke Elmers <luke.elmers@elastic.co>
Date: Wed, 17 Jun 2020 12:34:09 -0600
Subject: [PATCH] Fix build pipeline unit tests.

---
 .../public/legacy/build_pipeline.test.ts      | 36 ++++++++-----------
 1 file changed, 15 insertions(+), 21 deletions(-)

diff --git a/src/plugins/visualizations/public/legacy/build_pipeline.test.ts b/src/plugins/visualizations/public/legacy/build_pipeline.test.ts
index 9130581963800..9ecd321963e8a 100644
--- a/src/plugins/visualizations/public/legacy/build_pipeline.test.ts
+++ b/src/plugins/visualizations/public/legacy/build_pipeline.test.ts
@@ -28,7 +28,7 @@ import {
 } from './build_pipeline';
 import { Vis } from '..';
 import { dataPluginMock } from '../../../../plugins/data/public/mocks';
-import { IAggConfig } from '../../../../plugins/data/public';
+import { IndexPattern, IAggConfigs } from '../../../../plugins/data/public';
 
 describe('visualize loader pipeline helpers: build pipeline', () => {
   describe('prepareJson', () => {
@@ -344,23 +344,20 @@ describe('visualize loader pipeline helpers: build pipeline', () => {
   describe('buildVislibDimensions', () => {
     const dataStart = dataPluginMock.createStartContract();
 
-    let aggs: IAggConfig[];
+    let aggs: IAggConfigs;
     let vis: Vis;
     let params: any;
 
     beforeEach(() => {
-      aggs = [
+      aggs = dataStart.search.aggs.createAggConfigs({} as IndexPattern, [
         {
           id: '0',
           enabled: true,
-          type: {
-            type: 'metrics',
-            name: 'count',
-          },
+          type: 'count',
           schema: 'metric',
           params: {},
-        } as IAggConfig,
-      ];
+        },
+      ]);
 
       params = {
         searchSource: null,
@@ -393,11 +390,7 @@ describe('visualize loader pipeline helpers: build pipeline', () => {
             ],
           },
           data: {
-            aggs: {
-              getResponseAggs: () => {
-                return aggs;
-              },
-            } as any,
+            aggs,
             searchSource: {} as any,
           },
           isHierarchical: () => {
@@ -422,8 +415,13 @@ describe('visualize loader pipeline helpers: build pipeline', () => {
       });
 
       it('with two numeric metrics, mixed normal and percent mode should have corresponding formatters', async () => {
-        const aggConfig = aggs[0];
-        aggs = [{ ...aggConfig } as IAggConfig, { ...aggConfig, id: '5' } as IAggConfig];
+        aggs.createAggConfig({
+          id: '5',
+          enabled: true,
+          type: 'count',
+          schema: 'metric',
+          params: {},
+        });
 
         vis.params = {
           seriesParams: [
@@ -469,11 +467,7 @@ describe('visualize loader pipeline helpers: build pipeline', () => {
           },
           params: { gauge: {} },
           data: {
-            aggs: {
-              getResponseAggs: () => {
-                return aggs;
-              },
-            } as any,
+            aggs,
             searchSource: {} as any,
           },
           isHierarchical: () => {