ScadaBridge/tests/ZB.MOM.WW.ScadaBridge.ConfigurationDatabase.Tests/DataProtectionTests.cs

using Microsoft.AspNetCore.DataProtection;
using Microsoft.EntityFrameworkCore;
using Microsoft.Extensions.DependencyInjection;
using ZB.MOM.WW.ScadaBridge.ConfigurationDatabase;

namespace ZB.MOM.WW.ScadaBridge.ConfigurationDatabase.Tests;

public class DataProtectionTests : IDisposable
{
    private readonly string _dbPath;

    public DataProtectionTests()
    {
        _dbPath = Path.Combine(Path.GetTempPath(), $"scadabridge_dp_test_{Guid.NewGuid()}.db");
    }

    public void Dispose()
    {
        if (File.Exists(_dbPath))
            File.Delete(_dbPath);
    }

    [Fact]
    public void SharedDataProtection_ProtectAndUnprotect_AcrossContainers()
    {
        var connectionString = $"DataSource={_dbPath}";

        // Create the database schema
        var setupOptions = new DbContextOptionsBuilder<ScadaBridgeDbContext>()
            .UseSqlite(connectionString)
            .Options;
        using (var setupCtx = new ScadaBridgeDbContext(setupOptions))
        {
            setupCtx.Database.EnsureCreated();
        }

        // Container 1: protect some data
        var services1 = new ServiceCollection();
        services1.AddDbContext<ScadaBridgeDbContext>(opt => opt.UseSqlite(connectionString));
        services1.AddDataProtection()
            .SetApplicationName("ScadaBridge")
            .PersistKeysToDbContext<ScadaBridgeDbContext>();

        using var provider1 = services1.BuildServiceProvider();
        var protector1 = provider1.GetRequiredService<IDataProtectionProvider>()
            .CreateProtector("test-purpose");
        var protectedPayload = protector1.Protect("secret-data");

        // Container 2: unprotect using the same DB (shared keys)
        var services2 = new ServiceCollection();
        services2.AddDbContext<ScadaBridgeDbContext>(opt => opt.UseSqlite(connectionString));
        services2.AddDataProtection()
            .SetApplicationName("ScadaBridge")
            .PersistKeysToDbContext<ScadaBridgeDbContext>();

        using var provider2 = services2.BuildServiceProvider();
        var protector2 = provider2.GetRequiredService<IDataProtectionProvider>()
            .CreateProtector("test-purpose");
        var unprotected = protector2.Unprotect(protectedPayload);

        Assert.Equal("secret-data", unprotected);
    }
}