openiddict-core/test/OpenIddict.MongoDb.Tests/Resolvers/OpenIddictMongoDbTokenStoreResolverTests.cs

/*
 * Licensed under the Apache License, Version 2.0 (http://www.apache.org/licenses/LICENSE-2.0)
 * See https://github.com/openiddict/openiddict-core for more information concerning
 * the license and the contributors participating to this project.
 */

using Microsoft.Extensions.DependencyInjection;
using Microsoft.Extensions.Options;
using Moq;
using OpenIddict.MongoDb.Models;
using Xunit;

namespace OpenIddict.MongoDb.Tests;

public class OpenIddictMongoDbTokenStoreResolverTests
{
    [Fact]
    public void Get_ReturnsCustomStoreCorrespondingToTheSpecifiedTypeWhenAvailable()
    {
        // Arrange
        var services = new ServiceCollection();
        services.AddSingleton(Mock.Of<IOpenIddictTokenStore<CustomToken>>());

        var provider = services.BuildServiceProvider();
        var resolver = new OpenIddictMongoDbTokenStoreResolver(provider);

        // Act and assert
        Assert.NotNull(resolver.Get<CustomToken>());
    }

    [Fact]
    public void Get_ThrowsAnExceptionForInvalidEntityType()
    {
        // Arrange
        var services = new ServiceCollection();

        var provider = services.BuildServiceProvider();
        var resolver = new OpenIddictMongoDbTokenStoreResolver(provider);

        // Act and assert
        var exception = Assert.Throws<InvalidOperationException>(resolver.Get<CustomToken>);

        Assert.Equal(SR.GetResourceString(SR.ID0260), exception.Message);
    }

    [Fact]
    public void Get_ReturnsDefaultStoreCorrespondingToTheSpecifiedTypeWhenAvailable()
    {
        // Arrange
        var services = new ServiceCollection();
        services.AddSingleton(Mock.Of<IOpenIddictTokenStore<CustomToken>>());
        services.AddSingleton(CreateStore());

        var provider = services.BuildServiceProvider();
        var resolver = new OpenIddictMongoDbTokenStoreResolver(provider);

        // Act and assert
        Assert.NotNull(resolver.Get<MyToken>());
    }

    private static OpenIddictMongoDbTokenStore<MyToken> CreateStore()
        => new Mock<OpenIddictMongoDbTokenStore<MyToken>>(
            Mock.Of<IOpenIddictMongoDbContext>(),
            Mock.Of<IOptionsMonitor<OpenIddictMongoDbOptions>>()).Object;

    public class CustomToken { }

    public class MyToken : OpenIddictMongoDbToken { }
}