UnitTestBase.cs

using System;
using MccSoft.IntegreSql.EF;
using MccSoft.IntegreSql.EF.DatabaseInitialization;
using Microsoft.EntityFrameworkCore;

namespace ExampleWeb.UnitTests;

public class UnitTestBase : IDisposable
{
    protected readonly IDatabaseInitializer? _databaseInitializer;
    private DbContextOptions<ExampleDbContext>? _dbContextOptions;

    public UnitTestBase(
        DatabaseType? databaseType,
        DatabaseSeedingOptions<ExampleDbContext>? seedingOptions = null
    )
    {
        _databaseInitializer = CreateDatabaseInitializer(databaseType);

        _dbContextOptions = _databaseInitializer
            ?.CreateDatabaseGetDbContextOptionsBuilderSync<ExampleDbContext>(
                seedingOptions: seedingOptions
            )
            ?.Options;
    }

    private IDatabaseInitializer? CreateDatabaseInitializer(DatabaseType? databaseType)
    {
        return databaseType switch
        {
            null => null,
            DatabaseType.Postgres
                => new NpgsqlDatabaseInitializer(
                    // This is needed if you run tests NOT inside the container.
                    // 5434 is the public port number of Postgresql instance
                    connectionStringOverride: new() { Host = "localhost", Port = 5434 }
                )
                {
                    DropDatabaseOnRemove = true,
                },
            DatabaseType.Sqlite => new SqliteDatabaseInitializer(),
            _ => throw new ArgumentOutOfRangeException(nameof(databaseType), databaseType, null)
        };
    }

    public ExampleDbContext CreateDbContext()
    {
        if (_dbContextOptions == null)
            return null!;
        return new ExampleDbContext(_dbContextOptions);
    }

    public void Dispose()
    {
        _databaseInitializer?.RemoveDatabase(CreateDbContext().Database.GetConnectionString());
    }
}