Function makeRecordingTracer

apps/cloud/src/mcp/telemetry.test.ts:15–51

()

};

const makeRecordingTracer = (): {
  tracer: Tracer.Tracer;
  attributesOf: (name: string) => ReadonlyMap<string, unknown> | undefined;
} => {
  const spans: RecordedSpan[] = [];
  const tracer: Tracer.Tracer = {
    span: (options) => {
      const attributes = new Map<string, unknown>();
      spans.push({ name: options.name, attributes });
      let status: Tracer.SpanStatus = { _tag: "Started", startTime: options.startTime };
      return {
        _tag: "Span",
        name: options.name,
        spanId: `span-${spans.length}`,
        traceId: "trace-1",
        parent: options.parent,
        annotations: options.annotations,
        get status() {
          return status;
        },
        attributes,
        links: options.links,
        sampled: options.sampled,
        kind: options.kind,
        end: (endTime, exit) => {
          status = { _tag: "Ended", startTime: options.startTime, endTime, exit };
        },
        attribute: (key, value) => {
          attributes.set(key, value);
        },
        event: () => undefined,
        addLinks: () => undefined,
      };
    },
  };
  return { tracer, attributesOf: (name) => spans.find((span) => span.name === name)?.attributes };
};

const expectDefined: <T>(value: T) => asserts value is NonNullable<T> = (value) => {
  expect(value).toBeDefined();