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

using Akka.Actor;
using Shouldly;
using Xunit;
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>
///     Covers the bounded post-connect re-discovery loop: when an <see cref="ITagDiscovery"/> driver
///     reaches Connected, <see cref="DriverInstanceActor"/> runs repeated discovery passes (FOCAS-style:
///     the FixedTree is suppressed until the driver's cache populates ~0–2s after connect) and ships each
///     pass's captured nodes to its parent as <see cref="DriverInstanceActor.DiscoveredNodesReady"/>. The
///     loop STOPS once the non-empty discovered set stabilises (or the attempt cap is hit) — it must not
///     spin forever. A driver that does not implement <see cref="ITagDiscovery"/> produces no passes at all.
/// </summary>
[Trait("Category", "Unit")]
public sealed class DriverInstanceActorDiscoveryTests : RuntimeActorTestBase
{
    /// <summary>
    ///     A discoverable driver whose first two passes yield nothing (cache still warming) and whose third
    ///     pass onward yields a stable 3-node set: the actor ships every pass, then STOPS once the non-empty
    ///     set repeats. The final <see cref="DriverInstanceActor.DiscoveredNodesReady"/> carries the 3 nodes
    ///     and no further passes arrive — proving the loop is bounded.
    /// </summary>
    [Fact]
    public void Discovery_retries_until_set_stabilises_then_stops()
    {
        var driver = new DiscoverableStubDriver();
        var parent = CreateTestProbe();
        // Tiny interval so the bounded retry runs in well under a second (no real-time waits).
        var actor = parent.ChildActorOf(DriverInstanceActor.Props(
            driver, rediscoverInterval: TimeSpan.FromMilliseconds(20)));

        // Drive Connecting → Connected; the Connected entry kicks discovery.
        actor.Tell(new DriverInstanceActor.InitializeRequested("{}"));

        // Each discovery pass publishes one DiscoveredNodesReady. The fake stabilises after pass 4
        // (passes: 0,0,3,3), so exactly 4 messages arrive, then the stream stops.
        var msgs = new List<DriverInstanceActor.DiscoveredNodesReady>();
        for (var i = 0; i < 4; i++)
            msgs.Add(parent.ExpectMsg<DriverInstanceActor.DiscoveredNodesReady>(TimeSpan.FromSeconds(2)));

        // The loop must STOP once the non-empty set has stabilised — no fifth pass.
        parent.ExpectNoMsg(TimeSpan.FromMilliseconds(300));

        // Early passes were empty (FixedTree cache still populating).
        msgs[0].Nodes.Count.ShouldBe(0);
        msgs[1].Nodes.Count.ShouldBe(0);
        // The set then appears and stabilises at 3 nodes.
        msgs[2].Nodes.Count.ShouldBe(3);
        var final = msgs[^1];
        final.Nodes.Count.ShouldBe(3);
        final.DriverInstanceId.ShouldBe(driver.DriverInstanceId);
        final.Nodes.Select(n => n.FullReference).ShouldBe(new[] { "m.fixed.v0", "m.fixed.v1", "m.fixed.v2" });

        // The driver was asked exactly as many times as messages published — no extra zombie pass.
        driver.DiscoverCount.ShouldBe(4);
    }

    /// <summary>A driver that does not implement <see cref="ITagDiscovery"/> produces no discovery passes —
    /// the Connected entry's discovery kick is a no-op, so the parent receives no
    /// <see cref="DriverInstanceActor.DiscoveredNodesReady"/>.</summary>
    [Fact]
    public void Driver_without_ITagDiscovery_produces_no_discovery()
    {
        var driver = new SubscribableStubDriver(); // IDriver + ISubscribable, NOT ITagDiscovery
        var parent = CreateTestProbe();
        var actor = parent.ChildActorOf(DriverInstanceActor.Props(
            driver, rediscoverInterval: TimeSpan.FromMilliseconds(20)));

        actor.Tell(new DriverInstanceActor.InitializeRequested("{}"));
        AwaitCondition(() => driver.InitializeCount > 0, TimeSpan.FromSeconds(2));

        // No discovery capability ⇒ never any DiscoveredNodesReady to the parent.
        parent.ExpectNoMsg(TimeSpan.FromMilliseconds(300));
    }

    /// <summary>
    ///     Discovery RE-RUNS on every return to Connected: after the initial discovery settles, a
    ///     <see cref="DriverInstanceActor.ForceReconnect"/> drives the actor through Reconnecting and
    ///     back to Connected (via the auto-retry timer, the same path the existing reconnect tests use),
    ///     and a fresh bounded discovery loop fires — keeping the injected tree current if the backend's
    ///     capabilities changed across the reconnect. The new init bumps the generation, so any
    ///     pre-reconnect tick is discarded by the generation guard (the initial loop has already settled
    ///     here, so none are in flight).
    /// </summary>
    [Fact]
    public void Discovery_reruns_after_reconnect()
    {
        var driver = new DiscoverableStubDriver();
        var parent = CreateTestProbe();
        // Tiny reconnect + rediscover intervals so the whole reconnect-then-rediscover cycle runs fast.
        var actor = parent.ChildActorOf(DriverInstanceActor.Props(
            driver,
            reconnectInterval: TimeSpan.FromMilliseconds(50),
            rediscoverInterval: TimeSpan.FromMilliseconds(20)));

        actor.Tell(new DriverInstanceActor.InitializeRequested("{}"));

        // Drain the initial settling passes (0,0,3,3) and confirm the first loop stopped.
        for (var i = 0; i < 4; i++)
            parent.ExpectMsg<DriverInstanceActor.DiscoveredNodesReady>(TimeSpan.FromSeconds(2));
        parent.ExpectNoMsg(TimeSpan.FromMilliseconds(200));
        var passesBeforeReconnect = driver.DiscoverCount; // 4

        // Force a reconnect: Connected → Reconnecting → (auto retry-connect) → Connected again.
        actor.Tell(new DriverInstanceActor.ForceReconnect());

        // A fresh discovery pass must arrive after the reconnect — the cache is warm now, so it sees
        // the stable 3-node set immediately.
        var afterReconnect = parent.ExpectMsg<DriverInstanceActor.DiscoveredNodesReady>(TimeSpan.FromSeconds(3));
        afterReconnect.Nodes.Count.ShouldBe(3);
        afterReconnect.DriverInstanceId.ShouldBe(driver.DriverInstanceId);

        // The driver was discovered again — proves a fresh loop ran, not a replay of the old one.
        driver.DiscoverCount.ShouldBeGreaterThan(passesBeforeReconnect);
    }

    /// <summary>
    ///     A <see cref="StubDriver"/> that also exposes <see cref="ITagDiscovery"/>. Each <c>DiscoverAsync</c>
    ///     pass is counted; passes 1–2 yield nothing (cache warming), passes 3+ yield a stable 3-node set —
    ///     modelling FOCAS, whose FixedTree appears once a few seconds after connect and then stays put.
    /// </summary>
    private sealed class DiscoverableStubDriver : StubDriver, ITagDiscovery
    {
        private int _passCount;

        /// <summary>Number of <see cref="DiscoverAsync"/> passes the actor has driven.</summary>
        public int DiscoverCount => Volatile.Read(ref _passCount);

        /// <summary>Streams a growing-then-stable node set into the builder (0,0,3,3,…).</summary>
        public Task DiscoverAsync(IAddressSpaceBuilder builder, CancellationToken cancellationToken)
        {
            var pass = Interlocked.Increment(ref _passCount); // 1-based pass number
            var count = pass >= 3 ? 3 : 0;
            var fixedTree = builder.Folder("FixedTree", "FixedTree");
            for (var i = 0; i < count; i++)
            {
                fixedTree.Variable($"v{i}", $"v{i}", new DriverAttributeInfo(
                    FullName: $"m.fixed.v{i}",
                    DriverDataType: DriverDataType.Float64,
                    IsArray: false,
                    ArrayDim: null,
                    SecurityClass: SecurityClassification.ViewOnly,
                    IsHistorized: false));
            }
            return Task.CompletedTask;
        }
    }
}