sentry/static/app/utils/profiling/spanTree.spec.tsx

import {EntrySpans} from 'sentry/types/event';

function s(partial: Partial<EntrySpans['data'][0]>): EntrySpans['data'][0] {
  return {
    timestamp: 0,
    start_timestamp: 0,
    exclusive_time: 0,
    description: '',
    op: '',
    span_id: '',
    parent_span_id: '',
    trace_id: '',
    hash: '',
    data: {},
    ...partial,
  };
}

import {SpanTree} from './spanTree';

describe('SpanTree', () => {
  it('appends to parent that contains span', () => {
    const tree = new SpanTree([
      s({span_id: '1', timestamp: 1, start_timestamp: 0}),
      s({span_id: '2', timestamp: 0.5, start_timestamp: 0}),
    ]);
    expect(tree.spanTree.children[0].span.span_id).toBe('1');
    expect(tree.spanTree.children[0].children[0].span.span_id).toBe('2');
  });
  it('pushes consequtive span', () => {
    const tree = new SpanTree([
      s({span_id: '1', timestamp: 1, start_timestamp: 0}),
      s({span_id: '2', timestamp: 0.5, start_timestamp: 0}),
      s({span_id: '3', timestamp: 0.8, start_timestamp: 0.5}),
    ]);
    expect(tree.spanTree.children[0].children[0].span.span_id).toBe('2');
    expect(tree.spanTree.children[0].children[1].span.span_id).toBe('3');
  });
  it('marks span as orphaned if end overlaps', () => {
    const tree = new SpanTree([
      s({span_id: '1', timestamp: 1, start_timestamp: 0}),
      s({span_id: '2', timestamp: 1.1, start_timestamp: 0.1}),
    ]);
    expect(tree.orphanedSpans[0].span_id).toBe('2');
  });

  it('iterates over all spans with depth', () => {
    const tree = new SpanTree([
      s({span_id: '1', timestamp: 1, start_timestamp: 0}),
      s({span_id: '2', timestamp: 0.5, start_timestamp: 0}),
      s({span_id: '3', timestamp: 0.2, start_timestamp: 0}),
      s({span_id: '4', timestamp: 1, start_timestamp: 0.5}),
    ]);

    expect(tree.spanTree.children[0].children[1].span.span_id).toBe('4');

    tree.forEach(span => {
      if (span.node.span.span_id === '1') {
        expect(span.depth).toBe(0);
      } else if (span.node.span.span_id === '2') {
        expect(span.depth).toBe(1);
      } else if (span.node.span.span_id === '3') {
        expect(span.depth).toBe(2);
      } else if (span.node.span.span_id === '4') {
        expect(span.depth).toBe(1);
      }
    });
  });
});