// Licensed to the .NET Foundation under one or more agreements.
// The .NET Foundation licenses this file to you under the MIT license.
using Aspire.Dashboard.Model;
using Aspire.Dashboard.Otlp.Model;
using Aspire.Tests.Shared.Telemetry;
using Microsoft.Extensions.Logging.Abstractions;
using Xunit;
namespace Aspire.Dashboard.Tests.Model;
public sealed class TraceHelpersTests
{
[Fact]
public void GetOrderedResources_SingleSpan_GroupedResult()
{
// Arrange
var context = new OtlpContext { Logger = NullLogger.Instance, Options = new() };
var app1 = new OtlpResource("app1", "instance", uninstrumentedPeer: false, context);
var trace = new OtlpTrace(new byte[] { 1, 2, 3 }, DateTime.MinValue);
var scope = TelemetryTestHelpers.CreateOtlpScope(context);
trace.AddSpan(TelemetryTestHelpers.CreateOtlpSpan(app1, trace, scope, spanId: "1", parentSpanId: null, startDate: new DateTime(2001, 1, 1, 1, 1, 1, DateTimeKind.Utc)));
// Act
var results = TraceHelpers.GetOrderedResources(trace);
// Assert
Assert.Collection(results,
g =>
{
Assert.Equal(app1, g.Resource);
});
}
[Fact]
public void GetOrderedResources_MultipleUnparentedSpans_GroupedResult()
{
// Arrange
var context = new OtlpContext { Logger = NullLogger.Instance, Options = new() };
var app1 = new OtlpResource("app1", "instance", uninstrumentedPeer: false, context);
var app2 = new OtlpResource("app2", "instance", uninstrumentedPeer: false, context);
var trace = new OtlpTrace(new byte[] { 1, 2, 3 }, DateTime.MinValue);
var scope = TelemetryTestHelpers.CreateOtlpScope(context);
trace.AddSpan(TelemetryTestHelpers.CreateOtlpSpan(app2, trace, scope, spanId: "1-2", parentSpanId: "1", startDate: new DateTime(2001, 1, 1, 1, 1, 2, DateTimeKind.Utc)));
trace.AddSpan(TelemetryTestHelpers.CreateOtlpSpan(app1, trace, scope, spanId: "1-1", parentSpanId: "1", startDate: new DateTime(2001, 1, 1, 1, 1, 1, DateTimeKind.Utc)));
// Act
var results = TraceHelpers.GetOrderedResources(trace);
// Assert
Assert.Collection(results,
g =>
{
Assert.Equal(app1, g.Resource);
},
g =>
{
Assert.Equal(app2, g.Resource);
});
}
[Fact]
public void GetOrderedResources_ChildSpanAfterParentSpan_GroupedResult()
{
// Arrange
var context = new OtlpContext { Logger = NullLogger.Instance, Options = new() };
var app1 = new OtlpResource("app1", "instance", uninstrumentedPeer: false, context);
var app2 = new OtlpResource("app2", "instance", uninstrumentedPeer: false, context);
var trace = new OtlpTrace(new byte[] { 1, 2, 3 }, DateTime.MinValue);
var scope = TelemetryTestHelpers.CreateOtlpScope(context);
trace.AddSpan(TelemetryTestHelpers.CreateOtlpSpan(app1, trace, scope, spanId: "1", parentSpanId: null, startDate: new DateTime(2001, 1, 1, 1, 1, 2, DateTimeKind.Utc)));
trace.AddSpan(TelemetryTestHelpers.CreateOtlpSpan(app2, trace, scope, spanId: "1-1", parentSpanId: "1", startDate: new DateTime(2001, 1, 1, 1, 1, 1, DateTimeKind.Utc)));
// Act
var results = TraceHelpers.GetOrderedResources(trace);
// Assert
Assert.Collection(results,
g =>
{
Assert.Equal(app1, g.Resource);
},
g =>
{
Assert.Equal(app2, g.Resource);
});
}
[Fact]
public void GetOrderedResources_ChildSpanDifferentStartTime_GroupedResult()
{
// Arrange
var context = new OtlpContext { Logger = NullLogger.Instance, Options = new() };
var app1 = new OtlpResource("app1", "instance", uninstrumentedPeer: false, context);
var app2 = new OtlpResource("app2", "instance", uninstrumentedPeer: false, context);
var app3 = new OtlpResource("app3", "instance", uninstrumentedPeer: false, context);
var trace = new OtlpTrace(new byte[] { 1, 2, 3 }, DateTime.MinValue);
var scope = TelemetryTestHelpers.CreateOtlpScope(context);
trace.AddSpan(TelemetryTestHelpers.CreateOtlpSpan(app1, trace, scope, spanId: "1", parentSpanId: null, startDate: new DateTime(2001, 1, 1, 1, 1, 2, DateTimeKind.Utc)));
trace.AddSpan(TelemetryTestHelpers.CreateOtlpSpan(app2, trace, scope, spanId: "1-1", parentSpanId: "1", startDate: new DateTime(2001, 1, 1, 1, 1, 3, DateTimeKind.Utc)));
trace.AddSpan(TelemetryTestHelpers.CreateOtlpSpan(app3, trace, scope, spanId: "1-1-1", parentSpanId: "1-1", startDate: new DateTime(2001, 1, 1, 1, 1, 2, DateTimeKind.Utc)));
trace.AddSpan(TelemetryTestHelpers.CreateOtlpSpan(app3, trace, scope, spanId: "1-2", parentSpanId: "1", startDate: new DateTime(2001, 1, 1, 1, 1, 2, DateTimeKind.Utc)));
// Act
var results = TraceHelpers.GetOrderedResources(trace);
// Assert
Assert.Collection(results,
g =>
{
Assert.Equal(app1, g.Resource);
},
g =>
{
Assert.Equal(app3, g.Resource);
},
g =>
{
Assert.Equal(app2, g.Resource);
});
}
[Fact]
public void GetOrderedResources_HasUninstrumentedPeer_AddedToResults()
{
// Arrange
var context = new OtlpContext { Logger = NullLogger.Instance, Options = new() };
var app1 = new OtlpResource("app1", "instance", uninstrumentedPeer: false, context);
var app2 = new OtlpResource("app2", "instance", uninstrumentedPeer: false, context);
var app3 = new OtlpResource("app3", "instance", uninstrumentedPeer: true, context);
var trace = new OtlpTrace(new byte[] { 1, 2, 3 }, DateTime.MinValue);
var scope = TelemetryTestHelpers.CreateOtlpScope(context);
trace.AddSpan(TelemetryTestHelpers.CreateOtlpSpan(app1, trace, scope, spanId: "1", parentSpanId: null, startDate: new DateTime(2001, 1, 1, 1, 1, 2, DateTimeKind.Utc)));
trace.AddSpan(TelemetryTestHelpers.CreateOtlpSpan(app2, trace, scope, spanId: "1-1", parentSpanId: "1", startDate: new DateTime(2001, 1, 1, 1, 1, 3, DateTimeKind.Utc), uninstrumentedPeer: app3));
// Act
var results = TraceHelpers.GetOrderedResources(trace);
// Assert
Assert.Collection(results,
g =>
{
Assert.Equal(app1, g.Resource);
},
g =>
{
Assert.Equal(app2, g.Resource);
},
g =>
{
Assert.Equal(app3, g.Resource);
});
}
[Fact]
public void GetOrderedResources_SameStartTime_UninstrumentedPeerAfterInstrumentedResources()
{
var context = new OtlpContext { Logger = NullLogger.Instance, Options = new() };
var app1 = new OtlpResource("app1", "instance", uninstrumentedPeer: false, context);
var app2 = new OtlpResource("app2", "instance", uninstrumentedPeer: false, context);
var peer = new OtlpResource("peer", instanceId: null, uninstrumentedPeer: true, context);
var trace = new OtlpTrace(new byte[] { 1, 2, 3 }, DateTime.MinValue);
var scope = TelemetryTestHelpers.CreateOtlpScope(context);
var startTime = new DateTime(2001, 1, 1, 1, 1, 1, DateTimeKind.Utc);
trace.AddSpan(TelemetryTestHelpers.CreateOtlpSpan(app1, trace, scope, spanId: "1", parentSpanId: null, startDate: startTime, uninstrumentedPeer: peer));
trace.AddSpan(TelemetryTestHelpers.CreateOtlpSpan(app2, trace, scope, spanId: "2", parentSpanId: null, startDate: startTime));
var results = TraceHelpers.GetOrderedResources(trace);
Assert.Collection(results,
resource => Assert.False(resource.Resource.UninstrumentedPeer),
resource => Assert.False(resource.Resource.UninstrumentedPeer),
resource => Assert.Same(peer, resource.Resource));
}
[Fact]
public void GetOrderedResources_DifferentResourceInstancesWithSameKey_GroupedResult()
{
var context = new OtlpContext { Logger = NullLogger.Instance, Options = new() };
var app1 = new OtlpResource("app1", "instance", uninstrumentedPeer: false, context);
var equivalentApp1 = new OtlpResource("app1", "instance", uninstrumentedPeer: true, context);
var trace = new OtlpTrace(new byte[] { 1, 2, 3 }, DateTime.MinValue);
var scope = TelemetryTestHelpers.CreateOtlpScope(context);
trace.AddSpan(TelemetryTestHelpers.CreateOtlpSpan(app1, trace, scope, spanId: "1", parentSpanId: null, startDate: new DateTime(2001, 1, 1, 1, 1, 1, DateTimeKind.Utc), uninstrumentedPeer: equivalentApp1));
var results = TraceHelpers.GetOrderedResources(trace);
var result = Assert.Single(results);
Assert.Same(app1, result.Resource);
Assert.Equal(2, result.TotalSpans);
}
}