From b552cfc48212019ab52861a471d25f7b082ceb70 Mon Sep 17 00:00:00 2001
From: Timothy Sullivan <tsullivan@elastic.co>
Date: Thu, 18 Jun 2020 16:25:30 -0700
Subject: [PATCH] fix tests

---
 .../test_helpers/create_mock_levellogger.ts   | 23 +++++++++++++++
 .../create_mock_reportingplugin.ts            | 28 +++++++++++++++----
 2 files changed, 46 insertions(+), 5 deletions(-)
 create mode 100644 x-pack/plugins/reporting/server/test_helpers/create_mock_levellogger.ts

diff --git a/x-pack/plugins/reporting/server/test_helpers/create_mock_levellogger.ts b/x-pack/plugins/reporting/server/test_helpers/create_mock_levellogger.ts
new file mode 100644
index 000000000000000..f5e9a44281cb691
--- /dev/null
+++ b/x-pack/plugins/reporting/server/test_helpers/create_mock_levellogger.ts
@@ -0,0 +1,23 @@
+/*
+ * Copyright Elasticsearch B.V. and/or licensed to Elasticsearch B.V. under one
+ * or more contributor license agreements. Licensed under the Elastic License;
+ * you may not use this file except in compliance with the Elastic License.
+ */
+
+import { LevelLogger } from '../lib';
+
+export function createMockLevelLogger() {
+  // eslint-disable-next-line no-console
+  const consoleLogger = (tag: string) => (message: unknown) => console.log(tag, message);
+  const innerLogger = {
+    get: () => innerLogger,
+    debug: consoleLogger('debug'),
+    info: consoleLogger('info'),
+    warn: consoleLogger('warn'),
+    trace: consoleLogger('trace'),
+    error: consoleLogger('error'),
+    fatal: consoleLogger('fatal'),
+    log: consoleLogger('log'),
+  };
+  return new LevelLogger(innerLogger);
+}
diff --git a/x-pack/plugins/reporting/server/test_helpers/create_mock_reportingplugin.ts b/x-pack/plugins/reporting/server/test_helpers/create_mock_reportingplugin.ts
index 79e1708571c242b..427a6362a725818 100644
--- a/x-pack/plugins/reporting/server/test_helpers/create_mock_reportingplugin.ts
+++ b/x-pack/plugins/reporting/server/test_helpers/create_mock_reportingplugin.ts
@@ -20,7 +20,8 @@ import {
 } from '../browsers';
 import { ReportingInternalSetup, ReportingInternalStart } from '../core';
 import { ReportingStartDeps } from '../types';
-import { ReportingStore, LevelLogger } from '../lib';
+import { ReportingStore } from '../lib';
+import { createMockLevelLogger } from './create_mock_levellogger';
 
 (initializeBrowserDriverFactory as jest.Mock<
   Promise<HeadlessChromiumDriverFactory>
@@ -38,8 +39,12 @@ const createMockPluginSetup = (setupMock?: any): ReportingInternalSetup => {
   };
 };
 
-const createMockPluginStart = (startMock?: any): ReportingInternalStart => {
-  const store = new ReportingStore({} as ReportingCore, {} as LevelLogger);
+const createMockPluginStart = (
+  mockReportingCore: ReportingCore,
+  startMock?: any
+): ReportingInternalStart => {
+  const logger = createMockLevelLogger();
+  const store = new ReportingStore(mockReportingCore, logger);
   return {
     browserDriverFactory: startMock.browserDriverFactory,
     enqueueJob: startMock.enqueueJob,
@@ -63,9 +68,22 @@ export const createMockStartDeps = (startMock?: any): ReportingStartDeps => ({
 
 export const createMockReportingCore = async (
   config: ReportingConfig,
-  setupDepsMock: ReportingInternalSetup | undefined = createMockPluginSetup({}),
-  startDepsMock: ReportingInternalStart | undefined = createMockPluginStart({})
+  setupDepsMock: ReportingInternalSetup | undefined = undefined,
+  startDepsMock: ReportingInternalStart | undefined = undefined
 ) => {
+  if (!setupDepsMock) {
+    setupDepsMock = createMockPluginSetup({});
+  }
+
+  const mockReportingCore = {
+    getConfig: () => config,
+    getElasticsearchService: () => setupDepsMock?.elasticsearch,
+  } as ReportingCore;
+
+  if (!startDepsMock) {
+    startDepsMock = createMockPluginStart(mockReportingCore, {});
+  }
+
   config = config || {};
   const core = new ReportingCore();