using Microsoft.Extensions.Logging.Abstractions;
using Serilog;
using Shouldly;
using Xunit;
using ZB.MOM.WW.OtOpcUa.Core.Scripting;

namespace ZB.MOM.WW.OtOpcUa.Driver.Calculation.Tests;

/// <summary>
///     <see cref="CalculationEvaluator"/> runs every compiled script through a <see cref="ScriptRunGuard"/>
///     keyed by calc-tag RawPath. A timed-out run is abandoned on its pool thread; before the guard, every
///     later trigger of the same tag stacked another orphan behind it (one leaked thread per trigger). Now
///     the next trigger is refused ("previous evaluation still running"), the third consecutive
///     non-completion quarantines the tag, and a healthy sibling on the same evaluator keeps computing.
/// </summary>
public sealed class CalculationEvaluatorRunGuardTests
{
    private static ScriptRootLogger NoOpScriptRoot() => new(new LoggerConfiguration().CreateLogger());

    /// <summary>Outlives the 60 ms budget by a wide margin but still ends, so the orphan frees itself.</summary>
    private const string SpinScript =
        "var end = System.DateTime.UtcNow.AddSeconds(6); while (System.DateTime.UtcNow < end) { } return 0;";

    private static readonly Dictionary<string, object?> NoDeps = new();

    [Fact]
    public async Task A_trigger_during_the_previous_orphan_is_refused_then_the_tag_is_quarantined_and_a_sibling_keeps_computing()
    {
        using var sut = new CalculationEvaluator(
            NullLogger<CalculationEvaluator>.Instance, NoOpScriptRoot(), TimeSpan.FromMilliseconds(60));
        var ct = TestContext.Current.CancellationToken;

        var first = await sut.EvaluateAsync("calc/loop", SpinScript, Snap(NoDeps), ct).WaitAsync(TimeSpan.FromSeconds(30), ct);
        first.Success.ShouldBeFalse();
        first.Reason.ShouldNotBeNull();
        first.Reason.ShouldContain("timed out");

        var second = await sut.EvaluateAsync("calc/loop", SpinScript, Snap(NoDeps), ct);
        second.Success.ShouldBeFalse();
        second.Reason.ShouldBe(ScriptRunGuard.PreviousRunStillRunningReason);

        var third = await sut.EvaluateAsync("calc/loop", SpinScript, Snap(NoDeps), ct);
        third.Success.ShouldBeFalse();
        third.Reason.ShouldNotBeNull();
        third.Reason.ShouldStartWith(ScriptRunGuard.QuarantinedReasonPrefix);
        sut.QuarantinedScripts.ShouldBe(new[] { "calc/loop" });

        // Exactly one run was ever dispatched for the runaway tag — the leak is one orphan, not one per trigger.
        sut.RunGuard.StartedRunCount.ShouldBe(1);
        sut.RunGuard.InFlightCount.ShouldBe(1);

        // Positive control: a sibling tag on the same evaluator still computes.
        var sibling = await sut.EvaluateAsync("calc/ok", "return 42;", Snap(NoDeps), ct);
        sibling.Success.ShouldBeTrue(sibling.Reason);
        sibling.Value.ShouldBe(42);
    }

    [Fact]
    public async Task ClearCompiledScripts_lifts_the_quarantine_as_a_redeploy_does()
    {
        using var sut = new CalculationEvaluator(
            NullLogger<CalculationEvaluator>.Instance, NoOpScriptRoot(), TimeSpan.FromMilliseconds(60),
            new ScriptingOptions { QuarantineAfterConsecutiveTimeouts = 1 });
        var ct = TestContext.Current.CancellationToken;

        var r = await sut.EvaluateAsync("calc/loop", SpinScript, Snap(NoDeps), ct).WaitAsync(TimeSpan.FromSeconds(30), ct);
        r.Reason.ShouldNotBeNull();
        r.Reason.ShouldStartWith(ScriptRunGuard.QuarantinedReasonPrefix);

        sut.ClearCompiledScripts();

        sut.QuarantinedScripts.ShouldBeEmpty();
        // The fixed script (new source) runs; the old orphan is a sunk cost that no longer blocks the tag.
        var fixedRun = await sut.EvaluateAsync("calc/loop", "return 7;", Snap(NoDeps), ct);
        fixedRun.Success.ShouldBeTrue(fixedRun.Reason);
        fixedRun.Value.ShouldBe(7);
    }

    private static Dictionary<string, Core.Abstractions.DataValueSnapshot> Snap(Dictionary<string, object?> deps)
    {
        var now = DateTime.UtcNow;
        var d = new Dictionary<string, Core.Abstractions.DataValueSnapshot>(StringComparer.Ordinal);
        foreach (var kv in deps) d[kv.Key] = new Core.Abstractions.DataValueSnapshot(kv.Value, 0u, now, now);
        return d;
    }
}