fix types

elastic · Jul 16, 2019 · dd49f8a · dd49f8a
1 parent d0ae61c
commit dd49f8a
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/x-pack/legacy/plugins/lens/public/editor_frame_plugin/mocks.tsx b/x-pack/legacy/plugins/lens/public/editor_frame_plugin/mocks.tsx
@@ -13,7 +13,7 @@ export function createMockVisualization(): jest.Mocked<Visualization> {
   return {
     getPersistableState: jest.fn(_state => ({})),
     getSuggestions: jest.fn(_options => []),
-    initialize: jest.fn(_state => ({})),
+    initialize: jest.fn((_datasource, _state?) => ({})),
     renderConfigPanel: jest.fn(),
     toExpression: jest.fn((_state, _datasource) => null),
   };
@@ -38,7 +38,7 @@ export function createMockDatasource(): DatasourceMock {
     getDatasourceSuggestionsFromCurrentState: jest.fn(_state => []),
     getPersistableState: jest.fn(),
     getPublicAPI: jest.fn((_state, _setState) => publicAPIMock),
-    initialize: jest.fn(_state => Promise.resolve()),
+    initialize: jest.fn((_state?) => Promise.resolve()),
     renderDataPanel: jest.fn(),
     toExpression: jest.fn(_state => null),