using StellaOps.AirGap.Controller.Services;
using StellaOps.AirGap.Controller.Stores;
using StellaOps.AirGap.Time.Models;
using StellaOps.AirGap.Time.Services;
using Xunit;

namespace StellaOps.AirGap.Controller.Tests;

public class AirGapStateServiceTests
{
    private readonly AirGapStateService _service;
    private readonly InMemoryAirGapStateStore _store = new();
    private readonly StalenessCalculator _calculator = new();

    public AirGapStateServiceTests()
    {
        _service = new AirGapStateService(_store, _calculator);
    }

    [Fact]
    public async Task Seal_sets_state_and_computes_staleness()
    {
        var now = DateTimeOffset.UtcNow;
        var anchor = new TimeAnchor(now.AddMinutes(-2), "roughtime", "roughtime", "fp", "digest");
        var budget = new StalenessBudget(60, 120);

        await _service.SealAsync("tenant-a", "policy-1", anchor, budget, now);
        var status = await _service.GetStatusAsync("tenant-a", now);

        Assert.True(status.State.Sealed);
        Assert.Equal("policy-1", status.State.PolicyHash);
        Assert.Equal("tenant-a", status.State.TenantId);
        Assert.True(status.Staleness.AgeSeconds > 0);
        Assert.True(status.Staleness.IsWarning);
        Assert.Equal(120 - status.Staleness.AgeSeconds, status.Staleness.SecondsRemaining);
    }

    [Fact]
    public async Task Unseal_clears_sealed_flag_and_updates_timestamp()
    {
        var now = DateTimeOffset.UtcNow;
        await _service.SealAsync("default", "hash", TimeAnchor.Unknown, StalenessBudget.Default, now);

        var later = now.AddMinutes(1);
        await _service.UnsealAsync("default", later);
        var status = await _service.GetStatusAsync("default", later);

        Assert.False(status.State.Sealed);
        Assert.Equal(later, status.State.LastTransitionAt);
    }
}