✅ Align tests with new API

elastic · Nov 4, 2020 · 11caf4a · 11caf4a
1 parent cdb7054
commit 11caf4a
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/x-pack/plugins/lens/public/pie_visualization/visualization.test.ts b/x-pack/plugins/lens/public/pie_visualization/visualization.test.ts
@@ -4,15 +4,20 @@
  * you may not use this file except in compliance with the Elastic License.
  */
 
-import { pieVisualization } from './visualization';
+import { getPieVisualization } from './visualization';
 import { PieVisualizationState } from './types';
 import { createMockDatasource, createMockFramePublicAPI } from '../editor_frame_service/mocks';
 import { DatasourcePublicAPI, FramePublicAPI } from '../types';
+import { chartPluginMock } from '../../../../../src/plugins/charts/public/mocks';
 
 jest.mock('../id_generator');
 
 const LAYER_ID = 'l1';
 
+const pieVisualization = getPieVisualization({
+  paletteService: chartPluginMock.createPaletteRegistry(),
+});
+
 function exampleState(): PieVisualizationState {
   return {
     shape: 'pie',