Merge pull request #8897 from abpframework/liangshiwei/replaceDbContext

Introduce ReplaceDbContextAttribute

docs/en/Entity-Framework-Core.md

@@ -724,7 +724,19 @@ One advantage of using an interface for a DbContext is then it will be replaceab
 
 ### Replace Other DbContextes
 
-Once you properly define and use an interface for DbContext, then any other implementation can replace it using the `ReplaceDbContext` option:
+Once you properly define and use an interface for DbContext, then any other implementation can use the following ways to replace it:
+
+**ReplaceDbContextAttribute**
+
+```csharp
+[ReplaceDbContext(typeof(IBookStoreDbContext))]
+public class OtherDbContext : AbpDbContext<OtherDbContext>, IBookStoreDbContext
+{
+    //...
+}
+```
+
+**ReplaceDbContext option**
 
 ````csharp
 context.Services.AddAbpDbContext<OtherDbContext>(options =>

docs/en/MongoDB.md

@@ -391,7 +391,19 @@ One advantage of using interface for a MongoDbContext is then it becomes replace
 
 #### Replace Other DbContextes
 
-Once you properly define and use an interface for a MongoDbContext , then any other implementation can replace it using the `ReplaceDbContext` option:
+Once you properly define and use an interface for a MongoDbContext , then any other implementation can use the following ways to replace it:
+
+**ReplaceDbContextAttribute**
+
+```csharp
+[ReplaceDbContext(typeof(IBookStoreMongoDbContext))]
+public class OtherMongoDbContext : AbpMongoDbContext, IBookStoreMongoDbContext
+{
+    //...
+}
+```
+
+**ReplaceDbContext option**
 
 ```csharp
 context.Services.AddMongoDbContext<OtherMongoDbContext>(options =>

docs/zh-Hans/Entity-Framework-Core.md

@@ -433,7 +433,19 @@ public class BookRepository : EfCoreRepository<IBookStoreDbContext, Book, Guid>,
 
 ### 替换其他仓储
 
-正确定义并使用DbContext接口后,任何其他实现都可以使用以下ReplaceDbContext options 替换它:
+正确定义并使用DbContext接口后,任何其他实现都可以使用以下方法替换它:
+
+**ReplaceDbContextAttribute**
+
+```csharp
+[ReplaceDbContext(typeof(IBookStoreDbContext))]
+public class OtherDbContext : AbpDbContext<OtherDbContext>, IBookStoreDbContext
+{
+    //...
+}
+```
+
+**ReplaceDbContext option**
 
 ````csharp
 context.Services.AddAbpDbContext<OtherDbContext>(options =>

docs/zh-Hans/MongoDB.md

@@ -329,7 +329,19 @@ public class BookRepository
 
 ##### 替换其他的DbContexts
 
-一旦你正确定义并为MongoDbContext使用了接口,其他的实现就可以使用`ReplaceDbContext`来替换:
+一旦你正确定义并为MongoDbContext使用了接口,任何其他实现都可以使用以下方法替换它:
+
+**ReplaceDbContextAttribute**
+
+```csharp
+[ReplaceDbContext(typeof(IBookStoreMongoDbContext))]
+public class OtherMongoDbContext : AbpMongoDbContext, IBookStoreMongoDbContext
+{
+    //...
+}
+```
+
+**ReplaceDbContext option**
 
 ```csharp
 context.Services.AddMongoDbContext<OtherMongoDbContext>(options =>

framework/Volo.Abp.sln

@@ -395,6 +395,8 @@ Project("{FAE04EC0-301F-11D3-BF4B-00C04F79EFBC}") = "Volo.Abp.TextTemplating.Raz
 EndProject
 Project("{FAE04EC0-301F-11D3-BF4B-00C04F79EFBC}") = "Volo.Abp.TextTemplating.Scriban.Tests", "test\Volo.Abp.TextTemplating.Scriban.Tests\Volo.Abp.TextTemplating.Scriban.Tests.csproj", "{75D8DADB-3FA9-4C1D-B23A-DBFD08133B7C}"
 EndProject
+Project("{FAE04EC0-301F-11D3-BF4B-00C04F79EFBC}") = "Volo.Abp.MongoDB.Tests.SecondContext", "test\Volo.Abp.MongoDB.Tests.SecondContext\Volo.Abp.MongoDB.Tests.SecondContext.csproj", "{90B1866A-EF99-40B9-970E-B898E5AA523F}"
+EndProject
 Global
 	GlobalSection(SolutionConfigurationPlatforms) = preSolution
 		Debug|Any CPU = Debug|Any CPU
@@ -1177,6 +1179,10 @@ Global
 		{75D8DADB-3FA9-4C1D-B23A-DBFD08133B7C}.Debug|Any CPU.Build.0 = Debug|Any CPU
 		{75D8DADB-3FA9-4C1D-B23A-DBFD08133B7C}.Release|Any CPU.ActiveCfg = Release|Any CPU
 		{75D8DADB-3FA9-4C1D-B23A-DBFD08133B7C}.Release|Any CPU.Build.0 = Release|Any CPU
+		{90B1866A-EF99-40B9-970E-B898E5AA523F}.Debug|Any CPU.ActiveCfg = Debug|Any CPU
+		{90B1866A-EF99-40B9-970E-B898E5AA523F}.Debug|Any CPU.Build.0 = Debug|Any CPU
+		{90B1866A-EF99-40B9-970E-B898E5AA523F}.Release|Any CPU.ActiveCfg = Release|Any CPU
+		{90B1866A-EF99-40B9-970E-B898E5AA523F}.Release|Any CPU.Build.0 = Release|Any CPU
 	EndGlobalSection
 	GlobalSection(SolutionProperties) = preSolution
 		HideSolutionNode = FALSE
@@ -1376,6 +1382,7 @@ Global
 		{42EA6F06-2D78-4D18-8AC4-8F2AB7E6DA19} = {5DF0E140-0513-4D0D-BE2E-3D4D85CD70E6}
 		{C996F458-98FB-483D-9306-4701290E2FC1} = {447C8A77-E5F0-4538-8687-7383196D04EA}
 		{75D8DADB-3FA9-4C1D-B23A-DBFD08133B7C} = {447C8A77-E5F0-4538-8687-7383196D04EA}
+		{90B1866A-EF99-40B9-970E-B898E5AA523F} = {447C8A77-E5F0-4538-8687-7383196D04EA}
 	EndGlobalSection
 	GlobalSection(ExtensibilityGlobals) = postSolution
 		SolutionGuid = {BB97ECF4-9A84-433F-A80B-2A3285BDD1D5}

framework/src/Volo.Abp.Ddd.Domain/Volo/Abp/DependencyInjection/AbpCommonDbContextRegistrationOptions.cs

@@ -55,7 +55,7 @@ namespace Volo.Abp.DependencyInjection
                 throw new AbpException($"{OriginalDbContextType.AssemblyQualifiedName} should inherit/implement {otherDbContextType.AssemblyQualifiedName}!");
             }
 
-            ReplacedDbContextTypes.Add(otherDbContextType);
+            ReplacedDbContextTypes.AddIfNotContains(otherDbContextType);
 
             return this;
         }
@@ -93,9 +93,9 @@ namespace Volo.Abp.DependencyInjection
         public IAbpCommonDbContextRegistrationOptionsBuilder AddDefaultRepository(Type entityType)
         {
             EntityHelper.CheckEntity(entityType);
-            
+
             SpecifiedDefaultRepositories.AddIfNotContains(entityType);
-            
+
             return this;
         }
 

framework/src/Volo.Abp.Ddd.Domain/Volo/Abp/DependencyInjection/ReplaceDbContextAttribute.cs

@@ -0,0 +1,15 @@
+using System;
+
+namespace Volo.Abp.DependencyInjection
+{
+    [AttributeUsage(AttributeTargets.Class, AllowMultiple = true)]
+    public class ReplaceDbContextAttribute : Attribute
+    {
+        public Type[] ReplacedDbContextTypes { get; }
+
+        public ReplaceDbContextAttribute(params Type[] replacedDbContextTypes)
+        {
+            ReplacedDbContextTypes = replacedDbContextTypes;
+        }
+    }
+}

framework/src/Volo.Abp.EntityFrameworkCore/Microsoft/Extensions/DependencyInjection/AbpEfCoreServiceCollectionExtensions.cs

@@ -1,5 +1,8 @@
 using System;
+using System.Linq;
+using System.Reflection;
 using Microsoft.Extensions.DependencyInjection.Extensions;
+using Volo.Abp.DependencyInjection;
 using Volo.Abp.EntityFrameworkCore;
 using Volo.Abp.EntityFrameworkCore.DependencyInjection;
 
@@ -8,13 +11,22 @@ namespace Microsoft.Extensions.DependencyInjection
     public static class AbpEfCoreServiceCollectionExtensions
     {
         public static IServiceCollection AddAbpDbContext<TDbContext>(
-            this IServiceCollection services, 
+            this IServiceCollection services,
             Action<IAbpDbContextRegistrationOptionsBuilder> optionsBuilder = null)
             where TDbContext : AbpDbContext<TDbContext>
         {
             services.AddMemoryCache();
 
             var options = new AbpDbContextRegistrationOptions(typeof(TDbContext), services);
+
+            var replacedDbContextTypes = typeof(TDbContext).GetCustomAttributes<ReplaceDbContextAttribute>(true)
+                .SelectMany( x => x.ReplacedDbContextTypes).ToList();
+
+            foreach (var dbContextType in replacedDbContextTypes)
+            {
+                options.ReplaceDbContext(dbContextType);
+            }
+
             optionsBuilder?.Invoke(options);
 
             services.TryAddTransient(DbContextOptionsFactory.Create<TDbContext>);

framework/src/Volo.Abp.EntityFrameworkCore/Volo/Abp/EntityFrameworkCore/AbpEntityFrameworkCoreModule.cs

@@ -1,6 +1,8 @@
 using Microsoft.EntityFrameworkCore.Diagnostics;
+using Microsoft.Extensions.DependencyInjection;
 using Microsoft.Extensions.DependencyInjection.Extensions;
 using Volo.Abp.Domain;
+using Volo.Abp.EntityFrameworkCore.DependencyInjection;
 using Volo.Abp.Modularity;
 using Volo.Abp.Uow.EntityFrameworkCore;
 

framework/src/Volo.Abp.MongoDB/Microsoft/Extensions/DependencyInjection/AbpMongoDbServiceCollectionExtensions.cs

@@ -1,5 +1,8 @@
 using System;
+using System.Linq;
+using System.Reflection;
 using Microsoft.Extensions.DependencyInjection.Extensions;
+using Volo.Abp.DependencyInjection;
 using Volo.Abp.MongoDB;
 using Volo.Abp.MongoDB.DependencyInjection;
 
@@ -11,13 +14,22 @@ namespace Microsoft.Extensions.DependencyInjection
             where TMongoDbContext : AbpMongoDbContext
         {
             var options = new AbpMongoDbContextRegistrationOptions(typeof(TMongoDbContext), services);
+
+            var replacedDbContextTypes = typeof(TMongoDbContext).GetCustomAttributes<ReplaceDbContextAttribute>(true)
+                .SelectMany( x => x.ReplacedDbContextTypes).ToList();
+
+            foreach (var dbContextType in replacedDbContextTypes)
+            {
+                options.ReplaceDbContext(dbContextType);
+            }
+
             optionsBuilder?.Invoke(options);
 
             foreach (var dbContextType in options.ReplacedDbContextTypes)
             {
                 services.Replace(ServiceDescriptor.Transient(dbContextType, typeof(TMongoDbContext)));
             }
-            
+
             foreach (var dbContextType in options.ReplacedDbContextTypes)
             {
                 services.Replace(

framework/src/Volo.Abp.MongoDB/Volo/Abp/MongoDB/DependencyInjection/AbpMongoDbConventionalRegistrar.cs

@@ -1,5 +1,7 @@
 using System;
+using System.Reflection;
 using Microsoft.Extensions.DependencyInjection;
+using Microsoft.Extensions.DependencyInjection.Extensions;
 using Volo.Abp.DependencyInjection;
 
 namespace Volo.Abp.MongoDB.DependencyInjection

framework/test/Volo.Abp.EntityFrameworkCore.Tests.SecondContext/Volo/Abp/EntityFrameworkCore/TestApp/FourthContext/FourthDbContext.cs

@@ -0,0 +1,17 @@
+using Microsoft.EntityFrameworkCore;
+using Volo.Abp.EntityFrameworkCore.TestApp.ThirdDbContext;
+
+namespace Volo.Abp.EntityFrameworkCore.TestApp.FourthContext
+{
+    /* This dbcontext is just for testing to replace dbcontext from the application using ReplaceDbContextAttribute
+     */
+    public class FourthDbContext : AbpDbContext<FourthDbContext>, IFourthDbContext
+    {
+        public DbSet<FourthDbContextDummyEntity> FourthDummyEntities { get; set; }
+
+        public FourthDbContext(DbContextOptions<FourthDbContext> options)
+            : base(options)
+        {
+        }
+    }
+}

framework/test/Volo.Abp.EntityFrameworkCore.Tests.SecondContext/Volo/Abp/EntityFrameworkCore/TestApp/FourthContext/FourthDbContextDummyEntity.cs

@@ -0,0 +1,10 @@
+using System;
+using Volo.Abp.Domain.Entities;
+
+namespace Volo.Abp.EntityFrameworkCore.TestApp.FourthContext
+{
+    public class FourthDbContextDummyEntity : AggregateRoot<Guid>
+    {
+        public string Value { get; set; }
+    }
+}

framework/test/Volo.Abp.EntityFrameworkCore.Tests.SecondContext/Volo/Abp/EntityFrameworkCore/TestApp/FourthContext/IFourthDbContext.cs

@@ -0,0 +1,9 @@
+using Microsoft.EntityFrameworkCore;
+
+namespace Volo.Abp.EntityFrameworkCore.TestApp.FourthContext
+{
+    public interface IFourthDbContext : IEfCoreDbContext
+    {
+        DbSet<FourthDbContextDummyEntity> FourthDummyEntities { get; set; }
+    }
+}

framework/test/Volo.Abp.EntityFrameworkCore.Tests.SecondContext/Volo/Abp/EntityFrameworkCore/TestApp/SecondContext/AbpEfCoreTestSecondContextModule.cs

@@ -1,4 +1,5 @@
 using Microsoft.Extensions.DependencyInjection;
+using Volo.Abp.EntityFrameworkCore.TestApp.FourthContext;
 using Volo.Abp.EntityFrameworkCore.TestApp.ThirdDbContext;
 using Volo.Abp.Modularity;
 using Volo.Abp.Threading;
@@ -19,6 +20,11 @@ namespace Volo.Abp.EntityFrameworkCore.TestApp.SecondContext
             {
                 options.AddDefaultRepositories<IThirdDbContext>();
             });
+
+            context.Services.AddAbpDbContext<FourthDbContext>(options =>
+            {
+                options.AddDefaultRepositories<IFourthDbContext>();
+            });
         }
 
         public override void OnApplicationInitialization(ApplicationInitializationContext context)
@@ -36,4 +42,4 @@ namespace Volo.Abp.EntityFrameworkCore.TestApp.SecondContext
             }
         }
     }
-}
+}

framework/test/Volo.Abp.EntityFrameworkCore.Tests/Volo/Abp/EntityFrameworkCore/AbpEntityFrameworkCoreTestModule.cs

@@ -39,7 +39,7 @@ namespace Volo.Abp.EntityFrameworkCore
                 {
                     opt.DefaultWithDetailsFunc = q => q.Include(p => p.Phones);
                 });
-                
+
                 options.Entity<Author>(opt =>
                 {
                     opt.DefaultWithDetailsFunc = q => q.Include(p => p.Books);
@@ -76,7 +76,7 @@ namespace Volo.Abp.EntityFrameworkCore
                     @"CREATE VIEW View_PersonView AS 
                       SELECT Name, CreationTime, Birthday, LastActive FROM People");
             }
-            
+
             return connection;
         }
     }

framework/test/Volo.Abp.EntityFrameworkCore.Tests/Volo/Abp/EntityFrameworkCore/DbContext_Replace_Tests.cs

@@ -4,6 +4,7 @@ using Microsoft.Extensions.DependencyInjection;
 using Microsoft.Extensions.Options;
 using Shouldly;
 using Volo.Abp.Domain.Repositories;
+using Volo.Abp.EntityFrameworkCore.TestApp.FourthContext;
 using Volo.Abp.EntityFrameworkCore.TestApp.ThirdDbContext;
 using Volo.Abp.TestApp.Domain;
 using Volo.Abp.TestApp.EntityFrameworkCore;
@@ -15,13 +16,15 @@ namespace Volo.Abp.EntityFrameworkCore
     public class DbContext_Replace_Tests : EntityFrameworkCoreTestBase
     {
         private readonly IBasicRepository<ThirdDbContextDummyEntity, Guid> _dummyRepository;
-        private readonly IPersonRepository _personRepository; 
+        private readonly IBasicRepository<FourthDbContextDummyEntity, Guid> _fourthDummyRepository;
+        private readonly IPersonRepository _personRepository;
         private readonly IUnitOfWorkManager _unitOfWorkManager;
         private readonly AbpDbContextOptions _options;
 
         public DbContext_Replace_Tests()
         {
             _dummyRepository = GetRequiredService<IBasicRepository<ThirdDbContextDummyEntity, Guid>>();
+            _fourthDummyRepository = GetRequiredService<IBasicRepository<FourthDbContextDummyEntity, Guid>>();
             _personRepository = GetRequiredService<IPersonRepository>();
             _unitOfWorkManager = GetRequiredService<IUnitOfWorkManager>();
             _options = GetRequiredService<IOptions<AbpDbContextOptions>>().Value;
@@ -31,8 +34,10 @@ namespace Volo.Abp.EntityFrameworkCore
         public async Task Should_Replace_DbContext()
         {
             _options.GetReplacedTypeOrSelf(typeof(IThirdDbContext)).ShouldBe(typeof(TestAppDbContext));
-            
+            _options.GetReplacedTypeOrSelf(typeof(IFourthDbContext)).ShouldBe(typeof(TestAppDbContext));
+
             (ServiceProvider.GetRequiredService<IThirdDbContext>() is TestAppDbContext).ShouldBeTrue();
+            (ServiceProvider.GetRequiredService<IFourthDbContext>() is TestAppDbContext).ShouldBeTrue();
 
             using (var uow = _unitOfWorkManager.Begin())
             {
@@ -44,11 +49,19 @@ namespace Volo.Abp.EntityFrameworkCore
 
                 var instance3 = await _personRepository.GetDbContextAsync();
                 (instance3 is TestAppDbContext).ShouldBeTrue();
-                
+
+                var instance4 = await _fourthDummyRepository.GetDbContextAsync();
+                (instance4 is IFourthDbContext).ShouldBeTrue();
+
+                var instance5 = await _fourthDummyRepository.GetDbContextAsync();
+                (instance5 is TestAppDbContext).ShouldBeTrue();
+
                 // All instances should be the same!
                 instance3.ShouldBe(instance1);
                 instance3.ShouldBe(instance2);
-                
+                instance3.ShouldBe(instance4);
+                instance3.ShouldBe(instance5);
+
                 await uow.CompleteAsync();
             }
         }

framework/test/Volo.Abp.EntityFrameworkCore.Tests/Volo/Abp/TestApp/EntityFrameworkCore/TestAppDbContext.cs

@@ -1,12 +1,17 @@
 using Microsoft.EntityFrameworkCore;
+using Volo.Abp.DependencyInjection;
 using Volo.Abp.EntityFrameworkCore;
+using Volo.Abp.EntityFrameworkCore.TestApp.FourthContext;
 using Volo.Abp.EntityFrameworkCore.TestApp.ThirdDbContext;
 using Volo.Abp.TestApp.Domain;
 
 namespace Volo.Abp.TestApp.EntityFrameworkCore
 {
-    public class TestAppDbContext : AbpDbContext<TestAppDbContext>, IThirdDbContext
+    [ReplaceDbContext(typeof(IFourthDbContext))]
+    public class TestAppDbContext : AbpDbContext<TestAppDbContext>, IThirdDbContext, IFourthDbContext
     {
+        private DbSet<FourthDbContextDummyEntity> _dummyEntities;
+        private DbSet<FourthDbContextDummyEntity> _dummyEntities1;
         public DbSet<Person> People { get; set; }
 
         public DbSet<City> Cities { get; set; }
@@ -16,10 +21,12 @@ namespace Volo.Abp.TestApp.EntityFrameworkCore
         public DbSet<ThirdDbContextDummyEntity> DummyEntities { get; set; }
 
         public DbSet<EntityWithIntPk> EntityWithIntPks { get; set; }
-        
+
         public DbSet<Author> Author { get; set; }
 
-        public TestAppDbContext(DbContextOptions<TestAppDbContext> options) 
+        public DbSet<FourthDbContextDummyEntity> FourthDummyEntities { get; set; }
+
+        public TestAppDbContext(DbContextOptions<TestAppDbContext> options)
             : base(options)
         {
 

framework/test/Volo.Abp.MongoDB.Tests.SecondContext/Volo.Abp.MongoDB.Tests.SecondContext.csproj

@@ -0,0 +1,19 @@
+<Project Sdk="Microsoft.NET.Sdk">
+
+<Import Project="..\..\..\common.test.props" />
+
+<PropertyGroup>
+    <TargetFramework>net5.0</TargetFramework>
+    <AssetTargetFallback>$(AssetTargetFallback);portable-net45+win8+wp8+wpa81;</AssetTargetFallback>
+    <PreserveCompilationContext>true</PreserveCompilationContext>
+    <GenerateRuntimeConfigurationFiles>true</GenerateRuntimeConfigurationFiles>
+    <AutoGenerateBindingRedirects>true</AutoGenerateBindingRedirects>
+    <GenerateBindingRedirectsOutputType>true</GenerateBindingRedirectsOutputType>
+    <RootNamespace />
+</PropertyGroup>
+
+<ItemGroup>
+    <ProjectReference Include="..\..\src\Volo.Abp.MongoDB\Volo.Abp.MongoDB.csproj" />
+</ItemGroup>
+
+</Project>

framework/test/Volo.Abp.MongoDB.Tests.SecondContext/Volo/Abp/MongoDB/TestApp/FourthContext/FourthDbContext.cs

@@ -0,0 +1,13 @@
+using MongoDB.Driver;
+using Volo.Abp.EntityFrameworkCore.TestApp.FourthContext;
+
+namespace Volo.Abp.MongoDB.TestApp.FourthContext
+{
+    /* This dbcontext is just for testing to replace dbcontext from the application using ReplaceDbContextAttribute
+     */
+    public class FourthDbContext : AbpMongoDbContext, IFourthDbContext
+    {
+        public IMongoCollection<FourthDbContextDummyEntity> FourthDummyEntities => Collection<FourthDbContextDummyEntity>();
+
+    }
+}

framework/test/Volo.Abp.MongoDB.Tests.SecondContext/Volo/Abp/MongoDB/TestApp/FourthContext/FourthDbContextDummyEntity.cs

@@ -0,0 +1,10 @@
+using System;
+using Volo.Abp.Domain.Entities;
+
+namespace Volo.Abp.EntityFrameworkCore.TestApp.FourthContext
+{
+    public class FourthDbContextDummyEntity : AggregateRoot<Guid>
+    {
+        public string Value { get; set; }
+    }
+}

framework/test/Volo.Abp.MongoDB.Tests.SecondContext/Volo/Abp/MongoDB/TestApp/FourthContext/IFourthDbContext.cs

@@ -0,0 +1,10 @@
+using MongoDB.Driver;
+using Volo.Abp.EntityFrameworkCore.TestApp.FourthContext;
+
+namespace Volo.Abp.MongoDB.TestApp.FourthContext
+{
+    public interface IFourthDbContext : IAbpMongoDbContext
+    {
+        IMongoCollection<FourthDbContextDummyEntity> FourthDummyEntities { get;}
+    }
+}

framework/test/Volo.Abp.MongoDB.Tests.SecondContext/Volo/Abp/MongoDB/TestApp/SecondContext/AbpMongoDbTestSecondContextModule.cs

@@ -0,0 +1,45 @@
+using Microsoft.Extensions.DependencyInjection;
+using Volo.Abp.Modularity;
+using Volo.Abp.MongoDB.TestApp.FourthContext;
+using Volo.Abp.MongoDB.TestApp.ThirdDbContext;
+using Volo.Abp.Threading;
+
+namespace Volo.Abp.MongoDB.TestApp.SecondContext
+{
+    [DependsOn(typeof(AbpMongoDbModule))]
+    public class AbpMongoDbTestSecondContextModule : AbpModule
+    {
+        public override void ConfigureServices(ServiceConfigurationContext context)
+        {
+            context.Services.AddMongoDbContext<SecondDbContext>(options =>
+            {
+                options.AddDefaultRepositories();
+            });
+
+            context.Services.AddMongoDbContext<ThirdDbContext.ThirdDbContext>(options =>
+            {
+                options.AddDefaultRepositories<IThirdDbContext>();
+            });
+
+            context.Services.AddMongoDbContext<FourthDbContext>(options =>
+            {
+                options.AddDefaultRepositories<IFourthDbContext>();
+            });
+        }
+
+        public override void OnApplicationInitialization(ApplicationInitializationContext context)
+        {
+            SeedTestData(context);
+        }
+
+        private static void SeedTestData(ApplicationInitializationContext context)
+        {
+            using (var scope = context.ServiceProvider.CreateScope())
+            {
+                AsyncHelper.RunSync(() => scope.ServiceProvider
+                    .GetRequiredService<SecondContextTestDataBuilder>()
+                    .BuildAsync());
+            }
+        }
+    }
+}

framework/test/Volo.Abp.MongoDB.Tests.SecondContext/Volo/Abp/MongoDB/TestApp/SecondContext/BookInSecondDbContext.cs

@@ -0,0 +1,21 @@
+using System;
+using Volo.Abp.Domain.Entities;
+
+namespace Volo.Abp.MongoDB.TestApp.SecondContext
+{
+    public class BookInSecondDbContext : AggregateRoot<Guid>
+    {
+        public string Name { get; set; }
+
+        public BookInSecondDbContext()
+        {
+
+        }
+
+        public BookInSecondDbContext(Guid id, string name)
+            : base(id)
+        {
+            Name = name;
+        }
+    }
+}

framework/test/Volo.Abp.MongoDB.Tests.SecondContext/Volo/Abp/MongoDB/TestApp/SecondContext/PhoneInSecondDbContext.cs

@@ -0,0 +1,18 @@
+using System;
+using System.ComponentModel.DataAnnotations.Schema;
+using Volo.Abp.Domain.Entities;
+
+namespace Volo.Abp.MongoDB.TestApp.SecondContext
+{
+    public class PhoneInSecondDbContext : AggregateRoot
+    {
+        public virtual Guid PersonId { get; set; }
+
+        public virtual string Number { get; set; }
+
+        public override object[] GetKeys()
+        {
+            return new object[] {PersonId, Number};
+        }
+    }
+}

framework/test/Volo.Abp.MongoDB.Tests.SecondContext/Volo/Abp/MongoDB/TestApp/SecondContext/SecondContextTestDataBuilder.cs

@@ -0,0 +1,30 @@
+using System;
+using System.Threading.Tasks;
+using Volo.Abp.DependencyInjection;
+using Volo.Abp.Domain.Repositories;
+using Volo.Abp.Guids;
+
+namespace Volo.Abp.MongoDB.TestApp.SecondContext
+{
+    public class SecondContextTestDataBuilder : ITransientDependency
+    {
+        private readonly IBasicRepository<BookInSecondDbContext, Guid> _bookRepository;
+        private readonly IGuidGenerator _guidGenerator;
+
+        public SecondContextTestDataBuilder(IBasicRepository<BookInSecondDbContext, Guid> bookRepository, IGuidGenerator guidGenerator)
+        {
+            _bookRepository = bookRepository;
+            _guidGenerator = guidGenerator;
+        }
+
+        public async Task BuildAsync()
+        {
+            await _bookRepository.InsertAsync(
+                new BookInSecondDbContext(
+                    _guidGenerator.Create(),
+                    "TestBook1"
+                )
+            );
+        }
+    }
+}

framework/test/Volo.Abp.MongoDB.Tests.SecondContext/Volo/Abp/MongoDB/TestApp/SecondContext/SecondDbContext.cs

@@ -0,0 +1,11 @@
+using MongoDB.Driver;
+
+namespace Volo.Abp.MongoDB.TestApp.SecondContext
+{
+    public class SecondDbContext : AbpMongoDbContext
+    {
+        public IMongoCollection<BookInSecondDbContext> Books => Collection<BookInSecondDbContext>();
+
+        public IMongoCollection<PhoneInSecondDbContext> Phones => Collection<PhoneInSecondDbContext>();
+    }
+}

framework/test/Volo.Abp.MongoDB.Tests.SecondContext/Volo/Abp/MongoDB/TestApp/ThirdDbContext/IThirdDbContext.cs

@@ -0,0 +1,9 @@
+using MongoDB.Driver;
+
+namespace Volo.Abp.MongoDB.TestApp.ThirdDbContext
+{
+    public interface IThirdDbContext : IAbpMongoDbContext
+    {
+        IMongoCollection<ThirdDbContextDummyEntity> DummyEntities { get; }
+    }
+}

framework/test/Volo.Abp.MongoDB.Tests.SecondContext/Volo/Abp/MongoDB/TestApp/ThirdDbContext/ThirdDbContext.cs

@@ -0,0 +1,11 @@
+using MongoDB.Driver;
+
+namespace Volo.Abp.MongoDB.TestApp.ThirdDbContext
+{
+    /* This dbcontext is just for testing to replace dbcontext from the application using AbpDbContextRegistrationOptions.ReplaceDbContext
+     */
+    public class ThirdDbContext : AbpMongoDbContext, IThirdDbContext
+    {
+        public IMongoCollection<ThirdDbContextDummyEntity> DummyEntities => Collection<ThirdDbContextDummyEntity>();
+    }
+}

framework/test/Volo.Abp.MongoDB.Tests.SecondContext/Volo/Abp/MongoDB/TestApp/ThirdDbContext/ThirdDbContextDummyEntity.cs

@@ -0,0 +1,10 @@
+using System;
+using Volo.Abp.Domain.Entities;
+
+namespace Volo.Abp.MongoDB.TestApp.ThirdDbContext
+{
+    public class ThirdDbContextDummyEntity : AggregateRoot<Guid>
+    {
+        public string Value { get; set; }
+    }
+}

framework/test/Volo.Abp.MongoDB.Tests/Volo.Abp.MongoDB.Tests.csproj

@@ -9,8 +9,8 @@
 
   <ItemGroup>
     <ProjectReference Include="..\..\src\Volo.Abp.Autofac\Volo.Abp.Autofac.csproj" />
-    <ProjectReference Include="..\..\src\Volo.Abp.MongoDB\Volo.Abp.MongoDB.csproj" />
     <ProjectReference Include="..\AbpTestBase\AbpTestBase.csproj" />
+    <ProjectReference Include="..\Volo.Abp.MongoDB.Tests.SecondContext\Volo.Abp.MongoDB.Tests.SecondContext.csproj" />
     <ProjectReference Include="..\Volo.Abp.TestApp\Volo.Abp.TestApp.csproj" />
   </ItemGroup>
 

framework/test/Volo.Abp.MongoDB.Tests/Volo/Abp/MongoDB/AbpMongoDbTestModule.cs

@@ -1,21 +1,18 @@
 using System;
-using System.Linq;
-using System.Threading;
 using Microsoft.Extensions.DependencyInjection;
-using MongoDB.Driver;
-using MongoDB.Driver.Core.Servers;
 using Volo.Abp.Data;
 using Volo.Abp.Modularity;
+using Volo.Abp.MongoDB.TestApp.SecondContext;
+using Volo.Abp.MongoDB.TestApp.ThirdDbContext;
 using Volo.Abp.TestApp;
 using Volo.Abp.TestApp.Domain;
 using Volo.Abp.TestApp.MongoDB;
-using Volo.Abp.Uow;
 
 namespace Volo.Abp.MongoDB
 {
     [DependsOn(
-        typeof(AbpMongoDbModule),
-        typeof(TestAppModule)
+        typeof(TestAppModule),
+        typeof(AbpMongoDbTestSecondContextModule)
         )]
     public class AbpMongoDbTestModule : AbpModule
     {
@@ -35,6 +32,8 @@ namespace Volo.Abp.MongoDB
             {
                 options.AddDefaultRepositories<ITestAppMongoDbContext>();
                 options.AddRepository<City, CityRepository>();
+
+                options.ReplaceDbContext<IThirdDbContext>();
             });
         }
     }

framework/test/Volo.Abp.MongoDB.Tests/Volo/Abp/MongoDB/DbContext_Replace_Tests.cs

@@ -0,0 +1,31 @@
+using Microsoft.Extensions.DependencyInjection;
+using Microsoft.Extensions.Options;
+using Shouldly;
+using Volo.Abp.MongoDB.TestApp.FourthContext;
+using Volo.Abp.MongoDB.TestApp.ThirdDbContext;
+using Volo.Abp.TestApp.MongoDB;
+using Xunit;
+
+namespace Volo.Abp.MongoDB
+{
+    [Collection(MongoTestCollection.Name)]
+    public class DbContext_Replace_Tests : MongoDbTestBase
+    {
+        private readonly AbpMongoDbContextOptions _options;
+
+        public DbContext_Replace_Tests()
+        {
+            _options = GetRequiredService<IOptions<AbpMongoDbContextOptions>>().Value;
+        }
+
+        [Fact]
+        public void Should_Replace_DbContext()
+        {
+            _options.GetReplacedTypeOrSelf(typeof(IThirdDbContext)).ShouldBe(typeof(TestAppMongoDbContext));
+            _options.GetReplacedTypeOrSelf(typeof(IFourthDbContext)).ShouldBe(typeof(TestAppMongoDbContext));
+
+            (ServiceProvider.GetRequiredService<IThirdDbContext>() is TestAppMongoDbContext).ShouldBeTrue();
+            (ServiceProvider.GetRequiredService<IFourthDbContext>() is TestAppMongoDbContext).ShouldBeTrue();
+        }
+    }
+}

framework/test/Volo.Abp.MongoDB.Tests/Volo/Abp/TestApp/MongoDb/TestAppMongoDbContext.cs

@@ -1,12 +1,17 @@
 using MongoDB.Driver;
 using Volo.Abp.Data;
+using Volo.Abp.DependencyInjection;
+using Volo.Abp.EntityFrameworkCore.TestApp.FourthContext;
 using Volo.Abp.MongoDB;
+using Volo.Abp.MongoDB.TestApp.FourthContext;
+using Volo.Abp.MongoDB.TestApp.ThirdDbContext;
 using Volo.Abp.TestApp.Domain;
 
 namespace Volo.Abp.TestApp.MongoDB
 {
     [ConnectionStringName("TestApp")]
-    public class TestAppMongoDbContext : AbpMongoDbContext, ITestAppMongoDbContext
+    [ReplaceDbContext(typeof(IFourthDbContext))]
+    public class TestAppMongoDbContext : AbpMongoDbContext, ITestAppMongoDbContext, IThirdDbContext, IFourthDbContext
     {
         [MongoCollection("Persons")] //Intentionally changed the collection name to test it
         public IMongoCollection<Person> People => Collection<Person>();
@@ -15,6 +20,10 @@ namespace Volo.Abp.TestApp.MongoDB
 
         public IMongoCollection<City> Cities => Collection<City>();
 
+        public IMongoCollection<ThirdDbContextDummyEntity> DummyEntities  => Collection<ThirdDbContextDummyEntity>();
+
+        public IMongoCollection<FourthDbContextDummyEntity> FourthDummyEntities => Collection<FourthDbContextDummyEntity>();
+
         protected internal override void CreateModel(IMongoModelBuilder modelBuilder)
         {
             base.CreateModel(modelBuilder);