Updated CustomWebApplicationDbContextFactory to remove current dbcont…

…ext before adding test ones.
DFE-Digital · Oct 11, 2024 · 5b6ffda · 5b6ffda
1 parent 7f4a841
commit 5b6ffda
Showing 1 changed file with 15 additions and 0 deletions.
diff --git a/src/DfE.CoreLibs.Testing/Mocks/WebApplicationFactory/CustomWebApplicationDbContextFactory.cs b/src/DfE.CoreLibs.Testing/Mocks/WebApplicationFactory/CustomWebApplicationDbContextFactory.cs
@@ -27,6 +27,21 @@ protected override void ConfigureWebHost(IWebHostBuilder builder)
                     var dbContextType = entry.Key;
                     var seedAction = entry.Value;
 
+                    var dbContextDescriptor = services.SingleOrDefault(
+                        d => d.ServiceType == typeof(DbContextOptions<>).MakeGenericType(dbContextType));
+                    if (dbContextDescriptor != null)
+                    {
+                        services.Remove(dbContextDescriptor);
+                    }
+
+                    var dbConnectionDescriptor = services.SingleOrDefault(
+                        d => d.ServiceType == typeof(DbConnection));
+                    if (dbConnectionDescriptor != null)
+                    {
+                        services.Remove(dbConnectionDescriptor);
+                    }
+
+
                     var createDbContextMethod = typeof(DbContextHelper).GetMethod(nameof(DbContextHelper.CreateDbContext))
                         ?.MakeGenericMethod(dbContextType);