using System;

using System;

using Xunit;

using Xunit.Abstractions;

using Xunit.Sdk;

using Moq;

using Microsoft.Extensions.DependencyInjection;

using Microsoft.Extensions.Logging;

using Microsoft.EntityFrameworkCore;

using System.Threading.Tasks;

​

// Include your original test class and method here directly.

using Xunit;

using Moq;

using Microsoft.Extensions.DependencyInjection;

using Microsoft.Extensions.Logging;

using Microsoft.EntityFrameworkCore;

using Microsoft.Extensions.Hosting;

​

public class MigrationTests

{

    [Theory]

    [InlineData("true", true)]   // Azure primary: Flag set true -> migrate

    [InlineData(null, true)]     // AWS region: Flag unset -> migrate (desired)

    [InlineData("false", false)] // Explicitly false -> don't migrate

    public async Task ApplyMigrations_CurrentLogic_Behavior(string envValue, bool shouldMigrate)

    {

        // Arrange

        Environment.SetEnvironmentVariable("ENABLE_DB_MIGRATION_IN_PRIMARY_REGION", envValue);

​

        var migrateCalled = false;

​

        var mockDbContext = new Mock<ApplicationDbContext>();

        mockDbContext.Setup(db => db.Database.MigrateAsync(default))

                     .Callback(() => migrateCalled = true)

                     .Returns(Task.CompletedTask);

​

        var services = new ServiceCollection()

            .AddLogging()

            .AddSingleton(mockDbContext.Object)

            .BuildServiceProvider();

​

        var mockHost = new Mock<IHost>();

        mockHost.Setup(h => h.Services).Returns(services);

​

        // Act

        await ApplyMigrationsAsync(mockHost.Object);

​

        // Assert

        Assert.Equal(shouldMigrate, migrateCalled);

    }

​

    [Fact]

    public async Task AzureSecondary_NoFlagSet_ShouldSkipMigration_ButCurrentLogicIncorrectlyRunsMigration()

    {

        // Azure secondary scenario (no flag) currently migrates, but desired behavior is skip migration.

​

        // Arrange

        Environment.SetEnvironmentVariable("ENABLE_DB_MIGRATION_IN_PRIMARY_REGION", null);

​

        var migrateCalled = false;

​

        var mockDbContext = new Mock<ApplicationDbContext>();

        mockDbContext.Setup(db => db.Database.MigrateAsync(default))

                     .Callback(() => migrateCalled = true)

                     .Returns(Task.CompletedTask);

​

        var services = new ServiceCollection()

            .AddLogging()

            .AddSingleton(mockDbContext.Object)

            .BuildServiceProvider();

​

        var mockHost = new Mock<IHost>();

        mockHost.Setup(h => h.Services).Returns(services);

​

        // Act

        await ApplyMigrationsAsync(mockHost.Object);

​

        // Assert (currently failing test to demonstrate issue)

        Assert.False(migrateCalled); // THIS TEST WILL FAIL, proving current logic doesn't satisfy Azure-secondary

    }

​

    private static async Task ApplyMigrationsAsync(IHost host)

    {

        using var scope = host.Services.CreateScope();

        var services = scope.ServiceProvider;

        var logger = services.GetRequiredService<ILogger<MigrationTests>>();

​

        try

        {

            var envValue = Environment.GetEnvironmentVariable("ENABLE_DB_MIGRATION_IN_PRIMARY_REGION");

            if (envValue != null && bool.TryParse(envValue, out var isMigrationEnabled) && !isMigrationEnabled)

            {

                logger.LogInformation("Skipping database migration as it is explicitly disabled.");

                return;

            }

​

            var context = services.GetRequiredService<ApplicationDbContext>();

            await context.Database.MigrateAsync();

​

            logger.LogInformation("Database migration applied successfully in primary region.");

        }

        catch (Exception ex)

        {

            logger.LogError(ex, "An error occurred while applying database migrations in primary region.");

        }

    }

}

​

public class Program

{

    public static async Task Main()

    {

        var testAssembly = typeof(Program).Assembly;

        var testRunner = AssemblyRunner.WithoutAppDomain(testAssembly.Location);

        testRunner.OnTestFailed = info =>

        {

            Console.WriteLine($"[FAIL] {info.TestDisplayName}: {info.ExceptionMessage}");

        };

        testRunner.OnTestPassed = info =>

        {

            Console.WriteLine($"[PASS] {info.TestDisplayName}");

        };

        testRunner.Start();

​

        while (testRunner.Status != AssemblyRunnerStatus.Idle)

        {

            await Task.Delay(100);

        }

    }

}

​