test: 💍 fix expression mocks

elastic · Feb 11, 2020 · 3d04a47 · 3d04a47
1 parent 8550b8a
commit 3d04a47
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/src/plugins/expressions/public/mocks.tsx b/src/plugins/expressions/public/mocks.tsx
@@ -31,9 +31,11 @@ export type Start = jest.Mocked<ExpressionsStart>;
 
 const createSetupContract = (): Setup => {
   const setupContract: Setup = {
+    getFunction: jest.fn(),
     getFunctions: jest.fn(),
     getRenderer: jest.fn(),
     getRenderers: jest.fn(),
+    getType: jest.fn(),
     getTypes: jest.fn(),
     registerFunction: jest.fn(),
     registerRenderer: jest.fn(),
@@ -66,9 +68,11 @@ const createStartContract = (): Start => {
     ExpressionDataHandler: jest.fn(),
     ExpressionLoader: jest.fn(),
     ExpressionRenderHandler: jest.fn(),
+    getFunction: jest.fn(),
     getFunctions: jest.fn(),
     getRenderer: jest.fn(),
     getRenderers: jest.fn(),
+    getType: jest.fn(),
     getTypes: jest.fn(),
     loader: jest.fn(),
     ReactExpressionRenderer: jest.fn(props => <></>),