From 40bfd12cc55ebfb1641ef21133fb009c23b0106f Mon Sep 17 00:00:00 2001
From: Mark Hopkin <mark.hopkin@elastic.co>
Date: Tue, 15 Oct 2024 23:27:50 +0100
Subject: [PATCH] [Entity Analytics] Allow task status to be "claiming" in
 disable/enable test (#196172)

## Summary

Closes https://github.com/elastic/kibana/issues/196166

The test is checking that when we disable the risk engine, the risk
ewngine task is registered but not actively running.

This check originally checked if the task status was "idle".

We have had a failure where the task status is "claiming", reading the
docs about this task status (below) this is also an acceptable
"non-running" status

```
  // idle: Task Instance isn't being worked on
  // claiming: A Kibana instance has claimed ownership but hasn't started running
  //           the Task Instance yet
```
---
 .../init_and_status_apis.ts                            | 10 +++++++---
 1 file changed, 7 insertions(+), 3 deletions(-)

diff --git a/x-pack/test/security_solution_api_integration/test_suites/entity_analytics/risk_engine/trial_license_complete_tier/init_and_status_apis.ts b/x-pack/test/security_solution_api_integration/test_suites/entity_analytics/risk_engine/trial_license_complete_tier/init_and_status_apis.ts
index 19a9bb85326fa..3224caa24d5e2 100644
--- a/x-pack/test/security_solution_api_integration/test_suites/entity_analytics/risk_engine/trial_license_complete_tier/init_and_status_apis.ts
+++ b/x-pack/test/security_solution_api_integration/test_suites/entity_analytics/risk_engine/trial_license_complete_tier/init_and_status_apis.ts
@@ -19,6 +19,10 @@ import {
 } from '../../utils';
 import { FtrProviderContext } from '../../../../ftr_provider_context';
 
+const expectTaskIsNotRunning = (taskStatus?: string) => {
+  expect(['idle', 'claiming']).contain(taskStatus);
+};
+
 export default ({ getService }: FtrProviderContext) => {
   const es = getService('es');
   const supertest = getService('supertest');
@@ -355,7 +359,7 @@ export default ({ getService }: FtrProviderContext) => {
         expect(status2.body.legacy_risk_engine_status).to.be('NOT_INSTALLED');
 
         expect(status2.body.risk_engine_task_status?.runAt).to.be.a('string');
-        expect(status2.body.risk_engine_task_status?.status).to.be('idle');
+        expectTaskIsNotRunning(status2.body.risk_engine_task_status?.status);
         expect(status2.body.risk_engine_task_status?.startedAt).to.be(undefined);
 
         await riskEngineRoutes.disable();
@@ -373,7 +377,7 @@ export default ({ getService }: FtrProviderContext) => {
         expect(status4.body.legacy_risk_engine_status).to.be('NOT_INSTALLED');
 
         expect(status4.body.risk_engine_task_status?.runAt).to.be.a('string');
-        expect(status4.body.risk_engine_task_status?.status).to.be('idle');
+        expectTaskIsNotRunning(status4.body.risk_engine_task_status?.status);
         expect(status4.body.risk_engine_task_status?.startedAt).to.be(undefined);
       });
 
@@ -394,7 +398,7 @@ export default ({ getService }: FtrProviderContext) => {
         expect(status2.body.legacy_risk_engine_status).to.be('NOT_INSTALLED');
 
         expect(status2.body.risk_engine_task_status?.runAt).to.be.a('string');
-        expect(status2.body.risk_engine_task_status?.status).to.be('idle');
+        expectTaskIsNotRunning(status2.body.risk_engine_task_status?.status);
         expect(status2.body.risk_engine_task_status?.startedAt).to.be(undefined);
       });
     });