test(usage_collection): expose a function to mock plugin start (opens…

…earch-project#8235) (opensearch-project#8314) (cherry picked from commit 9fd0f0b) Signed-off-by: Yulong Ruan <[email protected]> Signed-off-by: github-actions[bot] <github-actions[bot]@users.noreply.github.com> Co-authored-by: github-actions[bot] <github-actions[bot]@users.noreply.github.com>
yubonluo · Sep 25, 2024 · a4a283c · a4a283c
1 parent 0f0bf99
commit a4a283c
Showing 1 changed file with 12 additions and 1 deletion.
diff --git a/src/plugins/usage_collection/public/mocks.ts b/src/plugins/usage_collection/public/mocks.ts
@@ -28,9 +28,10 @@
  * under the License.
  */
 
-import { UsageCollectionSetup, METRIC_TYPE } from '.';
+import { UsageCollectionSetup, UsageCollectionStart, METRIC_TYPE } from '.';
 
 export type Setup = jest.Mocked<UsageCollectionSetup>;
+export type Start = jest.Mocked<UsageCollectionStart>;
 
 const createSetupContract = (): Setup => {
   const setupContract: Setup = {
@@ -45,6 +46,16 @@ const createSetupContract = (): Setup => {
   return setupContract;
 };
 
+const createStartContract = (): Start => {
+  const startContract: Start = {
+    reportUiStats: jest.fn(),
+    METRIC_TYPE,
+  };
+
+  return startContract;
+};
+
 export const usageCollectionPluginMock = {
   createSetupContract,
+  createStartContract,
 };