Replace DbContext in WebApplicationFactory for unit testing

Question

I need to replace context in WebApplicationFactory. I have MyDbContext which I want to replace with SQLite context for testing.

The replace part works fine

.ConfigureServices(services =>
  {
    // Remove the app's ApplicationDbContext registration.
    var descriptor = services.SingleOrDefault(d => d.ServiceType == typeof(DbContextOptions<MyDbContext>));

    if (descriptor != null)
    {
      services.Remove(descriptor);
    }
    
    services.AddDbContext<MyDbContext>(builder =>
    {
      builder.UseSqlite(CreateInMemoryDatabase());
    });
  });

But because I moved from Npgsql to SQLite in tests I need to override some default values in OnModelCreating. I created a new db context class

public class MySqlLiteDbContext: MyDbContext
    {
        public MySqlLiteDbContext(DbContextOptions<MyDbContext> options) : base(options)
        {
        }

        protected override void OnModelCreating(ModelBuilder modelBuilder)
        {
            base.OnModelCreating(modelBuilder);

            modelBuilder.Entity<Record>()
                .Property(b => b.DateCreated)
                .HasDefaultValueSql("(datetime('now'))");

            ...
        }
    }

Is there any way to inject MySqlLiteDbContext instead of MyDbContext to force EnsureCreated use OnModelCreating for SQLite? Is extracting IMyDbContext an option? What else can I do to solve this problem?

kalucki23 · Accepted Answer · 2024-05-17 22:13:00Z

Ok, looks like I managed to solve it like this

Add non-generic DBContext constructor override, make it protected

public class MyDbContext: DbContext {        
    public MyDbContext(DbContextOptions<MyDbContext> options): base(options) {}
    // new
    protected MyDbContext(DbContextOptions options): base(options) {}
        
    // rest context stuff...
}

Remove DBContext and DBContextOptions in WebApplicationFactory on ConfigureServices

.ConfigureServices(services => {
    // Remove the app's ApplicationDbContext registration.
    var descriptor = services.SingleOrDefault(d => d.ServiceType == typeof (DbContextOptions<MyDbContext>));

    if (descriptor != null) {
        services.Remove(descriptor);
    }

    var descriptor2 = services.SingleOrDefault(d => d.ServiceType == typeof (MyDbContext));

    if (descriptor2 != null) {
        services.Remove(descriptor2);
    }

    //...
})

Add your descendant context as implementation

.ConfigureServices(services => {
    // Remove the app's ApplicationDbContext registration.

    // ...
    services.AddDbContext<MyDbContext, MySqliteDbContext>(builder => {
        builder.UseSqlite(CreateInMemoryDatabase());
    }, ServiceLifetime.Singleton);
})

Note: If you use InMemory Sqlite consider Singleton scope.

Done. Now when DocumentComparisonContext injected DocumentComparisonSqlLiteContext will be used as implementation, and on EnsureCreated sqlite-specific logic will be used.

Tupac · Accepted Answer · 2021-06-08 09:25:52Z

2

Register DbContext with the Services Container. Open Startup.cs and in the ConfigureServices function, we are going to use the AddDbContext extension method to add our new DbContext and tell it to use SQLite with the connection string from our appsettings.json. The following is the full function with the first two lines being the ones we added.

public void ConfigureServices(IServiceCollection services)
{
    services.AddDbContext<ContactsDbContext>(options =>
        options.UseSqlite(Configuration.GetConnectionString("DefaultConnection")));
    services.AddControllers();
    services.AddOpenApiDocument(document => 
        document.PostProcess = d => d.Info.Title = "Contacts API");
}

answered Jun 8, 2021 at 9:25

Tupac

2,9382 gold badges8 silver badges20 bronze badges

Comments

joacar · Accepted Answer · 2024-05-07 20:42:52Z

1

Thanks for guidance. There is no need to remove the context, it suffice with the options class.

var contextOptions = new DbContextOptionsBuilder<MyDbContext>()
            .UseInMemoryDatabase("<name>");
services.AddScoped<DbContextOptions<MyDbContext>>(_ => contextOptions.Options);

answered May 7, 2024 at 20:42

joacar

96111 silver badges28 bronze badges

Collectives™ on Stack Overflow

Replace DbContext in WebApplicationFactory for unit testing

3 Answers 3

Comments

Comments

Comments

Your Answer

Hot Network Questions

Collectives™ on Stack Overflow

3 Answers 3

Comments

Comments

Comments

Your Answer

Sign up or log in

Post as a guest

Related