lmxopcua/tests/Server/ZB.MOM.WW.OtOpcUa.Runtime.Tests/Drivers/DriverInstanceActorResilienceWiringTests.cs

using System.Collections.Concurrent;
using Akka.Actor;
using Shouldly;
using Xunit;
using ZB.MOM.WW.OtOpcUa.Commons.Types;
using ZB.MOM.WW.OtOpcUa.Core.Abstractions;
using ZB.MOM.WW.OtOpcUa.Runtime.Drivers;
using ZB.MOM.WW.OtOpcUa.Runtime.Tests.Harness;

namespace ZB.MOM.WW.OtOpcUa.Runtime.Tests.Drivers;

/// <summary>
///     Runtime guard for the Phase 6.1 resilience wiring (arch-review #10 / RESILIENCE-DISPATCH-GAP):
///     proves the <see cref="DriverInstanceActor"/> routes each driver-capability call through the
///     injected <see cref="IDriverCapabilityInvoker"/> at RUNTIME, with the correct
///     <see cref="DriverCapability"/> and resolved host key — not just that the calls compile wrapped
///     (the <c>OTOPCUA0001</c> analyzer is the compile-time guard; this is the behavioural one).
///     A recording invoker stands in for the real <c>CapabilityInvoker</c> so the test stays in the
///     Polly-free Runtime layer.
/// </summary>
public sealed class DriverInstanceActorResilienceWiringTests : RuntimeActorTestBase
{
    /// <summary>
    ///     A bulk subscribe issued on connect (via <c>ResubscribeDesired</c>) routes through the invoker
    ///     with <see cref="DriverCapability.Subscribe"/> keyed on the driver-instance id (single-host
    ///     fallback for a set that may span hosts). The driver's <c>SubscribeAsync</c> only runs from
    ///     inside the invoker's call-site, so <c>SubscribeCount &gt;= 1</c> already proves the route.
    /// </summary>
    [Fact]
    public async Task Subscribe_routes_through_the_capability_invoker()
    {
        var invoker = new RecordingInvoker();
        var driver = new SubscribableStubDriver();
        var actor = Sys.ActorOf(DriverInstanceActor.Props(driver, invoker: invoker));

        actor.Tell(new DriverInstanceActor.SetDesiredSubscriptions(
            new[] { "tag-a", "tag-b" }, TimeSpan.FromMilliseconds(100)));
        actor.Tell(new DriverInstanceActor.InitializeRequested("{}"));
        AwaitCondition(() => driver.SubscribeCount >= 1, TimeSpan.FromSeconds(3));

        // Barrier: round-trip a Connected-only Ask so the async subscribe handler has fully recorded.
        await actor.Ask<DriverInstanceActor.SubscriptionEstablished>(
            new DriverInstanceActor.Subscribe(new[] { "tag-a" }, TimeSpan.FromMilliseconds(100)),
            TimeSpan.FromSeconds(3));

        invoker.Calls.ShouldContain(c => c.Capability == DriverCapability.Subscribe && c.Host == driver.DriverInstanceId);
    }

    /// <summary>
    ///     A write routes through the invoker's <c>ExecuteWriteAsync</c> path (recorded as
    ///     <see cref="DriverCapability.Write"/>) with the host resolved via
    ///     <see cref="IPerCallHostResolver"/> — proving per-host breaker keying is wired, not the
    ///     single-host fallback. A write that arrives before <c>Connected</c> fast-fails WITHOUT
    ///     touching the invoker, so the loop drives to Connected and the recorded call is the real one.
    /// </summary>
    [Fact]
    public async Task Write_routes_through_the_invoker_with_resolved_host()
    {
        var invoker = new RecordingInvoker();
        var driver = new HostResolvingWritableDriver();   // ResolveHost("tag-1") => "plc-7"
        var actor = Sys.ActorOf(DriverInstanceActor.Props(driver, invoker: invoker));
        actor.Tell(new DriverInstanceActor.InitializeRequested("{}"));

        // Retry until the write succeeds (i.e. the actor reached Connected). Connecting-state writes
        // fast-fail on a synchronous path that never reaches the invoker, so they don't pollute Calls.
        DriverInstanceActor.WriteAttributeResult? reply = null;
        for (var i = 0; i < 60 && reply is not { Success: true }; i++)
        {
            reply = await actor.Ask<DriverInstanceActor.WriteAttributeResult>(
                new DriverInstanceActor.WriteAttribute("tag-1", 42), TimeSpan.FromSeconds(1));
            if (reply is not { Success: true }) await Task.Delay(25);
        }

        reply.ShouldNotBeNull();
        reply!.Success.ShouldBeTrue();
        driver.Writes.Count.ShouldBeGreaterThanOrEqualTo(1);   // the invoker's call-site actually ran
        invoker.Calls.ShouldContain(c => c.Capability == DriverCapability.Write && c.Host == "plc-7");
    }

    /// <summary>Records every capability + resolved-host key routed through it, then delegates to the
    /// real call-site so the driver still runs. Stands in for <c>CapabilityInvoker</c> without pulling
    /// Polly / Core into the Runtime test.</summary>
    private sealed class RecordingInvoker : IDriverCapabilityInvoker
    {
        public ConcurrentQueue<(DriverCapability Capability, string Host)> Calls { get; } = new();

        public ValueTask<TResult> ExecuteAsync<TResult>(
            DriverCapability capability, string hostName,
            Func<CancellationToken, ValueTask<TResult>> callSite, CancellationToken cancellationToken)
        {
            Calls.Enqueue((capability, hostName));
            return callSite(cancellationToken);
        }

        public ValueTask ExecuteAsync(
            DriverCapability capability, string hostName,
            Func<CancellationToken, ValueTask> callSite, CancellationToken cancellationToken)
        {
            Calls.Enqueue((capability, hostName));
            return callSite(cancellationToken);
        }

        public ValueTask<TResult> ExecuteWriteAsync<TResult>(
            string hostName, bool isIdempotent,
            Func<CancellationToken, ValueTask<TResult>> callSite, CancellationToken cancellationToken)
        {
            Calls.Enqueue((DriverCapability.Write, hostName));
            return callSite(cancellationToken);
        }
    }

    /// <summary>A writable driver that also resolves a per-tag host, so the write path exercises
    /// <see cref="DriverInstanceActor"/>'s <c>ResolveHost</c> (via <see cref="IPerCallHostResolver"/>).</summary>
    private sealed class HostResolvingWritableDriver : IDriver, IWritable, IPerCallHostResolver
    {
        public List<WriteRequest> Writes { get; } = new();

        public string DriverInstanceId => "host-resolving-driver";
        public string DriverType => "Stub";

        public string ResolveHost(string fullReference) => "plc-7";

        public Task InitializeAsync(string driverConfigJson, CancellationToken cancellationToken) => Task.CompletedTask;
        public Task ReinitializeAsync(string driverConfigJson, CancellationToken cancellationToken) => Task.CompletedTask;
        public Task ShutdownAsync(CancellationToken cancellationToken) => Task.CompletedTask;
        public DriverHealth GetHealth() => new(DriverState.Healthy, DateTime.UtcNow, null);
        public long GetMemoryFootprint() => 0;
        public Task FlushOptionalCachesAsync(CancellationToken cancellationToken) => Task.CompletedTask;

        public Task<IReadOnlyList<WriteResult>> WriteAsync(
            IReadOnlyList<WriteRequest> writes, CancellationToken cancellationToken)
        {
            Writes.AddRange(writes);
            IReadOnlyList<WriteResult> results = writes.Select(_ => new WriteResult(0u)).ToList();
            return Task.FromResult(results);
        }
    }
}