Move task partitions tests

x-pack/test/plugin_api_integration/plugins/sample_task_plugin/server/init_routes.ts

@@ -16,6 +16,7 @@ import {
 } from '@kbn/core/server';
 import { EventEmitter } from 'events';
 import { TaskManagerStartContract } from '@kbn/task-manager-plugin/server';
+import { BACKGROUND_TASK_NODE_SO_NAME } from '@kbn/task-manager-plugin/server/saved_objects';
 
 const scope = 'testing';
 const taskManagerQuery = {
@@ -398,4 +399,40 @@ export function initRoutes(
       }
     }
   );
+
+  router.post(
+    {
+      path: `/api/update_kibana_node`,
+      validate: {
+        body: schema.object({
+          id: schema.string(),
+          lastSeen: schema.string(),
+        }),
+      },
+    },
+    async function (
+      context: RequestHandlerContext,
+      req: KibanaRequest<any, any, any, any>,
+      res: KibanaResponseFactory
+    ): Promise<IKibanaResponse<any>> {
+      const { id, lastSeen } = req.body;
+
+      const client = (await context.core).savedObjects.getClient({
+        includedHiddenTypes: [BACKGROUND_TASK_NODE_SO_NAME],
+      });
+      const node = await client.update(
+        BACKGROUND_TASK_NODE_SO_NAME,
+        id,
+        {
+          id,
+          last_seen: lastSeen,
+        },
+        { upsert: { id, last_seen: lastSeen }, refresh: false, retryOnConflict: 3 }
+      );
+
+      return res.ok({
+        body: node,
+      });
+    }
+  );
 }

x-pack/test/plugin_api_integration/test_suites/task_manager/index.ts

@@ -18,6 +18,7 @@ export default function ({ loadTestFile }: FtrProviderContext) {
     loadTestFile(require.resolve('./task_management_removed_types'));
     loadTestFile(require.resolve('./check_registered_task_types'));
     loadTestFile(require.resolve('./kibana_discovery_service'));
+    loadTestFile(require.resolve('./task_partitions'));
 
     loadTestFile(require.resolve('./migrations'));
   });

x-pack/test/plugin_api_integration/test_suites/task_manager/task_partitions.ts

@@ -166,7 +166,7 @@ export default function ({ getService }: FtrProviderContext) {
       await updateKibanaNodes();
       await setTimeoutAsync(10000);
 
-      const tasksToSchedule = [];
+      const tasksToSchedule: Array<Promise<SerializedConcreteTaskInstance>> = [];
       for (let i = 0; i < 3; i++) {
         tasksToSchedule.push(
           scheduleTask({

x-pack/test/task_manager_claimer_update_by_query/test_suites/task_manager/index.ts

@@ -16,7 +16,6 @@ export default function ({ loadTestFile }: FtrProviderContext) {
     loadTestFile(require.resolve('./task_management'));
     loadTestFile(require.resolve('./task_management_scheduled_at'));
     loadTestFile(require.resolve('./task_management_removed_types'));
-    loadTestFile(require.resolve('./task_partitions'));
 
     loadTestFile(require.resolve('./migrations'));
   });