[Security Solution] add agent field to generator (#111428) (#111460)

Co-authored-by: Joey F. Poon <joey.poon@elastic.co>
elastic · Sep 7, 2021 · d4d52b0 · d4d52b0
1 parent 48b25a6
commit d4d52b0
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 0 deletions.
diff --git a/x-pack/plugins/security_solution/common/endpoint/data_generators/fleet_agent_generator.ts b/x-pack/plugins/security_solution/common/endpoint/data_generators/fleet_agent_generator.ts
@@ -82,6 +82,10 @@ export class FleetAgentGenerator extends BaseDataGenerator<Agent> {
           action_seq_no: -1,
           active: true,
           enrolled_at: now,
+          agent: {
+            id: this.randomUUID(),
+            version: this.randomVersion(),
+          },
           local_metadata: {
             elastic: {
               agent: {

diff --git a/x-pack/plugins/security_solution/common/endpoint/data_loaders/index_fleet_agent.ts b/x-pack/plugins/security_solution/common/endpoint/data_loaders/index_fleet_agent.ts
@@ -48,6 +48,10 @@ export const indexFleetAgentForHost = async (
 ): Promise<IndexedFleetAgentResponse> => {
   const agentDoc = fleetAgentGenerator.generateEsHit({
     _source: {
+      agent: {
+        id: endpointHost.agent.id,
+        version: endpointHost.agent.version,
+      },
       local_metadata: {
         elastic: {
           agent: {