From d90a48306cca84225b0e173eb06fe4c1c5d78791 Mon Sep 17 00:00:00 2001
From: Gidi Morris <github@gidi.io>
Date: Tue, 12 Nov 2019 11:24:43 -0500
Subject: [PATCH] feat(task-manager): use ensureScheduling whenever the ID of
 the task is included in lens, maps and telemetry

---
 x-pack/legacy/plugins/lens/server/usage/task.ts                 | 2 +-
 .../legacy/plugins/maps/server/maps_telemetry/telemetry_task.js | 2 +-
 x-pack/legacy/plugins/oss_telemetry/index.d.ts                  | 2 +-
 x-pack/legacy/plugins/oss_telemetry/server/lib/tasks/index.ts   | 2 +-
 x-pack/legacy/plugins/oss_telemetry/test_utils/index.ts         | 2 +-
 5 files changed, 5 insertions(+), 5 deletions(-)

diff --git a/x-pack/legacy/plugins/lens/server/usage/task.ts b/x-pack/legacy/plugins/lens/server/usage/task.ts
index 3cb857a453e1d..0a07c828cf8c9 100644
--- a/x-pack/legacy/plugins/lens/server/usage/task.ts
+++ b/x-pack/legacy/plugins/lens/server/usage/task.ts
@@ -82,7 +82,7 @@ function scheduleTasks(server: Server) {
     // function block.
     (async () => {
       try {
-        await taskManager.schedule({
+        await taskManager.ensureScheduling({
           id: TASK_ID,
           taskType: TELEMETRY_TASK_TYPE,
           state: { byDate: {}, suggestionsByDate: {}, saved: {}, runs: 0 },
diff --git a/x-pack/legacy/plugins/maps/server/maps_telemetry/telemetry_task.js b/x-pack/legacy/plugins/maps/server/maps_telemetry/telemetry_task.js
index 3702bc8e29539..00eca827c4a1a 100644
--- a/x-pack/legacy/plugins/maps/server/maps_telemetry/telemetry_task.js
+++ b/x-pack/legacy/plugins/maps/server/maps_telemetry/telemetry_task.js
@@ -29,7 +29,7 @@ export function scheduleTask(server) {
     // function block.
     (async () => {
       try {
-        await taskManager.schedule({
+        await taskManager.ensureScheduling({
           id: TASK_ID,
           taskType: TELEMETRY_TASK_TYPE,
           state: { stats: {}, runs: 0 },
diff --git a/x-pack/legacy/plugins/oss_telemetry/index.d.ts b/x-pack/legacy/plugins/oss_telemetry/index.d.ts
index 9f735c676fe6d..ef55a66e68fcb 100644
--- a/x-pack/legacy/plugins/oss_telemetry/index.d.ts
+++ b/x-pack/legacy/plugins/oss_telemetry/index.d.ts
@@ -46,7 +46,7 @@ export interface HapiServer {
     };
     task_manager: {
       registerTaskDefinitions: (opts: any) => void;
-      schedule: (opts: any) => Promise<void>;
+      ensureScheduling: (opts: any) => Promise<void>;
       fetch: (
         opts: any
       ) => Promise<{
diff --git a/x-pack/legacy/plugins/oss_telemetry/server/lib/tasks/index.ts b/x-pack/legacy/plugins/oss_telemetry/server/lib/tasks/index.ts
index eaa8cc7405821..9ea783486f58e 100644
--- a/x-pack/legacy/plugins/oss_telemetry/server/lib/tasks/index.ts
+++ b/x-pack/legacy/plugins/oss_telemetry/server/lib/tasks/index.ts
@@ -42,7 +42,7 @@ export function scheduleTasks(server: HapiServer) {
     // function block.
     (async () => {
       try {
-        await taskManager.schedule({
+        await taskManager.ensureScheduling({
           id: `${PLUGIN_ID}-${VIS_TELEMETRY_TASK}`,
           taskType: VIS_TELEMETRY_TASK,
           state: { stats: {}, runs: 0 },
diff --git a/x-pack/legacy/plugins/oss_telemetry/test_utils/index.ts b/x-pack/legacy/plugins/oss_telemetry/test_utils/index.ts
index 7168f598dca23..67b331f4d4d5c 100644
--- a/x-pack/legacy/plugins/oss_telemetry/test_utils/index.ts
+++ b/x-pack/legacy/plugins/oss_telemetry/test_utils/index.ts
@@ -50,7 +50,7 @@ export const getMockKbnServer = (
     xpack_main: {},
     task_manager: {
       registerTaskDefinitions: (opts: any) => undefined,
-      schedule: (opts: any) => Promise.resolve(),
+      ensureScheduling: (opts: any) => Promise.resolve(),
       fetch: mockTaskFetch,
     },
   },