update mock and types (#95771)

x-pack/plugins/lens/public/index.ts

@@ -19,7 +19,9 @@ export type {
   SeriesType,
   ValueLabelConfig,
   YAxisMode,
+  XYCurveType,
 } from './xy_visualization/types';
+export type { DataType } from './types';
 export type {
   PieVisualizationState,
   PieLayerState,

x-pack/plugins/lens/public/mocks.tsx

@@ -5,16 +5,20 @@
  * 2.0.
  */
 
+import React from 'react';
 import { LensPublicStart } from '.';
+import { visualizationTypes } from './xy_visualization/types';
 
 export type Start = jest.Mocked<LensPublicStart>;
 
 const createStartContract = (): Start => {
   const startContract: Start = {
-    EmbeddableComponent: jest.fn(() => null),
+    EmbeddableComponent: jest.fn(() => {
+      return <span>Lens Embeddable Component</span>;
+    }),
     canUseEditor: jest.fn(() => true),
     navigateToPrefilledEditor: jest.fn(),
-    getXyVisTypes: jest.fn(),
+    getXyVisTypes: jest.fn().mockReturnValue(new Promise(() => visualizationTypes)),
   };
   return startContract;
 };