adjustments from PR review

elastic · May 26, 2021 · 0ffdafe · 0ffdafe
1 parent fd57e28
commit 0ffdafe
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 6 deletions.
diff --git a/x-pack/plugins/security_solution/common/endpoint/generate_data.ts b/x-pack/plugins/security_solution/common/endpoint/generate_data.ts
@@ -439,6 +439,8 @@ export class EndpointDocGenerator extends BaseDataGenerator {
 
   private createHostData(): HostInfo {
     const hostName = this.randomHostname();
+    const isIsolated = this.randomBoolean();
+
     return {
       agent: {
         version: this.randomVersion(),
@@ -465,10 +467,10 @@ export class EndpointDocGenerator extends BaseDataGenerator {
           applied: this.randomChoice(APPLIED_POLICIES),
         },
         configuration: {
-          isolation: this.randomBoolean(),
+          isolation: isIsolated,
         },
         state: {
-          isolation: this.randomBoolean(),
+          isolation: isIsolated,
         },
       },
     };

diff --git a/x-pack/plugins/security_solution/common/endpoint/index_data.ts b/x-pack/plugins/security_solution/common/endpoint/index_data.ts
@@ -430,10 +430,7 @@ const indexFleetActionsForHost = async (
     const unIsolateAction = fleetActionGenerator.generateResponse({
       action_id: isolateAction.action_id,
       agent_id: agentId,
-      action_data: {
-        command: isolateAction.data.command,
-        comment: 'data generator: bad host isolated',
-      },
+      action_data: isolateAction.data,
     });
 
     await esClient.index(