fix mocks

pgayvallet · pgayvallet · commit f23af79b4c77 · 2020-04-28T11:05:24.000+02:00
diff --git a/x-pack/plugins/licensing/server/licensing.mocks.ts b/x-pack/plugins/licensing/server/licensing.mocks.ts
@@ -4,15 +4,17 @@
  * you may not use this file except in compliance with the Elastic License.
  */
 import { BehaviorSubject } from 'rxjs';
-import { LicensingPluginSetup } from './types';
+import { LicensingPluginSetup, LicensingPluginStart } from './types';
 import { licenseMock } from '../common/licensing.mock';
+import { featureUsageMock } from './services/feature_usage_service.mock';
 
 const createSetupMock = () => {
   const license = licenseMock.createLicense();
   const mock: jest.Mocked<LicensingPluginSetup> = {
     license$: new BehaviorSubject(license),
     refresh: jest.fn(),
     createLicensePoller: jest.fn(),
+    featureUsage: featureUsageMock.createSetup(),
   };
   mock.refresh.mockResolvedValue(license);
   mock.createLicensePoller.mockReturnValue({
@@ -23,7 +25,16 @@ const createSetupMock = () => {
   return mock;
 };
 
+const createStartMock = (): jest.Mocked<LicensingPluginStart> => {
+  const mock = {
+    featureUsage: featureUsageMock.createStart(),
+  };
+
+  return mock;
+};
+
 export const licensingMock = {
   createSetup: createSetupMock,
+  createStart: createStartMock,
   createLicense: licenseMock.createLicense,
 };
