// Copyright (c) 2018 Uber Technologies, Inc. // // Licensed under the Apache License, Version 2.0 (the "License"); // you may not use this file except in compliance with the License. // You may obtain a copy of the License at // // http://www.apache.org/licenses/LICENSE-2.0 // // Unless required by applicable law or agreed to in writing, software // distributed under the License is distributed on an "AS IS" BASIS, // WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. // See the License for the specific language governing permissions and // limitations under the License. import { TraceSpan } from '../types/trace'; import spanAncestorIdsSpy from './span-ancestor-ids'; describe('spanAncestorIdsSpy', () => { const ownSpanID = 'ownSpanID'; const firstParentSpanID = 'firstParentSpanID'; const firstParentFirstGrandparentSpanID = 'firstParentFirstGrandparentSpanID'; const firstParentSecondGrandparentSpanID = 'firstParentSecondGrandparentSpanID'; const secondParentSpanID = 'secondParentSpanID'; const rootSpanID = 'rootSpanID'; const span = { references: [ { span: { spanID: firstParentSpanID, references: [ { span: { spanID: firstParentFirstGrandparentSpanID, references: [ { span: { spanID: rootSpanID, }, }, ], }, refType: 'not an ancestor ref type', }, { span: { spanID: firstParentSecondGrandparentSpanID, references: [ { span: { spanID: rootSpanID, }, refType: 'FOLLOWS_FROM', }, ], }, refType: 'CHILD_OF', }, ], }, refType: 'CHILD_OF', }, { span: { spanID: secondParentSpanID, }, refType: 'CHILD_OF', }, ], spanID: ownSpanID, }; const expectedAncestorIds = [firstParentSpanID, firstParentSecondGrandparentSpanID, rootSpanID]; it('returns an empty array if given falsy span', () => { expect(spanAncestorIdsSpy(null)).toEqual([]); }); it('returns an empty array if span has no references', () => { const spanWithoutReferences = { spanID: 'parentlessSpanID', references: [], }; expect(spanAncestorIdsSpy(spanWithoutReferences as unknown as TraceSpan)).toEqual([]); }); it('returns all unique spanIDs from first valid CHILD_OF or FOLLOWS_FROM reference up to the root span', () => { expect(spanAncestorIdsSpy(span as TraceSpan)).toEqual(expectedAncestorIds); }); it('ignores references without a span', () => { const spanWithSomeEmptyReferences = { ...span, references: [{ refType: 'CHILD_OF' }, { refType: 'FOLLOWS_FROM', span: {} }, ...span.references], }; expect(spanAncestorIdsSpy(spanWithSomeEmptyReferences as TraceSpan)).toEqual(expectedAncestorIds); }); });