MysqlDbContext.cs

// Licensed to the .NET Foundation under one or more agreements.
// The .NET Foundation licenses this file to you under the MIT license.
// See the LICENSE file in the project root for more information.

using Microsoft.EntityFrameworkCore;
using Xunit.Abstractions;

namespace CommunityToolkit.Datasync.TestCommon.Databases;

[ExcludeFromCodeCoverage]
public class MysqlDbContext(DbContextOptions<MysqlDbContext> options) : BaseDbContext<MysqlDbContext, MysqlEntityMovie>(options)
{
    public static async Task<MysqlDbContext> CreateContextAsync(string connectionString, ITestOutputHelper output = null, bool clearEntities = true)
    {
        if (string.IsNullOrEmpty(connectionString))
        {
            throw new ArgumentNullException(nameof(connectionString));
        }

        DbContextOptionsBuilder<MysqlDbContext> optionsBuilder = new DbContextOptionsBuilder<MysqlDbContext>()
            .UseMySql(connectionString: connectionString, serverVersion: ServerVersion.AutoDetect(connectionString), options => options.EnableRetryOnFailure())
            .EnableLogging(output);
        MysqlDbContext context = new(optionsBuilder.Options);

        await context.InitializeDatabaseAsync(clearEntities);
        await context.PopulateDatabaseAsync();
        return context;
    }

    internal async Task InitializeDatabaseAsync(bool clearEntities)
    {
        await Database.EnsureCreatedAsync();

        if (clearEntities)
        {
            await ExecuteRawSqlOnEachEntityAsync("DELETE FROM {0}");
        }
    }

    protected override void OnModelCreating(ModelBuilder modelBuilder)
    {
        modelBuilder.Entity<MysqlEntityMovie>().Property(m => m.UpdatedAt)
            .ValueGeneratedOnAddOrUpdate();

        modelBuilder.Entity<MysqlEntityMovie>().Property(m => m.Version)
            .IsRowVersion();

        base.OnModelCreating(modelBuilder);
    }
}