Added auth scheme for diagnostic entries

Author: bhazen

identity-server/src/IdentityServer/Configuration/DependencyInjection/BuilderExtensions/Core.cs

@@ -213,6 +213,7 @@ public static IIdentityServerBuilder AddCoreServices(this IIdentityServerBuilder
         builder.Services.AddSingleton<LicenseExpirationChecker>();
 
         builder.Services.AddSingleton<IDiagnosticEntry, AssemblyInfoDiagnosticEntry>();
+        builder.Services.AddSingleton<IDiagnosticEntry, AuthSchemeInfoDiagnosticEntry>();
         builder.Services.AddSingleton<DiagnosticSummary>();
         builder.Services.AddHostedService<DiagnosticHostedService>();
 

identity-server/src/IdentityServer/Licensing/V2/Diagnostics/DiagnosticEntries/AuthSchemeInfoDiagnosticEntry.cs

@@ -0,0 +1,26 @@
+// Copyright (c) Duende Software. All rights reserved.
+// See LICENSE in the project root for license information.
+
+using System.Text.Json;
+using Microsoft.AspNetCore.Authentication;
+
+namespace Duende.IdentityServer.Licensing.V2.Diagnostics.DiagnosticEntries;
+
+internal class AuthSchemeInfoDiagnosticEntry(IAuthenticationSchemeProvider authenticationSchemeProvider) : IDiagnosticEntry
+{
+    public async Task WriteAsync(Utf8JsonWriter writer)
+    {
+        var schemes = await authenticationSchemeProvider.GetAllSchemesAsync();
+
+        writer.WriteStartObject("AuthSchemeInfo");
+        writer.WriteStartArray("Schemes");
+        foreach (var scheme in schemes)
+        {
+            writer.WriteStartObject();
+            writer.WriteString(scheme.Name, scheme.HandlerType.FullName ?? "Unknown");
+            writer.WriteEndObject();
+        }
+        writer.WriteEndArray();
+        writer.WriteEndObject();
+    }
+}

identity-server/test/IdentityServer.UnitTests/Licensing/v2/DiagnosticEntries/AuthSchemeInfoDiagnosticEntryTests.cs

@@ -0,0 +1,50 @@
+// Copyright (c) Duende Software. All rights reserved.
+// See LICENSE in the project root for license information.
+
+using Duende.IdentityServer.Licensing.V2.Diagnostics.DiagnosticEntries;
+using Microsoft.AspNetCore.Authentication;
+using UnitTests.Common;
+
+namespace IdentityServer.UnitTests.Licensing.V2.DiagnosticEntries;
+
+public class AuthSchemeInfoDiagnosticEntryTests
+{
+
+    private readonly MockAuthenticationSchemeProvider _mockAuthenticationSchemeProvider;
+    private readonly AuthSchemeInfoDiagnosticEntry _subject;
+
+    public AuthSchemeInfoDiagnosticEntryTests()
+    {
+        _mockAuthenticationSchemeProvider = new MockAuthenticationSchemeProvider();
+        _subject = new AuthSchemeInfoDiagnosticEntry(_mockAuthenticationSchemeProvider);
+    }
+
+    [Fact]
+    public async Task WriteAsync_ShouldWriteAuthSchemeInfo()
+    {
+        var testAuthenticationScheme = new AuthenticationScheme("TestScheme", "Test Scheme", typeof(MockAuthenticationHandler));
+        _mockAuthenticationSchemeProvider.RemoveScheme("scheme");
+        _mockAuthenticationSchemeProvider.AddScheme(testAuthenticationScheme);
+
+        var result = await DiagnosticEntryTestHelper.WriteEntryToJson(_subject);
+
+        var authSchemeInfo = result.RootElement.GetProperty("AuthSchemeInfo");
+        var authSchemes = authSchemeInfo.GetProperty("Schemes");
+        var firstEntry = authSchemes.EnumerateArray().First();
+        firstEntry.GetProperty("TestScheme").GetString().ShouldBe("UnitTests.Common.MockAuthenticationHandler");
+    }
+
+    [Fact]
+    public async Task WriteAsync_ShouldWriteAllRegisteredAuthSchemes()
+    {
+        _mockAuthenticationSchemeProvider.RemoveScheme("scheme");
+        _mockAuthenticationSchemeProvider.AddScheme(new AuthenticationScheme("FirstTestScheme", "First Test Scheme", typeof(MockAuthenticationHandler)));
+        _mockAuthenticationSchemeProvider.AddScheme(new AuthenticationScheme("SecondTestScheme", "Second Test Scheme", typeof(MockAuthenticationHandler)));
+
+        var result = await DiagnosticEntryTestHelper.WriteEntryToJson(_subject);
+
+        var authSchemeInfo = result.RootElement.GetProperty("AuthSchemeInfo");
+        var authSchemes = authSchemeInfo.GetProperty("Schemes");
+        authSchemes.GetArrayLength().ShouldBe(2);
+    }
+}