squidex/backend/tests/Squidex.Data.Tests/EntityFramework/TestHelpers/TestDbContext.cs

// ==========================================================================
//  Squidex Headless CMS
// ==========================================================================
//  Copyright (c) Squidex UG (haftungsbeschraenkt)
//  All rights reserved. Licensed under the MIT license.
// ==========================================================================

using Microsoft.EntityFrameworkCore;
using PhenX.EntityFrameworkCore.BulkInsert.MySql;
using PhenX.EntityFrameworkCore.BulkInsert.PostgreSql;
using PhenX.EntityFrameworkCore.BulkInsert.SqlServer;
using Squidex.Infrastructure;
using Squidex.Infrastructure.Json;
using Squidex.Infrastructure.Queries;
using Squidex.Infrastructure.States;
using Squidex.Providers.MySql;
using Squidex.Providers.Postgres;
using Squidex.Providers.SqlServer;
using Squidex.Shared;

#pragma warning disable CS9107 // Parameter is captured into the state of the enclosing type and its value is also passed to the base constructor. The value might be captured by the base class as well.
#pragma warning disable MA0048 // File name must match type name

namespace Squidex.EntityFramework.TestHelpers;

public class TestDbContextMySql(DbContextOptions options, IJsonSerializer jsonSerializer)
    : TestDbContext(options, jsonSerializer)
{
    public override SqlDialect Dialect => MySqlDialect.Instance;

    protected override void OnConfiguring(DbContextOptionsBuilder optionsBuilder)
    {
        optionsBuilder.UseBulkInsertMySql();
        base.OnConfiguring(optionsBuilder);
    }
}

public class TestDbContextPostgres(DbContextOptions options, IJsonSerializer jsonSerializer)
    : TestDbContext(options, jsonSerializer)
{
    public override SqlDialect Dialect => PostgresDialect.Instance;

    protected override void OnConfiguring(DbContextOptionsBuilder optionsBuilder)
    {
        optionsBuilder.UseBulkInsertPostgreSql();
        base.OnConfiguring(optionsBuilder);
    }
}

public class TestDbContextSqlServer(DbContextOptions options, IJsonSerializer jsonSerializer)
    : TestDbContext(options, jsonSerializer)
{
    public override SqlDialect Dialect => SqlServerDialect.Instance;

    protected override void OnConfiguring(DbContextOptionsBuilder optionsBuilder)
    {
        optionsBuilder.UseBulkInsertSqlServer();
        base.OnConfiguring(optionsBuilder);
    }
}

public abstract class TestDbContext(DbContextOptions options, IJsonSerializer jsonSerializer)
    : AppDbContext(options, jsonSerializer)
{
    protected override void OnModelCreating(ModelBuilder builder)
    {
        builder.UseSnapshot<SnapshotValue, EFState<SnapshotValue>>(jsonSerializer, Dialect.JsonColumnType());

        builder.Entity<TestEntity>(b =>
        {
            b.Property(x => x.Json).AsJsonString(jsonSerializer, Dialect.JsonColumnType());
        });

        base.OnModelCreating(builder);
    }
}