Unistream.TestTask.Tests/TransientContext/Common/DatabaseFixture.cs - master - d3h5/Unistream.TestTask

Unistream.TestTask.Tests/TransientContext/Common/DatabaseFixture.cs

71 строка2 KB

WORKSTATION-H\mrthe

Первый залив

03 июн 2025, 11:20

03 июн 2025, 11:2033c9164

using Grimtod.Articles.Tests.TransientContext.Postgresql;
using Microsoft.EntityFrameworkCore;
using Microsoft.Extensions.Configuration;
using Unistream.TestTask.DataAccessImpl.DataContexts;

namespace Unistream.TestTask.Tests.TransientContext.Common;

public class DatabaseFixture : IDisposable
{
    public ApplicationDbContext DbContext { get; }

    public ITestDatabase? TestDatabase { get; }

    private bool _disposed = false;

    /// <summary>
    /// Конструктор вызывается только один раз перед запуском всех тестов.
    /// </summary>
    public DatabaseFixture()
    {
        var configuration = new ConfigurationBuilder()
            .AddJsonFile("appsettings.json", optional: false, reloadOnChange: true)
            .Build();
        
        TestDatabase = new TestDatabaseBuilder()
            .WithConfiguration(configuration)
            .Build();

        TestDatabase.Create();

        var builder = new DbContextOptionsBuilder<ApplicationDbContext>();
        configuration["ConnectionStrings:Default"] = TestDatabase.ConnectionString;
        DbContext = new ApplicationDbContext(builder.Options, configuration);
        _ = DbContext.Database.EnsureCreated();

    }

    public string TemplateDatabaseName { get; } = string.Empty;

    public string Connection { get; } = string.Empty;

    protected virtual void Dispose(bool disposing)
    {
        if (!_disposed)
        {
            if (disposing && TestDatabase is not null)
            {
                // Удаляем базу данных
                TestDatabase.Drop();
            }

            _disposed = true;
        }
    }

    ~DatabaseFixture()
    {
        Dispose(false);
    }

    public void Dispose()
    {
        Dispose(true);
        GC.SuppressFinalize(this);
    }
}

[CollectionDefinition("Database")]
public class DatabaseCollectionFixture : ICollectionFixture<DatabaseFixture>
{
}