abp/framework/test/Volo.Abp.Data.Tests/Volo/Abp/Data/ConnectionStringResolver_Tests.cs

using System.Threading.Tasks;
using Microsoft.Extensions.DependencyInjection;
using Shouldly;
using Volo.Abp.Modularity;
using Volo.Abp.Testing;
using Xunit;

namespace Volo.Abp.Data
{
    public class ConnectionStringResolver_Tests : AbpIntegratedTest<ConnectionStringResolver_Tests.TestModule>
    {
        private const string DefaultConnString = "default-value";
        private const string Database1Name = "Database1";
        private const string Database1ConnString = "database-1-value";
        private const string Database2Name = "Database2";

        private readonly IConnectionStringResolver _connectionStringResolver;

        public ConnectionStringResolver_Tests()
        {
            _connectionStringResolver = ServiceProvider.GetRequiredService<IConnectionStringResolver>();
        }

        [Fact]
        public async Task Should_Get_Default_ConnString_By_Default()
        {
            (await _connectionStringResolver.ResolveAsync()).ShouldBe(DefaultConnString);
        }

        [Fact]
        public async Task Should_Get_Specific_ConnString_IfDefined()
        {
            (await _connectionStringResolver.ResolveAsync(Database1Name)).ShouldBe(Database1ConnString);
        }

        [Fact]
        public async Task Should_Get_Default_ConnString_If_Not_Specified()
        {
            (await _connectionStringResolver.ResolveAsync(Database2Name)).ShouldBe(DefaultConnString);
        }

        [DependsOn(typeof(AbpDataModule))]
        public class TestModule : AbpModule
        {
            public override void ConfigureServices(ServiceConfigurationContext context)
            {
                Configure<AbpDbConnectionOptions>(options =>
                {
                    options.ConnectionStrings.Default = DefaultConnString;
                    options.ConnectionStrings[Database1Name] = Database1ConnString;
                });
            }
        }
    }
}