DodoSSH/tests/DodoSSH.Api.Tests/StubIdentityProvider.cs

using System.IdentityModel.Tokens.Jwt;
using System.Security.Cryptography;
using System.Text.Json;
using Microsoft.IdentityModel.Tokens;
using WireMock.RequestBuilders;
using WireMock.ResponseBuilders;
using WireMock.Server;

namespace DodoSSH.Api.Tests;

/// <summary>
/// A stand-in identity provider: OIDC discovery, JWKS, and token minting.
/// </summary>
/// <remarks>
/// Deliberately not a <c>TestAuthHandler</c> that short-circuits authentication. Tokens are signed
/// with a real RSA key and validated by the application's own JwtBearer pipeline, so issuer,
/// audience, signature, lifetime and claim handling are all genuinely exercised. Bypassing that
/// would hide precisely the claim-mapping mistakes that cause real authorization holes.
/// </remarks>
public sealed class StubIdentityProvider : IDisposable
{
    private const string KeyId = "dodossh-test-key";

    private readonly WireMockServer server;
    private readonly RsaSecurityKey signingKey;

    public StubIdentityProvider()
    {
        var rsa = RSA.Create(2048);
        signingKey = new RsaSecurityKey(rsa) { KeyId = KeyId };

        server = WireMockServer.Start();
        Authority = server.Url!.TrimEnd('/');

        StubDiscovery();
        StubJwks();
    }

    /// <summary>Issuer URL, matching what the tokens claim.</summary>
    public string Authority { get; }

    /// <summary>Audience the API is configured to expect.</summary>
    public static string Audience => "dodossh-api";

    /// <summary>
    /// Mints a signed access token.
    /// </summary>
    /// <param name="subject">The <c>sub</c> claim — the stable user identifier.</param>
    /// <param name="email">Optional email claim.</param>
    /// <param name="name">Optional display name claim.</param>
    /// <param name="audience">Override the audience, to test rejection.</param>
    /// <param name="issuer">Override the issuer, to test rejection.</param>
    /// <param name="expires">Override expiry, to test rejection.</param>
    public string MintToken(
        string subject,
        string? email = null,
        string? name = null,
        string? audience = null,
        string? issuer = null,
        DateTime? expires = null)
    {
        var now = TimeProvider.System.GetUtcNow().UtcDateTime;

        var claims = new List<System.Security.Claims.Claim>
        {
            new("sub", subject),
        };

        if (email is not null)
        {
            claims.Add(new System.Security.Claims.Claim("email", email));
        }

        if (name is not null)
        {
            claims.Add(new System.Security.Claims.Claim("name", name));
        }

        var expiry = expires ?? now.AddMinutes(15);

        // The earlier of now-1min and exp-1min. A fixed now-1min would sit after the expiry of a
        // deliberately-expired test token (rejected at construction), while exp-1min alone would
        // put nbf in the future for a normal token and make every valid token unauthorized.
        var notBefore = expiry < now ? expiry.AddMinutes(-1) : now.AddMinutes(-1);

        var token = new JwtSecurityToken(
            issuer: issuer ?? Authority,
            audience: audience ?? Audience,
            claims: claims,
            notBefore: notBefore,
            expires: expiry,
            signingCredentials: new SigningCredentials(signingKey, SecurityAlgorithms.RsaSha256));

        return new JwtSecurityTokenHandler().WriteToken(token);
    }

    /// <summary>Mints a token signed by a different key, which must be rejected.</summary>
    public string MintTokenWithForeignKey(string subject)
    {
        var foreign = new RsaSecurityKey(RSA.Create(2048)) { KeyId = KeyId };
        var now = TimeProvider.System.GetUtcNow().UtcDateTime;

        var token = new JwtSecurityToken(
            issuer: Authority,
            audience: Audience,
            claims: [new System.Security.Claims.Claim("sub", subject)],
            notBefore: now.AddMinutes(-1),
            expires: now.AddMinutes(15),
            signingCredentials: new SigningCredentials(foreign, SecurityAlgorithms.RsaSha256));

        return new JwtSecurityTokenHandler().WriteToken(token);
    }

    private void StubDiscovery()
    {
        var document = new Dictionary<string, object>(StringComparer.Ordinal)
        {
            ["issuer"] = Authority,
            ["jwks_uri"] = $"{Authority}/.well-known/jwks.json",
            ["authorization_endpoint"] = $"{Authority}/connect/authorize",
            ["token_endpoint"] = $"{Authority}/connect/token",
            ["response_types_supported"] = new[] { "code" },
            ["subject_types_supported"] = new[] { "public" },
            ["id_token_signing_alg_values_supported"] = new[] { "RS256" },
            ["code_challenge_methods_supported"] = new[] { "S256" },
        };

        server
            .Given(Request.Create().WithPath("/.well-known/openid-configuration").UsingGet())
            .RespondWith(Response.Create()
                .WithStatusCode(200)
                .WithHeader("Content-Type", "application/json")
                .WithBody(JsonSerializer.Serialize(document)));
    }

    private void StubJwks()
    {
        var parameters = signingKey.Rsa.ExportParameters(includePrivateParameters: false);

        var jwks = new
        {
            keys = new[]
            {
                new
                {
                    kty = "RSA",
                    use = "sig",
                    kid = KeyId,
                    alg = "RS256",
                    n = Base64UrlEncoder.Encode(parameters.Modulus),
                    e = Base64UrlEncoder.Encode(parameters.Exponent),
                },
            },
        };

        server
            .Given(Request.Create().WithPath("/.well-known/jwks.json").UsingGet())
            .RespondWith(Response.Create()
                .WithStatusCode(200)
                .WithHeader("Content-Type", "application/json")
                .WithBody(JsonSerializer.Serialize(jwks)));
    }

    /// <inheritdoc />
    public void Dispose()
    {
        server.Stop();
        server.Dispose();
        signingKey.Rsa.Dispose();
    }
}