Add scenario for trace with orphan items

packages/kbn-apm-synthtrace/src/scenarios/trace_with_orphan_items.ts

@@ -0,0 +1,149 @@
+/*
+ * Copyright Elasticsearch B.V. and/or licensed to Elasticsearch B.V. under one
+ * or more contributor license agreements. Licensed under the Elastic License
+ * 2.0 and the Server Side Public License, v 1; you may not use this file except
+ * in compliance with, at your election, the Elastic License 2.0 or the Server
+ * Side Public License, v 1.
+ */
+
+import { apm, ApmFields, httpExitSpan, Serializable } from '@kbn/apm-synthtrace-client';
+import { Readable } from 'stream';
+import { Scenario } from '../cli/scenario';
+
+import { RunOptions } from '../cli/utils/parse_run_cli_flags';
+import { getSynthtraceEnvironment } from '../lib/utils/get_synthtrace_environment';
+
+const ENVIRONMENT = getSynthtraceEnvironment(__filename);
+
+const scenario: Scenario<ApmFields> = async (runOptions: RunOptions) => {
+  return {
+    generate: ({ range }) => {
+      const transactionName = 'trace with orphans';
+      const successfulTimestamps = range.interval('1s').rate(3);
+
+      const synthRum = apm
+        .service({ name: 'synth-rum', environment: ENVIRONMENT, agentName: 'rum-js' })
+        .instance('my-instance');
+      const synthNode = apm
+        .service({ name: 'synth-node', environment: ENVIRONMENT, agentName: 'nodejs' })
+        .instance('my-instance');
+      const synthGo = apm
+        .service({ name: 'synth-go', environment: ENVIRONMENT, agentName: 'go' })
+        .instance('my-instance');
+
+      const traces = successfulTimestamps.generator((timestamp) => {
+        // synth-rum
+        return synthGo
+          .transaction({ transactionName })
+          .duration(400)
+          .timestamp(timestamp)
+          .children(
+            // synth-rum -> synth-node
+            synthRum
+              .span(
+                httpExitSpan({
+                  spanName: 'GET /api/products/top',
+                  destinationUrl: 'http://synth-node:3000',
+                })
+              )
+              .duration(300)
+              .timestamp(timestamp)
+              .children(
+                synthRum
+                  .transaction({ transactionName: 'Child Transaction' })
+                  .timestamp(timestamp)
+                  .duration(200)
+                  .children(
+                    synthGo
+                      .span({ spanName: 'custom_operation', spanType: 'custom' })
+                      .timestamp(timestamp)
+                      .duration(100)
+                      .success()
+                  ),
+                // synth-node
+                synthNode
+                  .transaction({ transactionName: 'Initial transaction in synth-node' })
+                  .duration(300)
+                  .timestamp(timestamp)
+                  .children(
+                    synthNode
+                      // synth-node -> synth-go
+                      .span(
+                        httpExitSpan({
+                          spanName: 'GET synth-go:3000',
+                          destinationUrl: 'http://synth-go:3000',
+                        })
+                      )
+                      .timestamp(timestamp)
+                      .duration(400)
+
+                      .children(
+                        // synth-go
+                        synthGo
+                          .transaction({ transactionName: 'Child Transaction' })
+                          .timestamp(timestamp)
+                          .duration(200)
+                          .children(
+                            synthGo
+                              .span({ spanName: 'custom_operation', spanType: 'custom' })
+                              .timestamp(timestamp)
+                              .duration(100)
+                              .success(),
+                            synthGo
+                              .span({ spanName: 'custom_new_operation', spanType: 'custom' })
+                              .timestamp(timestamp)
+                              .duration(100)
+                              .success()
+                          )
+                      )
+                  )
+              )
+          );
+      });
+
+      const successfulTraceEvents = Array.from(
+        successfulTimestamps.generator((timestamp) =>
+          synthNode
+            .transaction({ transactionName: 'successful trace' })
+            .timestamp(timestamp)
+            .duration(1000)
+            .success()
+            .children(
+              synthNode
+                .span({
+                  spanName: 'GET apm-*/_search',
+                  spanType: 'db',
+                  spanSubtype: 'elasticsearch',
+                })
+                .duration(1000)
+                .success()
+                .destination('elasticsearch')
+                .timestamp(timestamp),
+              synthNode
+                .span({ spanName: 'custom_operation', spanType: 'custom' })
+                .duration(100)
+                .success()
+                .timestamp(timestamp)
+            )
+        )
+      );
+
+      const unserialized = Array.from(traces);
+
+      const serialized = unserialized
+        .flatMap((event) => event.serialize())
+        .filter((trace) => trace['transaction.name'] !== 'Child Transaction');
+
+      const unserializedChanged = serialized.map((event) => ({
+        fields: event,
+        serialize: () => {
+          return [event];
+        },
+      })) as Array<Serializable<ApmFields>>;
+
+      return Readable.from([...unserializedChanged, ...successfulTraceEvents]);
+    },
+  };
+};
+
+export default scenario;