|
// 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.Model.Otlp;
using Aspire.Dashboard.Otlp.Model;
using Aspire.Tests.Shared.Telemetry;
using Microsoft.Extensions.Logging;
using Microsoft.Extensions.Logging.Abstractions;
using Microsoft.Extensions.Logging.Testing;
using OpenTelemetry.Proto.Common.V1;
using OpenTelemetry.Proto.Resource.V1;
using Xunit;
namespace Aspire.Dashboard.Tests.Model;
public sealed class ResourcesSelectHelpersTests
{
[Fact]
public void GetResource_SameNameAsReplica_GetInstance()
{
// Arrange
var appVMs = ResourcesSelectHelpers.CreateResources(new List<OtlpResource>
{
CreateOtlpResource(name: "multiple", instanceId: "instance"),
CreateOtlpResource(name: "multiple", instanceId: "instanceabc"),
CreateOtlpResource(name: "singleton", instanceId: "instanceabc")
});
Assert.Collection(appVMs,
app =>
{
Assert.Equal("multiple", app.Name);
Assert.Equal(OtlpResourceType.ResourceGrouping, app.Id!.Type);
Assert.Null(app.Id!.InstanceId);
},
app =>
{
Assert.Equal("multiple-instance", app.Name);
Assert.Equal(OtlpResourceType.Instance, app.Id!.Type);
Assert.Equal("multiple-instance", app.Id!.InstanceId);
},
app =>
{
Assert.Equal("multiple-instanceabc", app.Name);
Assert.Equal(OtlpResourceType.Instance, app.Id!.Type);
Assert.Equal("multiple-instanceabc", app.Id!.InstanceId);
},
app =>
{
Assert.Equal("singleton", app.Name);
Assert.Equal(OtlpResourceType.Singleton, app.Id!.Type);
Assert.Equal("singleton-instanceabc", app.Id!.InstanceId);
});
// Act
var app = appVMs.GetResource(NullLogger.Instance, "multiple-instanceabc", canSelectGrouping: false, null!);
// Assert
Assert.Equal("multiple-instanceabc", app.Id!.InstanceId);
Assert.Equal(OtlpResourceType.Instance, app.Id!.Type);
}
[Fact]
public void GetResource_NameDifferentByCase_Merge()
{
// Arrange
var appVMs = ResourcesSelectHelpers.CreateResources(new List<OtlpResource>
{
CreateOtlpResource(name: "name", instanceId: "instance"),
CreateOtlpResource(name: "NAME", instanceId: "instanceabc")
});
Assert.Collection(appVMs,
app =>
{
Assert.Equal("name", app.Name);
Assert.Equal(OtlpResourceType.ResourceGrouping, app.Id!.Type);
Assert.Null(app.Id!.InstanceId);
},
app =>
{
Assert.Equal("NAME-instance", app.Name);
Assert.Equal(OtlpResourceType.Instance, app.Id!.Type);
Assert.Equal("name-instance", app.Id!.InstanceId);
},
app =>
{
Assert.Equal("NAME-instanceabc", app.Name);
Assert.Equal(OtlpResourceType.Instance, app.Id!.Type);
Assert.Equal("name-instanceabc", app.Id!.InstanceId);
});
var testSink = new TestSink();
var factory = LoggerFactory.Create(b => b.AddProvider(new TestLoggerProvider(testSink)));
// Act
var app = appVMs.GetResource(factory.CreateLogger("Test"), "name-instance", canSelectGrouping: false, null!);
// Assert
Assert.Equal("name-instance", app.Id!.InstanceId);
Assert.Equal(OtlpResourceType.Instance, app.Id!.Type);
Assert.Empty(testSink.Writes);
}
[Fact]
public void GetResource_MultipleMatches_UseFirst()
{
// Arrange
var apps = new Dictionary<string, OtlpResource>();
var appVMs = new List<SelectViewModel<ResourceTypeDetails>>
{
new SelectViewModel<ResourceTypeDetails>() { Name = "test", Id = ResourceTypeDetails.CreateSingleton("test-abc", "test") },
new SelectViewModel<ResourceTypeDetails>() { Name = "test", Id = ResourceTypeDetails.CreateSingleton("test-def", "test") }
};
var testSink = new TestSink();
var factory = LoggerFactory.Create(b => b.AddProvider(new TestLoggerProvider(testSink)));
// Act
var app = appVMs.GetResource(factory.CreateLogger("Test"), "test", canSelectGrouping: false, null!);
// Assert
Assert.Equal("test-abc", app.Id!.InstanceId);
Assert.Equal(OtlpResourceType.Singleton, app.Id!.Type);
Assert.Single(testSink.Writes);
}
[Fact]
public void GetResource_SelectGroup_NotEnabled_ReturnNull()
{
// Arrange
var appVMs = ResourcesSelectHelpers.CreateResources(new List<OtlpResource>
{
CreateOtlpResource(name: "app", instanceId: "123"),
CreateOtlpResource(name: "app", instanceId: "456")
});
Assert.Collection(appVMs,
app =>
{
Assert.Equal("app", app.Name);
Assert.Equal(OtlpResourceType.ResourceGrouping, app.Id!.Type);
Assert.Null(app.Id!.InstanceId);
},
app =>
{
Assert.Equal("app-123", app.Name);
Assert.Equal(OtlpResourceType.Instance, app.Id!.Type);
Assert.Equal("app-123", app.Id!.InstanceId);
},
app =>
{
Assert.Equal("app-456", app.Name);
Assert.Equal(OtlpResourceType.Instance, app.Id!.Type);
Assert.Equal("app-456", app.Id!.InstanceId);
});
// Act
var app = appVMs.GetResource(NullLogger.Instance, "app", canSelectGrouping: false, null!);
// Assert
Assert.Null(app);
}
[Fact]
public void GetResource_SelectGroup_Enabled_ReturnGroup()
{
// Arrange
var appVMs = ResourcesSelectHelpers.CreateResources(new List<OtlpResource>
{
CreateOtlpResource(name: "app", instanceId: "123"),
CreateOtlpResource(name: "app", instanceId: "456")
});
Assert.Collection(appVMs,
app =>
{
Assert.Equal("app", app.Name);
Assert.Equal(OtlpResourceType.ResourceGrouping, app.Id!.Type);
Assert.Null(app.Id!.InstanceId);
},
app =>
{
Assert.Equal("app-123", app.Name);
Assert.Equal(OtlpResourceType.Instance, app.Id!.Type);
Assert.Equal("app-123", app.Id!.InstanceId);
},
app =>
{
Assert.Equal("app-456", app.Name);
Assert.Equal(OtlpResourceType.Instance, app.Id!.Type);
Assert.Equal("app-456", app.Id!.InstanceId);
});
// Act
var app = appVMs.GetResource(NullLogger.Instance, "app", canSelectGrouping: true, null!);
// Assert
Assert.Equal("app", app.Name);
Assert.Equal(OtlpResourceType.ResourceGrouping, app.Id!.Type);
}
private static OtlpResource CreateOtlpResource(string name, string instanceId)
{
var resource = new Resource
{
Attributes =
{
new KeyValue { Key = "service.name", Value = new AnyValue { StringValue = name } },
new KeyValue { Key = "service.instance.id", Value = new AnyValue { StringValue = instanceId } }
}
};
var key = OtlpHelpers.GetResourceKey(resource);
return new OtlpResource(key.Name, key.InstanceId!, uninstrumentedPeer: false, TelemetryTestHelpers.CreateContext());
}
}
|